Exemplo n.º 1
0
def test_stopping_invalid_worker_pipeline():
    worker_validator.validate_hosts()

    assert_first_argument_contained(executor.execute, 'pipe stop ')
    assert_first_argument_contained(executor.execute, HOST2)
    assert_first_argument_not_contained(executor.execute, HOST1)
    assert_first_argument_not_contained(executor.execute, HOST3)
Exemplo n.º 2
0
def test_kill_jobs():
    jobs = [
        GridEngineJob(id=1, name='', user='', state='', datetime=''),
        GridEngineJob(id=2, name='', user='', state='', datetime='')
    ]

    grid_engine.kill_jobs(jobs)
    assert_first_argument_contained(executor.execute, 'qdel ')
    assert_first_argument_contained(executor.execute, ' 1 2')
    assert_first_argument_not_contained(executor.execute, '-f')
Exemplo n.º 3
0
def test_scaling_down_if_host_has_no_running_jobs():
    submit_datetime = datetime(2018, 12, 29, 11, 00, 00)
    jobs = [
        GridEngineJob(id=1,
                      name='name1',
                      user='******',
                      state=GridEngineJobState.RUNNING,
                      datetime=submit_datetime,
                      host=ANOTHER_HOSTNAME)
    ]
    grid_engine.get_jobs = MagicMock(return_value=jobs)

    assert scale_down_handler.scale_down(HOSTNAME)

    grid_engine.disable_host.assert_called()
    grid_engine.enable_host.assert_not_called()
    grid_engine.delete_host.assert_called()
    assert_first_argument_contained(cmd_executor.execute, HOSTNAME)
Exemplo n.º 4
0
def test_updating_default_hostfile():
    scale_up_handler.scale_up()

    assert_first_argument_contained(cmd_executor.execute, HOSTNAME)
    assert_first_argument_contained(cmd_executor.execute, default_hostfile)
Exemplo n.º 5
0
def test_updating_hosts():
    scale_up_handler.scale_up()

    assert_first_argument_contained(cmd_executor.execute, '%s\t%s' % (POD_IP, HOSTNAME))
    assert_first_argument_contained(cmd_executor.execute, '/etc/hosts')
Exemplo n.º 6
0
def test_scaling_down_updates_hosts():
    scale_down_handler.scale_down(HOSTNAME)

    assert_first_argument_contained(cmd_executor.execute, HOSTNAME)
    assert_first_argument_contained(cmd_executor.execute, '/etc/hosts')
Exemplo n.º 7
0
def test_scaling_down_stops_pipeline():
    scale_down_handler.scale_down(HOSTNAME)

    assert_first_argument_contained(cmd_executor.execute, 'pipe stop')