Exemplos de instance_task em Python

Exemplo n.º 1

Arquivo: test_instance.py Projeto: eric-erki/An-open-distributed-platform-as-a-service

def test_instance_task(mock_get_task, mock_instance_status,
                       mock_add_slave_info, mock_add_executor_info):
    mock_request = mock.Mock(swagger_data={
        "task_id": "123",
        "slave_hostname": "host1"
    })
    mock_instance_status.return_value = {"marathon": {"app_id": "app1"}}

    mock_task_1 = mock.Mock()
    mock_get_task.return_value = mock_task_1
    ret = instance.instance_task(mock_request)
    assert not mock_add_slave_info.called
    assert not mock_add_executor_info.called
    assert ret == mock_task_1._Task__items

    mock_request = mock.Mock(swagger_data={
        "task_id": "123",
        "slave_hostname": "host1",
        "verbose": True
    })
    ret = instance.instance_task(mock_request)
    mock_add_slave_info.assert_called_with(mock_task_1)
    mock_add_executor_info.assert_called_with(mock_add_slave_info.return_value)
    expected = mock_add_executor_info.return_value._Task__items
    assert ret == expected

Exemplo n.º 2

def test_instance_task(mock_get_task, mock_instance_status,
                       mock_add_slave_info, mock_add_executor_info):
    mock_request = mock.Mock(swagger_data={
        'task_id': '123',
        'slave_hostname': 'host1'
    })
    mock_instance_status.return_value = {'marathon': {'app_id': 'app1'}}

    mock_task_1 = mock.Mock()
    mock_get_task.return_value = mock_task_1
    ret = instance.instance_task(mock_request)
    assert not mock_add_slave_info.called
    assert not mock_add_executor_info.called
    assert ret == mock_task_1._Task__items

    mock_request = mock.Mock(swagger_data={
        'task_id': '123',
        'slave_hostname': 'host1',
        'verbose': True
    })
    ret = instance.instance_task(mock_request)
    mock_add_slave_info.assert_called_with(mock_task_1)
    mock_add_executor_info.assert_called_with(mock_add_slave_info.return_value)
    expected = mock_add_executor_info.return_value._Task__items
    assert ret == expected

    mock_instance_status.return_value = {'chronos': {}}
    with raises(ApiFailure):
        ret = instance.instance_task(mock_request)

Exemplo n.º 3

Arquivo: test_instance.py Projeto: oktopuz/paasta

def test_instance_task(mock_get_task, mock_instance_status, mock_add_slave_info, mock_add_executor_info):
    mock_request = mock.Mock(swagger_data={'task_id': '123', 'slave_hostname': 'host1'})
    mock_instance_status.return_value = {'marathon': {'app_id': 'app1'}}

    mock_task_1 = mock.Mock()
    mock_get_task.return_value = mock_task_1
    ret = instance.instance_task(mock_request)
    assert not mock_add_slave_info.called
    assert not mock_add_executor_info.called
    assert ret == mock_task_1._Task__items

    mock_request = mock.Mock(swagger_data={'task_id': '123', 'slave_hostname': 'host1', 'verbose': True})
    ret = instance.instance_task(mock_request)
    mock_add_slave_info.assert_called_with(mock_task_1)
    mock_add_executor_info.assert_called_with(mock_add_slave_info.return_value)
    expected = mock_add_executor_info.return_value._Task__items
    assert ret == expected

    mock_instance_status.return_value = {'chronos': {}}
    with raises(ApiFailure):
        ret = instance.instance_task(mock_request)