118 lines
4.5 KiB
Python
118 lines
4.5 KiB
Python
import json
|
|
import os
|
|
import re
|
|
|
|
from playwright.sync_api import BrowserContext, expect
|
|
|
|
from pytest_abra.dir_manager import DirManager
|
|
from pytest_abra.utils import BaseUrl
|
|
|
|
ADMIN_USER = os.environ["ADMIN_USER"]
|
|
ADMIN_PASS = os.environ["ADMIN_PASS"]
|
|
|
|
|
|
TESTUSER = {"username": "testuser", "name": "Test User", "password": "test123", "email": "test@example.com"}
|
|
|
|
|
|
def setup_admin_state(context: BrowserContext, env_config: dict[str, str], DIR: DirManager):
|
|
# go to page
|
|
page = context.new_page()
|
|
url = "https://" + env_config["DOMAIN"]
|
|
page.goto(url)
|
|
|
|
# check welcome message
|
|
welcome_message = env_config.get("welcome_message")
|
|
if welcome_message:
|
|
expect(page.get_by_text(welcome_message)).to_be_visible()
|
|
|
|
# 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()
|
|
|
|
# save state
|
|
context.storage_state(path=DIR.STATES / "authentik_admin_state.json")
|
|
|
|
|
|
def check_if_user_exists(admin_context: BrowserContext, env_config: dict[str, str], URL: BaseUrl):
|
|
# go to admin page
|
|
page = admin_context.new_page()
|
|
page.goto(URL.get())
|
|
page.get_by_role("link", name="Admin Interface").click()
|
|
nav = page.locator("ak-sidebar-item", has_text=re.compile(r"Directory|Verzeichnis"))
|
|
nav.click()
|
|
nav.get_by_role("link", name=re.compile(r"Users|Benutzer")).click()
|
|
|
|
user = page.get_by_text(TESTUSER["username"])
|
|
user.wait_for(state="visible")
|
|
return user.is_visible()
|
|
|
|
|
|
def create_invite_link(admin_context: BrowserContext, env_config: dict[str, str], URL: BaseUrl):
|
|
# go to admin page
|
|
page = admin_context.new_page()
|
|
page.goto(URL.get())
|
|
page.get_by_role("link", name="Admin Interface").click()
|
|
|
|
nav = page.locator("ak-sidebar-item", has_text=re.compile(r"Directory|Verzeichnis"))
|
|
nav.click()
|
|
nav.get_by_role("link", name=re.compile(r"Invitations|Einladungen")).click()
|
|
|
|
# todo: only works if no links have been created yet (empty list)
|
|
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()
|
|
|
|
# force, because else we get "intercepts pointer events"
|
|
page.locator("footer").locator("ak-spinner-button").first.click(force=True)
|
|
|
|
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(user_context: BrowserContext, invitelink):
|
|
# warning: only works on german site
|
|
page = user_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 setup_user_state(context: BrowserContext, env_config: dict[str, str], DIR: DirManager, URL: BaseUrl):
|
|
# load admin cookies to context
|
|
state_file = DIR.STATES / "authentik_admin_state.json"
|
|
storage_state = json.loads(state_file.read_bytes())
|
|
context.add_cookies(storage_state["cookies"])
|
|
|
|
if check_if_user_exists(context, env_config, URL):
|
|
# just login with user
|
|
pass
|
|
context.clear_cookies()
|
|
else:
|
|
# get invite_link
|
|
invite_link = create_invite_link(context, env_config, URL)
|
|
# create user
|
|
context.clear_cookies()
|
|
create_user(context, invite_link)
|
|
|
|
context.storage_state(path=DIR.STATES / "authentik_user_state.json")
|