fix: 3n-alt missing in regressions and progressions
[csit.git] / resources / libraries / python / TGSetup.py
1 # Copyright (c) 2021 Cisco and/or its affiliates.
2 # Licensed under the Apache License, Version 2.0 (the "License");
3 # you may not use this file except in compliance with the License.
4 # You may obtain a copy of the License at:
5 #
6 #     http://www.apache.org/licenses/LICENSE-2.0
7 #
8 # Unless required by applicable law or agreed to in writing, software
9 # distributed under the License is distributed on an "AS IS" BASIS,
10 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
11 # See the License for the specific language governing permissions and
12 # limitations under the License.
13
14 """TG Setup library."""
15
16 from resources.libraries.python.InterfaceUtil import InterfaceUtil
17 from resources.libraries.python.topology import NodeType
18
19
20 class TGSetup:
21     """TG setup before test."""
22
23     @staticmethod
24     def all_tgs_set_interface_default_driver(nodes):
25         """Setup interfaces default driver for all TGs in given topology.
26
27         :param nodes: Nodes in topology.
28         :type nodes: dict
29         """
30         for node in nodes.values():
31             if node[u"type"] == NodeType.TG:
32                 InterfaceUtil.tg_set_interfaces_default_driver(node)