test(firestore): add cases for set() method

Author: AsifArmanRahman

tests/config.py

@@ -14,6 +14,7 @@
 	"apiKey": config('FIREBASE_API_KEY'),
 	"authDomain": config('FIREBASE_AUTH_DOMAIN'),
 	"databaseURL": config('FIREBASE_DATABASE_URL'),
+	"projectId": config('FIREBASE_PROJECT_ID'),
 	"storageBucket": config('FIREBASE_STORAGE_BUCKET'),
 }
 

tests/conftest.py

@@ -6,7 +6,7 @@
 
 import pytest
 
-from tests.tools import make_auth, make_db, make_storage
+from tests.tools import make_auth, make_db, make_ds, make_storage
 from tests.config import TEST_USER_EMAIL, TEST_USER_PASSWORD
 
 
@@ -35,6 +35,16 @@ def email():
 	return TEST_USER_EMAIL
 
 
+@pytest.fixture(scope='session')
+def ds():
+	return make_ds()
+
+
+@pytest.fixture(scope='session')
+def ds_admin():
+	return make_ds(True)
+
+
 @pytest.fixture(scope='session')
 def password():
 	return TEST_USER_PASSWORD

tests/test_firestore.py

@@ -0,0 +1,60 @@
+
+#   Copyright (c) 2022 Asif Arman Rahman
+#   Licensed under MIT (https://github.com/AsifArmanRahman/firebase/blob/main/LICENSE)
+
+# --------------------------------------------------------------------------------------
+
+
+class TestFirestoreAdmin:
+	movies = {
+		'name': 'Iron Man',
+		'lead': {'name': 'Robert Downey Jr.'},
+		'released': False,
+		'year': 2008,
+		'rating': 7.9,
+		'prequel': None,
+		'cast': ['Jon Favreau', 'Gwyneth Paltrow', 'Jeff Bridges', b'J.A.R.V.I.S', 'Terrence Howard']
+	}
+
+	def test_manual_doc_set(self, ds_admin):
+		assert ds_admin.collection('Marvels').document('Movies').set(self.__class__.movies) is None
+
+
+class TestFirestoreAuth:
+	movies = {
+		'name': 'Dr. Strange',
+		'lead': {'name': 'Benedict Cumberbatch'},
+		'released': False,
+		'year': 2016,
+		'rating': 7.5,
+		'prequel': None,
+		'cast': ['Tilda Swinton', 'Rachel McAdams', 'Mads Mikkelsen', 'Chiwetel Ejiofor', 'Benedict Wong']
+	}
+	user = None
+
+	def test_create_test_user(self, auth):
+		user = auth.sign_in_anonymous()
+		self.__class__.user = user
+		assert user
+		assert user.get('idToken')
+
+	def test_manual_doc_set(self, ds):
+		assert ds.collection('Marvels').document('Movies').set(self.__class__.movies, token=self.__class__.user.get('idToken')) is None
+
+	def test_delete_test_user(self, auth):
+		assert auth.delete_user_account(self.__class__.user.get('idToken'))
+
+
+class TestFirestore:
+	series = {
+		'name': 'Moon Knight',
+		'lead': {'name': 'Oscar Issac'},
+		'released': False,
+		'year': 2022,
+		'rating': 7.4,
+		'prequel': None,
+		'cast': ['Ethan Hawke', 'May Calamawy', 'F. Murray Abraham']
+	}
+
+	def test_manual_doc_set(self, ds):
+		assert ds.collection('Marvels').document('Series').set(self.__class__.series) is None

tests/tools.py

@@ -31,6 +31,15 @@ def make_db(service_account=False):
 	return initialize_app(c).database()
 
 
+def make_ds(service_account=False):
+	if service_account:
+		c = config.SERVICE_CONFIG
+	else:
+		c = config.SIMPLE_CONFIG
+
+	return initialize_app(c).firestore()
+
+
 def make_storage(service_account=False):
 	if service_account:
 		c = config.SERVICE_CONFIG