diff options
| -rw-r--r-- | functest_kubernetes/netperf/netperf.py | 26 |
1 files changed, 24 insertions, 2 deletions
diff --git a/functest_kubernetes/netperf/netperf.py b/functest_kubernetes/netperf/netperf.py index 7832e6d1..5923cb78 100644 --- a/functest_kubernetes/netperf/netperf.py +++ b/functest_kubernetes/netperf/netperf.py @@ -19,16 +19,21 @@ import shutil import subprocess import time +from kubernetes import client +from kubernetes import config from xtesting.core import testcase class Netperf(testcase.TestCase): """Run Benchmarking Kubernetes Networking Performance""" + ns_generate_name = "netperf-" __logger = logging.getLogger(__name__) def __init__(self, **kwargs): super().__init__(**kwargs) + config.load_kube_config() + self.corev1 = client.CoreV1Api() self.output_log_name = 'functest-kubernetes.log' self.output_debug_log_name = 'functest-kubernetes.debug.log' @@ -45,13 +50,22 @@ class Netperf(testcase.TestCase): if not os.path.exists(self.res_dir): os.makedirs(self.res_dir) os.chdir(self.res_dir) + api_response = self.corev1.create_namespace( + client.V1Namespace(metadata=client.V1ObjectMeta( + generate_name=self.ns_generate_name, + labels={ + "pod-security.kubernetes.io/enforce": "baseline"}))) + self.namespace = api_response.metadata.name + self.__logger.debug("create_namespace: %s", api_response) cmd = ['launch', '-iterations', '1', '-kubeConfig', - f'{Path.home()}/.kube/config', '-v', '3'] + f'{Path.home()}/.kube/config', '-v', '3', + '-namespace', self.namespace] output = subprocess.check_output( cmd, stderr=subprocess.STDOUT, timeout=3600) self.__logger.info("%s\n%s", " ".join(cmd), output.decode("utf-8")) lfiles = glob.glob(os.path.join( - 'results_netperf-latest', 'netperf-latest*.csv')) + f'results_{self.namespace}-latest', + f'{self.namespace}-latest*.csv')) results = max(lfiles, key=os.path.getmtime) cmd = ['plotperf', '-c', results, '-o', self.res_dir, '-s', 'netperf'] @@ -69,3 +83,11 @@ class Netperf(testcase.TestCase): status = testcase.TestCase.EX_RUN_ERROR self.stop_time = time.time() return status + + def clean(self): + if self.namespace: + try: + api_response = self.corev1.delete_namespace(self.namespace) + self.__logger.debug("delete_namespace: %s", self.namespace) + except client.rest.ApiException: + pass |