aaron/nextcloud-docker
0
0
Fork 0
mirror of https://github.com/nextcloud/docker.git synced 2025-03-15 19:05:09 +01:00
Code Activity

Support for firefox and wrapper for errors

Browse source
This commit is contained in:
Danilo Ješić 2023-07-11 15:28:45 +02:00
parent 9a9b0f85df
commit 012bc9552b
1 changed files with 70 additions and 44 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

114
nextcloud_talk.py
View file

@ -1,8 +1,10 @@
import contextlib
import random import random
import string import string
from time import time_ns import sys
from time import sleep, time_ns
from playwright.sync_api import Playwright, sync_playwright, expect from playwright.sync_api import Playwright, sync_playwright, expect, Error
def log_note(message: str) -> None: def log_note(message: str) -> None:
timestamp = str(time_ns())[:16] timestamp = str(time_ns())[:16]
@ -19,60 +21,75 @@ def send_message(sender, message):
sender.get_by_role("textbox", name="Write message, @ to mention someone …").fill(message) sender.get_by_role("textbox", name="Write message, @ to mention someone …").fill(message)
sender.get_by_role("textbox", name="Write message, @ to mention someone …").press("Enter") sender.get_by_role("textbox", name="Write message, @ to mention someone …").press("Enter")
def create_conversation(playwright: Playwright) -> str: def create_conversation(playwright: Playwright, browser_name: str) -> str:
headless = True headless = True
log_note("Launch browser") log_note(f"Launch browser {browser_name}")
browser = playwright.chromium.launch(headless=headless) if browser_name == "firefox":
browser = playwright.firefox.launch(headless=headless)
else:
browser = playwright.chromium.launch(headless=headless)
context = browser.new_context() context = browser.new_context()
page = context.new_page() page = context.new_page()
log_note("Login as admin") try:
page.goto("http://nc/") log_note("Login as admin")
page.get_by_label("Account name or email").click() page.goto("http://nc/")
page.get_by_label("Account name or email").fill("Crash") page.get_by_label("Account name or email").click()
page.get_by_label("Account name or email").press("Tab") page.get_by_label("Account name or email").fill("Crash")
page.get_by_label("Password", exact=True).fill("Override") page.get_by_label("Account name or email").press("Tab")
page.get_by_role("button", name="Log in").click() page.get_by_label("Password", exact=True).fill("Override")
page.get_by_role("button", name="Log in").click()
# Wait for the modal to load. As it seems you can't close it while it is showing the opening animation. # Wait for the modal to load. As it seems you can't close it while it is showing the opening animation.
log_note("Close first-time run popup") log_note("Close first-time run popup")
page.get_by_role("button", name="Close modal").click(timeout=15_000) with contextlib.suppress(Exception):
sleep(3)
page.get_by_role("button", name="Close modal").click(timeout=15_000)
log_note("Open Talk app") log_note("Open Talk app")
page.get_by_role("link", name="Talk", exact=True).click() page.get_by_role("link", name="Talk", exact=True).click()
page.wait_for_url("**/apps/spreed/") page.wait_for_url("**/apps/spreed/")
# Headless browsers trigger a warning in Nextcloud, however they actually work fine # Headless browsers trigger a warning in Nextcloud, however they actually work fine
if headless: if headless:
log_note("Close headless warning") log_note("Close headless warning")
page.wait_for_selector('.toast-close') with contextlib.suppress(Exception):
page.click('.toast-close') page.wait_for_selector('.toast-close')
page.click('.toast-close')
page.wait_for_selector('.toast-close') page.wait_for_selector('.toast-close')
page.click('.toast-close') page.click('.toast-close')
log_note("Create conversation") log_note("Create conversation")
page.get_by_role("button", name="Create a new group conversation").click() page.get_by_role("button", name="Create a new group conversation").click()
page.get_by_placeholder("Conversation name").fill("Random talk") page.get_by_placeholder("Conversation name").fill("Random talk")
page.locator("label").filter(has_text="Allow guests to join via link").locator("svg").click() page.locator("label").filter(has_text="Allow guests to join via link").locator("svg").click()
page.get_by_role("button", name="Create conversation").click() page.get_by_role("button", name="Create conversation").click()
page.get_by_role("button", name="Copy conversation link").click() page.get_by_role("button", name="Copy conversation link").click()
log_note("Close browser") log_note("Close browser")
# --------------------- # ---------------------
page.close() page.close()
context.close() context.close()
browser.close() browser.close()
return page.url return page.url
def talk(playwright: Playwright, url: str) -> None: except Error as e:
log_note(f"Exception occurred: {e.message}")
log_note(f"Page content was: {page.content()}")
raise e
def talk(playwright: Playwright, url: str, browser_name: str) -> None:
headless = True headless = True
action_delay_ms = 300 action_delay_ms = 300
browser_count = 5 browser_count = 5
# Launch browsers # Launch browsers
log_note(f"Launching {browser_count} browsers") log_note(f"Launching {browser_count} {browser_name} browsers")
browsers = [playwright.chromium.launch(headless=headless, slow_mo=action_delay_ms) for _ in range(browser_count)] if browser_name == "firefox":
browsers = [playwright.firefox.launch(headless=headless, slow_mo=action_delay_ms) for _ in range(browser_count)]
else:
browsers = [playwright.chromium.launch(headless=headless, slow_mo=action_delay_ms) for _ in range(browser_count)]
contexts = [browser.new_context() for browser in browsers] contexts = [browser.new_context() for browser in browsers]
pages = [context.new_page() for context in contexts] pages = [context.new_page() for context in contexts]
@ -84,8 +101,9 @@ def talk(playwright: Playwright, url: str) -> None:
# Close toast messages for headless browsers # Close toast messages for headless browsers
if headless: if headless:
log_note("Close headless warning") log_note("Close headless warning")
for page in pages: with contextlib.suppress(Exception):
page.wait_for_selector('.toast-close').click() for page in pages:
page.wait_for_selector('.toast-close').click()
# Perform actions for all users # Perform actions for all users
log_note("Set guest usernames") log_note("Set guest usernames")
@ -128,5 +146,13 @@ def talk(playwright: Playwright, url: str) -> None:
with sync_playwright() as playwright: with sync_playwright() as playwright:
conversation_link = create_conversation(playwright) if len(sys.argv) > 1:
talk(playwright, conversation_link) browser_name = sys.argv[1].lower()
if browser_name not in ["chromium", "firefox"]:
print("Invalid browser name. Please choose either 'chromium' or 'firefox'.")
sys.exit(1)
else:
browser_name = "chromium"
conversation_link = create_conversation(playwright, browser_name)
talk(playwright, conversation_link, browser_name)