0
0
Fork 0
mirror of https://github.com/nextcloud/docker.git synced 2025-03-25 23:46:09 +01:00
nextcloud-docker/energy-tests/nextcloud_docs_collaboration.py
2023-12-30 13:56:33 +01:00

113 lines
4.8 KiB
Python

import contextlib
import random
import string
import sys
from time import time_ns, sleep
from playwright.sync_api import Playwright, sync_playwright, expect, TimeoutError
def get_random_text() -> str:
size_in_bytes = 1024
characters = string.ascii_letters + string.digits
return ''.join(random.choice(characters) for _ in range(size_in_bytes))
def log_note(message: str) -> None:
timestamp = str(time_ns())[:16]
print(f"{timestamp} {message}")
def collaborate(playwright: Playwright, browser_name: str) -> None:
log_note(f"Launch two {browser_name} browsers")
if browser_name == "firefox":
browser = playwright.firefox.launch(headless=True)
else:
# this leverages new headless mode by Chromium: https://developer.chrome.com/articles/new-headless/
# The mode is however ~40% slower: https://github.com/microsoft/playwright/issues/21216
browser = playwright.chromium.launch(headless=False,args=["--headless=new"])
context = browser.new_context()
admin_user = context.new_page()
if browser_name == "firefox":
browser_two = playwright.firefox.launch(headless=True)
else:
# this leverages new headless mode by Chromium: https://developer.chrome.com/articles/new-headless/
# The mode is however ~40% slower: https://github.com/microsoft/playwright/issues/21216
browser_two = playwright.chromium.launch(headless=False,args=["--headless=new"])
context_two = browser_two.new_context()
docs_user = context_two.new_page()
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(TimeoutError):
sleep(5)
docs_user.locator('button.first-run-wizard__close-button').click(timeout=15_000)
admin_user.get_by_role("link", name="Files", exact=True).click()
docs_user.get_by_role("link", name="Files", exact=True).click()
admin_user.get_by_role("link", name="Shares", exact=True).click()
docs_user.get_by_role("link", name="Shares", exact=True).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()
log_note("GMT_SCI_R=1")
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(timeout=15_000)
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(timeout=15_000)
log_note("GMT_SCI_R=1")
log_note("Sleeping for 5 seconds")
sleep(5)
log_note("Closing browsers")
# ---------------------
admin_user.close()
docs_user.close()
context.close()
context_two.close()
browser.close()
browser_two.close()
except Exception as e:
if hasattr(e, 'message'): # only Playwright error class has this member
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")
page.get_by_label("Account name or email").click()
page.get_by_label("Account name or email").fill(username)
page.get_by_label("Account name or email").press("Tab")
page.get_by_label("Password", exact=True).fill(password)
page.get_by_label("Password", exact=True).press("Enter")
with sync_playwright() as 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)