Question

[编辑：此问题仅在Python 2.7中存在。使用3.7，它可以按预期工作。知道有什么区别，以及是否有可能使它也适用于2.7，出于某些原因，我不得不使用它]

我想使用Mock覆盖套接字的方法sendto。我的第一种方法是这样的：

class TestClass01(TestCase):
    @mock.patch.object(socket.socket, "sendto", new=mock_sendto)
    def test_function1(self):
        ...

但是，这导致了异常：

AttributeError：'_socketobject'对象属性'sendto'是只读的。

对于其他类，此方法效果很好，但套接字不适用。我认为，原因是__init__为所有成员函数调用setattr，例如：

def __init__(self, family=AF_INET, type=SOCK_STREAM, proto=0, _sock=None):
    if _sock is None:
        _sock = _realsocket(family, type, proto)
    self._sock = _sock
    for method in _delegate_methods:
        setattr(self, method, getattr(_sock, method))      <<<<<<<<<<<<<

然后，我尝试了另一种方法：

class TestClass01(TestCase):
    @mock.patch("socket.socket")
    def test_function1(self, mock_class):
        s = mock.Mock(socket.socket)
        s.sendto = mock_sendto
        s.close = mock_close

        mock_class.return_value = s

        # ... run my tests here ...

        args, kwargs = mock_class.call_args_list[0]

        # Check arguments passed to sendto

那行得通，但是，调用参数列表只给了我

((2, 2), {})

对应于构造器AF_INET=2和SOCK_DGRAM=2。我没有打过sendto的电话。我的解决方法是使用全局变量，该变量是通过mock_sendto函数设置的。不过不是很优雅。

我有三个问题：

我可以解决“只读”问题并获得最初的工作方法吗？
为什么all_args_list仅跟踪构造的调用而不跟踪sendto？
有没有比使用全局变量更好的方法？

提前感谢您的回答！

单元测试模拟/覆盖套接字的sendto

0 个答案: