【Python Web自动化】04_03unittest项目中使用断言assert

代码优化，将判断语句修改为断言语句
GitHub代码commits id：767e0ac

---

#!/bin/usr/env python3
# -*- coding: utf-8 -*-
# --------------------------------
# ProjectName: 
# Author: Crisimple
# CreateTime: 2019/6/12 13:08
# FileName: first_case.py
# Description: 
# Question: 
# --------------------------------
from code_package.business.register_business import RegisterBusiness
from selenium import webdriver
import unittest


class FirstCase(unittest.TestCase):

    def setUp(self) -> None:
        self.driver = webdriver.Chrome("../tools/chromedriver.exe")
        # self.driver = webdriver.Edge("../tools/msedgedriver.exe")
        self.driver.maximize_window()
        self.driver.get("http://www.5itest.cn/register")
        self.register_business = RegisterBusiness(self.driver)

    def tearDown(self) -> None:
        self.driver.close()

    def test_login_email_error(self):
        email_error = self.register_business.login_email_error('34', 'U1212', '12Test12', 'der56')
        self.assertTrue(email_error, "注册成功了，此条 case 执行失败")
        # if email_error == True:
        #     print("注册成功了，此条 case 执行失败")
        # login('223', '1111')
        # 通过 assert 判断是否为 error

    def test_login_username_error(self):
        username = self.register_business.login_username_error("qqww@qq.com", "2", "Test@123", 'sdsds')
        self.assertFalse(username)
        # if username == True:
        #     print("注册成功了，此条 case 执行失败")

    def test_login_password_error(self):
        password = self.register_business.login_password_error("qqww@qq.com", "2gw", "2", 'sdsds')
        self.assertFalse(password)
        # if password == True:
        #     print("注册成功了，此条 case 执行失败")

    def test_login_code_error(self):
        code = self.register_business.login_code_error("qqww@qq.com", "2gw", "Test@123", 's')
        self.assertFalse(code, "注册成功了，此条 case 执行失败")
        # if code == True:
        #     print("注册成功了，此条 case 执行失败")

    def test_login_success(self):
        success = self.register_business.user_base("crisimple@163.com", "Crisimple", "Test@123", "test1")
        if success == True:
            print("注册成功")
        else:
            print("注册失败")


# def main():
#     first = FirstCase()
#     first.test_login_email_error()
#     first.test_login_username_error()
#     first.test_login_password_error()
#     first.test_login_success()


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