def setUp(self): self.today = datetime.utcnow().date() self.port = 9999 self.path = '/api/v1/sms/vas2nets/receive/' self.config = { 'transport_name': 'vas2nets', 'url': 'http://localhost:%s%s' % (self.port, self.path), 'username': '******', 'password': '******', 'owner': 'owner', 'service': 'service', 'subservice': 'subservice', 'web_receive_path': '/receive', 'web_receipt_path': '/receipt', 'web_port': 9998, } self.fail_config = { 'transport_name': 'vas2nets', 'retry_routing_key': '%(transport_name)s.outbound', 'failures_routing_key': '%(transport_name)s.failures', } self.workers = [] self.broker = FakeAMQPBroker() self.redis = FakeRedis() self.worker = yield self.mk_transport_worker(self.config, self.broker) self.fail_worker = yield self.mk_failure_worker( self.fail_config, self.broker, self.redis)
def _setup_workers(self, bucketters, aggregators, bucket_size): broker = FakeAMQPBroker() self.broker = BrokerWrapper(broker) bucket_config = { 'buckets': aggregators, 'bucket_size': bucket_size, } for _i in range(bucketters): worker = get_stubbed_worker(metrics_workers.MetricTimeBucket, config=bucket_config, broker=broker) yield worker.startWorker() self.bucket_workers.append(worker) aggregator_config = { 'bucket_size': bucket_size, } for i in range(aggregators): config = aggregator_config.copy() config['bucket'] = i worker = get_stubbed_worker(metrics_workers.MetricAggregator, config=config, broker=broker) worker._time = self.fake_time yield worker.startWorker() self.aggregator_workers.append(worker)
def setUp(self): warnings.warn( "VumiWorkerTestCase and its subclasses are deprecated. " "Use VumiTestCase and other tools from " "vumi.tests.helpers instead.", category=DeprecationWarning) self._workers = [] self._amqp = FakeAMQPBroker()
def test_publish_heartbeat(self): self.broker = FakeAMQPBroker() channel = yield get_stubbed_channel(self.broker) pub = MockHeartBeatPublisher(self.gen_fake_attrs) pub.start(channel) pub._beat() [msg] = self.broker.get_dispatched("vumi.health", "heartbeat.inbound") self.assertEqual(json.loads(msg.body), self.gen_fake_attrs())
def test_publish_heartbeat(self): broker = FakeAMQPBroker() client = WorkerHelper.get_fake_amqp_client(broker) channel = yield client.get_channel() pub = MockHeartBeatPublisher(self.gen_fake_attrs) pub.start(channel) pub._beat() [msg] = broker.get_dispatched("vumi.health", "heartbeat.inbound") self.assertEqual(json.loads(msg.body), self.gen_fake_attrs())
def __init__(self, spec): super(FakeAmqpClient, self).__init__(TwistedDelegate(), '', spec) self.broker = FakeAMQPBroker()
def setUp(self): self._workers = [] self._amqp = FakeAMQPBroker()