python之一测试用例登陆

import unittest
# import datetime
#
# date = datetime.date.today()
# count = 0
# while count < 3:
#     count += 1
#     username = input('请输入用户名：' + '\n')
#     passwd = input('请输入密码：' + '\n')
#     hopeusername = 'nihao'
#     hopepasswd = '123456'
#     if not username.strip() or not passwd.strip():
#         print('用户名或者密码不能为空' + '\t' + '还剩下' + 3 - count, '次机会')
#     elif username == hopeusername and passwd == hopepasswd:
#         print('欢迎{}登陆，今天的日期是{}'.format(username, date))
#     else:
#         print('用户名或者密码密码错误' + '\t' + '还剩下' + 3 - count, '次机会')
# else:
#     print("超过2次，登陆失败")

'''
1)测试用例类：自己定义的类只要继承于unittest中的Teascase，那么这个类就是测试用例类
2)测试用例类中，以test开头的就是方法就是一条测试用例
3)用例执行通没有通过的评判标准：断言异常
'''


def login_check(username, passwd):
    if (not username) or (not passwd):
        return {"code": 2, "messg": "账号和密码不能为空"}

    else:
        if username == "ceshi" and passwd == "123456":
            return {"code": 0, "messg": "登陆成功"}
        else:
            return {"code": 1, "messg": "账号和密码不正确"}


class LoginTestcase(unittest.TestCase):
    def test_login_pass(self):
        # 第一步准备测试用例数据
        # 1.用例的参数
        data = ("ceshi", "123456")
        # 2.预期结果
        expected = {"code": 0, "messg": "登陆成功"}
        # 第二步执行用例函数，获取实际结果
        result = login_check(*data)
        # print("result==",result)
        # 第三步：对比实际结果和预期结果
        # assert data == result
        self.assertEqual(expected, result)
        pass

    def test_login_pwd_error(self):
        # 第一步准备测试用例数据
        # 1.用例的参数
        data = ("ceshi", "1234567")
        # 2.预期结果
        expected = {"code": 1, "messg": "账号和密码不正确"}
        # 第二步执行用例函数，获取实际结果
        result = login_check(*data)
        print("result==", result)
        # 第三步：对比实际结果和预期结果
        # assert data == result
        self.assertEqual(expected, result)
        pass

    def test_login_pwd_none(self):
        # 第一步准备测试用例数据
        # 1.用例的参数
        data = ("ceshi", "")
        # 2.预期结果
        expected = {"code": 2, "messg": "账号和密码不能为空"}
        # 第二步执行用例函数，获取实际结果
        result = login_check(*data)
        print("result==", result)
        # 第三步：对比实际结果和预期结果
        # assert data == result
        self.assertEqual(expected, result)
        pass
if __name__=='__main__':
    unittest.main()