import unittest
from selenium import webdriver

import time


class ExMailTest(unittest.TestCase):
    username = ''
    password = ''

    def setUp(self):
        self.driver = webdriver.Chrome()
        self.driver.maximize_window()
        self.driver.get('https://mail.ex.ua/')

    def test_login(self):
        self.driver.find_element_by_xpath("//input[@name='login']").send_keys(self.username)
        self.driver.find_element_by_css_selector('.fex-form__group > input[name="password"]').send_keys(self.password)
        self.driver.find_element_by_css_selector('.fex-login__submitBtn').click()
        assert self.driver.find_element_by_css_selector('.ib_user_name').text == self.username + "@ex.ua"

    def tearDown(self):
        self.driver.quit()


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