Skip to content

Commit c853cd0

Browse files
test(firestore): cases for get method of collection
1 parent 67581b8 commit c853cd0

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

tests/test_firestore.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,9 @@ def test_manual_doc_set(self, ds_admin):
2222
def test_manual_doc_get(self, ds_admin):
2323
assert ds_admin.collection('Marvels').document('Movies').get() == self.__class__.movies
2424

25+
def test_collection_get(self, ds_admin):
26+
assert ds_admin.collection('Marvels').get() == [{'Movies': self.__class__.movies}]
27+
2528
def test_manual_doc_update(self, ds_admin):
2629
update_data = {'released': True}
2730
assert ds_admin.collection('Marvels').document('Movies').update(update_data) is None
@@ -58,6 +61,9 @@ def test_manual_doc_set(self, ds):
5861
def test_manual_doc_get(self, ds):
5962
assert ds.collection('Marvels').document('Movies').get(token=self.__class__.user.get('idToken')) == self.__class__.movies
6063

64+
def test_collection_get(self, ds):
65+
assert ds.collection('Marvels').get(token=self.__class__.user.get('idToken')) == [{'Movies': self.__class__.movies}]
66+
6167
def test_manual_doc_get_filtered(self, ds):
6268
assert ds.collection('Marvels').document('Movies').get(field_paths=['name'], token=self.__class__.user.get('idToken')) == {'name': self.__class__.movies['name']}
6369

@@ -90,6 +96,9 @@ def test_manual_doc_set(self, ds):
9096
def test_manual_doc_get(self, ds):
9197
assert ds.collection('Marvels').document('Series').get() == self.__class__.series
9298

99+
def test_collection_get(self, ds):
100+
assert ds.collection('Marvels').get() == [{'Series': self.__class__.series}]
101+
93102
def test_manual_doc_get_filtered(self, ds):
94103
assert ds.collection('Marvels').document('Series').get(field_paths=['name']) == {'name': self.__class__.series['name']}
95104

0 commit comments

Comments
 (0)