"""
1、单元测试框架：自动校验结果
    python:unittest或者pytest、Java：Junit
    怎么写用例：
        必须以test开头
    查找用例
    参数化
"""
import unittest
import myFunction
import HTMLTestRunner
import HTMLTestRunnerNew  # 测试报告丰富版本
import parameterized  # 参数化


class TestAdd(unittest.TestCase):

    def testAddNormal1(self):
        """正常的测试加法，by huozi"""
        result = myFunction.add(1, 2)
        self.assertEqual(3, result, )

    def testAddNormal2(self):
        """正常的测试加法，带有msg返回信息"""
        result = myFunction.add(4, 2)
        self.assertEqual(6, result, '正常case通过')

    def testAddError1(self):
        """测试失败使用,by huozi"""
        result = myFunction.add(0, 2)
        self.assertEqual(4, result)

    def testAddError2(self):
        """测试失败使用带有msg返回信息的"""
        result = myFunction.add(1, 2)
        self.assertEqual(0, result, '正常整数加法，没有通过')

    @parameterized.parameterized.expand(  # 传参为二维数组
        [[1, 2, 3, '参数化1'],
         [-1, 2, 3, '参数化2'],
         [2, 4, 7, '参数化3']]
    )
    def testParamAdd(self, a, b, c, desc):
        self._testMethodDoc = desc  # 使用这个_testMethodDoc参数传递
        result = myFunction.add(a, b)
        self.assertEqual(c, result, '预期结果是%s,实际结果是%s' % (c, result))


if __name__ == '__main__':
    # 写法0：不产生测试报告
    # unittest.main()  # 执行所有用例

    # 写法1：运行单个测试用例
    testSuite1 = unittest.TestSuite()
    testSuite1.addTest(TestAdd('testAddNormal'))  # 运行单个测试用例
    # testSuite.addTest(TestAdd('testAddError1'))
    # testSuite.addTest(TestAdd('testAddError1'))

    # 写法2：运行某个类里面的测试用例
    testSuite2 = unittest.makeSuite(TestAdd)  # 运行某个类(如TestAdd)里面所有的测试用例

    # 写法3：查找某个目录下的测试用例(绝对路径)，文件必须以test开头，所有文件就是：*.py
    testSuite3 = unittest.defaultTestLoader.discover('/Users/ray/PycharmProjects/tmz/day9/cases', 'test*.py')

    with open('report.html', 'wb') as fw:
        # runner = HTMLTestRunner.HTMLTestRunner(stream=fw, title='天马测试报告', description='天马测试',verbosity=3)
        runner = HTMLTestRunnerNew.HTMLTestRunner(stream=fw, title='天马测试报告', description='天马测试', verbosity=3)
        runner.run(testSuite2)