diff --git a/src/tests_authentik/setup_authentik.py b/src/tests_authentik/setup_authentik.py index 66a9070..4602d2b 100644 --- a/src/tests_authentik/setup_authentik.py +++ b/src/tests_authentik/setup_authentik.py @@ -75,14 +75,30 @@ def create_user(context: BrowserContext, invitelink, STATES: Path): expect(page.locator("ak-library")).to_be_visible() -def test_create_user_session(browser: Browser, dotenv_config: dict[str, str], STATES: Path): +# def test_create_user_session(browser: Browser, dotenv_config: dict[str, str], STATES: Path): +# # create invite_link +# admin_context = browser.new_context(storage_state=f"{STATES}/admin_state.json") +# admin_context.set_default_timeout(TIMEOUT) +# invite_link = create_invite_link(admin_context, dotenv_config) + +# # create user +# user_context = browser.new_context() +# user_context.set_default_timeout(TIMEOUT) +# create_user(user_context, invite_link, STATES) +# user_context.storage_state(path=f"{STATES}/user_state.json") + + +def test_create_user_session_one_context(context: BrowserContext, dotenv_config: dict[str, str], STATES: Path): # create invite_link - admin_context = browser.new_context(storage_state=f"{STATES}/admin_state.json") - admin_context.set_default_timeout(TIMEOUT) - invite_link = create_invite_link(admin_context, dotenv_config) + context.set_default_timeout(TIMEOUT) + storage_state = json.loads(f"{STATES}/admin_state.json") + # admin_context = browser.new_context(storage_state=f"{STATES}/admin_state.json") + context.add_cookies(*storage_state["cookies"]) + invite_link = create_invite_link(context, dotenv_config) # create user - user_context = browser.new_context() - user_context.set_default_timeout(TIMEOUT) - create_user(user_context, invite_link, STATES) - user_context.storage_state(path=f"{STATES}/user_state.json") + context.clear_cookies() + # user_context = browser.new_context() + context.set_default_timeout(TIMEOUT) + create_user(context, invite_link, STATES) + context.storage_state(path=f"{STATES}/user_state.json")