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)
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'])