Selenium自动化-Unitest实战之登录模块功能

登录用例-简单版

from selenium import webdriver
import unittest

class LoginTest(unittest.TestCase):
    """登录模块"""
    @classmethod
    def setUpClass(cls):
        cls.driver = webdriver.Chrome()
        cls.driver.implicitly_wait(5)

    # 用例每次执行时运行
    def setUp(self):
        self.driver.get("xxx测试的网址xx")

    # 用例每次结束时执行
    def tearDown(self):
        # 清空cookie
        self.driver.delete_all_cookies()
        self.driver.refresh()

    @classmethod
    def tearDownClass(cls):
        cls.driver.quit()

    def get_error_message(self):
        msg = self.driver.find_element_by_id("error-box").text
        return msg

    # 登录函数
    def login(self, username, password):
        # 输入用户名
        self.driver.find_elements_by_tag_name("input")[0].send_keys(username)
        # 输入密码
        self.driver.find_elements_by_tag_name("input")[1].send_keys(password)
        # 点击确认按钮
        self.driver.find_element_by_class_name("login-form-button").click()
        self.driver.find_element_by_class_name("ant-btn-primary").click()

    def