Skip to content

Commit 9ec30c4

Browse files
committed
Add __init__ file to tests module.
1 parent 5e852f6 commit 9ec30c4

File tree

2 files changed

+15
-14
lines changed

2 files changed

+15
-14
lines changed

tests/__init__.py

Whitespace-only changes.

tests/worker.py

Lines changed: 15 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ def stateless_task() -> None:
4242
pass
4343

4444
stateless_task.enqueue()
45-
assert Job.objects.filter(task="worker.stateless_task").first()
45+
assert Job.objects.filter(task="tests.worker.stateless_task").first()
4646

4747
worker.run_available_tasks()
4848
assert not Job.objects.exists()
@@ -53,7 +53,7 @@ def stateful_task() -> dict[str, str]:
5353
return {"some": "data"}
5454

5555
enqueued_task = stateful_task.enqueue()
56-
assert Job.objects.filter(task="worker.stateful_task").first()
56+
assert Job.objects.filter(task="tests.worker.stateful_task").first()
5757

5858
worker.run_available_tasks()
5959

@@ -88,7 +88,7 @@ def stateful_task_that_throws() -> None:
8888
worker.run_available_tasks()
8989

9090
assert not Job.objects.exists()
91-
failed_job = FailedJob.objects.get(task="worker.stateful_task_that_throws")
91+
failed_job = FailedJob.objects.get(task="tests.worker.stateful_task_that_throws")
9292
assert failed_job.failure_reason == "exception"
9393
assert failed_job.job_id == enqueued_task.job.id
9494

@@ -116,7 +116,7 @@ def stateless_task_that_throws() -> None:
116116
worker.run_available_tasks()
117117

118118
assert not Job.objects.exists()
119-
failed_job = FailedJob.objects.get(task="worker.stateless_task_that_throws")
119+
failed_job = FailedJob.objects.get(task="tests.worker.stateless_task_that_throws")
120120
assert failed_job.failure_reason == "exception"
121121
assert failed_job.job_id == enqueued_task.job.id
122122

@@ -136,7 +136,7 @@ def expired_task() -> None:
136136

137137
expire_at = timezone.now() - timedelta(minutes=1)
138138
expired_task.enqueue(expire_at=expire_at)
139-
job = Job.objects.get(task="worker.expired_task")
139+
job = Job.objects.get(task="tests.worker.expired_task")
140140
worker.run_available_tasks()
141141

142142
assert not mock.whatever.called
@@ -157,7 +157,7 @@ def retrying_task() -> None:
157157
job = Job.objects.get()
158158
worker.run_available_tasks()
159159

160-
retried_job = Job.objects.get(task="worker.retrying_task")
160+
retried_job = Job.objects.get(task="tests.worker.retrying_task")
161161
assert job.id != retried_job.id
162162
assert retried_job.retry_count == 1
163163
assert retried_job.execute_at > job.execute_at
@@ -184,20 +184,21 @@ def task_that_exceeds_its_time_limit() -> None:
184184
time.sleep(5)
185185

186186
enqueued_task = task_that_exceeds_its_time_limit.enqueue()
187-
Job.objects.get(task="worker.task_that_exceeds_its_time_limit")
187+
Job.objects.get(task="tests.worker.task_that_exceeds_its_time_limit")
188188

189189
worker.run_available_tasks()
190190
log_records = [
191191
r.message
192192
for r in caplog.records
193193
if (
194-
r.message == "worker.task_that_exceeds_its_time_limit reached its time limit" and r.levelno == logging.ERROR
194+
r.message == "tests.worker.task_that_exceeds_its_time_limit reached its time limit"
195+
and r.levelno == logging.ERROR
195196
)
196197
]
197198
assert len(log_records) == 1
198199

199200
assert not Job.objects.exists()
200-
failed_job = FailedJob.objects.get(task="worker.task_that_exceeds_its_time_limit")
201+
failed_job = FailedJob.objects.get(task="tests.worker.task_that_exceeds_its_time_limit")
201202
assert failed_job.failure_reason == "exception"
202203
assert failed_job.job_id == enqueued_task.job.id
203204

@@ -211,7 +212,7 @@ def task_that_takes_an_unexpectedly_long_time() -> None:
211212
time.sleep(3)
212213

213214
task_that_takes_an_unexpectedly_long_time.enqueue()
214-
Job.objects.get(task="worker.task_that_takes_an_unexpectedly_long_time")
215+
Job.objects.get(task="tests.worker.task_that_takes_an_unexpectedly_long_time")
215216

216217
worker.run_available_tasks()
217218
log_records = [
@@ -237,12 +238,12 @@ def atomic_task() -> None:
237238
raise AssertionError
238239

239240
enqueued_task = atomic_task.enqueue()
240-
job = Job.objects.get(task="worker.atomic_task")
241+
job = Job.objects.get(task="tests.worker.atomic_task")
241242

242243
worker.run_available_tasks()
243244

244245
assert not Job.objects.exists()
245-
failed_job = FailedJob.objects.get(task="worker.atomic_task")
246+
failed_job = FailedJob.objects.get(task="tests.worker.atomic_task")
246247
assert failed_job.failure_reason == "exception"
247248
assert failed_job.job_id == job.id == enqueued_task.job.id
248249

@@ -259,12 +260,12 @@ def non_atomic_task() -> None:
259260
raise AssertionError
260261

261262
enqueued_task = non_atomic_task.enqueue()
262-
job = Job.objects.get(task="worker.non_atomic_task")
263+
job = Job.objects.get(task="tests.worker.non_atomic_task")
263264

264265
worker.run_available_tasks()
265266

266267
assert not Job.objects.exists()
267-
failed_job = FailedJob.objects.get(task="worker.non_atomic_task")
268+
failed_job = FailedJob.objects.get(task="tests.worker.non_atomic_task")
268269
assert failed_job.failure_reason == "exception"
269270
assert failed_job.job_id == job.id == enqueued_task.job.id
270271

0 commit comments

Comments
 (0)