练习 | 断言

1. 目标

需求：使用UnitTest框架对tpshop项目测试
1). 点击登录，进入登录页面
2). 输入用户名和密码，不输入验证码，直接点击登录按钮
3). 获取错误提示信息
4). 断言错误提示信息是否为“验证码不能为空!”，如果断言失败则保存截图

2. 实现方法

# 定义初始化 [方法级别]
def setUp(self):
    # 获取浏览器驱动对象
    # 打开url
    # 最大化
    # 隐式等待
    pass
# 定义 tearDown [方法级别]
def tearDown(self):
    # 关闭浏览器
    pass
# 定义登录测试方法 验证码为空
def test_login_code_null(self):
    # 点击登录链接
    # 输入用户名
    # 输入密码
    # 输入验证码
    # 点击登录
    # 获取登录后的信息
    # 断言
    # 截图

3. 实现代码

"""
需求：使用UnitTest框架对tpshop项目测试
1). 点击登录，进入登录页面
2). 输入用户名和密码，不输入验证码，直接点击登录按钮
3). 获取错误提示信息
4). 断言错误提示信息是否为“验证码不能为空!”，如果断言失败则保存截图

方法
    # 定义初始化 [方法级别]
    def setUp(self):
        # 获取浏览器驱动对象
        # 打开url
        # 最大化
        # 隐式等待
        pass
    # 定义 tearDown [方法级别]
    def tearDown(self):
        # 关闭浏览器
        pass
    # 定义登录测试方法 验证码为空
    def test_login_code_null(self):
        # 点击登录链接
        # 输入用户名
        # 输入密码
        # 输入验证码
        # 点击登录
        # 获取登录后的信息
        # 断言
        # 截图
注意点
    driver在各个方法都要访问到：要加上 self. 实现跨函数使用
"""

# 导包
import time
import unittest
from time import sleep

from selenium import webdriver
from selenium.webdriver.common.by import By
# 定义测试类
class TestTpshopLogin(unittest.TestCase):
    # 定义初始化 [方法级别]
    def setUp(self):

        # 获取浏览器驱动对象
        self.driver = webdriver.Chrome()
        # 打开url
        self.driver.get("http://demo6.tp-shop.cn/")
        # 最大化
        self.driver.maximize_window()
        # 隐式等待
        self.driver.implicitly_wait(30)
    # 定义 tearDown [方法级别]
    def tearDown(self):
        sleep(3)
        # 关闭浏览器
        self.driver.quit()
        pass
    # 定义登录测试方法 验证码为空
    def test_login_code_null(self):
        driver = self.driver
        # 点击登录链接
        driver.find_element(By.LINK_TEXT, "登录").click()
        # 输入用户名
        driver.find_element(By.ID, "username").send_keys("1329269620@qq.com")
        # 输入密码
        driver.find_element(By.ID, "password").send_keys("000000")
        # 输入验证码
        # 点击登录
        driver.find_element(By.CSS_SELECTOR, ".J-login-submit").click()
        # 获取登录后的信息
        msg = driver.find_element(By.CSS_SELECTOR, ".layui-layer-content").text
        print(msg)
        rightMsg = "验证码不能为空!" # 预期输出结果
        try:
            # 断言
            self.assertEqual(msg, rightMsg)
        except AssertionError:
            # 截图
            driver.get_screenshot_as_file("./%s.png"%(time.strftime("%Y_%m_%d %H_%M_%S")))
            # 抛异常
            raise