Ejemplo n.º 1
0
def assert_start_page(web_driver: webdriver):
    """Test the MyakuWeb start page with the web driver."""
    viewport_width = web_driver.get_window_size()['width']
    _go_to_start_page(web_driver)

    assert_css_loaded(web_driver)
    assert_element_by_tag(web_driver, 'title', 'Myaku', False)

    header_element = web_driver.find_element_by_tag_name('header')
    assert_search_header(header_element, viewport_width)

    main_element = web_driver.find_element_by_tag_name('main')
    assert_start_tiles(main_element)
Ejemplo n.º 2
0
    def get_screenshot(self, name: str, driver: webdriver, dictionary_path: str) -> None:
        original_size = driver.get_window_size()
        required_width = driver.execute_script('return document.body.parentNode.scrollWidth')
        required_height = driver.execute_script('return document.body.parentNode.scrollHeight')
        driver.set_window_size(required_width, required_height)

        if not os.path.exists(dictionary_path):
            os.makedirs(dictionary_path)

        t = time.localtime()
        current_time = time.strftime("%H-%M-%S", t)
        path = f"{dictionary_path}/screenshot_{name}_{current_time}.png"
        time.sleep(1)
        driver.get_screenshot_as_file(path)
        time.sleep(1)
        driver.set_window_size(original_size['width'], original_size['height'])