test(auth): cases for admin and non admin

Author: AsifArmanRahman

tests/conftest.py

@@ -10,6 +10,16 @@
 from tests.config import TEST_USER_EMAIL, TEST_USER_PASSWORD
 
 
+@pytest.fixture(scope='session')
+def auth():
+	return make_auth()
+
+
+@pytest.fixture(scope='session')
+def auth_admin():
+	return make_auth(True)
+
+
 @pytest.fixture(scope='session')
 def db():
 	# To make it easier to test, we keep the test restricted to firebase_tests
@@ -20,11 +30,6 @@ def db():
 		make_db(service_account=True).child('firebase_tests').remove()
 
 
-@pytest.fixture(scope='session')
-def auth():
-	return make_auth(True)
-
-
 @pytest.fixture(scope='session')
 def email():
 	return TEST_USER_EMAIL

tests/test_auth.py

@@ -6,31 +6,31 @@
 
 
 import pytest
+import requests.exceptions
 
 
 class TestAuth:
 
 	user = None
 	anonymous_user = None
-	custom_token = None
-	custom_token_with_claims = None
-	custom_user = None
-	custom_user_with_claims = None
 
-	@pytest.mark.xfail
 	def test_sign_in_with_non_existing_account_email_and_password(self, auth, email, password):
-		assert auth.sign_in_with_email_and_password(email, password)
+		with pytest.raises(requests.exceptions.HTTPError) as exc_info:
+			auth.sign_in_with_email_and_password(email, password)
+		assert "EMAIL_NOT_FOUND" in str(exc_info.value)
 
 	def test_create_user_with_email_and_password(self, auth, email, password):
 		assert auth.create_user_with_email_and_password(email, password)
 
-	@pytest.mark.xfail
 	def test_create_user_with_existing_email_and_password(self, auth, email, password):
-		assert auth.create_user_with_email_and_password(email, password)
+		with pytest.raises(requests.exceptions.HTTPError) as exc_info:
+			auth.create_user_with_email_and_password(email, password)
+		assert "EMAIL_EXISTS" in str(exc_info.value)
 
-	@pytest.mark.xfail
 	def test_sign_in_with_email_and_wrong_password(self, auth, email):
-		assert auth.sign_in_with_email_and_password(email, 'WrongPassword123')
+		with pytest.raises(requests.exceptions.HTTPError) as exc_info:
+			auth.sign_in_with_email_and_password(email, 'WrongPassword123')
+		assert "INVALID_PASSWORD" in str(exc_info.value)
 
 	def test_sign_in_with_email_and_password(self, auth, email, password):
 		user = auth.sign_in_with_email_and_password(email, password)
@@ -43,24 +43,16 @@ def test_sign_in_anonymous(self, auth):
 		assert user
 
 	def test_create_custom_token(self, auth):
-		token = auth.create_custom_token('CreateCustomToken1')
-		self.__class__.custom_token = token
-		assert token
+		with pytest.raises(AttributeError):
+			auth.create_custom_token('CreateCustomToken1')
 
 	def test_create_custom_token_with_claims(self, auth):
-		token = auth.create_custom_token('CreateCustomToken2', {'premium': True})
-		self.__class__.custom_token_with_claims = token
-		assert token
+		with pytest.raises(AttributeError):
+			auth.create_custom_token('CreateCustomToken2', {'premium': True})
 
 	def test_sign_in_with_custom_token(self, auth):
-		user1 = auth.sign_in_with_custom_token(self.__class__.custom_token)
-		user2 = auth.sign_in_with_custom_token(self.__class__.custom_token_with_claims)
-
-		self.__class__.custom_user = user1
-		self.__class__.custom_user_with_claims = user2
-
-		assert user1
-		assert user2
+		with pytest.raises(requests.exceptions.HTTPError):
+			auth.sign_in_with_custom_token(None)
 
 	def test_refresh(self, auth):
 		assert auth.refresh(self.__class__.user.get('refreshToken'))
@@ -87,5 +79,89 @@ def test_update_profile_display_name(self, auth):
 	def test_delete_user_account(self, auth):
 		assert auth.delete_user_account(self.__class__.user.get('idToken'))
 		assert auth.delete_user_account(self.__class__.anonymous_user.get('idToken'))
