File tree Expand file tree Collapse file tree 1 file changed +24
-0
lines changed Expand file tree Collapse file tree 1 file changed +24
-0
lines changed Original file line number Diff line number Diff line change @@ -130,3 +130,27 @@ def test_bulk_enqueues_jobs(self) -> None:
130
130
job .as_dict (),
131
131
"Second job" ,
132
132
)
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
+ )
You can’t perform that action at this time.
0 commit comments