def setUp(self): self.conn = TwistedProtocolConnection() self.conn._impl = mock.Mock()
def buildProtocol(self, addr): parameters = pika.ConnectionParameters() protocol = TwistedProtocolConnection(parameters) protocol.add_on_open_callback(on_connect) return protocol