Python policies.ProtocolWrapper方法代碼示例

# 需要導入模塊: from twisted.protocols import policies [as 別名]
# 或者: from twisted.protocols.policies import ProtocolWrapper [as 別名]
def test_wrappingBehavior(self):
        """
        Any modifications performed by the underlying L{ProtocolWrapper}
        propagate through to the wrapped L{Protocol}.
        """
        connecting = self.wrapper.connect(self.factory)
        pump = self.completer.succeedOnce()
        proto = self.successResultOf(connecting)
        pump.server.transport.write(b'5:hello,')
        pump.flush()
        self.assertEqual(proto.strings, [b'HELLO']) 

# 需要導入模塊: from twisted.protocols import policies [as 別名]
# 或者: from twisted.protocols.policies import ProtocolWrapper [as 別名]
def __init__(self, protocol, factory):
        policies.ProtocolWrapper.__init__(self, protocol, factory)
        self.deferred = defer.Deferred() 

# 需要導入模塊: from twisted.protocols import policies [as 別名]
# 或者: from twisted.protocols.policies import ProtocolWrapper [as 別名]
def connectionLost(self, reason):
        policies.ProtocolWrapper.connectionLost(self, reason)
        self.deferred.callback(None) 

# 需要導入模塊: from twisted.protocols import policies [as 別名]
# 或者: from twisted.protocols.policies import ProtocolWrapper [as 別名]
def __init__(self, factory, wrappedProtocol):
        policies.ProtocolWrapper.__init__(self, factory, wrappedProtocol)
        self._proxyInfo = None
        self._parser = None 

# 需要導入模塊: from twisted.protocols import policies [as 別名]
# 或者: from twisted.protocols.policies import ProtocolWrapper [as 別名]
def __init__(self, factory, wrappedProtocol):
        policies.ProtocolWrapper.__init__(self, factory, wrappedProtocol) 

# 需要導入模塊: from twisted.protocols import policies [as 別名]
# 或者: from twisted.protocols.policies import ProtocolWrapper [as 別名]
def connectionLost(self, reason):
        policies.ProtocolWrapper.connectionLost(self, reason)
        self.factory.onDisconnect.callback(None) 

# 需要導入模塊: from twisted.protocols import policies [as 別名]
# 或者: from twisted.protocols.policies import ProtocolWrapper [as 別名]
def test_transportInterfaces(self):
        """
        The transport wrapper passed to the wrapped protocol's
        C{makeConnection} provides the same interfaces as are provided by the
        original transport.
        """
        class IStubTransport(Interface):
            pass

        @implementer(IStubTransport)
        class StubTransport:
            pass

        # Looking up what ProtocolWrapper implements also mutates the class.
        # It adds __implemented__ and __providedBy__ attributes to it.  These
        # prevent __getattr__ from causing the IStubTransport.providedBy call
        # below from returning True.  If, by accident, nothing else causes
        # these attributes to be added to ProtocolWrapper, the test will pass,
        # but the interface will only be provided until something does trigger
        # their addition.  So we just trigger it right now to be sure.
        implementedBy(policies.ProtocolWrapper)

        proto = protocol.Protocol()
        wrapper = policies.ProtocolWrapper(policies.WrappingFactory(None), proto)

        wrapper.makeConnection(StubTransport())
        self.assertTrue(IStubTransport.providedBy(proto.transport)) 

# 需要導入模塊: from twisted.protocols import policies [as 別名]
# 或者: from twisted.protocols.policies import ProtocolWrapper [as 別名]
def test_protocolLogPrefix(self):
        """
        L{ProtocolWrapper.logPrefix} is customized to mention both the original
        protocol and the wrapper.
        """
        server = Server()
        factory = policies.WrappingFactory(server)
        protocol = factory.buildProtocol(
            address.IPv4Address('TCP', '127.0.0.1', 35))
        self.assertEqual("EchoProtocol (ProtocolWrapper)",
                         protocol.logPrefix()) 

# 需要導入模塊: from twisted.protocols import policies [as 別名]
# 或者: from twisted.protocols.policies import ProtocolWrapper [as 別名]
def test_protocolLogPrefixFallback(self):
        """
        If the wrapped protocol doesn't have a L{logPrefix} method,
        L{ProtocolWrapper.logPrefix} falls back to the protocol class name.
        """
        class NoProtocol(object):
            pass

        server = Server()
        server.protocol = NoProtocol
        factory = policies.WrappingFactory(server)
        protocol = factory.buildProtocol(
            address.IPv4Address('TCP', '127.0.0.1', 35))
        self.assertEqual("NoProtocol (ProtocolWrapper)",
                         protocol.logPrefix()) 

# 需要導入模塊: from twisted.protocols import policies [as 別名]
# 或者: from twisted.protocols.policies import ProtocolWrapper [as 別名]
def test_getHost(self):
        """
        L{policies.ProtocolWrapper.getHost} calls C{getHost} on the underlying
        transport.
        """
        wrapper = self._getWrapper()
        self.assertEqual(wrapper.getHost(), wrapper.transport.getHost()) 

# 需要導入模塊: from twisted.protocols import policies [as 別名]
# 或者: from twisted.protocols.policies import ProtocolWrapper [as 別名]
def test_getPeer(self):
        """
        L{policies.ProtocolWrapper.getPeer} calls C{getPeer} on the underlying
        transport.
        """
        wrapper = self._getWrapper()
        self.assertEqual(wrapper.getPeer(), wrapper.transport.getPeer()) 

# 需要導入模塊: from twisted.protocols import policies [as 別名]
# 或者: from twisted.protocols.policies import ProtocolWrapper [as 別名]
def test_registerProducer(self):
        """
        L{policies.ProtocolWrapper.registerProducer} calls C{registerProducer}
        on the underlying transport.
        """
        wrapper = self._getWrapper()
        producer = object()
        wrapper.registerProducer(producer, True)
        self.assertIs(wrapper.transport.producer, producer)
        self.assertTrue(wrapper.transport.streaming) 

# 需要導入模塊: from twisted.protocols import policies [as 別名]
# 或者: from twisted.protocols.policies import ProtocolWrapper [as 別名]
def test_unregisterProducer(self):
        """
        L{policies.ProtocolWrapper.unregisterProducer} calls
        C{unregisterProducer} on the underlying transport.
        """
        wrapper = self._getWrapper()
        producer = object()
        wrapper.registerProducer(producer, True)
        wrapper.unregisterProducer()
        self.assertIsNone(wrapper.transport.producer)
        self.assertIsNone(wrapper.transport.streaming) 

# 需要導入模塊: from twisted.protocols import policies [as 別名]
# 或者: from twisted.protocols.policies import ProtocolWrapper [as 別名]
def test_stopConsuming(self):
        """
        L{policies.ProtocolWrapper.stopConsuming} calls C{stopConsuming} on
        the underlying transport.
        """
        wrapper = self._getWrapper()
        result = []
        wrapper.transport.stopConsuming = lambda: result.append(True)
        wrapper.stopConsuming()
        self.assertEqual(result, [True])