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"]
