adjust project directories
[bottlenecks.git] / testsuites / vstf / vstf_scripts / vstf / controller / settings / flows_settings.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 import logging
11
12 import vstf.controller.settings.settings as sets
13
14 LOG = logging.getLogger(__name__)
15
16
17 class FlowsSettings(sets.Settings):
18     def __init__(self, path="/etc/vstf/perf/",
19                  filename="sw_perf.flownodes-settings",
20                  mode=sets.SETS_SINGLE):
21         self._check_actors = {'namespaces', 'senders', 'receivers', 'watchers'}
22         self._nocheck_actors = {"cpu_listens"}
23         super(FlowsSettings, self).__init__(path, filename, mode)
24
25     def _register_func(self):
26         super(FlowsSettings, self)._register_func()
27         for actor in self._check_actors:
28             actor = actor.encode()
29             func_name = "add_%s" % actor
30             setattr(self, func_name, self._adding_file(func_name, self._mset, self._fset, actor, self._check_add))
31             func_name = "madd_%s" % actor
32             setattr(self, func_name, self._adding_memory(func_name, self._mset, actor, self._check_add))
33
34         for actor in self._nocheck_actors:
35             actor = actor.encode()
36             func_name = "add_%s" % actor
37             setattr(self, func_name, self._adding_file(func_name, self._mset, self._fset, actor))
38             func_name = "madd_%s" % actor
39             setattr(self, func_name, self._adding_memory(func_name, self._mset, actor))
40
41         LOG.debug(self.__dict__.keys())
42
43     def clear_all(self):
44         actors = self._check_actors | self._nocheck_actors
45         for actor in actors:
46             func_name = "set_%s" % actor
47             func = getattr(self, func_name)
48             func([])
49
50     def mclear_all(self):
51         actors = self._check_actors | self._nocheck_actors
52         for actor in actors:
53             func_name = "mset_%s" % actor
54             func = getattr(self, func_name)
55             func([])
56
57     def _check_add(self, value):
58         flows = ['agent', 'dev']
59         if not isinstance(value, dict):
60             raise Exception("type is error: %s" % (str(value)))
61         for flow in flows:
62             if flow not in value.keys():
63                 raise Exception("keys[%s] is missing: %s" % (flow, str(value)))
64
65         items = ["ip", "namespace", "mac", "iface", "bdf"]
66         for item in items:
67             if item not in value['dev'].keys():
68                 raise Exception("keys[%s] is error: %s" % (item, str(value)))
69
70
71 def unit_test():
72     from vstf.common.log import setup_logging
73     setup_logging(level=logging.DEBUG, log_file="/var/log/vstf/vstf-flows-settings.log", clevel=logging.INFO)
74
75     flows_settings = FlowsSettings()
76     LOG.info(flows_settings.settings)
77
78     flows_settings.clear_all()
79     flows_settings.set_flows(2)
80     LOG.info(flows_settings.settings)
81
82     flow_1 = {
83         "agent": "192.168.188.14",
84         "dev": {
85             "ip": "192.168.1.100",
86             "namespace": "vstf-space-1",
87             "mac": "90:e2:ba:20:1f:d8",
88             "iface": "eth4",
89             "bdf": "04:00.0"
90         }
91     }
92     flow_2 = {
93         "agent": "192.168.188.14",
94         "dev": {
95             "ip": "192.168.1.101",
96             "namespace": "vstf-space-2",
97             "mac": "90:e2:ba:20:1f:d9",
98             "iface": "p57p2",
99             "bdf": "04:00.1"
100         }
101     }
102
103     flows_settings.add_senders(flow_1)
104     flows_settings.add_senders(flow_2)
105     flows_settings.add_receivers(flow_2)
106     flows_settings.add_receivers(flow_1)
107
108     flows_settings.add_watchers(flow_1)
109     flows_settings.add_watchers(flow_2)
110
111     flows_settings.add_namespaces(flow_1)
112     flows_settings.add_namespaces(flow_2)
113
114     cpu = {
115         "agent": "192.168.188.16",
116         "affctl":{
117             "policy": 2
118         }
119     }
120     flows_settings.add_cpu_listens(cpu)
121     LOG.info(flows_settings.settings)
122
123
124 if __name__ == '__main__':
125     unit_test()