TestON/tests/USECASE/SegmentRouting/INT/dependencies/IntTest.py

# SPDX-FileCopyrightText: Copyright 2021-present Open Networking Foundation.
# SPDX-License-Identifier: GPL-2.0-or-later

import os
from tests.dependencies.Network import Network
from tests.USECASE.SegmentRouting.dependencies.Testcaselib import \
    Testcaselib as run

class IntTest:

    def __init__(self, scapy=False):
        self.hosts = ["h1", "h2"]
        self.scapy = scapy

    def setUpTest(self, main):
        main.Network = Network()
        main.Network.connectToNet()
        main.configPath = os.path.join(os.path.dirname(main.testFile), "dependencies")

        for host in self.hosts:
            main.Network.createHostComponent(host)
            if self.scapy:
                hostHandle = getattr(main, host)
                hostHandle.sudoRequired = True
                hostHandle.startScapy()

    def cleanUp(self, main):
        run.saveOnosDiagsIfFailure(main)
        for host in self.hosts:
            if self.scapy:
                hostHandle = getattr(main, host)
                hostHandle.stopScapy()
            main.Network.removeHostComponent(host)
        main.Network.disconnectFromNet()