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_empty_worker(self): sqs = SQSConnection(region=get_sqs_region('us-east-1')) self.assertEqual(len(sqs.get_all_queues()), 0) with spawn_worker(Worker()): # Worker is empty, not registering any queues self.assertEqual(len(sqs.get_all_queues()), 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 test_region(self): region = get_sqs_region('us-east-1') self.assertEqual(region.name, 'us-east-1')
def _create_queue(self, name): sqs = SQSConnection(region=get_sqs_region('us-east-1')) self.assertEqual(len(sqs.get_all_queues()), 0) queue = sqs.create_queue(name) return queue