python - 如何在重新连接测试的单元测试中终止套接字

Question

我正在尝试测试一些在断开连接后重新连接到服务器的代码。这在测试之外工作得很好，但它无法确认套接字在运行测试时已断开连接。

我正在使用 Gevent Stream Server 模拟真实的监听服务器:

import gevent.server
from gevent import queue


class TestServer(gevent.server.StreamServer):

    def __init__(self, *args, **kwargs):
        super(TestServer, self).__init__(*args, **kwargs)
        self.sockets = {}

    def handle(self, socket, address):
        self.sockets[address] = (socket, queue.Queue())
        socket.sendall('testing the connection\r\n')
        gevent.spawn(self.recv, address)

    def recv(self, address):
        socket = self.sockets[address][0]
        queue = self.sockets[address][1]
        print 'Connection accepted %s:%d' % address
        try:
            for data in socket.recv(1024):
                queue.put(data)
        except:
            pass

    def murder(self):
        self.stop()
        for sock in self.sockets.iteritems():
            print sock
            sock[1][0].shutdown(socket.SHUT_RDWR)
            sock[1][0].close()
        self.sockets = {}


def run_server():
    test_server = TestServer(('127.0.0.1', 10666))
    test_server.start()
    return test_server

我的测试是这样的:

def test_can_reconnect(self):
    test_server = run_server()
    client_config = {'host': '127.0.0.1', 'port': 10666}
    client = Connection('test client', client_config, get_config())
    client.connect()
    assert client.socket_connected
    test_server.murder()
    #time.sleep(4) #tried sleeping. no dice.
    assert not client.socket_connected
    assert client.server_disconnect
    test_server = run_server()
    client.reconnect()
    assert client.socket_connected

它在 assert not client.socket_connected 处失败。

我在接收过程中检测到“非数据”。如果是 None，那么我设置一些变量，以便其他代码可以决定是否重新连接(如果是 user_disconnect 等则不要重新连接)。这种行为有效并且过去一直对我有效，直到现在我才尝试对其进行测试。套接字连接和局部函数范围之类的东西有什么奇怪的吗？就好像即使在停止服务器后连接仍然存在。

我要测试的代码是开放的:https://github.com/kyleterry/tenyks.git

如果您运行测试，您将看到我尝试修复的测试失败。

Answer 1

尝试使用真实的套接字运行单元测试是一项艰巨的任务。这将很棘手，因为一次只能运行一组测试，因为将使用服务器端口，并且随着套接字的设置和拆除，它会很慢。最重要的是，如果这真的是一个单元测试，您不想测试套接字，而只想测试使用套接字的代码。

如果您模拟套接字调用，您可以从模拟代码中随意抛出异常，并确保使用套接字的代码做正确的事情。您不需要真正的套接字来确保被测类做正确的事情，如果您可以将套接字调用包装在一个对象中，则可以伪造它。在构造类时传入对套接字对象的引用，然后就可以开始了。

我的建议是将套接字调用包装在一个支持 sendall、recv 以及您在套接字上调用的所有方法的类中。然后您可以用 TestReconnectSocket(或其他)替换实际的 Socket 类并运行您的测试。

看看mox ，一个 Python 模拟框架。