Differential D5281 Diff 18920 swh/auth/tests/test_utils.py

Changeset View

Standalone View

swh/auth/tests/test_utils.py

# Copyright (C) 2021 The Software Heritage developers		# Copyright (C) 2021 The Software Heritage developers
# See the AUTHORS file at the top-level directory of this distribution		# See the AUTHORS file at the top-level directory of this distribution
# License: GNU General Public License version 3, or any later version		# License: GNU General Public License version 3, or any later version
# See top-level LICENSE file for more information		# See top-level LICENSE file for more information

from copy import copy		from copy import copy
from datetime import datetime		from datetime import datetime

		import pytest

from swh.auth.django.utils import oidc_user_from_decoded_token, oidc_user_from_profile		from swh.auth.django.utils import oidc_user_from_decoded_token, oidc_user_from_profile
from swh.auth.tests.sample_data import CLIENT_ID, DECODED_TOKEN, OIDC_PROFILE		from swh.auth.tests.sample_data import CLIENT_ID, DECODED_TOKEN, OIDC_PROFILE


def test_oidc_user_from_decoded_token():		def test_oidc_user_from_decoded_token():
user = oidc_user_from_decoded_token(DECODED_TOKEN)		user = oidc_user_from_decoded_token(DECODED_TOKEN)

assert user.id == 338521271020811424925120118444075479552		assert user.id == 338521271020811424925120118444075479552
Show All 20 Lines	def test_oidc_user_from_decoded_token2():
assert user.first_name == "John"		assert user.first_name == "John"
assert user.last_name == "Doe"		assert user.last_name == "Doe"
assert user.email == "john.doe@example.com"		assert user.email == "john.doe@example.com"
assert user.is_staff is True		assert user.is_staff is True
assert user.permissions == {"read-api"}		assert user.permissions == {"read-api"}
assert user.sub == "feacd344-b468-4a65-a236-14f61e6b7200"		assert user.sub == "feacd344-b468-4a65-a236-14f61e6b7200"


		@pytest.mark.parametrize(
		"key,mapped_key",
		[
		("preferred_username", "username"),
		("given_name", "first_name"),
		("family_name", "last_name"),
		("email", "email"),
		],
		)
		def test_oidc_user_from_decoded_token_empty_fields_ok(key, mapped_key):
		decoded_token = copy(DECODED_TOKEN)
		decoded_token.pop(key, None)

		user = oidc_user_from_decoded_token(decoded_token, client_id=CLIENT_ID)

		assert user.id == 338521271020811424925120118444075479552
		assert user.password == ""
		assert user.is_staff is False
		assert user.permissions == set()
		assert user.sub == "feacd344-b468-4a65-a236-14f61e6b7200"
		# Ensure the missing field is mapped to an empty value
		assert getattr(user, mapped_key) == ""


def test_oidc_user_from_profile(keycloak_mock):		def test_oidc_user_from_profile(keycloak_mock):
date_now = datetime.now()		date_now = datetime.now()

user = oidc_user_from_profile(keycloak_mock, OIDC_PROFILE)		user = oidc_user_from_profile(keycloak_mock, OIDC_PROFILE)

assert user.id == 338521271020811424925120118444075479552		assert user.id == 338521271020811424925120118444075479552
assert user.username == "johndoe"		assert user.username == "johndoe"
assert user.password == ""		assert user.password == ""
Show All 11 Lines