我用 Günter Zöchbauer 解决方案的核心思想更新了我的代码。
library functionMockTest;
import "package:unittest/unittest.dart";
import "package:mock/mock.dart";
import "dart:math" as math;
typedef int BinaryIntToInt(int a, int b);
typedef double BinaryIntToDouble(int a, int b);
typedef int DoubleIntToInt(double a, int b);
int useBinaryIntToInt(BinaryIntToInt adder) => adder(1, 2);
void main() {
test('Mock [min] function from the "dart:math"', () {
var min = new FunctionMock(math.min);
expect(min(2,1), 1);
expect(min(1,2), 1);
min.calls('call', 1, 2).verify(happenedOnce);
});
test("Function Mock", () {
var adder2 = new FunctionMock<BinaryIntToInt>((a,b) => a + b);
var adder3 = new FunctionMock((a,b,c) => a + b + c);
expect(adder2 is Function, true);
expect(useBinaryIntToInt(adder2), 3);
expect(adder3(1,2,3), 6);
adder2.calls('call', 1, 2).verify(happenedOnce);
});
group("Type check on function mocks:", (){
test("Should throw [TypeError] \n "
"if function has wrong number of arguments", () {
TypeError error;
try {
var adder3 = new FunctionMock<BinaryIntToInt>((a,b,c) => a + b + c);
}
catch(exception) {
expect(exception is TypeError, true);
error = exception;
}
expect(error != null, true);
});
test("Should throw [TypeError] \n "
"if function has wrong type of arguments", () {
TypeError error;
try {
var adder3 = new FunctionMock<BinaryIntToInt>((double a,b) => 10);
}
catch(exception) {
expect(exception is TypeError, true);
error = exception;
}
expect(error != null, true);
});
test("Doesn't throw on typedef mismatch \n"
"without type annotation", () {
BinaryIntToDouble foo = (c,d) => c / d;
DoubleIntToInt bar = (c,d) => c + d;
var wrongTypedefReturnType = new FunctionMock<BinaryIntToInt>(foo);
var wrongTypedefArgumentType = new FunctionMock<BinaryIntToInt>(bar);
wrongTypedefReturnType(1.1,2.1);
wrongTypedefArgumentType(1.1,2.1);
});
test("Throws with type annotation", () {
double foo_ (int c, int d) => c / d;
BinaryIntToDouble foo = foo_;
int bar_ (double c, int d) => 10;
DoubleIntToInt bar = bar_;
TypeError returnTypeError, argumentTypeError;
try {
var wrongTypedefReturnType = new FunctionMock<BinaryIntToInt>(foo);
}
catch(exception) {
expect(exception is TypeError, true);
returnTypeError = exception;
}
try {
var wrongTypedefArgumentType = new FunctionMock<BinaryIntToInt>(bar);
}
catch(exception) {
expect(exception is TypeError, true);
argumentTypeError = exception;
}
expect(returnTypeError != null, true);
expect(argumentTypeError != null, true);
});
}
);
}
class _Sentinel {
const _Sentinel();
}
const NO_ARG = const _Sentinel();
class FunctionMock<FunctionTypedef> extends Mock implements Function{
final FunctionTypedef _callable;
FunctionMock._internal(FunctionTypedef function) : _callable = function {
when(callsTo('call')).alwaysCall(_callable);
}
//Place to 'shovel in' black magic if needed.
factory FunctionMock(FunctionTypedef function){
return new FunctionMock._internal(function);
}
call([arg0 = NO_ARG,
arg1 = NO_ARG,
arg2 = NO_ARG,
arg3 = NO_ARG,
arg4 = NO_ARG,
arg5 = NO_ARG,
arg6 = NO_ARG,
arg7 = NO_ARG,
arg8 = NO_ARG,
arg9 = NO_ARG]) {
if (identical(arg0, NO_ARG)) return super.call();
if (identical(arg1, NO_ARG)) return super.call(arg0);
if (identical(arg2, NO_ARG)) return super.call(arg0, arg1);
if (identical(arg3, NO_ARG)) return super.call(arg0, arg1, arg2);
if (identical(arg4, NO_ARG)) return super.call(arg0, arg1, arg2, arg3);
if (identical(arg5, NO_ARG)) return super.call(arg0, arg1, arg2, arg3, arg4);
if (identical(arg6, NO_ARG)) return super.call(arg0, arg1, arg2, arg3, arg4,
arg5);
if (identical(arg7, NO_ARG)) return super.call(arg0, arg1, arg2, arg3, arg4,
arg5, arg6);
if (identical(arg8, NO_ARG)) return super.call(arg0, arg1, arg2, arg3, arg4,
arg5, arg6, arg7);
if (identical(arg9, NO_ARG)) return super.call(arg0, arg1, arg2, arg3, arg4,
arg5, arg6, arg7, arg8);
return super.call(arg0, arg1, arg2, arg3, arg4, arg5, arg6, arg7, arg8, arg9);
}
}
*它适用于 0-9 个参数 - 与callsTo