e2e_tests/src/tests_authentik/setup_authentik.py
2023-11-24 22:14:34 +01:00

115 lines
4.6 KiB
Python

import json
import os
import re
from pathlib import Path
from playwright.sync_api import BrowserContext, expect
ADMIN_USER = os.environ["ADMIN_USER"]
ADMIN_PASS = os.environ["ADMIN_PASS"]
TESTUSER = {"username": "testuser", "name": "Test User", "password": "test123", "email": "test@example.com"}
TIMEOUT = 5000
def test_create_admin_login(context: BrowserContext, dotenv_config: dict[str, str], STATES: Path):
# go to page
page = context.new_page()
url = "https://" + dotenv_config["DOMAIN"]
page.goto(url)
# check welcome message
welcome_message = dotenv_config.get("welcome_message")
if welcome_message:
expect(page.get_by_text(welcome_message)).to_be_visible(timeout=TIMEOUT)
# login
page.locator('input[name="uidField"]').fill(ADMIN_USER)
page.locator('ak-stage-identification input[name="password"]').fill(ADMIN_PASS)
page.get_by_role("button", name="Log In").click()
expect(page.locator("ak-library")).to_be_visible(timeout=TIMEOUT)
# save state
context.storage_state(path=f"{STATES}/admin_state.json")
def create_invite_link(admin_context: BrowserContext, dotenv_config: dict[str, str]):
page = admin_context.new_page()
url = "https://" + dotenv_config["DOMAIN"]
page.goto(url)
page.get_by_role("link", name="Admin Interface").click()
# regex approach (shorter)
page.get_by_role("button", name=re.compile(r"Directory|Verzeichnis")).click()
# the or approach
# button_dir_en = page.get_by_role("button", name="Directory")
# page.get_by_role("button").get_by_text(re.compile("Verzeichnis|Directory")).click()
# button_dir_de = page.get_by_text("Verzeichnis")
# button_dir_en.or_(button_dir_de).click()
nav = page.locator("ak-sidebar-item", has_text=re.compile(r"Directory|Verzeichnis"))
nav.get_by_role("link", name=re.compile(r"Users|Benutzer")).click()
nav = page.locator("ak-sidebar-item", has_text="Directory")
nav.get_by_role("link", name=re.compile(r"Invitations|Einladungen")).click()
# button_invitation_en = page.get_by_role("link", name="Invitations")
# button_invitation_de = page.get_by_text("Einladungen")
# button_invitation_en.or_(button_invitation_de).click()
page.get_by_role("cell", name=re.compile(r"Keine Objekte|objects")).get_by_role(
"button"
).click() # todo: confirm "objects" for en lang
page.locator('input[name="name"]').click()
linkname = "test_link_123"
page.locator('input[name="name"]').fill(linkname)
page.get_by_placeholder("Wählen Sie ein Objekt aus.").click()
page.get_by_role("option", name=re.compile(r"invitation-enrollment-flow")).click()
page.get_by_role("dialog").get_by_role("button", name=re.compile(r"Erstellen|create")).click() # confirm "create"
# page.get_by_text("Erstellen", exact=True).first.click()
linklocator = page.get_by_role("rowgroup").filter(has=page.get_by_text(linkname))
linklocator.locator(".fa-angle-down").click()
# page.get_by_text(linkname).click()
invitelink = linklocator.get_by_role("textbox").get_attribute(name="value")
return invitelink
def create_user(context: BrowserContext, invitelink, STATES: Path):
# warning: only works on german site
page = context.new_page()
page.goto(invitelink)
page.get_by_placeholder("Benutzername").click()
page.get_by_placeholder("Benutzername").fill(TESTUSER["username"])
page.locator('input[name="name"]').click()
page.locator('input[name="name"]').fill(TESTUSER["name"])
page.locator('input[name="email"]').click()
page.locator('input[name="email"]').fill(TESTUSER["email"])
page.get_by_placeholder("Passwort", exact=True).click()
page.get_by_placeholder("Passwort", exact=True).fill(TESTUSER["password"])
page.get_by_placeholder("Passwort (wiederholen)").click()
page.get_by_placeholder("Passwort (wiederholen)").fill(TESTUSER["password"])
page.get_by_role("button", name="Weiter").click()
expect(page.locator("ak-library")).to_be_visible()
def test_create_user_session(context: BrowserContext, dotenv_config: dict[str, str], STATES: Path):
context.set_default_timeout(TIMEOUT)
# load admin cookies
state_file = STATES / "admin_state.json"
storage_state = json.loads(state_file.read_bytes())
context.add_cookies(storage_state["cookies"])
# create invite_link
invite_link = create_invite_link(context, dotenv_config)
# create user
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")