from selenium import webdriver
import time

def test_check_error_page_1():
    try:
        link = "http://suninjuly.github.io/registration1.html"
        browser = webdriver.Chrome()
        browser.get(link)

        # Проверяем Fisrt name*
        field_text = browser.find_element_by_xpath(
            '//body/div/form/div[@class="first_block"]/div[@class="form-group first_class"]/label').text
        first_input = browser.find_element_by_xpath(
            '//body/div/form/div[@class="first_block"]/div[@class="form-group first_class"]/input')
        first_input.send_keys('Иван')

        # assert field_text=='First name*'

        # Проверяем Last name*
        field_text = browser.find_element_by_xpath(
            '//body/div/form/div[@class="first_block"]/div[@class="form-group second_class"]/label').text
        first_input = browser.find_element_by_xpath(
            '//body/div/form/div[@class="first_block"]/div[@class="form-group second_class"]/input')
        first_input.send_keys('Петров')

        # assert field_text == 'Last name*'

        # Проверяем Email*
        field_text = browser.find_element_by_xpath(
            '//body/div/form/div[@class="first_block"]/div[@class="form-group third_class"]/label').text
        first_input = browser.find_element_by_xpath(
            '//body/div/form/div[@class="first_block"]/div[@class="form-group third_class"]/input')
        first_input.send_keys('email@email.test')

        # assert field_text == 'Email*'

        # Отправляем заполненную форму
        button = browser.find_element_by_css_selector("button.btn")
        button.click()

        # Проверяем, что смогли зарегистрироваться
        # ждем загрузки страницы
        time.sleep(1)

        # находим элемент, содержащий текст
        welcome_text_elt = browser.find_element_by_tag_name("h1")
        # записываем в переменную welcome_text текст из элемента welcome_text_elt
        welcome_text = welcome_text_elt.text

        # с помощью assert проверяем, что ожидаемый текст совпадает с текстом на странице сайта
        assert "Congratulations! You have successfully registered!" == welcome_text

    finally:
        # ожидание чтобы визуально оценить результаты прохождения скрипта
        time.sleep(10)
        # закрываем браузер после всех манипуляций
        browser.quit()

def test_check_error_page_2():
    try:
        link = "http://suninjuly.github.io/registration2.html"
        browser = webdriver.Chrome()
        browser.get(link)

        # Проверяем Fisrt name*
        field_text = browser.find_element_by_xpath(
            '//body/div/form/div[@class="first_block"]/div[@class="form-group first_class"]/label').text
        first_input = browser.find_element_by_xpath(
            '//body/div/form/div[@class="first_block"]/div[@class="form-group first_class"]/input')
        first_input.send_keys('Иван')

        # assert field_text=='First name*'

        # Проверяем Last name*
        field_text = browser.find_element_by_xpath(
            '//body/div/form/div[@class="first_block"]/div[@class="form-group second_class"]/label').text
        first_input = browser.find_element_by_xpath(
            '//body/div/form/div[@class="first_block"]/div[@class="form-group second_class"]/input')
        first_input.send_keys('Петров')

        # assert field_text == 'Last name*'

        # Проверяем Email*
        field_text = browser.find_element_by_xpath(
            '//body/div/form/div[@class="first_block"]/div[@class="form-group third_class"]/label').text
        first_input = browser.find_element_by_xpath(
            '//body/div/form/div[@class="first_block"]/div[@class="form-group third_class"]/input')
        first_input.send_keys('email@email.test')

        # assert field_text == 'Email*'

        # Отправляем заполненную форму
        button = browser.find_element_by_css_selector("button.btn")
        button.click()

        # Проверяем, что смогли зарегистрироваться
        # ждем загрузки страницы
        time.sleep(1)

        # находим элемент, содержащий текст
        welcome_text_elt = browser.find_element_by_tag_name("h1")
        # записываем в переменную welcome_text текст из элемента welcome_text_elt
        welcome_text = welcome_text_elt.text

        # с помощью assert проверяем, что ожидаемый текст совпадает с текстом на странице сайта
        assert "Congratulations! You have successfully registered!" == welcome_text

    finally:
        # ожидание чтобы визуально оценить результаты прохождения скрипта
        time.sleep(10)
        # закрываем браузер после всех манипуляций
        browser.quit()