import os

from selenium.webdriver.common.by import By

from screenshot_tests.utils.screenshots import TestCase
import time


class TestExample(TestCase):
    def test_main_page(self):
        self.driver.get("https://s57map.ivazh.ru/")

        def action():
            time.sleep(2)

        self.check_by_screenshot(None, full_page=True, action=action)

    def test_search_block(self):
        self.driver.get("https://s57.ivazh.ru/styles/r14/#12.13/71.2832/72.13405")

        def action():
            time.sleep(2)

        self.check_by_screenshot_file(None, screenshot_path=f'test_search_block.png',
                                      full_page=True, action=action)

    def test_save_screenshot(self):
        self.driver.get("https://s57.ivazh.ru/styles/r14/#12.13/71.2832/72.13405")

        def action():
            time.sleep(2)

        self.save_screenshot('test_search_block.png', full_page=True, action=action)