testsuites/vstf/vstf_scripts/vstf/agent/softagent.py

   1 ##############################################################################
   2 # Copyright (c) 2015 Huawei Technologies Co.,Ltd and others.
   3 #
   4 # All rights reserved. This program and the accompanying materials
   5 # are made available under the terms of the Apache License, Version 2.0
   6 # which accompanies this distribution, and is available at
   7 # http://www.apache.org/licenses/LICENSE-2.0
   8 ##############################################################################
   9
  10
  11 import logging
  12 import time
  13 from vstf.agent.env.basic.image_manager import ImageManager
  14 from vstf.agent.env.basic.source_manager import SourceCodeManager
  15 from vstf.agent.env.basic import commandline
  16 from vstf.agent.env.basic.device_manager import DeviceManager
  17 from vstf.agent.env.basic import collect as coll
  18 from vstf.agent.perf import netns, vnstat, vstfperf, sar, ethtool, affctl
  19 from vstf.agent.env import builder
  20 from vstf.agent.equalizer.get_info import GetPhyInfo
  21 from vstf.agent.equalizer.optimize import Optimize
  22 from vstf.agent.env.driver_plugins.manager import DriverPluginManager
  23
  24 LOG = logging.getLogger(__name__)
  25
  26
  27 class ENV(object):
  28
  29     def __init__(self):
  30         super(ENV, self).__init__()
  31         self.builder = builder.PluginManager()
  32
  33     def build_env(self, cfg_intent):
  34         return self.builder.build(cfg_intent)
  35
  36     def clean_env(self):
  37         return self.builder.clean()
  38
  39     @staticmethod
  40     def create_images(cfg):
  41         return ImageManager(cfg).create_all()
  42
  43     @staticmethod
  44     def clean_images(cfg):
  45         return ImageManager(cfg).clean_all()
  46
  47
  48 class Drivers(object):
  49
  50     def __init__(self):
  51         super(Drivers, self).__init__()
  52         self.dr_mgr = DriverPluginManager()
  53
  54     def install_drivers(self, drivers):
  55         LOG.info("install drivers:%s", drivers)
  56         self.dr_mgr.clean()
  57         ret = self.dr_mgr.load(drivers)
  58         return ret
  59
  60     def clean_drivers(self):
  61         return self.dr_mgr.clean()
  62
  63     def autoneg_on(self, iface, nspace):
  64         return ethtool.autoneg_on(iface, nspace)
  65
  66     def autoneg_off(self, iface, nspace):
  67         return ethtool.autoneg_off(iface, nspace)
  68
  69     def autoneg_query(self, iface, nspace):
  70         return ethtool.autoneg_query(iface, nspace)
  71
  72
  73 class Cpu(object):
  74
  75     def affctl_load(self, policy):
  76         return affctl.affctl_load(policy)
  77
  78     def affctl_list(self):
  79         return affctl.affctl_list()
  80
  81
  82 class Perf(object):
  83
  84     def __init__(self):
  85         super(Perf, self).__init__()
  86         self._vnstat = vnstat.VnStat()
  87         self._vstfperf = vstfperf.Vstfperf()
  88         self._sar = sar.Sar()
  89
  90     def run_vnstat(self, device, namespace=None):
  91         return self._vnstat.run_vnstat(device, namespace)
  92
  93     def kill_vnstat(self, pid, namespace=None):
  94         return self._vnstat.kill_vnstat(pid, namespace)
  95
  96     def perf_run(self, **kwargs):
  97         return self._vstfperf.run(**kwargs)
  98
  99     def run_cpuwatch(self, interval=2):
 100         return self._sar.start(interval)
 101
 102     def kill_cpuwatch(self, pid):
 103         return self._sar.stop(pid)
 104
 105     def force_clean(self):
 106         self._vstfperf.force_clean()
 107         self._sar.force_clean()
 108         self._vnstat.force_clean()
 109         return True
 110
 111
 112 class EqualizerOps(GetPhyInfo, Optimize):
 113
 114     def __init__(self):
 115         super(EqualizerOps, self).__init__()
 116
 117
 118 class BaseAgent(coll.Collect,
 119                 ENV,
 120                 Cpu,
 121                 Drivers,
 122                 DeviceManager,
 123                 commandline.CommandLine,
 124                 netns.NetnsManager,
 125                 SourceCodeManager
 126                 ):
 127
 128     def __init__(self):
 129         super(BaseAgent, self).__init__()
 130
 131
 132 class softAgent(BaseAgent, Perf, EqualizerOps):
 133
 134     def __init__(self):
 135         super(softAgent, self).__init__()
 136
 137
 138 if __name__ == '__main__':
 139     softAgent()