单元测试 - Python如何测试HTTP客户端程序？

Question

打算写一个模拟HTTP客户端的程序（通过urlopen来发送和接收HTTP请求）。

需要写单元测试，初步打算使用unittest测试框架写测试用例。但是受具体环境限制，不能真正发送HTTP请求。
现在，退而求其次，写一个测试用例，不用发送请求并验证服务器返回的数据，只需要检查HTTP请求的头部和数据部分是否正确就好。
怎么写这种测试用例？
初步的想法是mock一个HTTP Server

Answer 1

OK，找到了unittest.mock。
unittest.mock本來是用來在單元測試中hook第三方模組（類別/函數）的（繞過原來的模組），但是HTTP請求也是透過某個函數發送的，也可以用它來模擬。

---

以下為範例：
main.py 待測程式

from urllib.request import urlopen


def func():
    url_ip = 'http://ifconfig.me/ip'
    ip = None
    try:
        ip = urlopen(url_ip)
    except:
        pass
    return ip


if __name__ == '__main__':
    print(func())

test.py 測試案例

import unittest
from unittest import mock

from main import func


class FuncTestCase(unittest.TestCase):
    @mock.patch('main.urlopen')
    def test_func(self, mock_open):
        mock_open.return_value = '127.0.0.1'
        #设置模拟函数的返回值
        result = func()
        #调用待测功能，但已经模拟了urlopen函数
        mock_open.assert_called_with('http://ifconfig.me/ip')
        #验证调用参数
        self.assertEqual(result, '127.0.0.1')
        #验证返回结果


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

缺陷：如果在func中呼叫了多次urlopen，則上述程式碼無法正常測試。
Update：可以透過設定mock物件的side_effect屬性來用自己的函數取代系統函數（不過不太優雅）。

Answer 2

你可能需要這個。
http://docs.python-requests.org/en/latest/