如何在Dart/Flutter测试中Override/Mock库函数以自定义返回值

你可以通过两种主流方案实现需求，可根据自身业务场景选择：

方案1：依赖注入改造（推荐，长期可维护） #

该方案通过解耦业务代码和第三方依赖实现测试灵活配置，无黑魔法，兼容所有Dart运行环境，且后续替换第三方依赖成本极低，唯一要求是对现有业务代码做少量调整。

改造后业务代码： #

import 'package:foo_package/exposing_foo_function.dart';

class TestableClass {
  // 定义第三方函数作为可注入依赖
  final bool Function() fooFunc;
  // 构造函数默认使用原生foo实现，原有业务侧调用无需修改参数
  TestableClass({bool Function()? fooFunc}) : fooFunc = fooFunc ?? foo;

  bool bar() {
    return fooFunc(); // 调用注入的依赖实例
  }
}

对应测试代码： #

void main() {
  test('TestableClass.bar() when foo_package.foo() returns false', () {
    // 构造实例时传入自定义返回的mock函数即可
    TestableClass testableClass = TestableClass(fooFunc: () => false);
    expect(testableClass.bar(), isFalse);
  });
}

方案2：Zone覆盖实现（无需修改业务代码） #

如果不想改动现有业务代码，可以用Dart的Zone机制拦截重写第三方顶级函数的返回值，该方案无需改造业务逻辑，但属于运行时黑魔法，不推荐在生产逻辑中使用，且Dart版本升级可能存在兼容风险。

测试代码示例： #

import 'dart:async';
import 'package:foo_package/exposing_foo_function.dart' as foo_pkg;

void main() {
  test('TestableClass.bar() when foo_package.foo() returns false', () {
    // 创建自定义测试Zone，覆盖foo函数实现
    final testZone = Zone.current.fork(
      specification: ZoneSpecification(
        registerFunction: (self, parent, zone, name, func) {
          if (func == foo_pkg.foo) {
            return () => false;
          }
          return parent.registerFunction(zone, name, func);
        },
      ),
    );

    // 在自定义Zone内执行测试逻辑即可生效
    testZone.run(() {
      TestableClass testableClass = TestableClass();
      expect(testableClass.bar(), isFalse);
    });
  });
}

内容的提问来源于stack exchange，提问作者Roboroads