Python instance_tasks Examples

Example #1

def test_instance_tasks(mock_get_tasks_from_app_id, 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_task_2 = mock.Mock()
    mock_get_tasks_from_app_id.return_value = [mock_task_1, mock_task_2]
    ret = instance.instance_tasks(mock_request)
    assert not mock_add_slave_info.called
    assert not mock_add_executor_info.called

    mock_request = mock.Mock(swagger_data={'task_id': '123', 'slave_hostname': 'host1', 'verbose': True})
    ret = instance.instance_tasks(mock_request)
    mock_add_executor_info.assert_has_calls([mock.call(mock_task_1), mock.call(mock_task_2)])
    mock_add_slave_info.assert_has_calls([
        mock.call(mock_add_executor_info.return_value),
        mock.call(mock_add_executor_info.return_value),
    ])
    expected = [
        mock_add_slave_info.return_value._Task__items,
        mock_add_slave_info.return_value._Task__items,
    ]

    def ids(l):
        return {id(x) for x in l}
    assert len(ret) == len(expected) and ids(expected) == ids(ret)

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

Example #2

File: test_instance.py Project: eric-erki/An-open-distributed-platform-as-a-service

def test_instance_tasks(
    mock_get_tasks_from_app_id,
    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": {
            "desired_app_id": "app1"
        }
    }

    mock_task_1 = mock.Mock()
    mock_task_2 = mock.Mock()
    mock_get_tasks_from_app_id.return_value = [mock_task_1, mock_task_2]
    ret = instance.instance_tasks(mock_request)
    assert not mock_add_slave_info.called
    assert not mock_add_executor_info.called

    mock_request = mock.Mock(swagger_data={
        "task_id": "123",
        "slave_hostname": "host1",
        "verbose": True
    })
    ret = instance.instance_tasks(mock_request)
    mock_add_executor_info.assert_has_calls(
        [mock.call(mock_task_1),
         mock.call(mock_task_2)])
    mock_add_slave_info.assert_has_calls([
        mock.call(mock_add_executor_info.return_value),
        mock.call(mock_add_executor_info.return_value),
    ])
    expected = [
        mock_add_slave_info.return_value._Task__items,
        mock_add_slave_info.return_value._Task__items,
    ]

    def ids(l):
        return {id(x) for x in l}

    assert len(ret) == len(expected) and ids(expected) == ids(ret)

Example #3

File: test_instance.py Project: oktopuz/paasta

def test_instance_tasks(mock_get_tasks_from_app_id, 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_task_2 = mock.Mock()
    mock_get_tasks_from_app_id.return_value = [mock_task_1, mock_task_2]
    ret = instance.instance_tasks(mock_request)
    assert not mock_add_slave_info.called
    assert not mock_add_executor_info.called

    mock_request = mock.Mock(swagger_data={'task_id': '123', 'slave_hostname': 'host1', 'verbose': True})
    ret = instance.instance_tasks(mock_request)
    mock_add_executor_info.assert_has_calls([mock.call(mock_task_1), mock.call(mock_task_2)])
    mock_add_slave_info.assert_has_calls([mock.call(mock_add_executor_info.return_value),
                                          mock.call(mock_add_executor_info.return_value)])
    expected = [mock_add_slave_info.return_value._Task__items,
                mock_add_slave_info.return_value._Task__items]
    assert len(ret) == len(expected) and sorted(expected) == sorted(ret)

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