# -*- coding: utf-8 -*-
from selenium import webdriver
from selenium.webdriver.common.keys import Keys
import unittest

class UnitTest(unittest.TestCase):

    def setUp(self):
        self.driver = webdriver.Firefox()   #инициализируем браузер
        self.driver.implicitly_wait(5)  #ожидаем загрузки браузера в течении 5 секунд

    def test_case0(self):
        driver = self.driver    #объявляем переменную
        driver.get("https://www.yandex.ru") #переходим на страницу

        self.assertTrue("Яндекс" in driver.title)   #проверяем имя страницы, которая загрузилась после запуска браузера

        driver.find_element_by_name("text").click()   #находим элемент страницы по имени и кликаем по нему
        driver.find_element_by_name("text").send_keys("YouTube" + Keys.ENTER) #находим элемент страницы по имени и вводим поисковой запрос, нажимаем клавишу Enter
        driver.find_element_by_css_selector("b.needsclick").click() #находим элемент среди результатов поиска по css и кликаем по нему

        driver.quit()   #закрываем браузер

if __name__ == "__main__":
    unittest.main()