# -*- coding: utf-8 -*-
# @Time : 2020/2/12 8:40
# @File : learn_mock_8.py
# @Author: Hero Liu
# 单元测试结合mock改写
import unittest
from mock import mock
from day_20200208_mooc.base.interface_request_class_5 import RunMain


class TestMethod(unittest.TestCase):
    def setUp(self):
        # 避免每个用例都去执行一次实例化
        self.run = RunMain()

    def test_01(self):
        url = "https://www.baidu.com"
        data = {"response": "haha"}

        # 把data作为mock.Mock的返回值，传给self.run.run_main方法
        self.run.run_main = mock.Mock(return_value=data)
        res = self.run.run_main(url, "method", data)
        print(res)
        # self.assertEqual(res["errorCode"], "期望值 ", "这是期望值和预期值不相等时给出的提示语")


if __name__ == "__main__":
    suite = unittest.TestSuite()
    # 添加测试用例
    suite.addTest(TestMethod("test_01"))
    runner = unittest.TextTestRunner()
    runner.run(suite)