用python实现自动化测试用例_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。主要用于配合断言方法。