test(Auth): cases for set_custom_user_claims

Author: AsifArmanRahman

tests/test_auth.py

@@ -76,6 +76,13 @@ def test_update_profile_display_name(self, auth):
 		assert user
 		assert new_name == user['displayName']
 
+	def test_set_custom_user_claims(self, auth):
+		with pytest.raises(AttributeError) as exc_info:
+			auth.set_custom_user_claims(self.__class__.user.get('localId'), {'premium': True})
+			auth.set_custom_user_claims(self.__class__.anonymous_user.get('localId'), {'premium': True})
+
+		assert "'NoneType' object has no attribute 'valid'" in str(exc_info.value)
+
 	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'))
@@ -160,6 +167,10 @@ def test_update_profile_display_name(self, auth_admin):
 		assert user
 		assert new_name == user['displayName']
 
+	def test_set_custom_user_claims(self, auth_admin):
+		auth_admin.set_custom_user_claims(self.__class__.user.get('localId'), {'premium': True})
+		auth_admin.set_custom_user_claims(self.__class__.anonymous_user.get('localId'), {'premium': True})
+
 	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'))