0
0
Fork 0
mirror of https://github.com/nextcloud/docker.git synced 2025-04-20 02:46:10 +02:00
nextcloud-docker/energy-tests/nextcloud_install.py
2024-10-16 13:17:52 +02:00

62 lines
2.4 KiB
Python

import sys
import signal
from time import time_ns
from playwright.sync_api import sync_playwright
from helpers.helper_functions import log_note, get_random_text, login_nextcloud, close_modal, timeout_handler
def main(browser_name: str = "chromium"):
with sync_playwright() as playwright:
log_note(f"Launch browser {browser_name}")
signal.signal(signal.SIGALRM, timeout_handler)
signal.alarm(10)
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()
page = context.new_page()
signal.alarm(0) # remove timeout signal
try:
page.goto('http://nc/')
# 1. Create User
log_note("Create admin user")
page.locator('#adminlogin').fill('Crash')
page.locator('#adminpass').fill('Override')
page.locator('.primary').click()
# 2. Install all Apps
log_note("Install recommended apps")
install_selector = '.button-vue--vue-primary'
page.locator(install_selector).click()
# 3. Dashboard
page.locator('.app-dashboard').wait_for(timeout=240_000)
log_note("Installation complete")
browser.close()
except Exception as e:
if hasattr(e, 'message'): # only Playwright error class has this member
log_note(f"Exception occurred: {e.message}")
# set a timeout. Since the call to page.content() is blocking we need to defer it to the OS
signal.signal(signal.SIGALRM, timeout_handler)
signal.alarm(20)
log_note(f"Page content was: {page.content()}")
signal.alarm(0) # remove timeout signal
raise e
if __name__ == '__main__':
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"
main(browser_name)