def test_lose_connection(self): factory = ProtocolFactory() factory.protocol = Protocol p = factory.build(loop) self.assertTrue(isinstance(p, Protocol)) t = MockTransport() p.make_connection(t, 'test') p.lose_connection() self.assertTrue(t.closes==1)
def test_factory_build(self): factory = ProtocolFactory() factory.protocol = Protocol p = factory.build(loop) self.assertTrue(isinstance(p, Protocol))