def test_worker_consumes_queue(self):
sqs = SQSConnection(region=get_sqs_region('us-east-1'))
self.assertEqual(len(sqs.get_all_queues()), 0)
queue = sqs.create_queue('test_events')
queue.write(make_message({'test': '1'}))
self.assertEqual(queue.count(), 1)
worker = Worker()
worker.ctrl.wait_time_seconds = 0.1
worker.idle_time_seconds = 0.1
worker.add_consumer('test_events', dummy_consumer)
with spawn_worker(worker):
time.sleep(.2)
self.assertEqual(queue.count(), 0)
queue.write(make_message({'test': '2'}))
self.assertEqual(queue.count(), 1)
time.sleep(.2)
self.assertEqual(queue.count(), 0)
def test_worker_creates_queue(self):
sqs = SQSConnection(region=get_sqs_region('us-east-1'))
self.assertEqual(len(sqs.get_all_queues()), 0)
worker = Worker()
worker.ctrl.wait_time_seconds = 0.1
worker.idle_time_seconds = 0.1
worker.add_consumer('test_events', dummy_consumer)
with spawn_worker(worker):
time.sleep(.2)
all_queues = sqs.get_all_queues()
self.assertEqual(len(all_queues), 1)
self.assertEqual(
all_queues[0].name,
'test_events'
)
def launch_worker(args):
from zing.worker import Worker
import zing.consumers
try:
worker = Worker(credentials={
'aws_access_key_id': args.aws_key,
'aws_secret_access_key': args.aws_secret
})
worker.add_consumer(
queue_name='zing.event',
consumer=zing.consumers.consume_events,
)
worker.run()
except KeyboardInterrupt:
print 'Shutting down'
worker.stop()
