Exemplo n.º 1
0
 def test_tlsAfterStartTLS(self):
     """
     The C{TLS} attribute of a L{Connection} instance is C{True} after
     L{Connection.startTLS} is called.
     """
     skt = FakeSocket("")
     protocol = FakeProtocol()
     conn = Connection(skt, protocol, reactor=_FakeFDSetReactor())
     conn._tlsClientDefault = True
     conn.startTLS(ClientContextFactory(), True)
     self.assertTrue(conn.TLS)
Exemplo n.º 2
0
 def test_doReadWarningIsRaised(self):
     """
     When an L{IProtocol} implementation that returns a value from its
     C{dataReceived} method, a deprecated warning is emitted.
     """
     skt = FakeSocket("someData")
     protocol = FakeProtocol()
     conn = Connection(skt, protocol)
     conn.doRead()
     warnings = self.flushWarnings([FakeProtocol.dataReceived])
     self.assertEquals(warnings[0]['category'], DeprecationWarning)
     self.assertEquals(
         warnings[0]["message"], "Returning a value other than None from "
         "twisted.internet.test.test_tcp.FakeProtocol.dataReceived "
         "is deprecated since Twisted 11.0.0.")
     self.assertEquals(len(warnings), 1)
Exemplo n.º 3
0
 def test_doReadWarningIsRaised(self):
     """
     When an L{IProtocol} implementation that returns a value from its
     C{dataReceived} method, a deprecated warning is emitted.
     """
     skt = FakeSocket("someData")
     protocol = FakeProtocol()
     conn = Connection(skt, protocol)
     conn.doRead()
     warnings = self.flushWarnings([FakeProtocol.dataReceived])
     self.assertEqual(warnings[0]['category'], DeprecationWarning)
     self.assertEqual(
         warnings[0]["message"],
         "Returning a value other than None from "
         "twisted.internet.test.test_tcp.FakeProtocol.dataReceived "
         "is deprecated since Twisted 11.0.0.")
     self.assertEqual(len(warnings), 1)
Exemplo n.º 4
0
    def __init__(self, sock, protocol, addr, additionalData):
        Connection.__init__(self, sock, protocol)
        self.client = addr

        # Inform the protocol we've made a connection.
        protocol.makeConnection(self)

        # Now, we want to feed in the extra data BEFORE the reactor reads
        # anything additional from the socket. However, if we call this in
        # the other order, and the socket gets closed (or passed to something
        # non-twisted) after just the initial chunk, we'll be calling
        # startReading() on something we've already stopped reading. That won't
        # work too well... Fortunately, the reactor runs in this thread, so
        # merely adding it (with startReading()) can't cause a read to happen
        # immediately.
        self.startReading()
        self.connected = 1

        protocol.dataReceived(additionalData)
Exemplo n.º 5
0
    def __init__(self, sock, protocol, addr, additionalData):
        Connection.__init__(self, sock, protocol)
        self.client = addr

        # Inform the protocol we've made a connection.
        protocol.makeConnection(self)

        # Now, we want to feed in the extra data BEFORE the reactor reads
        # anything additional from the socket. However, if we call this in
        # the other order, and the socket gets closed (or passed to something
        # non-twisted) after just the initial chunk, we'll be calling
        # startReading() on something we've already stopped reading. That won't
        # work too well... Fortunately, the reactor runs in this thread, so
        # merely adding it (with startReading()) can't cause a read to happen
        # immediately.
        self.startReading()
        self.connected = 1

        protocol.dataReceived(additionalData)
Exemplo n.º 6
0
    def test_getWriters(self):
        """
        Check that L{interfaces.IReactorFDSet.getWriters} reflects the actions
        made with L{interfaces.IReactorFDSet.addWriter} and
        L{interfaces.IReactorFDSet.removeWriter}.
        """
        s = socket.socket()
        self.addCleanup(s.close)

        c = Connection(s, protocol.Protocol())
        self.assertNotIn(c, reactor.getWriters())

        reactor.addWriter(c)
        self.assertIn(c, reactor.getWriters())

        reactor.removeWriter(c)
        self.assertNotIn(c, reactor.getWriters())