summaryrefslogtreecommitdiffstats
path: root/scripts/automation/regression/stateless_tests/stl_general_test.py
blob: 64e93510fb6831ece0623fc1d4e192897ca23642 (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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
import os, sys
import unittest
from trex import CTRexScenario
from stateful_tests.trex_general_test import CTRexGeneral_Test
from trex_stl_lib.api import *
import time
from nose.tools import nottest


class CStlGeneral_Test(CTRexGeneral_Test):
    """This class defines the general stateless testcase of the T-Rex traffic generator"""

    #once for all tests under CStlGeneral_Test
    @classmethod
    def setUpClass(cls):
        cls.stl_trex = CTRexScenario.stl_trex

    def setUp(self):
        CTRexGeneral_Test.setUp(self)
        # check basic requirements, should be verified at test_connectivity, here only skip test
        if CTRexScenario.stl_init_error:
            self.skip(CTRexScenario.stl_init_error)

    def connect(self, timeout = 20):
        sys.stdout.write('Connecting')
        for i in range(timeout):
            try:
                sys.stdout.write('.')
                sys.stdout.flush()
                self.stl_trex.connect()
                print ''
                return True
            except:
                time.sleep(1)
        print ''
        return False

    def map_ports(self, timeout = 5):
        sys.stdout.write('Mapping ports')
        for i in range(timeout):
            sys.stdout.write('.')
            sys.stdout.flush()
            CTRexScenario.stl_ports_map = stl_map_ports(self.stl_trex)
            if self.verify_bidirectional(CTRexScenario.stl_ports_map):
                print ''
                return True
            time.sleep(1)
        print ''
        return False

    # verify all the ports are bidirectional
    @staticmethod
    def verify_bidirectional(mapping_dict):
        if len(mapping_dict['unknown']):
            return False
        if len(mapping_dict['bi']) * 2 == len(mapping_dict['map']):
            return True
        return False

    @staticmethod
    def get_port_count():
        return CTRexScenario.stl_trex.get_port_count()

    @staticmethod
    def is_connected():
        return CTRexScenario.stl_trex.is_connected()

class STLBasic_Test(CStlGeneral_Test):
    # will run it first explicitly, check connectivity and configure routing
    @nottest
    def test_connectivity(self):
        if not self.is_loopback:
            CTRexScenario.router.load_clean_config()
            CTRexScenario.router.configure_basic_interfaces()
            CTRexScenario.router.config_pbr(mode = "config")

        err = 'Client could not connect'
        CTRexScenario.stl_init_error = err
        if not self.connect():
            self.fail(err)
        err = 'Client could not map ports'
        CTRexScenario.stl_init_error = err
        if not self.map_ports():
            self.fail(err)
        print 'Got ports mapping: %s' % CTRexScenario.stl_ports_map
        CTRexScenario.stl_init_error = None