ddt 基础版小例子

test_login_case.py

import unittest
from testdir.login_logic import login_check
import ddt

# 测试数据
# 测试步骤
# 测试结果与预期结果比对
datas = [
    {"name": "nanwu", "password": "123456", "res": {"code": 0, "message": "登陆成功"}},
    {"name": "nanwu", "password": "qwer123", "res": {"code": 1, "message": "账号密码不正确"}},
    {"name": "nanwu", "password": "qwer1234", "res": {"code": 1, "message": "账号密码不正确"}}
]


@ddt.ddt
class MyTestCase(unittest.TestCase):

    @ddt.data(*datas)
    def test_login_ok(self, data):
        res = login_check(data["name"], data["password"])
        self.assertEqual(res, data["res"])

    # def test_login_ok(self):
    #     res = login_check("nanwu", "123456")
    #     # assert res == {"code": 0, "message": "登陆成功"}
    #     self.assertEqual(res, {"code": 0, "message": "登陆成功"})
    #
    # def test_login_wrong_Password(self):
    #     res = login_check("nanwu", "qwer123")
    #     self.assertEqual(res, {"code": 1, "message": "账号密码不正确"})
    #
    # def test_login_wrong_Passwords(self):
    #     res = login_check("nanwu", "qwer1234")
    #     self.assertEqual(res, {"code": 1, "message": "账号密码不正确"})
    #
    # def test_login_wrong_user(self):
    #     res = login_check("nanwu")
    #     self.assertEqual(res, {"code": 1, "message": "所有参数不能为空"})
    #
    # def test_login_wrong_users(self):
    #     res = login_check("nanwu", "123456")
    #     self.assertEqual(res, {"code": 1, "message": "所有参数不能为空"})


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

login_logic.py

def login_check(username=None, password=None):
    if username is not None and password is not None:
        if username == 'nanwu' and password == "123456":
            return {"code": 0, "message": "登陆成功"}
        else:
            return {"code": 1, "message": "账号密码不正确"}
    else:
        return {"code": 1, "message": "所有参数不能为空"}