diff --git a/mockfirestore/collection.py b/mockfirestore/collection.py index d2af224..4e6ae90 100644 --- a/mockfirestore/collection.py +++ b/mockfirestore/collection.py @@ -26,9 +26,8 @@ def document(self, document_id: Optional[str] = None) -> DocumentReference: return DocumentReference(self._data, new_path, parent=self) def get(self) -> Iterable[DocumentSnapshot]: - warnings.warn('Collection.get is deprecated, please use Collection.stream', - category=DeprecationWarning) - return self.stream() + # Stream uses a generator, so we need to convert it to a list for compatibility for .get() method with firestore library + return list(self.stream()) @property def path(self): @@ -127,9 +126,8 @@ def document(self, document_id: Optional[str] = None, path: List[str] = None) -> return ret def get(self) -> Iterable[DocumentSnapshot]: - warnings.warn('Collection.get is deprecated, please use Collection.stream', - category=DeprecationWarning) - return self.stream() + # Stream uses a generator, so we need to convert it to a list for compatibility for .get() method with firestore library + return list(self.stream()) def stream(self, transaction=None) -> Iterable[DocumentSnapshot]: for path in self._path: diff --git a/mockfirestore/query.py b/mockfirestore/query.py index 179ac46..5698bf0 100644 --- a/mockfirestore/query.py +++ b/mockfirestore/query.py @@ -57,12 +57,11 @@ def stream(self, transaction=None) -> Iterator[DocumentSnapshot]: if self._limit: doc_snapshots = islice(doc_snapshots, self._limit) - return iter(doc_snapshots) + return iter(list(doc_snapshots)) def get(self) -> Iterator[DocumentSnapshot]: - warnings.warn('Query.get is deprecated, please use Query.stream', - category=DeprecationWarning) - return self.stream() + # Stream uses a generator, so we need to convert it to a list for compatibility for .get() method with firestore library + return list(self.stream()) def _add_field_filter(self, field: str, op: str, value: Any): compare = self._compare_func(op)