def _makeOne( self, document_reference=None, firestore=None, target=None, comparator=None, snapshot_callback=None, snapshot_class=None, reference_class=None, ): # pragma: NO COVER from google.cloud.firestore_v1.watch import Watch if document_reference is None: document_reference = DummyDocumentReference() if firestore is None: firestore = DummyFirestore() if target is None: WATCH_TARGET_ID = 0x5079 # "Py" target = { "documents": { "documents": ["/"] }, "target_id": WATCH_TARGET_ID } if comparator is None: comparator = self._document_watch_comparator if snapshot_callback is None: snapshot_callback = self._snapshot_callback if snapshot_class is None: snapshot_class = DummyDocumentSnapshot if reference_class is None: reference_class = DummyDocumentReference inst = Watch( document_reference, firestore, target, comparator, snapshot_callback, snapshot_class, reference_class, BackgroundConsumer=DummyBackgroundConsumer, ResumableBidiRpc=DummyRpc, ) return inst
def _make_watch_no_mocks( snapshots=None, comparator=_document_watch_comparator, ): from google.cloud.firestore_v1.watch import Watch WATCH_TARGET_ID = 0x5079 # "Py" target = {"documents": {"documents": ["/"]}, "target_id": WATCH_TARGET_ID} if snapshots is None: snapshots = [] def snapshot_callback(*args): snapshots.append(args) return Watch( document_reference=DummyDocumentReference(), firestore=DummyFirestore(), target=target, comparator=comparator, snapshot_callback=snapshot_callback, document_snapshot_cls=DummyDocumentSnapshot, )