[cleanup] Fix some linting issues

This commit is contained in:
Bastien Vallet
2020-06-10 11:52:31 +02:00
parent 68d95a2924
commit 3569d2ea8c
5 changed files with 15 additions and 12 deletions

View File

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

View File

@@ -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 = []

View File

@@ -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

View File

@@ -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