File tree Expand file tree Collapse file tree 1 file changed +40
-0
lines changed Expand file tree Collapse file tree 1 file changed +40
-0
lines changed Original file line number Diff line number Diff line change @@ -213,6 +213,33 @@ def two_queues_neither_responsive(monkeypatch):
213
213
)
214
214
215
215
216
+ @pytest .fixture
217
+ def three_queues_two_candidates (monkeypatch ):
218
+ return _mock_up_queues (
219
+ monkeypatch ,
220
+ {
221
+ "queue1" : {
222
+ "workers" : {
223
+ "worker1_1" : {"responsive" : True },
224
+ },
225
+ "process_count" : 1 ,
226
+ },
227
+ "queue2" : {
228
+ "workers" : {
229
+ "worker2_1" : {"responsive" : True },
230
+ },
231
+ "process_count" : 1 ,
232
+ },
233
+ "queue3" : {
234
+ "workers" : {
235
+ "worker3_1" : {"responsive" : True },
236
+ },
237
+ "process_count" : 2 ,
238
+ },
239
+ },
240
+ )
241
+
242
+
216
243
@pytest .mark .parametrize (
217
244
"fixture" ,
218
245
[
@@ -240,6 +267,19 @@ def test_get_least_busy_queue_name(fixture, request):
240
267
get_least_busy_queue_name ()
241
268
242
269
270
+ def test_get_least_busy_queue_name_multiple_queues (
271
+ three_queues_two_candidates ,
272
+ ):
273
+ expected_queues = [
274
+ "queue1" ,
275
+ "queue2" ,
276
+ ]
277
+ unexpected_queue = "queue3"
278
+ queue = get_least_busy_queue_name ()
279
+ assert queue in expected_queues
280
+ assert queue != unexpected_queue
281
+
282
+
243
283
@pytest .fixture
244
284
def multi_queue_various_states (monkeypatch ):
245
285
return _mock_up_queues (
You can’t perform that action at this time.
0 commit comments