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.messages = []