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()
