move regression to trex-core

slight fixes of hltapi + vm in packet builder
update yaml lib version from 3.01 to 3.11

9 files changed, 419 insertions, 0 deletions

diff --git a/scripts/automation/regression/unit_tests/functional_tests/config.yaml b/scripts/automation/regression/unit_tests/functional_tests/config.yaml
new file mode 100644
index 00000000..4f4c7c40
--- /dev/null
+++ b/scripts/automation/regression/unit_tests/functional_tests/config.yaml
@@ -0,0 +1,74 @@
+################################################################
+####         T-Rex nightly test configuration file          ####
+################################################################
+
+
+### T-Rex configuration:
+# hostname       - can be DNS name or IP for the TRex machine for ssh to the box
+# password       - root password for TRex machine
+# is_dual        - should the TRex inject with -p ?
+# version_path   - path to the t-rex version and executable
+# cores          - how many cores should be used
+# latency        - rate of latency packets injected by the TRex
+
+### Router configuration:
+# hostname       - the router hostname as apears in ______# cli prefix
+# ip_address     - the router's ip that can be used to communicate with
+# image          - the desired imaged wished to be loaded as the router's running config
+# line_password  - router password when access via Telent
+# en_password    - router password when changing to "enable" mode
+# interfaces     - an array of client-server pairs, representing the interfaces configurations of the router
+# configurations - an array of configurations that could possibly loaded into the router during the test. 
+#                  The "clean" configuration is a mandatory configuration the router will load with to run the basic test bench
+
+### TFTP configuration:
+# hostname       - the tftp hostname
+# ip_address     - the tftp's ip address
+# images_path    - the tftp's relative path in which the router's images are located
+
+### Test_misc configuration:
+# expected_bw    - the "golden" bandwidth (in Gbps) results planned on receiving from the test
+
+trex:
+  hostname       : hostname
+  password       : root password
+  version_path   : not used
+  cores          : 1
+
+router:
+  model          : device model
+  hostname       : device hostname
+  ip_address     : device ip 
+  image          : device image name
+  line_password  : telnet pass
+  en_password    : enable pass
+  mgmt_interface : GigabitEthernet0/0/0
+  clean_config   : path to clean_config file
+  intf_masking   : 255.255.255.0
+  ipv6_mask      : 64
+  interfaces     :
+    - client : 
+        name          : GigabitEthernet0/0/1
+        src_mac_addr  : 0000.0001.0000
+        dest_mac_addr : 0000.1000.0000
+      server : 
+        name          : GigabitEthernet0/0/2
+        src_mac_addr  : 0000.0002.0000
+        dest_mac_addr : 0000.2000.0000
+      vrf_name      :   null
+    - client : 
+        name          : GigabitEthernet0/0/3
+        src_mac_addr  : 0000.0003.0000
+        dest_mac_addr : 0000.3000.0000
+      server : 
+        name          : GigabitEthernet0/0/4
+        src_mac_addr  : 0000.0004.0000
+        dest_mac_addr : 0000.4000.0000
+      vrf_name      : dup  
+    
+
+tftp:
+  hostname       : tftp hostname
+  ip_address     : tftp ip
+  root_dir       : tftp root dir
+  images_path    : path related to root dir
diff --git a/scripts/automation/regression/unit_tests/functional_tests/functional_general_test.py b/scripts/automation/regression/unit_tests/functional_tests/functional_general_test.py
new file mode 100755
index 00000000..525b58d2
--- /dev/null
+++ b/scripts/automation/regression/unit_tests/functional_tests/functional_general_test.py
@@ -0,0 +1,22 @@
+#!/router/bin/python
+
+from nose.tools import assert_equal
+from nose.tools import assert_not_equal
+from nose.tools import assert_raises
+from nose.tools import raises
+
+
+class CGeneralFunctional_Test(object): 
+    def __init__(self):
+        pass
+
+
+    def setUp(self):
+        pass
+
+
+    def tearDown(self):
+        pass
+
+if __name__ == "__main__":
+    pass
diff --git a/scripts/automation/regression/unit_tests/functional_tests/misc_methods_test.py b/scripts/automation/regression/unit_tests/functional_tests/misc_methods_test.py
new file mode 100755
index 00000000..096f86d8
--- /dev/null
+++ b/scripts/automation/regression/unit_tests/functional_tests/misc_methods_test.py
@@ -0,0 +1,61 @@
+#!/router/bin/python
+
+import functional_general_test
+import misc_methods
+from nose.tools import assert_equal
+from nose.tools import assert_not_equal
+from nose.tools import assert_raises
+from nose.tools import raises
+
+
+class MiscMethods_Test(functional_general_test.CGeneralFunctional_Test):
+
+    def setUp(self):
+        self.ipv4_gen = misc_methods.get_network_addr()
+        self.ipv6_gen = misc_methods.get_network_addr(ip_type = 'ipv6')
+        pass
+
+    def test_ipv4_gen(self):
+        for i in range(1, 255):
+            assert_equal( next(self.ipv4_gen), [".".join( map(str, [1, 1, i, 0])), '255.255.255.0'] )
+
+    def test_ipv6_gen(self):
+        tmp_ipv6_addr = ['2001', 'DB8', 0, '2222', 0, 0, 0, 0]
+        for i in range(0, 255):
+            tmp_ipv6_addr[2] = hex(i)[2:]
+            assert_equal( next(self.ipv6_gen), ":".join( map(str, tmp_ipv6_addr)) )
+
+    def test_get_ipv4_client_addr(self):
+        tmp_ipv4_addr = next(self.ipv4_gen)[0]
+        assert_equal ( misc_methods.get_single_net_client_addr(tmp_ipv4_addr), '1.1.1.1')
+        assert_raises (ValueError, misc_methods.get_single_net_client_addr, tmp_ipv4_addr, {'3' : 255} )
+    
+    def test_get_ipv6_client_addr(self):
+        tmp_ipv6_addr = next(self.ipv6_gen)
+        assert_equal ( misc_methods.get_single_net_client_addr(tmp_ipv6_addr, {'7' : 1}, ip_type = 'ipv6'), '2001:DB8:0:2222:0:0:0:1')    
+        assert_equal ( misc_methods.get_single_net_client_addr(tmp_ipv6_addr, {'7' : 2}, ip_type = 'ipv6'), '2001:DB8:0:2222:0:0:0:2')    
+        assert_raises (ValueError, misc_methods.get_single_net_client_addr, tmp_ipv6_addr, {'7' : 70000} )
+        
+
+    @raises(ValueError)
+    def test_ipv4_client_addr_exception(self):
+        tmp_ipv4_addr = next(self.ipv4_gen)[0]
+        misc_methods.get_single_net_client_addr(tmp_ipv4_addr, {'4' : 1})
+
+    @raises(ValueError)
+    def test_ipv6_client_addr_exception(self):
+        tmp_ipv6_addr = next(self.ipv6_gen)
+        misc_methods.get_single_net_client_addr(tmp_ipv6_addr, {'8' : 1}, ip_type = 'ipv6')
+
+    @raises(StopIteration)
+    def test_gen_ipv4_to_limit (self):
+        while(True):
+            next(self.ipv4_gen)
+
+    @raises(StopIteration)
+    def test_gen_ipv6_to_limit (self):
+        while(True):
+            next(self.ipv6_gen)
+
+    def tearDown(self):
+        pass
diff --git a/scripts/automation/regression/unit_tests/functional_tests/platform_cmd_cache_test.py b/scripts/automation/regression/unit_tests/functional_tests/platform_cmd_cache_test.py
new file mode 100755
index 00000000..24ccf7a5
--- /dev/null
+++ b/scripts/automation/regression/unit_tests/functional_tests/platform_cmd_cache_test.py
@@ -0,0 +1,60 @@
+#!/router/bin/python
+
+from platform_cmd_link import *
+import functional_general_test
+from nose.tools import assert_equal
+from nose.tools import assert_not_equal
+
+
+class CCommandCache_Test(functional_general_test.CGeneralFunctional_Test):
+
+    def setUp(self):
+    	self.cache = CCommandCache()
+        self.cache.add('IF', "ip nbar protocol-discovery", 'GigabitEthernet0/0/1')
+        self.cache.add('IF', "ip nbar protocol-discovery", 'GigabitEthernet0/0/2')
+        self.cache.add('conf', "arp 1.1.1.1 0000.0001.0000 arpa")
+        self.cache.add('conf', "arp 1.1.2.1 0000.0002.0000 arpa")
+        self.cache.add('exec', "show ip nbar protocol-discovery stats packet-count")
+
+    def test_add(self):
+        assert_equal(self.cache.cache['IF'],
+            {'GigabitEthernet0/0/1' : ['ip nbar protocol-discovery'],
+             'GigabitEthernet0/0/2' : ['ip nbar protocol-discovery']
+            })
+        assert_equal(self.cache.cache['CONF'],
+            ["arp 1.1.1.1 0000.0001.0000 arpa",
+             "arp 1.1.2.1 0000.0002.0000 arpa"]
+            )
+        assert_equal(self.cache.cache['EXEC'],
+            ["show ip nbar protocol-discovery stats packet-count"])
+
+    def test_dump_config (self):
+        import sys
+        from StringIO import StringIO
+        saved_stdout = sys.stdout
+        try:
+            out = StringIO()
+            sys.stdout = out
+            self.cache.dump_config()
+            output = out.getvalue().strip()
+            assert_equal(output, 
+                "configure terminal\ninterface GigabitEthernet0/0/1\nip nbar protocol-discovery\ninterface GigabitEthernet0/0/2\nip nbar protocol-discovery\nexit\narp 1.1.1.1 0000.0001.0000 arpa\narp 1.1.2.1 0000.0002.0000 arpa\nexit\nshow ip nbar protocol-discovery stats packet-count"
+                )
+        finally:
+            sys.stdout = saved_stdout
+
+    def test_get_config_list (self):
+        assert_equal(self.cache.get_config_list(),
+            ["configure terminal", "interface GigabitEthernet0/0/1", "ip nbar protocol-discovery", "interface GigabitEthernet0/0/2", "ip nbar protocol-discovery", "exit", "arp 1.1.1.1 0000.0001.0000 arpa", "arp 1.1.2.1 0000.0002.0000 arpa", "exit", "show ip nbar protocol-discovery stats packet-count"]
+            )
+
+    def test_clear_cache (self):
+        self.cache.clear_cache()
+        assert_equal(self.cache.cache,
+            {"IF"   : {},
+             "CONF" : [],
+             "EXEC" : []}
+            )
+
+    def tearDown(self):
+        self.cache.clear_cache()
diff --git a/scripts/automation/regression/unit_tests/functional_tests/platform_cmd_link_test.py b/scripts/automation/regression/unit_tests/functional_tests/platform_cmd_link_test.py
new file mode 100755
index 00000000..7a31815b
--- /dev/null
+++ b/scripts/automation/regression/unit_tests/functional_tests/platform_cmd_link_test.py
@@ -0,0 +1,62 @@
+#!/router/bin/python
+
+from platform_cmd_link import *
+import functional_general_test
+from nose.tools import assert_equal
+from nose.tools import assert_not_equal
+
+
+class CCommandLink_Test(functional_general_test.CGeneralFunctional_Test):
+
+    def setUp(self):
+        self.cache = CCommandCache()
+        self.cache.add('IF', "ip nbar protocol-discovery", 'GigabitEthernet0/0/1')
+        self.cache.add('IF', "ip nbar protocol-discovery", 'GigabitEthernet0/0/2')
+        self.cache.add('conf', "arp 1.1.1.1 0000.0001.0000 arpa")
+        self.cache.add('conf', "arp 1.1.2.1 0000.0002.0000 arpa")
+        self.cache.add('exec', "show ip nbar protocol-discovery stats packet-count")
+        self.com_link = CCommandLink()
+
+    def test_transmit(self):
+        # test here future implemntatin of platform physical link
+        pass
+
+    def test_run_cached_command (self):
+        self.com_link.run_command([self.cache])
+
+        assert_equal (self.com_link.get_history(), 
+            ["configure terminal", "interface GigabitEthernet0/0/1", "ip nbar protocol-discovery", "interface GigabitEthernet0/0/2", "ip nbar protocol-discovery", "exit", "arp 1.1.1.1 0000.0001.0000 arpa", "arp 1.1.2.1 0000.0002.0000 arpa", "exit", "show ip nbar protocol-discovery stats packet-count"]
+            )
+
+        self.com_link.clear_history()
+        self.com_link.run_single_command(self.cache)
+        assert_equal (self.com_link.get_history(), 
+            ["configure terminal", "interface GigabitEthernet0/0/1", "ip nbar protocol-discovery", "interface GigabitEthernet0/0/2", "ip nbar protocol-discovery", "exit", "arp 1.1.1.1 0000.0001.0000 arpa", "arp 1.1.2.1 0000.0002.0000 arpa", "exit", "show ip nbar protocol-discovery stats packet-count"]
+            )
+
+    def test_run_single_command(self):
+        self.com_link.run_single_command("show ip nbar protocol-discovery stats packet-count")
+        assert_equal (self.com_link.get_history(), 
+            ["show ip nbar protocol-discovery stats packet-count"]
+            )
+
+    def test_run_mixed_commands (self):
+        self.com_link.run_single_command("show ip nbar protocol-discovery stats packet-count")
+        self.com_link.run_command([self.cache])
+        self.com_link.run_command(["show ip interface brief"])
+
+        assert_equal (self.com_link.get_history(), 
+            ["show ip nbar protocol-discovery stats packet-count",
+             "configure terminal", "interface GigabitEthernet0/0/1", "ip nbar protocol-discovery", "interface GigabitEthernet0/0/2", "ip nbar protocol-discovery", "exit", "arp 1.1.1.1 0000.0001.0000 arpa", "arp 1.1.2.1 0000.0002.0000 arpa", "exit", "show ip nbar protocol-discovery stats packet-count",
+             "show ip interface brief"]
+            )
+
+    def test_clear_history (self):
+        self.com_link.run_command(["show ip interface brief"])
+        self.com_link.clear_history()
+        assert_equal (self.com_link.get_history(), [])
+
+    def tearDown(self):
+        self.cache.clear_cache()
+
+
diff --git a/scripts/automation/regression/unit_tests/functional_tests/platform_device_cfg_test.py b/scripts/automation/regression/unit_tests/functional_tests/platform_device_cfg_test.py
new file mode 100755
index 00000000..890d0cb9
--- /dev/null
+++ b/scripts/automation/regression/unit_tests/functional_tests/platform_device_cfg_test.py
@@ -0,0 +1,20 @@
+#!/router/bin/python
+
+from platform_cmd_link import *
+import functional_general_test
+from nose.tools import assert_equal
+from nose.tools import assert_not_equal
+
+
+class CDeviceCfg_Test(functional_general_test.CGeneralFunctional_Test):
+
+    def setUp(self):
+    	self.dev_cfg = CDeviceCfg('./unit_tests/functional_tests/config.yaml')
+
+    def test_get_interfaces_cfg(self):
+        assert_equal (self.dev_cfg.get_interfaces_cfg(), 
+        	[{'client': {'src_mac_addr': '0000.0001.0000', 'name': 'GigabitEthernet0/0/1', 'dest_mac_addr': '0000.1000.0000'}, 'vrf_name': None, 'server': {'src_mac_addr': '0000.0002.0000', 'name': 'GigabitEthernet0/0/2', 'dest_mac_addr': '0000.2000.0000'}}, {'client': {'src_mac_addr': '0000.0003.0000', 'name': 'GigabitEthernet0/0/3', 'dest_mac_addr': '0000.3000.0000'}, 'vrf_name': 'dup', 'server': {'src_mac_addr': '0000.0004.0000', 'name': 'GigabitEthernet0/0/4', 'dest_mac_addr': '0000.4000.0000'}}]
+        	)
+
+    def tearDown(self):
+        pass
diff --git a/scripts/automation/regression/unit_tests/functional_tests/platform_dual_if_obj_test.py b/scripts/automation/regression/unit_tests/functional_tests/platform_dual_if_obj_test.py
new file mode 100755
index 00000000..ff54b9ee
--- /dev/null
+++ b/scripts/automation/regression/unit_tests/functional_tests/platform_dual_if_obj_test.py
@@ -0,0 +1,31 @@
+#!/router/bin/python
+
+from platform_cmd_link import *
+import functional_general_test
+from nose.tools import assert_equal
+from nose.tools import assert_not_equal
+
+
+class CDualIfObj_Test(functional_general_test.CGeneralFunctional_Test):
+
+    def setUp(self):
+    	self.if_1   = CIfObj('gig0/0/1', '1.1.1.1', '2001:DB8:0:2222:0:0:0:1', '0000.0001.0000', '0000.0001.0000', IFType.Client)
+        self.if_2   = CIfObj('gig0/0/2', '1.1.2.1', '2001:DB8:1:2222:0:0:0:1', '0000.0002.0000', '0000.0002.0000', IFType.Server)
+        self.if_3   = CIfObj('gig0/0/3', '1.1.3.1', '2001:DB8:2:2222:0:0:0:1', '0000.0003.0000', '0000.0003.0000', IFType.Client)
+        self.if_4   = CIfObj('gig0/0/4', '1.1.4.1', '2001:DB8:3:2222:0:0:0:1', '0000.0004.0000', '0000.0004.0000', IFType.Server)
+        self.dual_1 = CDualIfObj(None, self.if_1, self.if_2)
+        self.dual_2 = CDualIfObj('dup', self.if_3, self.if_4)
+
+    def test_id_allocation(self):
+        assert (self.dual_1.get_id() < self.dual_2.get_id() < CDualIfObj._obj_id)
+
+    def test_get_vrf_name (self):
+        assert_equal ( self.dual_1.get_vrf_name() , None )
+        assert_equal ( self.dual_2.get_vrf_name() , 'dup' )
+
+    def test_is_duplicated (self):
+        assert_equal ( self.dual_1.is_duplicated() , False )
+        assert_equal ( self.dual_2.is_duplicated() , True )
+
+    def tearDown(self):
+        pass
\ No newline at end of file
diff --git a/scripts/automation/regression/unit_tests/functional_tests/platform_if_manager_test.py b/scripts/automation/regression/unit_tests/functional_tests/platform_if_manager_test.py
new file mode 100755
index 00000000..7ba6e66e
--- /dev/null
+++ b/scripts/automation/regression/unit_tests/functional_tests/platform_if_manager_test.py
@@ -0,0 +1,40 @@
+#!/router/bin/python
+
+from platform_cmd_link import *
+import functional_general_test
+from nose.tools import assert_equal
+from nose.tools import assert_not_equal
+
+
+class CIfManager_Test(functional_general_test.CGeneralFunctional_Test):
+
+    def setUp(self):
+        self.dev_cfg = CDeviceCfg('./unit_tests/functional_tests/config.yaml')
+        self.if_mng  = CIfManager()
+
+    # main testing method to check the entire class
+    def test_load_config (self):
+        self.if_mng.load_config(self.dev_cfg)
+
+        # check the number of items in each qeury
+        assert_equal( len(self.if_mng.get_if_list()), 4 )
+        assert_equal( len(self.if_mng.get_if_list(if_type = IFType.Client)), 2 )
+        assert_equal( len(self.if_mng.get_if_list(if_type = IFType.Client, is_duplicated = True)), 1 )
+        assert_equal( len(self.if_mng.get_if_list(if_type = IFType.Client, is_duplicated = False)), 1 )
+        assert_equal( len(self.if_mng.get_if_list(if_type = IFType.Server)), 2 )
+        assert_equal( len(self.if_mng.get_if_list(if_type = IFType.Server, is_duplicated = True)), 1 )
+        assert_equal( len(self.if_mng.get_if_list(if_type = IFType.Server, is_duplicated = False)), 1 )
+        assert_equal( len(self.if_mng.get_duplicated_if()), 2 )
+        assert_equal( len(self.if_mng.get_dual_if_list()), 2 )
+
+        # check the classification with intf name
+        assert_equal( map(CIfObj.get_name, self.if_mng.get_if_list() ), ['GigabitEthernet0/0/1','GigabitEthernet0/0/2','GigabitEthernet0/0/3','GigabitEthernet0/0/4'] )
+        assert_equal( map(CIfObj.get_name, self.if_mng.get_if_list(is_duplicated = True) ), ['GigabitEthernet0/0/3','GigabitEthernet0/0/4'] )
+        assert_equal( map(CIfObj.get_name, self.if_mng.get_if_list(is_duplicated = False) ), ['GigabitEthernet0/0/1','GigabitEthernet0/0/2'] )
+        assert_equal( map(CIfObj.get_name, self.if_mng.get_duplicated_if() ), ['GigabitEthernet0/0/3', 'GigabitEthernet0/0/4'] )
+
+        # check the classification with vrf name
+        assert_equal( map(CDualIfObj.get_vrf_name, self.if_mng.get_dual_if_list() ), [None, 'dup'] )
+
+    def tearDown(self):
+        pass
diff --git a/scripts/automation/regression/unit_tests/functional_tests/platform_if_obj_test.py b/scripts/automation/regression/unit_tests/functional_tests/platform_if_obj_test.py
new file mode 100755
index 00000000..534d4170
--- /dev/null
+++ b/scripts/automation/regression/unit_tests/functional_tests/platform_if_obj_test.py
@@ -0,0 +1,49 @@
+#!/router/bin/python
+
+from platform_cmd_link import *
+import functional_general_test
+from nose.tools import assert_equal
+from nose.tools import assert_not_equal
+
+
+class CIfObj_Test(functional_general_test.CGeneralFunctional_Test):
+    test_idx = 1
+
+    def setUp(self):
+        self.if_1 = CIfObj('gig0/0/1', '1.1.1.1', '2001:DB8:0:2222:0:0:0:1', '0000.0001.0000', '0000.0001.0000', IFType.Client)
+        self.if_2 = CIfObj('TenGig0/0/0', '1.1.2.1', '2001:DB8:1:2222:0:0:0:1', '0000.0002.0000', '0000.0002.0000', IFType.Server)
+        CIfObj_Test.test_idx += 1
+
+    def test_id_allocation(self):
+        assert (self.if_1.get_id() < self.if_2.get_id() < CIfObj._obj_id)
+
+    def test_isClient(self):
+        assert_equal (self.if_1.is_client(), True)
+
+    def test_isServer(self):
+        assert_equal (self.if_2.is_server(), True)
+
+    def test_get_name (self):
+        assert_equal (self.if_1.get_name(), 'gig0/0/1')
+        assert_equal (self.if_2.get_name(), 'TenGig0/0/0')
+
+    def test_get_src_mac_addr (self):
+        assert_equal (self.if_1.get_src_mac_addr(), '0000.0001.0000')
+
+    def test_get_dest_mac (self):
+        assert_equal (self.if_2.get_dest_mac(), '0000.0002.0000')
+
+    def test_get_ipv4_addr (self):
+        assert_equal (self.if_1.get_ipv4_addr(), '1.1.1.1' )
+        assert_equal (self.if_2.get_ipv4_addr(), '1.1.2.1' ) 
+
+    def test_get_ipv6_addr (self):
+        assert_equal (self.if_1.get_ipv6_addr(), '2001:DB8:0:2222:0:0:0:1' )
+        assert_equal (self.if_2.get_ipv6_addr(), '2001:DB8:1:2222:0:0:0:1' )
+
+    def test_get_type (self):
+        assert_equal (self.if_1.get_if_type(), IFType.Client)
+        assert_equal (self.if_2.get_if_type(), IFType.Server)
+
+    def tearDown(self):
+        pass