aboutsummaryrefslogtreecommitdiffstats
path: root/functest_kubernetes/netperf/netperf.py
blob: 2bd1e1bebb2c88befefab1098a2f58c5eb933306 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
#!/usr/bin/env python

# Copyright (c) 2021 Orange and others.
#
# All rights reserved. This program and the accompanying materials
# are made available under the terms of the Apache License, Version 2.0
# which accompanies this distribution, and is available at
# http://www.apache.org/licenses/LICENSE-2.0

"""
Benchmarking Kubernetes Networking Performance
"""

import glob
import logging
import os
import shutil
import subprocess
import time

from xtesting.core import testcase


class Netperf(testcase.TestCase):
    """Run Benchmarking Kubernetes Networking Performance"""

    __logger = logging.getLogger(__name__)

    def __init__(self, **kwargs):
        super().__init__(**kwargs)
        self.output_log_name = 'functest-kubernetes.log'
        self.output_debug_log_name = 'functest-kubernetes.debug.log'

    def check_requirements(self):
        """Check if launch is in $PATH"""
        self.is_skipped = not (
            shutil.which("launch") and shutil.which("plotperf"))
        if self.is_skipped:
            self.__logger.warning("launch or plotperf is missing")

    def run(self, **kwargs):
        self.start_time = time.time()
        try:
            if not os.path.exists(self.res_dir):
                os.makedirs(self.res_dir)
            cmd = ['launch', '-iterations', '1', '-kubeConfig',
                   '/root/.kube/config']
            output = subprocess.check_output(cmd, stderr=subprocess.STDOUT)
            self.__logger.info("%s\n%s", " ".join(cmd), output.decode("utf-8"))
            lfiles = glob.glob(os.path.join(
                'results_netperf-latest', 'netperf-latest*.csv'))
            results = max(lfiles, key=os.path.getmtime)
            shutil.move(results, os.path.join(self.res_dir, 'netperf.csv'))
            cmd = ['plotperf', '-c',
                   os.path.join(self.res_dir, 'netperf.csv'),
                   '-o', self.res_dir, '-s', 'netperf']
            output = subprocess.check_output(cmd, stderr=subprocess.STDOUT)
            self.__logger.info("%s\n%s", " ".join(cmd), output.decode("utf-8"))
            self.result = 100
            status = testcase.TestCase.EX_OK
        except Exception:  # pylint: disable=broad-except
            self.__logger.exception("Can not run Netperf")
            self.result = 0
            status = testcase.TestCase.EX_RUN_ERROR
        self.stop_time = time.time()
        return status