TGSetup.py revision d68951ac
1# Copyright (c) 2019 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
16from resources.libraries.python.InterfaceUtil import InterfaceUtil
17from resources.libraries.python.topology import NodeType
18
19
20class 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)
33