summaryrefslogtreecommitdiffstats
path: root/tools/pkt_gen/xena/json/xena_json_pairs.py
blob: 4f97f0ba11b473fce0bef06c596800bcb7414f4e (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
54
55
56
57
58
59
60
# Copyright 2017 Red Hat Inc & Xena Networks.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#   http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

# Contributors:
#   Dan Amzulescu, Xena Networks
#   Christian Trautman, Red Hat Inc.
#
# Usage can be seen below in unit test. This implementation is designed for one
# module two port Xena chassis runs only.

"""
Xena JSON module for pairs topology
"""

from tools.pkt_gen.xena.json.xena_json import XenaJSON


class XenaJSONPairs(XenaJSON):
    """
    Class to modify and read Xena JSON configuration files. Topology will be set
    as Blocks
    """
    def __init__(self):
        super(XenaJSONPairs, self).__init__()
        self.set_topology_pairs()

    def set_topology_pairs(self):
        """
        Set the test topology to loopback pairs for packets that need to return
        to the same port they are sent from
        :return: None
        """
        self.json_data['TestOptions']['TopologyConfig']['Topology'] = 'PAIRS'
        self.json_data['TestOptions']['TopologyConfig'][
            'Direction'] = 'WEST_EAST'
        self.json_data['PortHandler']['EntityList'][0][
            'PortGroup'] = "WEST"
        self.json_data['PortHandler']['EntityList'][1][
            'PortGroup'] = "WEST"

    def remove_port(self, port_number):
        """
        Remove a port from the config.
        :param port_number: port to remove from json config as int
        :return: None
        """
        del self.json_data['PortHandler']['EntityList'][port_number]
        del self.json_data['StreamHandler']['StreamConnectionList'][0][
            'Port{}Id'.format(port_number + 1)]