add flag for checking kube config is loaded

Signed-off-by: Javan Lacerda <[email protected]>

Author: javanlacerda

src/clusterfuzz/_internal/k8s/service.py

@@ -219,13 +219,10 @@ def _create_job_body(config: KubernetesJobConfig, input_url: str) -> dict:
 class KubernetesService(RemoteTaskInterface):
   """A remote task execution client for Kubernetes."""
 
-  def __init__(self):
+  def __init__(self, k8s_config_loaded: bool = False):
     # In e2e tests, the kubeconfig is already loaded by the test setup.
-    if not os.getenv('K8S_E2E'):
-      try:
-        k8s_config.load_kube_config()
-      except (k8s_config.ConfigException, TypeError):
-        self._load_gke_credentials()
+    if not k8s_config_loaded:
+      self._load_gke_credentials()
 
     self._core_api = k8s_client.CoreV1Api()
     self._batch_api = k8s_client.BatchV1Api()

src/clusterfuzz/_internal/tests/core/k8s/k8s_service_e2e_test.py

@@ -152,7 +152,7 @@ def get_batch_config(key):
       k8s_config.load_kube_config(config_file=cls.kubeconfig_path)
       cls.api_client = k8s_client.BatchV1Api()
 
-    cls.kubernetes_client = kubernetes_service.KubernetesService()
+    cls.kubernetes_client = kubernetes_service.KubernetesService(k8s_config_loaded=True)
     data_types.Job(name='test-job', platform='LINUX').put()
     data_types.Job(name='test-job1', platform='LINUX').put()
     data_types.Job(name='test-job2', platform='LINUX').put()