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))