import unittest
from selenium import webdriver
from login_test import NavBar


class TestLogin(unittest.TestCase):

    def test_login(self):
        # 1、打开浏览器
        driver = webdriver.Chrome()
        # 2、打开登录网址
        driver.get("http://39.107.96.138:3000/")
        # 3、导航到登录页面
        navBar = NavBar(driver)
        loginPage = navBar.go_to_login_page()
        # 4、使用正确的用户名和密码登录
        loginPage.login_with_username_pass(username="imtest11", passwd="123456")
        # 5.1、登录成功  期望页面跳转到首页
        URL = driver.current_url
        self.assertEqual(URL, "http://39.107.96.138:3000/")
        # 5.2、登录成功  期望首页页面上个人信息栏显示huihui
        username = navBar.go_to_username_page()
        self.assertEqual(username, 'imtest11')


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