-		assert auth.delete_user_account(self.__class__.custom_user.get('idToken'))
-		assert auth.delete_user_account(self.__class__.custom_user_with_claims.get('idToken'))
+
+
+class TestAuthAdmin:
+
+	user = None
+	anonymous_user = None
+	custom_token = None
+	custom_token_with_claims = None
+	custom_user = None
+	custom_user_with_claims = None
+
+	def test_sign_in_with_non_existing_account_email_and_password(self, auth_admin, email, password):
+		with pytest.raises(requests.exceptions.HTTPError) as exc_info:
+			auth_admin.sign_in_with_email_and_password(email, password)
+		assert "EMAIL_NOT_FOUND" in str(exc_info.value)
+
+	def test_create_user_with_email_and_password(self, auth_admin, email, password):
+		assert auth_admin.create_user_with_email_and_password(email, password)
+
+	def test_create_user_with_existing_email_and_password(self, auth_admin, email, password):
+		with pytest.raises(requests.exceptions.HTTPError) as exc_info:
+			auth_admin.create_user_with_email_and_password(email, password)
+		assert "EMAIL_EXISTS" in str(exc_info.value)
+
+	def test_sign_in_with_email_and_wrong_password(self, auth_admin, email):
+		with pytest.raises(requests.exceptions.HTTPError) as exc_info:
+			auth_admin.sign_in_with_email_and_password(email, 'WrongPassword123')
+		assert "INVALID_PASSWORD" in str(exc_info.value)
+
+	def test_sign_in_with_email_and_password(self, auth_admin, email, password):
+		user = auth_admin.sign_in_with_email_and_password(email, password)
+		self.__class__.user = user
+		assert user
+
+	def test_sign_in_anonymous(self, auth_admin):
+		user = auth_admin.sign_in_anonymous()
+		self.__class__.anonymous_user = user
+		assert user
+
+	def test_create_custom_token(self, auth_admin):
+		token = auth_admin.create_custom_token('CreateCustomToken1')
+		self.__class__.custom_token = token
+		assert token
+
+	def test_create_custom_token_with_claims(self, auth_admin):
+		token = auth_admin.create_custom_token('CreateCustomToken2', {'premium': True})
+		self.__class__.custom_token_with_claims = token
+		assert token
+
+	def test_sign_in_with_custom_token(self, auth_admin):
+		user1 = auth_admin.sign_in_with_custom_token(self.__class__.custom_token)
+		user2 = auth_admin.sign_in_with_custom_token(self.__class__.custom_token_with_claims)
+
+		self.__class__.custom_user = user1
+		self.__class__.custom_user_with_claims = user2
+
+		assert user1
+		assert user2
+
+	def test_refresh(self, auth_admin):
+		assert auth_admin.refresh(self.__class__.user.get('refreshToken'))
+
+	def test_get_account_info(self, auth_admin):
+		assert auth_admin.get_account_info(self.__class__.user.get('idToken'))
+
+	def test_send_email_verification(self, auth_admin):
+		assert auth_admin.send_email_verification(self.__class__.user.get('idToken'))
+
+	def test_send_password_reset_email(self, auth_admin):
+		assert auth_admin.send_password_reset_email(self.__class__.user.get('email'))
+
+	@pytest.mark.xfail
+	def test_verify_password_reset_code(self, auth_admin):
+		assert auth_admin.verify_password_reset_code('123456', 'NewTestPassword123')
+
+	def test_update_profile_display_name(self, auth_admin):
+		new_name = 'Test User'
+		user = auth_admin.update_profile(self.__class__.user.get('idToken'), display_name=new_name)
+		assert user
+		assert new_name == user['displayName']
+
+	def test_delete_user_account(self, auth_admin):
+		assert auth_admin.delete_user_account(self.__class__.user.get('idToken'))
+		assert auth_admin.delete_user_account(self.__class__.anonymous_user.get('idToken'))
+		assert auth_admin.delete_user_account(self.__class__.custom_user.get('idToken'))
+		assert auth_admin.delete_user_account(self.__class__.custom_user_with_claims.get('idToken'))

tests/test_setup.py

@@ -16,6 +16,13 @@ def test_initiate_app_with_service_account_file():
 
 
 def test_setup_auth():
+	auth = make_auth()
+	user = auth.sign_in_anonymous()
+
+	assert auth.delete_user_account(user['idToken'])
+
+
+def test_setup_auth_admin():
 	auth = make_auth(True)
 	user = auth.sign_in_anonymous()