add kwargs to doc ref

mdowds · Feb 5, 2024 · 04b3463 · 04b3463
1 parent 6457932
commit 04b3463
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/mockfirestore/document.py b/mockfirestore/document.py
@@ -67,7 +67,7 @@ def id(self):
     def path(self):
         return '/'.join(self._path)
 
-    def get(self) -> DocumentSnapshot:
+    def get(self, **kwargs) -> DocumentSnapshot:
         return DocumentSnapshot(self, get_by_path(self._data, self._path))
 
     def delete(self):

diff --git a/tests/test_document_reference.py b/tests/test_document_reference.py
@@ -16,6 +16,15 @@ def test_get_document_by_path(self):
         self.assertEqual({'id': 1}, doc.to_dict())
         self.assertEqual('first', doc.id)
 
+    def test_get_document_by_path_with_transaction_kwarg(self):
+        fs = MockFirestore()
+        fs._data = {'foo': {
+            'first': {'id': 1}
+        }}
+        doc = fs.document('foo/first').get(transaction='tx')
+        self.assertEqual({'id': 1}, doc.to_dict())
+        self.assertEqual('first', doc.id)
+
     def test_document_path_property(self):
         fs = MockFirestore()
         fs._data = {'foo': {