diff --git a/nextcloud_create_doc_and_share.py b/nextcloud_create_doc_and_share.py index 19b8d687..ee4e5f93 100644 --- a/nextcloud_create_doc_and_share.py +++ b/nextcloud_create_doc_and_share.py @@ -1,43 +1,65 @@ -from time import time_ns +import contextlib +import sys +from time import sleep, time_ns -from playwright.sync_api import Playwright, sync_playwright +from playwright.sync_api import Playwright, sync_playwright, Error def log_note(message: str) -> None: timestamp = str(time_ns())[:16] print(f"{timestamp} {message}") -def run(playwright: Playwright) -> None: - log_note("Launch browser") - browser = playwright.chromium.launch(headless=True) +def run(playwright: Playwright, browser_name: str) -> None: + log_note(f"Launch browser {browser_name}") + if browser_name == "firefox": + browser_type = playwright.firefox + else: + browser_type = playwright.chromium + browser = browser_type.launch(headless=True) context = browser.new_context() page = context.new_page() - log_note("Login") - page.goto("http://nc/") - page.get_by_label("Account name or email").click() - page.get_by_label("Account name or email").fill("Crash") - page.get_by_label("Account name or email").press("Tab") - page.get_by_label("Password", exact=True).fill("Override") - page.get_by_label("Password", exact=True).press("Enter") - log_note("Create new text file") - page.get_by_role("link", name="Files").click() - page.get_by_role("link", name="New file/folder menu").click() - page.get_by_role("link", name="New text file").click() - page.locator("#view9-input-file").fill("colab_meeting.md") - page.locator("#view9-input-file").press("Enter") - page.get_by_role("button", name="Create a new file with the selected template").click() - page.get_by_role("button", name="Close modal").click() - log_note("Share file with other user") - page.get_by_role("link", name="colab_meeting .md").get_by_role("link", name="Share").click() - page.get_by_placeholder("Name, email, or Federated Cloud ID …").click() - page.get_by_placeholder("Name, email, or Federated Cloud ID …").fill("docs") - page.get_by_text("docs_dude").first.click() - log_note("Close browser") - page.close() + try: + log_note("Login") + page.goto("http://nc/") + page.get_by_label("Account name or email").click() + page.get_by_label("Account name or email").fill("Crash") + page.get_by_label("Account name or email").press("Tab") + page.get_by_label("Password", exact=True).fill("Override") + page.get_by_label("Password", exact=True).press("Enter") + log_note("Create new text file") + page.get_by_role("link", name="Files").click() + page.get_by_role("link", name="New file/folder menu").click() + page.get_by_role("link", name="New text file").click() + page.locator("#view9-input-file").fill("colab_meeting.md") + page.locator("#view9-input-file").press("Enter") + page.get_by_role("button", name="Create a new file with the selected template").click() + sleep(5) + with contextlib.suppress(Exception): + page.get_by_role("button", name="Close modal").click(timeout=15_000) + page.keyboard.press("Escape") + log_note("Share file with other user") + page.get_by_role("link", name="colab_meeting .md").get_by_role("link", name="Share").click() + page.get_by_placeholder("Name, email, or Federated Cloud ID …").click() + page.get_by_placeholder("Name, email, or Federated Cloud ID …").fill("docs") + page.get_by_text("docs_dude").first.click() + log_note("Close browser") + page.close() - # --------------------- - context.close() - browser.close() + # --------------------- + context.close() + browser.close() + except Error as e: + log_note(f"Exception occurred: {e.message}") + log_note(f"Page content was: {page.content()}") + raise e with sync_playwright() as playwright: - run(playwright) + if len(sys.argv) > 1: + 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" + + run(playwright, browser_name) diff --git a/nextcloud_create_user.py b/nextcloud_create_user.py index 51c32121..516e1afc 100644 --- a/nextcloud_create_user.py +++ b/nextcloud_create_user.py @@ -1,43 +1,61 @@ +import sys import contextlib from time import time_ns -from playwright.sync_api import Playwright, sync_playwright +from playwright.sync_api import Playwright, sync_playwright, Error def log_note(message: str) -> None: timestamp = str(time_ns())[:16] print(f"{timestamp} {message}") -def create_user(playwright: Playwright, username: str, password: str) -> None: - log_note("Launch browser") - browser = playwright.chromium.launch(headless=True) +def create_user(playwright: Playwright, browser_name: str, username: str, password: str) -> None: + log_note(f"Launch browser {browser_name}") + if browser_name == "firefox": + browser_type = playwright.firefox + else: + browser_type = playwright.chromium + browser = browser_type.launch(headless=True) context = browser.new_context() - page = context.new_page() - log_note("Login") - page.goto("http://nc/") - page.get_by_label("Account name or email").click() - page.get_by_label("Account name or email").fill("Crash") - page.get_by_label("Account name or email").press("Tab") - page.get_by_label("Password", exact=True).fill("Override") - page.get_by_label("Password", exact=True).press("Enter") - with contextlib.suppress(Exception): - page.get_by_role("button", name="Close modal").click(timeout=15_000) - log_note("Create user") - page.get_by_role("link", name="Open settings menu").click() - page.get_by_role("link", name="Users").click() - page.get_by_role("button", name="New user").click() - page.get_by_placeholder("Username").click() - page.get_by_placeholder("Username").fill(username) - page.get_by_placeholder("Username").press("Tab") - page.get_by_placeholder("Display name").press("Tab") - page.get_by_placeholder("Password", exact=True).fill(password) - page.get_by_role("button", name="Add a new user").click() - log_note("Close browser") + try: + page = context.new_page() + log_note("Login") + page.goto("http://nc/") + page.get_by_label("Account name or email").click() + page.get_by_label("Account name or email").fill("Crash") + page.get_by_label("Account name or email").press("Tab") + page.get_by_label("Password", exact=True).fill("Override") + page.get_by_label("Password", exact=True).press("Enter") + with contextlib.suppress(Exception): + page.get_by_role("button", name="Close modal").click(timeout=15_000) + log_note("Create user") + page.get_by_role("link", name="Open settings menu").click() + page.get_by_role("link", name="Users").click() + page.get_by_role("button", name="New user").click() + page.get_by_placeholder("Username").click() + page.get_by_placeholder("Username").fill(username) + page.get_by_placeholder("Username").press("Tab") + page.get_by_placeholder("Display name").press("Tab") + page.get_by_placeholder("Password", exact=True).fill(password) + page.get_by_role("button", name="Add a new user").click() + log_note("Close browser") - # --------------------- - page.close() - context.close() - browser.close() + # --------------------- + page.close() + context.close() + browser.close() + except Error as e: + log_note(f"Exception occurred: {e.message}") + log_note(f"Page content was: {page.content()}") + raise e with sync_playwright() as playwright: - create_user(playwright, username="docs_dude", password="docsrule!12") + if len(sys.argv) > 1: + 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" + + create_user(playwright, browser_name, username="docs_dude", password="docsrule!12") diff --git a/nextcloud_docs_collaboration.py b/nextcloud_docs_collaboration.py index bf53c769..3b735680 100644 --- a/nextcloud_docs_collaboration.py +++ b/nextcloud_docs_collaboration.py @@ -1,8 +1,10 @@ +import contextlib import random import string +import sys from time import time_ns, sleep -from playwright.sync_api import Playwright, sync_playwright, expect +from playwright.sync_api import Playwright, sync_playwright, expect, Error def get_random_text() -> str: @@ -14,58 +16,72 @@ def log_note(message: str) -> None: timestamp = str(time_ns())[:16] print(f"{timestamp} {message}") -def collaborate(playwright: Playwright) -> None: - log_note("Launching browsers") - browser = playwright.chromium.launch(headless=True) +def collaborate(playwright: Playwright, browser_name: str) -> None: + log_note(f"Launch two {browser_name} browsers") + if browser_name == "firefox": + browser_type = playwright.firefox + else: + browser_type = playwright.chromium + + browser = browser_type.launch(headless=True) context = browser.new_context() admin_user = context.new_page() - browser_two = playwright.chromium.launch(headless=True) + browser_two = browser_type.launch(headless=True) context_two = browser_two.new_context() docs_user = context_two.new_page() - # Login and open the file for both users - log_note("Logging in with users") - login(admin_user, "Crash", "Override") - login(docs_user, "docs_dude", "docsrule!12") - log_note("Opening document with both users") - docs_user.get_by_role("button", name="Close modal").click(timeout=15_000) - admin_user.get_by_role("link", name="Files").click() - docs_user.get_by_role("link", name="Files").click() - admin_user.get_by_role("link", name="Shares").click() - docs_user.get_by_role("link", name="Shares").click() - admin_user.get_by_role("link", name="colab_meeting .md").click() - docs_user.get_by_role("link", name="colab_meeting .md").click() + try: + # Login and open the file for both users + log_note("Logging in with users") + login(admin_user, "Crash", "Override") + login(docs_user, "docs_dude", "docsrule!12") + log_note("Opening document with both users") + with contextlib.suppress(Exception): + sleep(5) + docs_user.get_by_role("button", name="Close modal").click(timeout=15_000) + admin_user.get_by_role("link", name="Files").click() + docs_user.get_by_role("link", name="Files").click() + admin_user.get_by_role("link", name="Shares").click() + docs_user.get_by_role("link", name="Shares").click() + admin_user.get_by_role("link", name="colab_meeting .md").click() + docs_user.get_by_role("link", name="colab_meeting .md").click() - # Write the first message and assert it's visible for the other user - log_note("Sending first validation message") - first_message = "FIRST_VALIDATION_MESSAGE" - admin_user.get_by_role("dialog", name="colab_meeting.md").get_by_role("document").locator("div").first.type(first_message) - expect(docs_user.get_by_text(first_message)).to_be_visible() + # Write the first message and assert it's visible for the other user + log_note("Sending first validation message") + first_message = "FIRST_VALIDATION_MESSAGE" + admin_user.get_by_role("dialog", name="colab_meeting.md").get_by_role("document").locator("div").first.type(first_message) + expect(docs_user.get_by_text(first_message)).to_be_visible() - for x in range(1, 7): - random_message = get_random_text() - # Admin sends on even, docs_dude on odd - if x % 2 == 0: - log_note("Admin adding text") - admin_user.get_by_role("dialog", name="colab_meeting.md").get_by_role("document").locator("div").first.type(random_message) - expect(docs_user.get_by_text(random_message)).to_be_visible() - else: - log_note("User adding text") - docs_user.get_by_role("dialog", name="colab_meeting.md").get_by_role("document").locator("div").first.type(random_message) - expect(admin_user.get_by_text(random_message)).to_be_visible() + for x in range(1, 7): + random_message = get_random_text() + # Admin sends on even, docs_dude on odd + if x % 2 == 0: + log_note("Admin adding text") + admin_user.get_by_role("dialog", name="colab_meeting.md").get_by_role("document").locator("div").first.type(random_message) + expect(docs_user.get_by_text(random_message)).to_be_visible() + else: + log_note("User adding text") + docs_user.get_by_role("dialog", name="colab_meeting.md").get_by_role("document").locator("div").first.type(random_message) + expect(admin_user.get_by_text(random_message)).to_be_visible() - log_note("Sleeping for 10 seconds") - sleep(10) + log_note("Sleeping for 10 seconds") + sleep(10) - log_note("Closing browsers") - # --------------------- - admin_user.close() - docs_user.close() - context.close() - context_two.close() - browser.close() - browser_two.close() + log_note("Closing browsers") + # --------------------- + admin_user.close() + docs_user.close() + context.close() + context_two.close() + browser.close() + browser_two.close() + + except Error as e: + log_note(f"Exception occurred: {e.message}") + log_note(f"Page content was: {docs_user.content()}") + log_note(f"Page content was: {admin_user.content()}") + raise e def login(page, username, password): page.goto("http://nc/login") @@ -77,4 +93,12 @@ def login(page, username, password): with sync_playwright() as playwright: - collaborate(playwright) + if len(sys.argv) > 1: + 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" + + collaborate(playwright, browser_name) diff --git a/nextcloud_install.py b/nextcloud_install.py index e1f124cd..de5b470e 100644 --- a/nextcloud_install.py +++ b/nextcloud_install.py @@ -36,14 +36,13 @@ def main(browser_name: str = "chromium"): # 3. Dashboard page.wait_for_selector('.app-dashboard') log_note("Installation complete") + browser.close() except Error as e: log_note(f"Exception occurred: {e.message}") log_note(f"Page content was: {page.content()}") raise e - browser.close() - if __name__ == '__main__': if len(sys.argv) > 1: