4.selenium 完成用户登录用例

直接上代码，细节其他篇文章再补，先执行识别用例，最后执行成功用例。
随便先执行哪条用例还得优化。

# coding:utf-8

from selenium import webdriver
from time import sleep
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.support.wait import WebDriverWait


class TestUserLogin(object):
    def __init__(self):
        self.driver = webdriver.Chrome()                                   # 打开谷歌浏览器
        self.driver.get("http://120.25.209.187:12306/jpress/user/login")   # 打开登录页
        self.driver.maximize_window()                                      # 页面最大化

    # 测试用户登录成功
    def test_user_login_success(self):
        # 测试数据：账号、密码、期望结果
        user = 'test'
        pwd = 123456
        expected = '用户中心'

        self.driver.find_element_by_name('user').clear()                  # 清空账号输入框
        self.driver.find_element_by_name('user').send_keys(user)          # 输入框输入账号 test
        self.driver.find_element_by_name('pwd').clear()
        self.driver.find_element_by_name('pwd').send_keys(pwd)
        self.driver.find_element_by_class_name('btn').click()             # 点击登录

        WebDriverWait(self.driver, 5).until(EC.title_is(expected))        # 等待 title 的提示框出现
        sleep(2)
        print(self.driver.title)
        assert self.driver.title == expected                              # 断言，断言 title

        self.driver.quit()                                                # 关闭浏览器

    # 测试账号正确，密码错误
    def test_pws_error(self):
        # 测试数据：账号、密码、期望结果
        user = 'test'
        pwd = 'test'
        expected = '用户名或密码不正确'

        self.driver.find_element_by_name('user').clear()                  # 清空账号输入框
        self.driver.find_element_by_name('user').send_keys(user)          # 输入框输入账号 test
        self.driver.find_element_by_name('pwd').clear()
        self.driver.find_element_by_name('pwd').send_keys(pwd)
        self.driver.find_element_by_class_name('btn').click()             # 点击登录

        WebDriverWait(self.driver, 5).until(EC.alert_is_present())        # 等待 alert 的提示框出现
        alert = self.driver.switch_to.alert                               # 获取 alert 弹窗
        print(alert.text)
        sleep(2)

        assert alert.text == expected                                     # 断言，断言 alert 弹窗
        alert.accept()                                                    # 关闭弹窗

        # self.driver.quit()                                              # 关闭浏览器


if __name__ == '__main__':
    jp = TestUserLogin()           # 实例化
    jp.test_pws_error()            # 调用密码失败用例
    jp.test_user_login_success()   # 调用登录成功用例