Beispiel #1
0
    def test_delete_queue(self):
        sid = uuid4().hex
        queue = Queue(sid)

        self.assertIsInstance(queue.queue, SQSQueue)
        queue.delete()

        assert not connect_to_region("us-east-1").get_queue(sid)
Beispiel #2
0
    def test_delete_queue(self):
        sid = uuid4().hex
        queue = Queue(sid)

        self.assertIsInstance(queue.queue, SQSQueue)
        queue.delete()

        assert not connect_to_region('us-east-1').get_queue(sid)
Beispiel #3
0
    def test_remove_job(self):
        sid = uuid4().hex
        queue = Queue(sid)

        queue.add_job(Job(test_job, "there"))
        queue.add_job(Job(test_job, arg1="test", arg2="test"))

        for job in queue.jobs:
            queue.remove_job(job)

        sleep(10)

        self.assertEqual(queue.num_jobs(), 0)
        queue.delete()
Beispiel #4
0
    def test_add_job(self):
        sid = uuid4().hex
        queue = Queue(sid)

        queue.add_job(Job(test_job, 'there'))
        queue.add_job(Job(test_job, arg1='test', arg2='test'))

        sleep(10)

        self.assertEqual(queue.num_jobs(), 2)
        queue.delete()
Beispiel #5
0
    def test_add_job(self):
        sid = uuid4().hex
        queue = Queue(sid)

        queue.add_job(Job(test_job, "there"))
        queue.add_job(Job(test_job, arg1="test", arg2="test"))

        sleep(10)

        self.assertEqual(queue.num_jobs(), 2)
        queue.delete()
    def handle(self, *args, **options):
        queue_name = options.get("queue_name") or getattr(
            settings, "QUEUE_NAME")
        if not queue_name:
            raise ValueError("Invalid queue_name")

        self.stdout.write(f"Listening to {queue_name} queue")
        self.queue = Queue(queue_name)

        self.stdout.write("Running simpleq worker")
        self.worker = Worker(queues=[self.queue])
        self.worker.work()
Beispiel #7
0
    def test_work(self):
        sid = uuid4().hex
        queue = Queue(sid)
        queue.add_job(Job(test_job, 'hi', 'there'))

        sleep(10)

        self.assertEqual(queue.num_jobs(), 1)

        worker = Worker([queue])
        worker.work(True)

        sleep(10)

        self.assertEqual(queue.num_jobs(), 0)
        queue.delete()
Beispiel #8
0
    def test_work(self):
        sid = uuid4().hex
        queue = Queue(sid)
        queue.add_job(Job(test_job, 'hi', 'there'))

        sleep(10)

        self.assertEqual(queue.num_jobs(), 1)

        worker = Worker([queue])
        worker.work(True)

        sleep(10)

        self.assertEqual(queue.num_jobs(), 0)
        queue.delete()
Beispiel #9
0
def run():
    try:
        job_id = str(uuid.uuid4())
        uid = str(uuid.uuid4())
        print("Creating queue")
        q = Queue("simpleq_test")
        job = Job(job_id, None, log_stuff, uid)
        print(f"job_id: {job_id}")
        print(f"uid: {uid}")

        for n in range(10):
            q.add_job(job, delay=5)

        sleep(10)
        print(f"q.num_jobs(): {q.num_jobs()}")

    finally:
        q.delete()
Beispiel #10
0
    def test_remove_job(self):
        sid = uuid4().hex
        queue = Queue(sid)

        queue.add_job(Job(test_job, 'there'))
        queue.add_job(Job(test_job, arg1='test', arg2='test'))

        for job in queue.jobs:
            queue.remove_job(job)

        sleep(10)

        self.assertEqual(queue.num_jobs(), 0)
        queue.delete()
Beispiel #11
0
    def test_create_queue(self):
        sid = uuid4().hex
        queue = Queue(sid)

        self.assertIsInstance(queue.queue, SQSQueue)
        queue.delete()
Beispiel #12
0
    def test_lazy_create_queue(self):
        sid = uuid4().hex
        Queue(sid)

        self.assertFalse(connect_to_region('us-east-1').get_queue(sid))
Beispiel #13
0
    def test_create_queue(self):
        sid = uuid4().hex
        queue = Queue(sid)

        self.assertIsInstance(queue.queue, SQSQueue)
        queue.delete()
Beispiel #14
0
    def test_initialize(self):
        sid = uuid4().hex
        queue = Queue(sid)
        worker = Worker([queue])

        self.assertIsInstance(worker, Worker)