diff --git a/authentik/core/tests/test_account_switch.py b/authentik/core/tests/test_account_switch.py index cd80701abe..7f32fa5eae 100644 --- a/authentik/core/tests/test_account_switch.py +++ b/authentik/core/tests/test_account_switch.py @@ -14,15 +14,15 @@ from authentik.core.tests.utils import ( create_test_user, ) from authentik.core.views.account_switch import QS_ACCOUNT_SWITCH_STALE +from authentik.events.models import Event, EventAction +from authentik.flows.markers import StageMarker +from authentik.flows.models import Flow, FlowDesignation, FlowStageBinding from authentik.flows.planner import ( PLAN_CONTEXT_ACCOUNT_SWITCH_FROM_USER, PLAN_CONTEXT_IS_ACCOUNT_SWITCH, PLAN_CONTEXT_PENDING_USER, FlowPlan, ) -from authentik.events.models import Event, EventAction -from authentik.flows.markers import StageMarker -from authentik.flows.models import Flow, FlowDesignation, FlowStageBinding from authentik.flows.tests import FlowTestCase from authentik.flows.views.executor import SESSION_KEY_PLAN from authentik.root.middleware import COOKIE_NAME_ACCOUNTS diff --git a/authentik/core/tests/test_sessions.py b/authentik/core/tests/test_sessions.py index 0f7eecc3d9..ec3e695d1c 100644 --- a/authentik/core/tests/test_sessions.py +++ b/authentik/core/tests/test_sessions.py @@ -44,7 +44,9 @@ class TestSessionSuperseding(TestCase): """Test a new login marks the browser's previous logins as not current""" browser_key = get_random_string(BROWSER_KEY_LENGTH) previous = create_test_session(self.user, browser_key=browser_key) - other_browser = create_test_session(self.user, browser_key=get_random_string(BROWSER_KEY_LENGTH)) + other_browser = create_test_session( + self.user, browser_key=get_random_string(BROWSER_KEY_LENGTH) + ) request = RequestFactory().get("/") request.browser_key = browser_key diff --git a/authentik/core/views/account_switch.py b/authentik/core/views/account_switch.py index 97652bfd6d..9d3b505792 100644 --- a/authentik/core/views/account_switch.py +++ b/authentik/core/views/account_switch.py @@ -58,9 +58,7 @@ class AccountSwitchView(View): response = plan.to_redirect(request, flow) if stale_user_uid: separator = "&" if "?" in response["Location"] else "?" - response["Location"] += separator + urlencode( - {QS_ACCOUNT_SWITCH_STALE: stale_user_uid} - ) + response["Location"] += separator + urlencode({QS_ACCOUNT_SWITCH_STALE: stale_user_uid}) return response @staticmethod