Пример #1
0
class TestBasicConnection(TestCase):
    timeout = 10

    def setUp(self):
        deferred1, deferred2 = [Deferred() for _ in range(2)]

        self.server_app = mock.MagicMock()
        self.server_deferred = Deferred()
        self.server_factory = ServerFactory(self.server_app)
        self.server_transport = proto_helpers.StringTransport()
        self.server_connection = self.server_factory.buildProtocol(server_protocol.SERVER_PORT)
        self.server_factory.onConnectionLost = self.server_deferred
        self.server_factory.onConnectionMade = deferred1

        self.client_app = mock.MagicMock()
        self.client_deferred = Deferred()
        self.client_factory = ClientFactory(self.client_app)
        self.client_connection = self.client_factory.buildProtocol((client_protocol.SERVER_IP,
                                                                    client_protocol.SERVER_PORT))
        self.client_transport = proto_helpers.StringTransport()
        self.client_factory.onConnectionLost = self.client_deferred
        self.client_factory.onConnectionMade = deferred2

        def on_server_connect(*args):
            log.msg("Server transport disconnected.")

        deferred1.addCallback(on_server_connect)

        def on_client_connect(*args):
            log.msg("Client transport connected.")

        deferred2.addCallback(on_client_connect)

        deferred_last = Deferred()

        def on_connect(connection):
            self.client = connection
            deferred_last.callback(True)
            log.msg("Client connected")

        self.client_app.on_connection = on_connect

        self.client_connection.makeConnection(self.client_transport)
        return gatherResults([deferred2, deferred1, deferred_last])

    def tearDown(self):
        d = maybeDeferred(self.server_connection.stopListening)
        self.client_connection.disconnect()
        self.client.abortConnection()
        d = gatherResults([self.client_deferred,
                           self.server_deferred,
                           d])
        log.msg("Returning " + str(d))
        from sys import stdout
        stdout.flush()
        return d
Пример #2
0
    def setUp(self):
        deferred1, deferred2 = [Deferred() for _ in range(2)]

        self.server_app = mock.MagicMock()
        self.server_deferred = Deferred()
        self.server_factory = ServerFactory(self.server_app)
        self.server_transport = proto_helpers.StringTransport()
        self.server_connection = self.server_factory.buildProtocol(server_protocol.SERVER_PORT)
        self.server_factory.onConnectionLost = self.server_deferred
        self.server_factory.onConnectionMade = deferred1

        self.client_app = mock.MagicMock()
        self.client_deferred = Deferred()
        self.client_factory = ClientFactory(self.client_app)
        self.client_connection = self.client_factory.buildProtocol((client_protocol.SERVER_IP,
                                                                    client_protocol.SERVER_PORT))
        self.client_transport = proto_helpers.StringTransport()
        self.client_factory.onConnectionLost = self.client_deferred
        self.client_factory.onConnectionMade = deferred2

        def on_server_connect(*args):
            log.msg("Server transport disconnected.")

        deferred1.addCallback(on_server_connect)

        def on_client_connect(*args):
            log.msg("Client transport connected.")

        deferred2.addCallback(on_client_connect)

        deferred_last = Deferred()

        def on_connect(connection):
            self.client = connection
            deferred_last.callback(True)
            log.msg("Client connected")

        self.client_app.on_connection = on_connect

        self.client_connection.makeConnection(self.client_transport)
        return gatherResults([deferred2, deferred1, deferred_last])