#!/router/bin/python
from .stl_general_test import CStlGeneral_Test, CTRexScenario
import os, sys
from misc_methods import run_command
from time import sleep


def setUpModule(self):
    # examples connect by their own
    if CTRexScenario.stl_trex.is_connected():
        CTRexScenario.stl_trex.disconnect()
        sleep(3)

def tearDownModule():
    # connect back at end of tests
    if not CTRexScenario.stl_trex.is_connected():
        CTRexScenario.stl_trex.connect()


class STLExamples_Test(CStlGeneral_Test):
    """This class defines the IMIX testcase of the T-Rex traffic generator"""

    def test_stl_examples(self):
        examples_dir = '../trex_control_plane/stl/examples'
        examples_to_test = [
                            'stl_imix.py',
                            ]

        for example in examples_to_test:
            return_code, stdout, stderr = run_command("sh -c 'cd %s; %s %s -s %s'" % (examples_dir, sys.executable, example, CTRexScenario.configuration.trex['trex_name']))
            assert return_code == 0, 'example %s failed.\nstdout: %s\nstderr: %s' % (return_code, stdout, stderr)