summaryrefslogtreecommitdiff
path: root/test-cli/test/tests/qiperf.py
blob: 126b6ee289cf1657847e8aea056c6477fe8c197c (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
from test.helpers.syscmd import SysCommand


class QIperf(object):
    __sip = None
    __raw_out = None
    __MB_req = None
    __MB_real = None
    __BW_real = None
    __dat_list = None
    __bind = None

    def __init__(self, sip = None):
        if sip is not None:
            self.__sip = sip
        self.__MB_req = '10'

    def execute(self, sip = None, bind = None):
        if sip is not None:
            self.__sip = sip

        if bind is None:
            str_cmd = "iperf -c {} -x CMSV -n {}M".format(self.__sip, self.__MB_req)
        else:
            self.__bind = bind
            str_cmd = "iperf -c {} -x CMSV -n {}M -B {}".format(self.__sip, self.__MB_req, self.__bind)        
        t = SysCommand("iperf", str_cmd)
        if t.execute() == 0:
            self.__raw_out = t.getOutput()
            if self.__raw_out == "":
                return -1
            lines = t.getOutput().splitlines()
            dat = lines[1]
            dat = dat.decode('ascii')
            dat_list = dat.split( )            
            for d in dat_list:
                a = dat_list.pop(0)
                if a == "sec":
                    break
            self.__MB_real = dat_list[0]
            self.__BW_real = dat_list[2]
            self.__dat_list = dat_list
            print(self.__MB_real)
            print(self.__BW_real)
        else:
            return -1
        return 0

    def get_Total_MB(self):
        return self.__MB_real;

    def get_Total_BW(self):
        return self.__MB_real;