Python + Selenium 使用 unittest 编写自动化测试用例

我们使用之前演示过的 showdoc，编写一个登录的用例，并检查自定义项目是否存在。

首先，我们把这个过程先罗列出来：

• 打开登录页面
• 选择登录
• 输入用户名
• 输入密码
• 点击登录
• 验证自定义项目是否存在

由于这个项目登录后没有任何用户名标识，所以要想判断登录的账号是否正确，可以通过该用户下的项目来判断。

import unittest
import time
from selenium import webdriver
from vcode import ocr_b64 # 处理验证码的程序


class TestLogin(unittest.TestCase):

    def setUp(self): # => setUp中处理浏览器初始化
        self.driver = webdriver.Chrome()
        self.driver.implicitly_wait(10)
        self.driver.maximize_window()

    def page_contains(self, text): # => 写一个方法用于断言
        '''
        判断页面上是否包含某个文本
        :params: text, 需要验证的文本
        :return: True, False
        '''
        try:
            self.driver.find_element_by_xpath('//*[contains(text(), %s)]' % text)
            return True
        except Exception:
            return False

    def test_login(self):	# => 测试用例
        driver = self.driver # => 少写点self 😀
        # 打开登录页面
        driver.get('https://www.showdoc.cc/user/login')

        # 处理验证码
        while True:
            driver.find_element_by_css_selector(
                '[placeholder="用户名/邮箱"]').send_keys('nemo1')
            driver.find_element_by_css_selector(
                '[placeholder="密码"]').send_keys('123456')
            v_code = ocr_b64(driver.find_element_by_class_name(
                'v_code_img').screenshot_as_base64)  # => 识别验证码
            driver.find_element_by_css_selector(
                '[placeholder="验证码"]').send_keys(v_code)
            driver.find_element_by_xpath('//button/*[text()="登录"]').click()
            try:
                # 如果弹出框出现，则说明验证码错误 
                driver.find_element_by_css_selector(
                    '.el-message-box .el-button--primary').click()
                # 刷新页面
                driver.refresh()
            except:
                # 如果弹出框不出现，则说明验证码正确
                break

        # 提取当前用户的项目
        self.assertTrue(self.page_contains('Nemo的项目'), '自定义项目未找到')

    def tearDown(self):  # => 处理浏览器
        time.sleep(3)  # => 等一下查看下结果
        self.driver.quit()

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

代码说明：
from vcode import ocr_b64：
关于验证码的处理，请参考文章：使用百度文字识别API识别验证码

setUp()：

测试固件中的初始化方法主要处理浏览器的初始化过程。

tearDown()：

测试固件中的清理方法仅用来处理浏览器的退出。

test_login()：

测试方法包含操作过程、断言。

page_contains(text)：

定义了一个方法，用来判断页面上某个文本是否存在，存在则返回 True。主要用于配合断言方法。