def test_clusterinit_get_cmk_node_list_all_hosts_error(caplog): fake_http_resp = FakeHTTPResponse(500, "fake reason", "fake body") fake_api_exception = K8sApiException(http_resp=fake_http_resp) with pytest.raises(SystemExit): with patch('intel.k8s.get_compute_nodes', MagicMock(side_effect=fake_api_exception)): clusterinit.get_cmk_node_list(None, True) exp_err = "Exception when getting the node list: {}" exp_log_err = get_expected_log_error(exp_err) caplog_tuple = caplog.record_tuples assert caplog_tuple[-2][2] == exp_log_err exp_err = "Aborting cluster-init ..." assert caplog_tuple[-1][2] == exp_err
def test_k8s_node_list_all(): fake_node_list_resp = [ {"metadata": {"name": "fakenode1"}, "spec": {}}, {"metadata": {"name": "fakenode2"}, "spec": {}}, {"metadata": {"name": "fakenode3"}, "spec": {}} ] with patch('intel.k8s.get_node_list', MagicMock(return_value=fake_node_list_resp)): node_list = clusterinit.get_cmk_node_list(None, True) assert node_list == ["fakenode1", "fakenode2", "fakenode3"]
def test_clusterinit_node_list_host_list(): fake_node_list = "fakenode1, fakenode2, fakenode3" node_list = clusterinit.get_cmk_node_list(fake_node_list, False) assert node_list == ["fakenode1", "fakenode2", "fakenode3"]