Skip to content

Commit d9bd471

Browse files
test(firestore): add cases for list_of_files()
1 parent 90bbff5 commit d9bd471

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
@@ -43,6 +43,9 @@ def test_manual_doc_get(self, ds_admin):
4343
def test_collection_get(self, ds_admin):
4444
assert ds_admin.collection('Marvels').document('Movies').collection('PhaseOne').get() == [{'001': self.__class__.movies1}, {self.__class__.auto_doc_id: self.__class__.movies2}]
4545

46+
def test_collection_list_document(self, ds_admin):
47+
assert ds_admin.collection('Marvels').document('Movies').collection('PhaseOne').list_of_documents() == ['001', self.__class__.auto_doc_id]
48+
4649
def test_collection_get_start_after(self, ds_admin):
4750
assert ds_admin.collection('Marvels').document('Movies').collection('PhaseOne').order_by('rating').start_after({'rating': 7.4}).get() == [{'001': self.__class__.movies1}]
4851
assert ds_admin.collection('Marvels').document('Movies').collection('PhaseOne').order_by('rating').start_after({'rating': 6.9}).get() == [{self.__class__.auto_doc_id: self.__class__.movies2}, {'001': self.__class__.movies1}]
@@ -140,6 +143,9 @@ def test_manual_doc_get(self, ds):
140143
def test_collection_get(self, ds):
141144
assert ds.collection('Marvels').document('Movies').collection('PhaseThree').get(token=self.__class__.user.get('idToken')) == [{'014': self.__class__.movies1}, {self.__class__.auto_doc_id: self.__class__.movies2}]
142145

146+
def test_collection_list_documents(self, ds):
147+
assert ds.collection('Marvels').document('Movies').collection('PhaseThree').list_of_documents(token=self.__class__.user.get('idToken')) == ['014', self.__class__.auto_doc_id]
148+
143149
def test_manual_doc_get_filtered(self, ds):
144150
assert ds.collection('Marvels').document('Movies').collection('PhaseThree').document('014').get(field_paths=['name'], token=self.__class__.user.get('idToken')) == {'name': self.__class__.movies1['name']}
145151
assert ds.collection('Marvels').document('Movies').collection('PhaseThree').document(self.__class__.auto_doc_id).get(field_paths=['name'], token=self.__class__.user.get('idToken')) == {'name': self.__class__.movies2['name']}
@@ -233,6 +239,9 @@ def test_manual_doc_get(self, ds):
233239
def test_collection_get(self, ds):
234240
assert ds.collection('Marvels').document('Series').collection('PhaseFour').get() == [{'003': self.__class__.series1}, {self.__class__.auto_doc_id: self.__class__.series2}]
235241

242+
def test_collection_list_documents(self, ds):
243+
assert ds.collection('Marvels').document('Series').collection('PhaseFour').list_of_documents() == ['003', self.__class__.auto_doc_id]
244+
236245
def test_manual_doc_get_filtered(self, ds):
237246
assert ds.collection('Marvels').document('Series').collection('PhaseFour').document('003').get(field_paths=['name']) == {'name': self.__class__.series1['name']}
238247
assert ds.collection('Marvels').document('Series').collection('PhaseFour').document(self.__class__.auto_doc_id).get(field_paths=['name']) == {'name': self.__class__.series2['name']}

0 commit comments

Comments
 (0)