def get_message_sender(self): '''Creates a new MessageSender object, with a fake amqp client''' message_sender = MessageSender('amqp-spec-0-8.xml', None) spec = get_spec(vumi_resource_path('amqp-spec-0-8.xml')) client = FakeAmqpClient(spec) message_sender.client = client return message_sender
def get_worker(self, **config): spec = get_spec(vumi_resource_path("amqp-spec-0-8.xml")) amq_client = fake_amqp.FakeAMQClient(spec, {}, self.broker) worker = TestWorker({}, config) worker._amqp_client = amq_client yield worker.startWorker() returnValue(worker)
def __init__( self, specfile, amqp_config, connected_callback, disconnected_callback): '''Factory that creates JunebugAMQClients. specfile - string of specfile name amqp_config - connection details for amqp server ''' self.connected_callback, self.disconnected_callback = ( connected_callback, disconnected_callback) self.amqp_config = amqp_config self.spec = get_spec(vumi_resource_path(specfile)) self.delegate = TwistedDelegate() super(AmqpFactory, self).__init__()
def test_fake_amqclient(self): spec = get_spec(make_vumi_path_abs("config/amqp-spec-0-8.xml")) amq_client = fake_amqp.FakeAMQClient(spec, {}, self.broker) d = Deferred() class TestWorker(Worker): @inlineCallbacks def startWorker(self): self.pub = yield self.publish_to('test.pub') self.conpub = yield self.publish_to('test.con') self.con = yield self.consume('test.con', self.consume_msg) def consume_msg(self, msg): # NOTE: One-shot. d.callback(msg) worker = TestWorker(amq_client, {}) yield worker.startWorker() yield worker.pub.publish_json({'message': 'foo'}) yield worker.conpub.publish_json({'message': 'bar'}) msg = yield d self.assertEqual({'message': 'bar'}, msg.payload)
def get_fake_amq_client(broker=None): spec = get_spec(vumi_resource_path("amqp-spec-0-8.xml")) return FakeAMQClient(spec, {}, broker)
def get_stubbed_channel(broker=None, id=0): spec = get_spec(vumi_resource_path("amqp-spec-0-8.xml")) amq_client = FakeAMQClient(spec, {}, broker) return amq_client.channel(id)
def _connect(self, worker, timeout, bindAddress): spec = get_spec(vumi_resource_path("amqp-spec-0-8.xml")) amq_client = FakeAMQClient(spec, self.options, self.broker) self.broker = amq_client.broker # So we use the same broker for all. reactor.callLater(0, worker._amqp_connected, amq_client)
def get_stubbed_worker(worker_class, config=None, broker=None): spec = get_spec(vumi_resource_path("amqp-spec-0-8.xml")) amq_client = FakeAMQClient(spec, {}, broker) worker = worker_class({}, config) worker._amqp_client = amq_client return worker
def get_stubbed_channel(broker=None, id=0): spec = get_spec(make_vumi_path_abs("config/amqp-spec-0-8.xml")) amq_client = FakeAMQClient(spec, {}, broker) return amq_client.channel(id)
def get_stubbed_worker(worker_class, config=None, broker=None): spec = get_spec(make_vumi_path_abs("config/amqp-spec-0-8.xml")) amq_client = FakeAMQClient(spec, {}, broker) worker = worker_class(amq_client, config) return worker