Skip to content

Commit 789f10a

Browse files
add test
Signed-off-by: Alex <aizquier@redhat.com>
1 parent f1412b2 commit 789f10a

File tree

1 file changed

+40
-0
lines changed

1 file changed

+40
-0
lines changed

tests/unit/test_orchestrator.py

Lines changed: 40 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -213,6 +213,33 @@ def two_queues_neither_responsive(monkeypatch):
213213
)
214214

215215

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+
216243
@pytest.mark.parametrize(
217244
"fixture",
218245
[
@@ -240,6 +267,19 @@ def test_get_least_busy_queue_name(fixture, request):
240267
get_least_busy_queue_name()
241268

242269

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+
243283
@pytest.fixture
244284
def multi_queue_various_states(monkeypatch):
245285
return _mock_up_queues(

0 commit comments

Comments
 (0)