Skip to content
Snippets Groups Projects

Launch jobs with variable nice value

Merged Philipp Schmidt requested to merge feat/degressive-slurm-priority into master
Files
3
+ 7
5
@@ -144,13 +144,15 @@ async def test_wait_on_transfer_exceptions(
('sim', ['DARK', '1', '2', '3', '4'], 1, "success: simulated"),
],
)
async def test_run_action(mode, cmd, retcode, expected):
async def test_run_action(mode, cmd, retcode, expected, monkeypatch):
job_db = mock.Mock()
async def mock_run_proc_async(*args):
return retcode, b'Submitted job: 42'
webservice.webservice.run_proc_async = mock_run_proc_async
monkeypatch.setattr(
webservice.webservice, 'run_proc_async', mock_run_proc_async
)
ret = await run_action(job_db, cmd, mode, 1, 1, 1)
assert ret.lower().startswith(expected)
@@ -191,9 +193,9 @@ async def test_get_slurm_partition(proposal_number,
async def test_get_slurm_nice(fake_process, cycle, num_jobs, expected_result):
fake_process.register(
['squeue', '-h', '-o', '%.20j', '-p', 'exfel', '--me'],
stdout=b'\n'.join(
[f'correct_SPB_{i}'.encode('ascii') for i in range(num_jobs)] +
[f'correct_FXE_{i}'.encode('ascii') for i in range(num_jobs*2)]),
stdout='\n'.join(
[f'correct_SPB_{i}' for i in range(num_jobs)] +
[f'correct_FXE_{i}' for i in range(num_jobs*2)]).encode('ascii'),
returncode=0)
ret = await get_slurm_nice(
Loading