【Python+unittest】再次使用mock.Mock()方法进行测试。根据返回的code状态码，进行assert断言

继续学习mock

# 文件名称：test40.py
# 引入requests，这是专门做接口自动化的模块
# 声明2个函数，第1个是返回code码，第2个是调取第1个函数、赋值、返回值
import requests
def send_request():
    r = request.get()
    return r.status_code
def visit():
    return send_request("https://www.baidu.com")
    return send_request("https://www.baidu.com/1.html")

执行文件：

import unittest
import mock
import test40

class TestClient(unittest.TestCase):
    def test_success_request(self):
        test40.send_request = mock.Mock(return_value='200')
        print (test40.visit())
        self.assertEqual(test40.visit(), "200")

    def test_fail_request(self):
        test40.send_request = mock.Mock(return_value='404')
        print (test40.visit())
        self.assertEqual(test40.visit(), '404')

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