From fe828e07a14638bc75bc233c6b803c7cdff88772 Mon Sep 17 00:00:00 2001
From: Sivert Kronen Hatteberg <skh@uio.no>
Date: Mon, 6 Dec 2021 22:16:10 +0100
Subject: [PATCH] Update tests to check auth_type
---
gregui/tests/api/views/test_userinfo.py | 3 +++
gregui/tests/conftest.py | 1 +
2 files changed, 4 insertions(+)
diff --git a/gregui/tests/api/views/test_userinfo.py b/gregui/tests/api/views/test_userinfo.py
index c5caaa30..6b9a2125 100644
--- a/gregui/tests/api/views/test_userinfo.py
+++ b/gregui/tests/api/views/test_userinfo.py
@@ -19,6 +19,7 @@ def test_userinfo_invited_get(client, invitation_link):
response = client.get(reverse("api-userinfo"))
assert response.status_code == status.HTTP_200_OK
assert response.json() == {
+ "auth_type": "invite",
"feide_id": None,
"sponsor_id": None,
"person_id": 1,
@@ -51,6 +52,7 @@ def test_userinfo_sponsor_get(client, log_in, user_sponsor):
response = client.get(reverse("api-userinfo"))
assert response.status_code == status.HTTP_200_OK
assert response.json() == {
+ "auth_type": "oidc",
"feide_id": "",
"person_id": None,
"roles": [],
@@ -65,6 +67,7 @@ def test_userinfo_guest_get(client, log_in, user_person):
response = client.get(reverse("api-userinfo"))
assert response.status_code == status.HTTP_200_OK
assert response.json() == {
+ "auth_type": "oidc",
"feide_id": "",
"sponsor_id": None,
"person_id": 1,
diff --git a/gregui/tests/conftest.py b/gregui/tests/conftest.py
index f30b5e00..01cff9ed 100644
--- a/gregui/tests/conftest.py
+++ b/gregui/tests/conftest.py
@@ -458,6 +458,7 @@ def log_in(client) -> Callable[[UserModel], APIClient]:
# It seems like the session was not updated automatically this way
session = client.session
session["oidc_id_token_payload"] = {"iat": time.time()}
+ session["oidc_states"] = {}
session.save()
return client
--
GitLab