Skip to content

Commit a5d0f5d

Browse files
committed
Validate behaviour when enqueuing more than the batch size
1 parent d805dd2 commit a5d0f5d

File tree

1 file changed

+24
-0
lines changed

1 file changed

+24
-0
lines changed

tests/test_task.py

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -130,3 +130,27 @@ def test_bulk_enqueues_jobs(self) -> None:
130130
job.as_dict(),
131131
"Second job",
132132
)
133+
134+
def test_bulk_enqueues_jobs_batch_size_boundary(self) -> None:
135+
self.assertEqual(0, self.backend.length(QUEUE), "Should initially be empty")
136+
137+
with dummy_task.bulk_enqueue(batch_size=3) as enqueue:
138+
enqueue(1)
139+
enqueue(2)
140+
enqueue(3)
141+
enqueue(4)
142+
143+
jobs = [
144+
self.backend.dequeue(QUEUE, WorkerNumber(0), 5)
145+
for _ in range(4)
146+
]
147+
148+
self.assertEqual(0, self.backend.length(QUEUE), "Should be empty after dequeuing all jobs")
149+
150+
args = [x.args for x in jobs if x is not None]
151+
152+
self.assertEqual(
153+
[[1], [2], [3], [4]],
154+
args,
155+
"Wrong jobs bulk enqueued",
156+
)

0 commit comments

Comments
 (0)