fix(auth): create_custom_token

The issue arose when updating dependencies

Author: AsifArmanRahman

firebase/auth/__init__.py

@@ -17,6 +17,7 @@
 import python_jwt as jwt
 import jwcrypto.jwk as jwk
 from urllib.parse import parse_qs
+from cryptography.hazmat.primitives.serialization import Encoding, NoEncryption, PrivateFormat
 
 from firebase._exception import raise_detailed_error
 
@@ -190,7 +191,7 @@ def create_custom_token(self, uid, additional_claims=None, expiry_minutes=60):
 		"""
 
 		service_account_email = self.credentials.service_account_email
-		private_key = jwk.JWK.from_pem(self.credentials._private_key_pkcs8_pem.encode('utf-8'))
+		private_key = jwk.JWK.from_pem(self.credentials.signer._key.private_bytes(encoding=Encoding.PEM, format=PrivateFormat.PKCS8, encryption_algorithm=NoEncryption()))
 
 		payload = {
 			"iss": service_account_email,
@@ -204,7 +205,7 @@ def create_custom_token(self, uid, additional_claims=None, expiry_minutes=60):
 
 		exp = datetime.timedelta(minutes=expiry_minutes)
 
-		return jwt.generate_jwt(payload, private_key, "RS256", exp, other_headers={'kid': self.credentials._private_key_id})
+		return jwt.generate_jwt(payload, private_key, "RS256", exp, other_headers={'kid': self.credentials.signer._key_id})
 
 	def sign_in_with_custom_token(self, token):
 		""" Exchange custom token for an ID and refresh token.

tests/test_auth.py

@@ -118,19 +118,16 @@ def test_sign_in_anonymous(self, auth_admin):
 		self.__class__.anonymous_user = user
 		assert user
 
-	@pytest.mark.xfail
 	def test_create_custom_token(self, auth_admin):
 		token = auth_admin.create_custom_token('CreateCustomToken1')
 		self.__class__.custom_token = token
 		assert token
 
-	@pytest.mark.xfail
 	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
 
-	@pytest.mark.xfail
 	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)
@@ -166,8 +163,5 @@ def test_update_profile_display_name(self, auth_admin):
 	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'))
-
-	@pytest.mark.xfail
-	def test_delete_custom_user_account(self, auth_admin):
 		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'))