Skip to content

Commit e7c093d

Browse files
test(firestore): add cases for get() method
1 parent dba57ea commit e7c093d

File tree

1 file changed

+18
-0
lines changed

1 file changed

+18
-0
lines changed

tests/test_firestore.py

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,12 @@ class TestFirestoreAdmin:
1919
def test_manual_doc_set(self, ds_admin):
2020
assert ds_admin.collection('Marvels').document('Movies').set(self.__class__.movies) is None
2121

22+
def test_manual_doc_get(self, ds_admin):
23+
assert ds_admin.collection('Marvels').document('Movies').get() == self.__class__.movies
24+
25+
def test_manual_doc_get_filtered(self, ds_admin):
26+
assert ds_admin.collection('Marvels').document('Movies').get(field_paths=['name']) == {'name': self.__class__.movies['name']}
27+
2228
def test_manual_doc_delete(self, ds_admin):
2329
assert ds_admin.collection('Marvels').document('Movies').delete() is None
2430

@@ -44,6 +50,12 @@ def test_create_test_user(self, auth):
4450
def test_manual_doc_set(self, ds):
4551
assert ds.collection('Marvels').document('Movies').set(self.__class__.movies, token=self.__class__.user.get('idToken')) is None
4652

53+
def test_manual_doc_get(self, ds):
54+
assert ds.collection('Marvels').document('Movies').get(token=self.__class__.user.get('idToken')) == self.__class__.movies
55+
56+
def test_manual_doc_get_filtered(self, ds):
57+
assert ds.collection('Marvels').document('Movies').get(field_paths=['name'], token=self.__class__.user.get('idToken')) == {'name': self.__class__.movies['name']}
58+
4759
def test_manual_doc_delete(self, ds):
4860
assert ds.collection('Marvels').document('Movies').delete(self.__class__.user.get('idToken')) is None
4961

@@ -65,5 +77,11 @@ class TestFirestore:
6577
def test_manual_doc_set(self, ds):
6678
assert ds.collection('Marvels').document('Series').set(self.__class__.series) is None
6779

80+
def test_manual_doc_get(self, ds):
81+
assert ds.collection('Marvels').document('Series').get() == self.__class__.series
82+
83+
def test_manual_doc_get_filtered(self, ds):
84+
assert ds.collection('Marvels').document('Series').get(field_paths=['name']) == {'name': self.__class__.series['name']}
85+
6886
def test_manual_doc_delete(self, ds):
6987
assert ds.collection('Marvels').document('Series').delete() is None

0 commit comments

Comments
 (0)