mirror of
https://github.com/django-q2/django-q2.git
synced 2026-09-15 13:37:56 +08:00
[cleanup] Fix some linting issues
This commit is contained in:
@@ -57,7 +57,6 @@ class Sqs(Broker):
|
||||
del(config['aws_region'])
|
||||
return Session(**config)
|
||||
|
||||
|
||||
def get_queue(self):
|
||||
self.sqs = self.connection.resource('sqs')
|
||||
return self.sqs.create_queue(QueueName=self.list_key)
|
||||
|
||||
@@ -60,7 +60,7 @@ class ORM(Broker):
|
||||
|
||||
def dequeue(self):
|
||||
tasks = self.get_connection().filter(key=self.list_key, lock__lt=_timeout())[
|
||||
0 : Conf.BULK
|
||||
0: Conf.BULK
|
||||
]
|
||||
if tasks:
|
||||
task_list = []
|
||||
|
||||
@@ -7,9 +7,8 @@ from django.db import IntegrityError
|
||||
from django.utils import timezone
|
||||
from multiprocessing import Value
|
||||
|
||||
from django_q.brokers import get_broker
|
||||
|
||||
# local
|
||||
from django_q.brokers import get_broker
|
||||
from django_q.conf import Conf, logger
|
||||
from django_q.humanhash import uuid
|
||||
from django_q.models import Schedule, Task
|
||||
|
||||
@@ -45,6 +45,7 @@ def test_sync(broker):
|
||||
task = async_task('django_q.tests.tasks.count_letters', DEFAULT_WORDLIST, broker=broker, sync=True)
|
||||
assert result(task) == 1506
|
||||
|
||||
|
||||
@pytest.mark.django_db
|
||||
def test_sync_raise_exception(broker):
|
||||
with pytest.raises(TaskError):
|
||||
@@ -401,6 +402,7 @@ def test_update_failed(broker):
|
||||
assert saved_task.success is True
|
||||
assert saved_task.result == 'result'
|
||||
|
||||
|
||||
@pytest.mark.django_db
|
||||
def test_acknowledge_failure_override():
|
||||
class VerifyAckMockBroker(Broker):
|
||||
@@ -434,10 +436,12 @@ def test_acknowledge_failure_override():
|
||||
|
||||
tag = uuid()
|
||||
task_success_ack = task_fail_ack.copy()
|
||||
task_success_ack.update({'id': tag[1],
|
||||
task_success_ack.update({
|
||||
'id': tag[1],
|
||||
'name': tag[0],
|
||||
'ack_id': 'test_success_ack_id',
|
||||
'success': True,})
|
||||
'success': True,
|
||||
})
|
||||
del task_success_ack['ack_failure']
|
||||
|
||||
result_queue = Queue()
|
||||
@@ -453,6 +457,7 @@ def test_acknowledge_failure_override():
|
||||
assert broker.acknowledgements.get('test_fail_no_ack_id') is None
|
||||
assert broker.acknowledgements.get('test_success_ack_id') == 1
|
||||
|
||||
|
||||
@pytest.mark.django_db
|
||||
def assert_result(task):
|
||||
assert task is not None
|
||||
|
||||
Reference in New Issue
Block a user