Skip to content

Commit 8f4f6ad

Browse files
test(firestore): add cases for update() method
1 parent b1af488 commit 8f4f6ad

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

tests/test_firestore.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,11 @@ 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_manual_doc_update(self, ds_admin):
26+
update_data = {'released': True}
27+
assert ds_admin.collection('Marvels').document('Movies').update(update_data) is None
28+
assert ds_admin.collection('Marvels').document('Movies').get(field_paths=['released']) == update_data
29+
2530
def test_manual_doc_get_filtered(self, ds_admin):
2631
assert ds_admin.collection('Marvels').document('Movies').get(field_paths=['name']) == {'name': self.__class__.movies['name']}
2732

@@ -56,6 +61,11 @@ def test_manual_doc_get(self, ds):
5661
def test_manual_doc_get_filtered(self, ds):
5762
assert ds.collection('Marvels').document('Movies').get(field_paths=['name'], token=self.__class__.user.get('idToken')) == {'name': self.__class__.movies['name']}
5863

64+
def test_manual_doc_update(self, ds):
65+
update_data = {'released': True}
66+
assert ds.collection('Marvels').document('Movies').update(update_data, token=self.__class__.user.get('idToken')) is None
67+
assert ds.collection('Marvels').document('Movies').get(field_paths=['released'], token=self.__class__.user.get('idToken')) == update_data
68+
5969
def test_manual_doc_delete(self, ds):
6070
assert ds.collection('Marvels').document('Movies').delete(self.__class__.user.get('idToken')) is None
6171

@@ -83,5 +93,10 @@ def test_manual_doc_get(self, ds):
8393
def test_manual_doc_get_filtered(self, ds):
8494
assert ds.collection('Marvels').document('Series').get(field_paths=['name']) == {'name': self.__class__.series['name']}
8595

96+
def test_manual_doc_update(self, ds):
97+
update_data = {'released': True}
98+
assert ds.collection('Marvels').document('Series').update(update_data) is None
99+
assert ds.collection('Marvels').document('Series').get(field_paths=['released']) == update_data
100+
86101
def test_manual_doc_delete(self, ds):
87102
assert ds.collection('Marvels').document('Series').delete() is None

0 commit comments

Comments
 (0)