mirror of https://github.com/AMT-Cheif/drift.git
53 lines
1.7 KiB
Dart
53 lines
1.7 KiB
Dart
import 'dart:async';
|
|
|
|
import 'package:mockito/mockito.dart';
|
|
import 'package:moor/moor.dart';
|
|
import 'package:moor/src/runtime/executor/stream_queries.dart';
|
|
|
|
export 'package:mockito/mockito.dart';
|
|
|
|
typedef Future<T> _EnsureOpenAction<T>(QueryExecutor e);
|
|
|
|
class MockExecutor extends Mock implements QueryExecutor {
|
|
final MockTransactionExecutor transactions = MockTransactionExecutor();
|
|
|
|
MockExecutor() {
|
|
when(runSelect(any, any)).thenAnswer((_) => Future.value([]));
|
|
when(runUpdate(any, any)).thenAnswer((_) => Future.value(0));
|
|
when(runDelete(any, any)).thenAnswer((_) => Future.value(0));
|
|
when(runInsert(any, any)).thenAnswer((_) => Future.value(0));
|
|
when(doWhenOpened(any)).thenAnswer((i) {
|
|
final action = i.positionalArguments.single as _EnsureOpenAction;
|
|
|
|
return action(this);
|
|
});
|
|
|
|
when(beginTransaction()).thenAnswer((_) => transactions);
|
|
}
|
|
}
|
|
|
|
class MockTransactionExecutor extends Mock implements TransactionExecutor {
|
|
MockTransactionExecutor() {
|
|
when(runSelect(any, any)).thenAnswer((_) => Future.value([]));
|
|
when(runUpdate(any, any)).thenAnswer((_) => Future.value(0));
|
|
when(runDelete(any, any)).thenAnswer((_) => Future.value(0));
|
|
when(runInsert(any, any)).thenAnswer((_) => Future.value(0));
|
|
when(doWhenOpened(any)).thenAnswer((i) {
|
|
final action = i.positionalArguments.single as _EnsureOpenAction;
|
|
|
|
return action(this);
|
|
});
|
|
|
|
when(send()).thenAnswer((_) => Future.value(null));
|
|
}
|
|
}
|
|
|
|
class MockStreamQueries extends Mock implements StreamQueryStore {}
|
|
|
|
// used so that we can mock the SqlExecutor typedef
|
|
abstract class SqlExecutorAsClass {
|
|
Future<void> call(String sql);
|
|
}
|
|
|
|
class MockQueryExecutor extends Mock implements SqlExecutorAsClass {}
|