sockfilter 1.4

用法

预期用途是确保单元测试不会建立网络连接。使用如下谓词激活 sockfilter：

# only allow http connections to localhost
def socket_address_allowed(address):
    return (address.host in ['localhost', '127.0.0.1'] and
            address.port == 80)

sockfilter.enable(socket_address_allowed)

如果后续尝试在地址不满足谓词的情况下打开连接，将引发 sockfilter.SockFilterError。

from sockfilter import with_sockfiltering

@with_sockfiltering(socket_address_allowed)
def test_foo():
    foo()

from sockfilter import sockfiltering

def test_foo():
    with sockfiltering(socket_address_allowed):
        foo()

import sockfilter
from unittest import TestCase

class FooTest(TestCase):

    def setUp(self):
        sockfilter.enable(socket_address_allowed)

    def tearDown(self):
        sockfilter.disable()

    def test_foo(self):
        foo()

内部原理

Sockfilter通过猴子补丁Python的 socket 库工作，灵感来自 HTTPretty。