test_marathon.py 文件源码

def test_kill_pod_instances_executes_successfully():
    pod_id = 'foo'
    instance_ids = ['instance1', 'instance2']
    path = 'v2/pods/foo::instances'
    response_json = {'some': ['instance', 'status']}

    mock_response = mock.create_autospec(requests.Response)
    mock_response.json.return_value = response_json

    marathon_client, rpc_client = _create_fixtures()
    rpc_client.http_req.return_value = mock_response

    path_format_method_name = 'dcos.marathon.Client._marathon_id_path_format'
    path_format = mock.MagicMock()
    path_format.return_value = path

    with mock.patch(path_format_method_name, new=path_format):
        actual_json = marathon_client.kill_pod_instances(pod_id, instance_ids)

    path_format.assert_called_with('v2/pods/{}::instances', pod_id)
    rpc_client.http_req.assert_called_with(
        http.delete, path, json=instance_ids)
    assert actual_json == response_json