[cleanup] Fix some linting issues

This commit is contained in:
Bastien Vallet
2020-06-10 12:00:55 +02:00
parent 68d95a2924
commit 3569d2ea8c
5 changed files with 15 additions and 12 deletions
-1
View File
@@ -57,7 +57,6 @@ class Sqs(Broker):
del(config['aws_region']) del(config['aws_region'])
return Session(**config) return Session(**config)
def get_queue(self): def get_queue(self):
self.sqs = self.connection.resource('sqs') self.sqs = self.connection.resource('sqs')
return self.sqs.create_queue(QueueName=self.list_key) return self.sqs.create_queue(QueueName=self.list_key)
+1 -2
View File
@@ -7,9 +7,8 @@ from django.db import IntegrityError
from django.utils import timezone from django.utils import timezone
from multiprocessing import Value from multiprocessing import Value
from django_q.brokers import get_broker
# local # local
from django_q.brokers import get_broker
from django_q.conf import Conf, logger from django_q.conf import Conf, logger
from django_q.humanhash import uuid from django_q.humanhash import uuid
from django_q.models import Schedule, Task from django_q.models import Schedule, Task
+7 -2
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) task = async_task('django_q.tests.tasks.count_letters', DEFAULT_WORDLIST, broker=broker, sync=True)
assert result(task) == 1506 assert result(task) == 1506
@pytest.mark.django_db @pytest.mark.django_db
def test_sync_raise_exception(broker): def test_sync_raise_exception(broker):
with pytest.raises(TaskError): with pytest.raises(TaskError):
@@ -401,6 +402,7 @@ def test_update_failed(broker):
assert saved_task.success is True assert saved_task.success is True
assert saved_task.result == 'result' assert saved_task.result == 'result'
@pytest.mark.django_db @pytest.mark.django_db
def test_acknowledge_failure_override(): def test_acknowledge_failure_override():
class VerifyAckMockBroker(Broker): class VerifyAckMockBroker(Broker):
@@ -434,10 +436,12 @@ def test_acknowledge_failure_override():
tag = uuid() tag = uuid()
task_success_ack = task_fail_ack.copy() task_success_ack = task_fail_ack.copy()
task_success_ack.update({'id': tag[1], task_success_ack.update({
'id': tag[1],
'name': tag[0], 'name': tag[0],
'ack_id': 'test_success_ack_id', 'ack_id': 'test_success_ack_id',
'success': True,}) 'success': True,
})
del task_success_ack['ack_failure'] del task_success_ack['ack_failure']
result_queue = Queue() 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_fail_no_ack_id') is None
assert broker.acknowledgements.get('test_success_ack_id') == 1 assert broker.acknowledgements.get('test_success_ack_id') == 1
@pytest.mark.django_db @pytest.mark.django_db
def assert_result(task): def assert_result(task):
assert task is not None assert task is not None