def __init__(self):
GeneratorWrapperProtocol.__init__(self)
self.bytes_read = 0
self.session_init_time = None
# handshaker gets instantiated on connection
self._hs = None
# demuxer and muxer get instantiated on successful handshake
self._demuxer = None
self.muxer = None
def __init__(self):
GeneratorWrapperProtocol.__init__(self)
self.bytes_read = 0
self.session_init_time = None
# handshaker gets instantiated on connection
self._hs = None
# demuxer and muxer get instantiated on successful handshake
self._demuxer = None
self.muxer = None
def test_changing_live(self):
t = StringTransport()
p = GeneratorWrapperProtocol()
def _123_gen_handler():
for n in (1, 2, 3):
s = yield n
self.messages.append(s.read(n)[:])
def _321_next_gen_handler(next):
for n in (3, 2, 1):
s = yield n
self.messages.append(s.read(n)[:])
next()
def next():
p.init_handler(_123_gen_handler())
p.init_handler(_321_next_gen_handler(next))
p.makeConnection(t)
p.dataReceived('aaabbcabbcccX')
self.assertEquals(self.pop_messages(), ['aaa', 'bb', 'c',
'a', 'bb', 'ccc'])
def test_late_init(self):
t = StringTransport()
p = GeneratorWrapperProtocol()
def _123_gen_handler():
for n in (1, 2, 3):
s = yield n
self.messages.append(s.read(n)[:])
p.init_handler(_123_gen_handler(), do_init=False)
p.makeConnection(t)
p.init_handler()
p.dataReceived('abbcccX')
self.assertEquals(self.pop_messages(), ['a', 'bb', 'ccc'])
def test_simple_123_partial_writes(self):
t = StringTransport()
p = GeneratorWrapperProtocol()
def _123_gen_handler():
for n in (1, 2, 3):
s = yield n
self.messages.append(s.read(n)[:])
p.init_handler(_123_gen_handler())
p.makeConnection(t)
p.dataReceived('a')
p.dataReceived('b')
p.dataReceived('bc')
p.dataReceived('cc')
self.assertEquals(self.pop_messages(), ['a', 'bb', 'ccc'])
def __init__(self, *a, **kw):
GeneratorWrapperProtocol.__init__(self, *a, **kw)
self.handshake_status = None
def connectionLost(self, reason=protocol.connectionDone):
del self.transport.writeSequence
del self.orig_writeSequence
GeneratorWrapperProtocol.connectionLost(self, reason)
def __init__(self, *a, **kw):
GeneratorWrapperProtocol.__init__(self, *a, **kw)
self.handshake_status = None
def connectionLost(self, reason=protocol.connectionDone):
del self.transport.writeSequence
del self.orig_writeSequence
GeneratorWrapperProtocol.connectionLost(self, reason)
def __init__(self, *a, **kw):
GeneratorWrapperProtocol.__init__(self, *a, **kw)
self.messages = []
