chore(ui_firestore): mock Collection#snapshots() (#10146)

firebase · Dec 22, 2022 · 4289881 · 4289881
1 parent dc264ed
commit 4289881
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/packages/firebase_ui_firestore/test/table_builder_test.dart b/packages/firebase_ui_firestore/test/table_builder_test.dart
@@ -344,6 +344,25 @@ class MockAggregateQuery extends Mock implements AggregateQuery {
 
 class MockCollection extends Mock
     implements CollectionReference<Map<String, Object?>> {
+  @override
+  Stream<QuerySnapshot<Map<String, Object?>>> snapshots({
+    bool includeMetadataChanges = false,
+  }) {
+    return super.noSuchMethod(
+      Invocation.method(#snapshots, null, {
+        #includeMetadataChanges: includeMetadataChanges,
+      }),
+      returnValue: Stream.fromIterable([
+        MockQuerySnapshot(),
+        MockQuerySnapshot(),
+      ]),
+      returnValueForMissingStub: Stream.fromIterable([
+        MockQuerySnapshot(),
+        MockQuerySnapshot(),
+      ]),
+    );
+  }
+
   @override
   CollectionReference<R> withConverter<R extends Object?>({
     FromFirestore<R>? fromFirestore,