def test_k8s_create_pod():
mock = MagicMock()
with patch('intel.k8s.client_from_config', MagicMock(return_value=mock)):
k8s.get_pod_list(None)
called_methods = mock.method_calls
assert len(called_methods) == 1
assert called_methods[0][0] == "list_pod_for_all_namespaces"
def test_k8s_create_pod():
mock = MagicMock()
with patch(CLIENT_CONFIG, MagicMock(return_value=mock)):
k8s.get_pod_list(None)
called_methods = mock.method_calls
assert len(called_methods) == 1
assert called_methods[0][0] == "list_pod_for_all_namespaces"
def wait_for_pod_phase(pod_name, phase_name):
wait = True
while wait:
try:
pod_list_resp = k8s.get_pod_list(None)
except K8sApiException as err:
logging.error("Exception while waiting for Pod [{}] status: {}"
.format(pod_name, err))
logging.error("Aborting cluster-init ...")
sys.exit(1)
for pod in pod_list_resp["items"]:
if ("metadata" in pod) and ("name" in pod["metadata"]) \
and pod_name in pod["metadata"]["name"]:
if pod["status"]["phase"] == phase_name:
wait = False
break
elif pod["status"]["phase"] == "Failed":
raise RuntimeError("The Pod {} went into Failed state"
.format(pod_name))