Example #1
0
 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
Example #2
0
 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
Example #3
0
    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)
Example #4
0
    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)
Example #5
0
 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__()
Example #6
0
    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)
Example #7
0
def get_fake_amq_client(broker=None):
    spec = get_spec(vumi_resource_path("amqp-spec-0-8.xml"))
    return FakeAMQClient(spec, {}, broker)
Example #8
0
def get_fake_amq_client(broker=None):
    spec = get_spec(vumi_resource_path("amqp-spec-0-8.xml"))
    return FakeAMQClient(spec, {}, broker)
Example #9
0
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)
Example #10
0
 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)
Example #11
0
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
Example #12
0
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)
Example #13
0
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)
Example #14
0
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
Example #15
0
 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)
Example #16
0
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