import os

from selenium import webdriver
from selenium.webdriver.chrome.service import Service
from selenium.webdriver.common.by import By


def test_homepage_loads():
    app_url = os.environ["APP_URL"]
    expected_title = os.environ["EXPECTED_TITLE"]

    options = webdriver.ChromeOptions()
    options.add_argument("--headless=new")
    options.add_argument("--window-size=1280,720")
    options.add_argument("--disable-dev-shm-usage")
    if os.environ.get("CHROME_NO_SANDBOX") == "1":
        options.add_argument("--no-sandbox")

    chrome_binary = os.environ.get("CHROME_BINARY")
    if chrome_binary:
        options.binary_location = chrome_binary

    service_path = os.environ.get("CHROMEDRIVER")
    service = Service(service_path) if service_path else None

    driver = webdriver.Chrome(service=service, options=options)
    try:
        driver.get(app_url)
        assert expected_title in driver.title
        assert driver.find_element(By.TAG_NAME, "body").is_displayed()
    finally:
        driver.quit()