Changeset View
Changeset View
Standalone View
Standalone View
swh/scheduler/tests/test_config.py
| Show All 39 Lines | |||||
| def test_get_available_slots_no_queue_length(mocker): | def test_get_available_slots_no_queue_length(mocker): | ||||
| mock = mocker.patch("swh.scheduler.celery_backend.config.get_queue_length") | mock = mocker.patch("swh.scheduler.celery_backend.config.get_queue_length") | ||||
| mock.return_value = None | mock.return_value = None | ||||
| actual_num = get_available_slots(app, "anything", max_length=100) | actual_num = get_available_slots(app, "anything", max_length=100) | ||||
| assert actual_num == MAX_NUM_TASKS | assert actual_num == MAX_NUM_TASKS | ||||
| assert mock.called | assert mock.called | ||||
| def test_get_available_slots_no_more_slots(mocker): | |||||
| mock = mocker.patch("swh.scheduler.celery_backend.config.get_queue_length") | |||||
| max_length = 100 | |||||
| queue_length = 9000 | |||||
| mock.return_value = queue_length | |||||
| actual_num = get_available_slots(app, "anything", max_length) | |||||
| assert actual_num == 0 | |||||
| assert mock.called | |||||
| def test_get_available_slots(mocker): | def test_get_available_slots(mocker): | ||||
| mock = mocker.patch("swh.scheduler.celery_backend.config.get_queue_length") | mock = mocker.patch("swh.scheduler.celery_backend.config.get_queue_length") | ||||
| max_length = 100 | max_length = 100 | ||||
| queue_length = 90 | queue_length = 90 | ||||
| mock.return_value = queue_length | mock.return_value = queue_length | ||||
| actual_num = get_available_slots(app, "anything", max_length) | actual_num = get_available_slots(app, "anything", max_length) | ||||
| assert actual_num == max_length - queue_length | assert actual_num == max_length - queue_length | ||||
| assert mock.called | assert mock.called | ||||