# 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 blocks topology
"""

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


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

    def set_topology_blocks(self):
        """
        Set the test topology to a West to East config for half duplex flow with
        port 0 as the sender and port 1 as the receiver.
        :return: None
        """
        self.json_data['TestOptions']['TopologyConfig']['Topology'] = 'BLOCKS'
        self.json_data['TestOptions']['TopologyConfig'][
            'Direction'] = 'WEST_EAST'
        self.json_data['PortHandler']['EntityList'][0][
            'PortGroup'] = "WEST"
        self.json_data['PortHandler']['EntityList'][1][
            'PortGroup'] = "EAST"