我们使用之前演示过的 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。主要用于配合断言方法。
