summaryrefslogtreecommitdiffstats
path: root/scripts/automation/regression
diff options
context:
space:
mode:
authorHanoh Haim <hhaim@cisco.com>2016-02-10 17:07:27 +0200
committerHanoh Haim <hhaim@cisco.com>2016-02-10 17:07:27 +0200
commit18d20a3b25f5f1077a872c84a7e1869de56c19fd (patch)
tree48c7fee90980ca2c191208fe83883364a6cf264d /scripts/automation/regression
parent15783cbf852a4733fff97cad00421d45de549d37 (diff)
parent82a5a558113c7111a6269e411513b83a410df596 (diff)
Merge functional tests
Diffstat (limited to 'scripts/automation/regression')
-rwxr-xr-xscripts/automation/regression/hltapi_example.py61
-rwxr-xr-xscripts/automation/regression/outer_packages.py2
-rwxr-xr-xscripts/automation/regression/unit_tests/functional_tests/hltapi_stream_builder_test.py33
-rwxr-xr-xscripts/automation/regression/unit_tests/functional_tests/payload_gen_test.py125
-rwxr-xr-xscripts/automation/regression/unit_tests/functional_tests/pkt_builder_test.py442
-rwxr-xr-xscripts/automation/regression/unit_tests/functional_tests/vm_test.py77
-rwxr-xr-xscripts/automation/regression/unit_tests/functional_tests/vm_variable_test.py63
7 files changed, 66 insertions, 737 deletions
diff --git a/scripts/automation/regression/hltapi_example.py b/scripts/automation/regression/hltapi_example.py
index 0040219a..9e69cb13 100755
--- a/scripts/automation/regression/hltapi_example.py
+++ b/scripts/automation/regression/hltapi_example.py
@@ -1,25 +1,25 @@
#!/router/bin/python
import outer_packages
-from client.trex_hltapi import CTRexHltApi
+from client.trex_hltapi import CTRexHltApi, CStreamsPerPort
import traceback
import sys, time
from pprint import pprint
import argparse
+def error(err = None):
+ if not err:
+ raise Exception('Unknown exception, look traceback')
+ if type(err) is str and not err.startswith('[ERR]'):
+ err = '[ERR] ' + err
+ print err
+ sys.exit(1)
+
def check_res(res):
if res['status'] == 0:
- print('Encountered error:\n%s' % res['log'])
- sys.exit(1)
+ error('Encountered error:\n%s' % res['log'])
return res
-def save_streams_id(res, streams_id_arr):
- stream_id = res.get('stream_id')
- if type(stream_id) in (int, long):
- streams_id_arr.append(stream_id)
- elif type(stream_id) is list:
- streams_id_arr.extend(stream_id)
-
def print_brief_stats(res):
title_str = ' '*3
tx_str = 'TX:'
@@ -48,37 +48,66 @@ if __name__ == "__main__":
parser.add_argument('--device', dest = 'device', default = 'localhost', help='Address of TRex server')
args = parser.parse_args()
hlt_client = CTRexHltApi(verbose = int(args.verbose))
- streams_id_arr = []
-
+
print('Connecting to %s...' % args.device)
res = check_res(hlt_client.connect(device = args.device, port_list = [0, 1], username = 'danklei', break_locks = True, reset = True))
port_handle = res['port_handle']
- print('Connected.')
+ print('Connected, got port handles %s' % port_handle)
+ ports_streams_dict = CStreamsPerPort(port_handle)
+
+ print('Imix should create 3 streams (forth ratio is 0)')
+ res = check_res(hlt_client.traffic_config(mode = 'create', l2_encap = 'ethernet_ii_vlan', bidirectional = True, length_mode = 'imix',
+ port_handle = port_handle, save_to_yaml = '/tmp/d1.yaml'))
+ ports_streams_dict.add_streams_from_res(res)
+
+ #print ports_streams_dict
+ #print hlt_client.trex_client._STLClient__get_all_streams(port_id = port_handle[0])
+ res = check_res(hlt_client.traffic_config(mode = 'modify', port_handle = port_handle[0], stream_id = ports_streams_dict[0][0],
+ mac_src = '1-2-3:4:5:6', l4_protocol = 'udp', save_to_yaml = '/tmp/d2.yaml'))
+ #print hlt_client.trex_client._STLClient__get_all_streams(port_id = port_handle[0])
+ #print hlt_client._streams_history
+ res = check_res(hlt_client.traffic_config(mode = 'modify', port_handle = port_handle[0], stream_id = ports_streams_dict[0][0],
+ mac_dst = '{ 7 7 7-7:7:7}', save_to_yaml = '/tmp/d3.yaml'))
+ #print hlt_client.trex_client._STLClient__get_all_streams(port_id = port_handle[0])
+ check_res(hlt_client.traffic_config(mode = 'reset', port_handle = port_handle))
+
+ res = check_res(hlt_client.traffic_config(mode = 'create', bidirectional = True, length_mode = 'fixed', port_handle = port_handle,
+ transmit_mode = 'single_burst', pkts_per_burst = 100, rate_pps = 100,
+ mac_src = '1-2-3-4-5-6',
+ mac_dst = '6:5:4:4:5:6',
+ save_to_yaml = '/tmp/imix.yaml'))
+ ports_streams_dict.add_streams_from_res(res)
print('Create single_burst 100 packets rate_pps=100 on port 0')
- res = check_res(hlt_client.traffic_config(mode = 'create', port_handle = port_handle[0], transmit_mode = 'single_burst', pkts_per_burst = 100, rate_pps = 100))
- #save_streams_id(res, streams_id_arr)
+ res = check_res(hlt_client.traffic_config(mode = 'create', port_handle = port_handle[0], transmit_mode = 'single_burst',
+ pkts_per_burst = 100, rate_pps = 100))
+ ports_streams_dict.add_streams_from_res(res)
# playground - creating various streams on port 1
res = check_res(hlt_client.traffic_config(mode = 'create', port_handle = port_handle[1], save_to_yaml = '/tmp/hlt2.yaml',
tcp_src_port_mode = 'decrement',
tcp_src_port_count = 10, tcp_dst_port_count = 10, tcp_dst_port_mode = 'random'))
+ ports_streams_dict.add_streams_from_res(res)
res = check_res(hlt_client.traffic_config(mode = 'create', port_handle = port_handle[1], save_to_yaml = '/tmp/hlt3.yaml',
l4_protocol = 'udp',
udp_src_port_mode = 'decrement',
udp_src_port_count = 10, udp_dst_port_count = 10, udp_dst_port_mode = 'random'))
+ ports_streams_dict.add_streams_from_res(res)
res = check_res(hlt_client.traffic_config(mode = 'create', port_handle = port_handle[1], save_to_yaml = '/tmp/hlt4.yaml',
length_mode = 'increment',
#ip_src_addr = '192.168.1.1', ip_src_mode = 'increment', ip_src_count = 5,
ip_dst_addr = '5.5.5.5', ip_dst_mode = 'random', ip_dst_count = 2))
+ ports_streams_dict.add_streams_from_res(res)
res = check_res(hlt_client.traffic_config(mode = 'create', port_handle = port_handle[1], save_to_yaml = '/tmp/hlt5.yaml',
length_mode = 'decrement', frame_size_min = 100, frame_size_max = 3000,
#ip_src_addr = '192.168.1.1', ip_src_mode = 'increment', ip_src_count = 5,
#ip_dst_addr = '5.5.5.5', ip_dst_mode = 'random', ip_dst_count = 2
))
+ ports_streams_dict.add_streams_from_res(res)
+
# remove the playground
check_res(hlt_client.traffic_config(mode = 'reset', port_handle = port_handle[1]))
@@ -99,7 +128,7 @@ if __name__ == "__main__":
check_res(hlt_client.traffic_control(action = 'stop', port_handle = port_handle[0]))
check_res(hlt_client.traffic_config(mode = 'reset', port_handle = port_handle[0]))
res = check_res(hlt_client.traffic_config(mode = 'create', port_handle = port_handle[0], rate_pps = 1000))
- save_streams_id(res, streams_id_arr)
+ ports_streams_dict.add_streams_from_res(res)
check_res(hlt_client.traffic_control(action = 'run', port_handle = port_handle[0]))
wait_with_progress(5)
print('Sample after another 5 seconds (only packets count)')
diff --git a/scripts/automation/regression/outer_packages.py b/scripts/automation/regression/outer_packages.py
index 05bedc71..6b7c58f9 100755
--- a/scripts/automation/regression/outer_packages.py
+++ b/scripts/automation/regression/outer_packages.py
@@ -9,6 +9,7 @@ if not TREX_PATH or not os.path.isfile('%s/trex_daemon_server' % TREX_PATH):
TREX_PATH = os.path.abspath(os.path.join(CURRENT_PATH, os.pardir, os.pardir))
PATH_TO_PYTHON_LIB = os.path.abspath(os.path.join(TREX_PATH, 'external_libs'))
PATH_TO_CTRL_PLANE = os.path.abspath(os.path.join(TREX_PATH, 'automation', 'trex_control_plane'))
+PATH_STL_API = os.path.abspath(os.path.join(PATH_TO_CTRL_PLANE, 'stl'))
NIGHTLY_MODULES = ['enum34-1.0.4',
'nose-1.3.4',
@@ -22,6 +23,7 @@ NIGHTLY_MODULES = ['enum34-1.0.4',
def import_nightly_modules ():
sys.path.append(TREX_PATH)
sys.path.append(PATH_TO_CTRL_PLANE)
+ sys.path.append(PATH_STL_API)
import_module_list(NIGHTLY_MODULES)
def import_module_list (modules_list):
diff --git a/scripts/automation/regression/unit_tests/functional_tests/hltapi_stream_builder_test.py b/scripts/automation/regression/unit_tests/functional_tests/hltapi_stream_builder_test.py
index eefa1947..3b09d8a1 100755
--- a/scripts/automation/regression/unit_tests/functional_tests/hltapi_stream_builder_test.py
+++ b/scripts/automation/regression/unit_tests/functional_tests/hltapi_stream_builder_test.py
@@ -1,10 +1,10 @@
#!/router/bin/python
-from client.trex_hltapi import CTRexHltApiPktBuilder
+from client.trex_hltapi import CTRexHltApiBuilder
import os
import unittest
-gen_stream = CTRexHltApiPktBuilder.generate_stream
+gen_stream = CTRexHltApiBuilder.generate_stream
def compare_yamls(yaml1, yaml2):
if type(yaml1) is not str:
@@ -29,8 +29,8 @@ class CTRexHltApi_Test(unittest.TestCase):
# Eth/IP/TCP, all values default, no VM instructions
def test_default(self):
- test_stream = gen_stream()
- self.test_yaml = test_stream.dump_to_yaml(self.yaml_save_location(), test_stream)
+ test_stream = gen_stream(name = 'stream-0')
+ self.test_yaml = test_stream.dump_to_yaml(self.yaml_save_location())
self.golden_yaml = '''
- name: stream-0
stream:
@@ -57,8 +57,9 @@ class CTRexHltApi_Test(unittest.TestCase):
ip_src_count = 5,
ip_dst_addr = '5.5.5.5',
ip_dst_count = 2,
- ip_dst_mode = 'random')
- self.test_yaml = test_stream.dump_to_yaml(self.yaml_save_location(), test_stream)
+ ip_dst_mode = 'random',
+ name = 'stream-0')
+ self.test_yaml = test_stream.dump_to_yaml(self.yaml_save_location())
self.golden_yaml = '''
- name: stream-0
stream:
@@ -111,8 +112,9 @@ class CTRexHltApi_Test(unittest.TestCase):
tcp_src_port_count = 10,
tcp_dst_port_mode = 'random',
tcp_dst_port_count = 10,
- tcp_dst_port = 1234)
- self.test_yaml = test_stream.dump_to_yaml(self.yaml_save_location(), test_stream)
+ tcp_dst_port = 1234,
+ name = 'stream-0')
+ self.test_yaml = test_stream.dump_to_yaml(self.yaml_save_location())
self.golden_yaml = '''
- name: stream-0
stream:
@@ -183,8 +185,9 @@ class CTRexHltApi_Test(unittest.TestCase):
udp_src_port = 1234,
udp_dst_port_mode = 'increment',
udp_dst_port_count = 10,
- udp_dst_port = 1234)
- self.test_yaml = test_stream.dump_to_yaml(self.yaml_save_location(), test_stream)
+ udp_dst_port = 1234,
+ name = 'stream-0')
+ self.test_yaml = test_stream.dump_to_yaml(self.yaml_save_location())
self.golden_yaml = '''
- name: stream-0
stream:
@@ -244,8 +247,9 @@ class CTRexHltApi_Test(unittest.TestCase):
frame_size_max = 3000)
test_stream = gen_stream(length_mode = 'decrement',
frame_size_min = 100,
- frame_size_max = 3000)
- self.test_yaml = test_stream.dump_to_yaml(self.yaml_save_location(), test_stream)
+ frame_size_max = 3000,
+ name = 'stream-0')
+ self.test_yaml = test_stream.dump_to_yaml(self.yaml_save_location())
self.golden_yaml = '''
- name: stream-0
stream:
@@ -293,8 +297,9 @@ class CTRexHltApi_Test(unittest.TestCase):
test_stream = gen_stream(l4_protocol = 'udp',
length_mode = 'random',
l3_length_min = 100,
- l3_length_max = 400)
- self.test_yaml = test_stream.dump_to_yaml(self.yaml_save_location(), test_stream)
+ l3_length_max = 400,
+ name = 'stream-0')
+ self.test_yaml = test_stream.dump_to_yaml(self.yaml_save_location())
self.golden_yaml = '''
- name: stream-0
stream:
diff --git a/scripts/automation/regression/unit_tests/functional_tests/payload_gen_test.py b/scripts/automation/regression/unit_tests/functional_tests/payload_gen_test.py
deleted file mode 100755
index 80d2b086..00000000
--- a/scripts/automation/regression/unit_tests/functional_tests/payload_gen_test.py
+++ /dev/null
@@ -1,125 +0,0 @@
-#!/router/bin/python
-
-import pkt_bld_general_test
-from client_utils.packet_builder import *
-from dpkt.ethernet import Ethernet
-from dpkt.ip import IP
-from dpkt.icmp import ICMP
-from nose.tools import assert_equal
-from nose.tools import assert_not_equal
-from nose.tools import assert_raises
-from nose.tools import raises
-import re
-import binascii
-
-class CTRexPayloadGen_Test(pkt_bld_general_test.CGeneralPktBld_Test):
-
- def setUp(self):
-# echo = dpkt.icmp.ICMP.Echo()
-# echo.id = random.randint(0, 0xffff)
-# echo.seq = random.randint(0, 0xffff)
-# echo.data = 'hello world'
-#
-# icmp = dpkt.icmp.ICMP()
-# icmp.type = dpkt.icmp.ICMP_ECHO
-# icmp.data = echo
-
- # packet generation is done directly using dpkt package and
- self.packet = Ethernet()
- ip = IP(src='\x01\x02\x03\x04', dst='\x05\x06\x07\x08', p=1)
- icmp = ICMP(type=8, data=ICMP.Echo(id=123, seq=1, data='foobar'))
- ip.data = icmp
- self.packet.src = "\x00\x00\x55\x55\x00\x00"
- self.packet.dst = "\x00\x00\x11\x11\x00\x00"
- self.packet.data = ip
- self.print_packet(self.packet)
- self.max_pkt_size = 1400
- self.pld_gen = CTRexPktBuilder.CTRexPayloadGen(self.packet, self.max_pkt_size)
-
- @staticmethod
- def special_match(strg, search=re.compile(r'[^a-zA-Z0-9]').search):
- return not bool(search(strg))
-
- @staticmethod
- def principal_period(s):
- # finds the string the repeats itself in the string
- i = (s+s).find(s, 1, -1)
- return None if i == -1 else s[:i]
-
- def test_gen_random_str(self):
- generated_str = self.pld_gen.gen_random_str()
- # print "\nGenerated string: {}".format(generated_str)
- # chech that the generated string is accorsing to rules.
- assert CTRexPayloadGen_Test.special_match(generated_str)
- assert_equal(len(generated_str), (self.max_pkt_size - len(self.packet)))
-
- def test_gen_repeat_ptrn(self):
- gen_len = self.max_pkt_size - len(self.packet)
- # case 1 - repeated string
- repeat_str = "HelloWorld"
- generated_str = self.pld_gen.gen_repeat_ptrn(repeat_str)
- for i in xrange(len(repeat_str)):
- if generated_str.endswith(repeat_str[:i+1]):
- # remove the string residue, if found
- generated_str = generated_str[:-(i+1)]
- # print generated_str
- break
- assert_equal(CTRexPayloadGen_Test.principal_period(generated_str), "HelloWorld")
-
- # case 2.1 - repeated single number - long number
- repeat_num = 0x645564646465
- generated_str = self.pld_gen.gen_repeat_ptrn(repeat_num)
- ptrn = binascii.unhexlify(hex(repeat_num)[2:])
- for i in xrange(len(ptrn)):
- if generated_str.endswith(ptrn[:i+1]):
- # remove the string residue, if found
- generated_str = generated_str[:-(i+1)]
- break
- assert_equal(CTRexPayloadGen_Test.principal_period(generated_str), ptrn)
- # case 2.2 - repeated single number - 1 byte
- repeat_num = 0x64
- generated_str = self.pld_gen.gen_repeat_ptrn(repeat_num)
- ptrn = binascii.unhexlify(hex(repeat_num)[2:])
- assert_equal(CTRexPayloadGen_Test.principal_period(generated_str), ptrn)
- assert_equal(len(generated_str), (self.max_pkt_size - len(self.packet)))
- # case 3 - repeated sequence
- repeat_seq = (0x55, 0x60, 0x65, 0x70, 0x85)
- ptrn = binascii.unhexlify(''.join(hex(x)[2:] for x in repeat_seq))
- generated_str = self.pld_gen.gen_repeat_ptrn(repeat_seq)
- # ptrn = binascii.unhexlify(hex(repeat_num)[2:])
- for i in xrange(len(ptrn)):
- if generated_str.endswith(ptrn[:i+1]):
- # remove the string residue, if found
- generated_str = generated_str[:-(i+1)]
- break
- assert_equal(CTRexPayloadGen_Test.principal_period(generated_str), ptrn)
-
- # in tuples, check that if any of the numbers exceeds limit
- assert_raises(ValueError, self.pld_gen.gen_repeat_ptrn, (0x1, -18))
- assert_raises(ValueError, self.pld_gen.gen_repeat_ptrn, (0xFFF, 5))
- # finally, check an exception is thrown in rest of cases
- assert_raises(ValueError, self.pld_gen.gen_repeat_ptrn, 5.5)
-
-
-
-
-
-
-
-
-
-
-
-
-
-
- pass
-
-
- def tearDown(self):
- pass
-
-
-if __name__ == "__main__":
- pass
-
diff --git a/scripts/automation/regression/unit_tests/functional_tests/pkt_builder_test.py b/scripts/automation/regression/unit_tests/functional_tests/pkt_builder_test.py
deleted file mode 100755
index d33d92fe..00000000
--- a/scripts/automation/regression/unit_tests/functional_tests/pkt_builder_test.py
+++ /dev/null
@@ -1,442 +0,0 @@
-#!/router/bin/python
-
-import pkt_bld_general_test
-from client_utils.packet_builder import *
-import dpkt
-from nose.tools import assert_equal
-from nose.tools import assert_not_equal
-from nose.tools import assert_raises
-from nose.tools import raises
-import os
-import random
-import pprint
-
-class CTRexPktBuilderSanity_Test(pkt_bld_general_test.CGeneralPktBld_Test):
-
- def setUp(self):
- pass
-
- def test_decode_ip_addr(self):
- # test ipv4 case
- assert_equal(CTRexPktBuilder._decode_ip_addr('1.2.3.4', "ipv4"), '\x01\x02\x03\x04')
- assert_equal(CTRexPktBuilder._decode_ip_addr('127.0.0.1', "ipv4"), '\x7F\x00\x00\x01')
- assert_raises(CTRexPktBuilder.IPAddressError, CTRexPktBuilder._decode_ip_addr, '1.2.3.4.5', "ipv4")
- assert_raises(CTRexPktBuilder.IPAddressError, CTRexPktBuilder._decode_ip_addr, '1.2.3.4', "ipv6")
- # test ipv6 case
- assert_equal(CTRexPktBuilder._decode_ip_addr("5001::DB8:1:3333:1:1", "ipv6"),
- 'P\x01\x00\x00\x00\x00\r\xb8\x00\x0133\x00\x01\x00\x01')
- assert_raises(CTRexPktBuilder.IPAddressError, CTRexPktBuilder._decode_ip_addr,
- '2001::DB8:1:2222::1:1', "ipv6")
-
- def test_decode_mac_addr(self):
- assert_equal(CTRexPktBuilder._decode_mac_addr('00:de:34:ef:2e:f4'), '\x00\xde4\xef.\xf4')
- assert_equal(CTRexPktBuilder._decode_mac_addr('00-de-55-ef-2e-f4'), '\x00\xdeU\xef.\xf4')
- assert_raises(CTRexPktBuilder.MACAddressError, CTRexPktBuilder._decode_mac_addr,
- '00:de:34:ef:2e:f4:f4')
- assert_raises(CTRexPktBuilder.MACAddressError, CTRexPktBuilder._decode_mac_addr,
- '1.2.3.4')
- assert_raises(CTRexPktBuilder.MACAddressError, CTRexPktBuilder._decode_mac_addr,
- '00 de 34 ef 2e f4 f4')
-
- def test_gen_layer_name(self):
- pkt = CTRexPktBuilder()
- assert_equal(pkt._gen_layer_name("eth"), "eth_1")
- pkt._pkt_by_hdr = {'eth':None} # mock header pointer data
- assert_equal(pkt._gen_layer_name("eth"), "eth_1")
- pkt._pkt_by_hdr.update({'eth_1':None}) # more mock header pointer data
- assert_equal(pkt._gen_layer_name("eth"), "eth_2")
-
- def test_set_layer_attr_basic(self):
- pkt = CTRexPktBuilder()
- pkt._pkt_by_hdr['ip'] = dpkt.ip.IP()
- # case 1 - test full value assignment
- pkt.set_layer_attr('ip', 'src', '\x01\x02\x03\x04')
- assert_equal(pkt._pkt_by_hdr['ip'].src, '\x01\x02\x03\x04')
- # case 2 - test bit assignment
- pkt.set_layer_bit_attr('ip', 'off', dpkt.ip.IP_DF)
- pkt.set_layer_bit_attr('ip', 'off', dpkt.ip.IP_MF)
- assert_equal(bin(pkt._pkt_by_hdr['ip'].off), '0b110000000000000')
- # case 3 - test assignment of not-exist attribute
- assert_raises(ValueError, pkt.set_layer_bit_attr, 'ip', 'src_dst', 0)
- # case 4.1 - test assignment of data attribute - without dpkt.Packet object
- assert_raises(ValueError, pkt.set_layer_bit_attr, 'ip', 'data', "Not a dpkt.Packet object")
- # case 4.2 - test assignment of data attribute - with dpkt.Packet object - tested under CTRexPktBuilder_Test class
-# tcp = dpkt.tcp.TCP()
- self.print_packet(pkt._pkt_by_hdr['ip'])
-# pkt.set_layer_attr('ip', 'data', tcp)
- # case 5 - test assignment of not-exist layer
- assert_raises(KeyError, pkt.set_layer_bit_attr, 'no_such_layer', 'src', 0)
-
- def tearDown(self):
- pass
-
-
-class CTRexPktBuilder_Test(pkt_bld_general_test.CGeneralPktBld_Test):
-
- def setUp(self):
- self.pkt_bld = CTRexPktBuilder()
- self.pkt_bld.add_pkt_layer("l2", dpkt.ethernet.Ethernet())
- self.pp = pprint.PrettyPrinter(indent=4)
-
- def test_add_pkt_layer(self):
- ip = dpkt.ip.IP(src='\x01\x02\x03\x04', dst='\x05\x06\x07\x08', p=1)
- self.pkt_bld.add_pkt_layer("l3", ip)
- tcp = dpkt.tcp.TCP(sport = 8080)
- self.pkt_bld.add_pkt_layer("l4_tcp", tcp)
- assert_equal(len(self.pkt_bld._pkt_by_hdr), 3)
- assert_equal(self.pkt_bld._pkt_by_hdr.keys(), ['l2', 'l3', 'l4_tcp'])
- self.print_packet(self.pkt_bld._packet)
- assert_raises(ValueError, self.pkt_bld.add_pkt_layer, 'l2', dpkt.ethernet.Ethernet())
-
- def test_set_ip_layer_addr(self):
- ip = dpkt.ip.IP()
- self.pkt_bld.add_pkt_layer("l3", ip)
- self.pkt_bld.set_ip_layer_addr("l3", "src", "1.2.3.4")
- self.print_packet(self.pkt_bld._packet)
- assert_equal(self.pkt_bld._pkt_by_hdr['l3'].src, '\x01\x02\x03\x04')
- # check that only IP layer is using this function
- assert_raises(ValueError, self.pkt_bld.set_ip_layer_addr, 'l2', "src", "1.2.3.4")
-
- def test_calc_offset(self):
- ip = dpkt.ip.IP()
- self.pkt_bld.add_pkt_layer("l3", ip)
- assert_equal(self.pkt_bld._calc_offset("l3", "src", 4), (14, 14+12))
-
- def test_set_ipv6_layer_addr(self):
- ip6 = dpkt.ip6.IP6()
- self.pkt_bld.add_pkt_layer("l3", ip6)
- self.pkt_bld.set_ipv6_layer_addr("l3", "src", "5001::DB8:1:3333:1:1")
- self.print_packet(self.pkt_bld._packet)
- assert_equal(self.pkt_bld._pkt_by_hdr['l3'].src, 'P\x01\x00\x00\x00\x00\r\xb8\x00\x0133\x00\x01\x00\x01')
- # check that only IP layer is using this function
- assert_raises(ValueError, self.pkt_bld.set_ipv6_layer_addr, 'l2', "src", "5001::DB8:1:3333:1:1")
-
- def test_set_eth_layer_addr(self):
- ip = dpkt.ip.IP()
- self.pkt_bld.add_pkt_layer("l3", ip)
- self.pkt_bld.set_eth_layer_addr("l2", "src", "00:de:34:ef:2e:f4")
- self.print_packet(self.pkt_bld._packet)
- assert_equal(self.pkt_bld._pkt_by_hdr['l2'].src, '\x00\xde4\xef.\xf4')
- # check that only IP layer is using this function
- assert_raises(ValueError, self.pkt_bld.set_eth_layer_addr, 'l3', "src", "\x00\xde4\xef.\xf4")
-
- def test_set_layer_attr(self):
- # extend the set_layer_attr_basic test by handling the following case:
- # replace some header data with another layer, causing other layers to disconnect
- # this also tests the _reevaluate_packet method
- ip = dpkt.ip.IP(src='\x01\x02\x03\x04', dst='\x05\x06\x07\x08', p=1)
- self.pkt_bld.add_pkt_layer("l3_ip", ip)
- tcp = dpkt.tcp.TCP(sport = 8080)
- self.pkt_bld.add_pkt_layer("l4_tcp", tcp)
- # sanity: try changing data attr with non-dpkt.Packet instance
- assert_raises(ValueError, self.pkt_bld.set_layer_attr, 'l2', 'data', "HelloWorld")
- # now, add different L3 layer instead of existting one, L4 would disconnect
- old_layer_count = len(self.pkt_bld._pkt_by_hdr)
- new_ip = dpkt.ip.IP(src='\x05\x06\x07\x08', dst='\x01\x02\x03\x04')
- print "\nBefore disconnecting layers:"
- print "============================",
- self.print_packet(self.pkt_bld._packet)
- self.pkt_bld.set_layer_attr('l2', 'data', new_ip)
- print "\nAfter disconnecting layers:"
- print "===========================",
- self.print_packet(self.pkt_bld._packet)
- assert_not_equal(old_layer_count, len(self.pkt_bld._pkt_by_hdr))
- assert_equal(len(self.pkt_bld._pkt_by_hdr), 1) # only Eth layer appears
-
- def test_set_pkt_payload(self):
- payload = "HelloWorld"
- # test for setting a payload to an empty packet
- empty_pkt = CTRexPktBuilder()
- assert_raises(AttributeError, empty_pkt.set_pkt_payload, payload)
- # add content to packet
- ip = dpkt.ip.IP(src='\x01\x02\x03\x04', dst='\x05\x06\x07\x08', p=1)
- self.pkt_bld.add_pkt_layer("l3_ip", ip)
- tcp = dpkt.tcp.TCP(sport = 8080)
- self.pkt_bld.add_pkt_layer("l4_tcp", tcp)
- # now, set a payload for the packet
- self.pkt_bld.set_pkt_payload(payload)
- self.print_packet(self.pkt_bld._packet)
- assert_equal(self.pkt_bld._pkt_by_hdr['l4_tcp'].data, payload)
-
- def test_load_packet(self):
- # add content to packet
- ip = dpkt.ip.IP(src='\x01\x02\x03\x04', dst='\x05\x06\x07\x08', p=1)
- self.pkt_bld.add_pkt_layer("l3_ip", ip)
- tcp = dpkt.tcp.TCP(sport = 8080)
- self.pkt_bld.add_pkt_layer("l4_tcp", tcp)
- self.pkt_bld.set_pkt_payload("HelloWorld")
-
- new_pkt = CTRexPktBuilder()
- new_pkt.load_packet(self.pkt_bld._packet)
- self.print_packet(new_pkt._packet)
- assert_equal(len(new_pkt._pkt_by_hdr), 4)
- assert_equal(new_pkt._pkt_by_hdr.keys(),
- ['ip_1',
- 'tcp_1',
- 'pkt_final_payload',
- 'ethernet_1'
- ]
- )
- assert_equal(new_pkt._pkt_by_hdr['pkt_final_payload'], "HelloWorld")
-
- def test_get_packet(self):
- # get a pointer to the packet
- assert(self.pkt_bld.get_packet(get_ptr=True) is self.pkt_bld._packet)
- # get a copy of the packet
- assert(not(self.pkt_bld.get_packet() is self.pkt_bld._packet))
-
- def test_get_layer(self):
- assert_equal(self.pkt_bld.get_layer('no_such_layer'), None)
- assert(not(self.pkt_bld.get_layer('l2') is self.pkt_bld._packet))
- assert_equal(type(self.pkt_bld.get_layer('l2')).__name__, "Ethernet")
-
- def test_dump_to_pcap(self):
- # set Ethernet layer attributes
- self.pkt_bld.set_eth_layer_addr("l2", "src", "00:15:17:a7:75:a3")
- self.pkt_bld.set_eth_layer_addr("l2", "dst", "e0:5f:b9:69:e9:22")
- self.pkt_bld.set_layer_attr("l2", "type", dpkt.ethernet.ETH_TYPE_IP)
- # set IP layer attributes
- self.pkt_bld.add_pkt_layer("l3_ip", dpkt.ip.IP())
- self.pkt_bld.set_ip_layer_addr("l3_ip", "src", "21.0.0.2")
- self.pkt_bld.set_ip_layer_addr("l3_ip", "dst", "22.0.0.12")
- self.pkt_bld.set_layer_attr("l3_ip", "p", dpkt.ip.IP_PROTO_TCP)
- # set TCP layer attributes
- self.pkt_bld.add_pkt_layer("l4_tcp", dpkt.tcp.TCP())
- self.pkt_bld.set_layer_attr("l4_tcp", "sport", 13311)
- self.pkt_bld.set_layer_attr("l4_tcp", "dport", 80)
- self.pkt_bld.set_layer_attr("l4_tcp", "flags", 0)
- self.pkt_bld.set_layer_attr("l4_tcp", "win", 32768)
- self.pkt_bld.set_layer_attr("l4_tcp", "seq", 0)
- # set packet payload, for example HTTP GET request
- self.pkt_bld.set_pkt_payload('GET /10k_60k HTTP/1.1\r\nHost: 22.0.0.3\r\nConnection: Keep-Alive\r\nUser-Agent: Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; SV1; .NET CLR 1.1.4322; .NET CLR 2.0.50727)\r\nAccept: */*\r\nAccept-Language: en-us\r\nAccept-Encoding: gzip, deflate, compress\r\n\r\n')
- # finally, set IP header len with relation to payload data
- self.pkt_bld.set_layer_attr("l3_ip", "len", len(self.pkt_bld.get_layer('l3_ip')))
-
- filepath = 'reports/test%s.pcap' % os.getenv('SETUP_DIR', '')
- self.pkt_bld.dump_pkt_to_pcap(filepath)
- assert os.path.isfile(filepath)
- # remove pcap after ensuring it exists
- os.remove(filepath)
- filepath = "/not/a/valid/path/test.pcap"
- assert_raises(IOError, self.pkt_bld.dump_pkt_to_pcap, filepath)
- # check that dump is not available for empty packet
- new_pkt = CTRexPktBuilder()
- assert_raises(CTRexPktBuilder.EmptyPacketError, new_pkt.dump_pkt_to_pcap, filepath)
-
- def test_dump_pkt(self):
- # check that dump is not available for empty packet
- new_pkt = CTRexPktBuilder()
- assert_raises(CTRexPktBuilder.EmptyPacketError, new_pkt.dump_pkt)
-
- # set Ethernet layer attributes
- self.pkt_bld.set_eth_layer_addr("l2", "src", "00:15:17:a7:75:a3")
- self.pkt_bld.set_eth_layer_addr("l2", "dst", "e0:5f:b9:69:e9:22")
- self.pkt_bld.set_layer_attr("l2", "type", dpkt.ethernet.ETH_TYPE_IP)
- # set IP layer attributes
- self.pkt_bld.add_pkt_layer("l3_ip", dpkt.ip.IP())
- self.pkt_bld.set_ip_layer_addr("l3_ip", "src", "21.0.0.2")
- self.pkt_bld.set_ip_layer_addr("l3_ip", "dst", "22.0.0.12")
- self.pkt_bld.set_layer_attr("l3_ip", "p", dpkt.ip.IP_PROTO_ICMP)
- # set ICMP layer attributes
- self.pkt_bld.add_pkt_layer("icmp", dpkt.icmp.ICMP())
- self.pkt_bld.set_layer_attr("icmp", "type", dpkt.icmp.ICMP_ECHO)
- # set Echo(ICMP) layer attributes
- self.pkt_bld.add_pkt_layer("icmp_echo", dpkt.icmp.ICMP.Echo())
- self.pkt_bld.set_layer_attr("icmp_echo", "id", 24528)
- self.pkt_bld.set_layer_attr("icmp_echo", "seq", 11482)
- self.pkt_bld.set_pkt_payload('hello world')
-
- # finally, set IP header len with relation to payload data
- self.pkt_bld.set_layer_attr("l3_ip", "len", len(self.pkt_bld.get_layer('l3_ip')))
-
- self.print_packet(self.pkt_bld.get_packet())
- assert_equal(self.pkt_bld.dump_pkt(), {
- 'binary': '4F+5aekiABUXp3WjCABFAAAnAAAAAEABT8kVAAACFgAADAgA2YZf0CzaaGVsbG8gd29ybGQ=',
- 'meta': '',
- })
-
- def test_set_vm_ip_range_ipv4(self):
- # set some mock packet
- ip = dpkt.ip.IP()
- self.pkt_bld.add_pkt_layer("l3", ip)
- self.pkt_bld.add_pkt_layer("l4_tcp", dpkt.tcp.TCP())
- self.pkt_bld.set_pkt_payload("HelloWorld")
-
- self.pkt_bld.set_vm_ip_range("l3",
- "src",
- "10.0.0.1",
- "10.0.0.255",
- "inc")
-# self.pkt_bld.set_vm_custom_range(layer_name="l3",
-# hdr_field="tos",
-# init_val="10", start_val="10", end_val="200", add_val=2, val_size=1,
-# operation="inc")
- print ''
- self.pp.pprint(self.pkt_bld.vm.dump())
- assert_equal(self.pkt_bld.vm.dump(),
- { 'instructions': [ { 'init_value': 167772161,
- 'max_value': 167772415,
- 'min_value': 167772161,
- 'name': 'l3__src',
- 'op': 'inc',
- 'size': 4,
- 'type': 'flow_var'},
- { 'add_value': 0,
- 'is_big_endian': True,
- 'name': 'l3__src',
- 'pkt_offset': 26,
- 'type': 'write_flow_var'},
- { 'pkt_offset': 14, 'type': 'fix_checksum_ipv4'}],
- 'split_by_var': ''}
- )
-
- def test_set_vm_ip_range_ipv4_no_checksum(self):
- # set some mock packet
- ip = dpkt.ip.IP()
- self.pkt_bld.add_pkt_layer("l3", ip)
- self.pkt_bld.add_pkt_layer("l4_tcp", dpkt.tcp.TCP())
- self.pkt_bld.set_pkt_payload("HelloWorld")
-
- self.pkt_bld.set_vm_ip_range(ip_layer_name="l3",
- ip_field="src",
- ip_init="10.0.0.1", ip_start="10.0.0.1", ip_end="10.0.0.255",
- add_value=1,
- operation="inc",
- add_checksum_inst=False)
- print ''
- self.pp.pprint(self.pkt_bld.vm.dump())
- assert_equal(self.pkt_bld.vm.dump(),
- { 'instructions': [ { 'init_value': 167772161,
- 'max_value': 167772415,
- 'min_value': 167772161,
- 'name': 'l3__src',
- 'op': 'inc',
- 'size': 4,
- 'type': 'flow_var'},
- { 'add_value': 1,
- 'is_big_endian': True,
- 'name': 'l3__src',
- 'pkt_offset': 26,
- 'type': 'write_flow_var'}],
- 'split_by_var': ''}
- )
-
- def test_set_vm_ip_range_ipv6(self):
- # set some mock packet
- ip6 = dpkt.ip6.IP6()
- self.pkt_bld.add_pkt_layer("l3", ip6)
- self.pkt_bld.add_pkt_layer("l4_tcp", dpkt.tcp.TCP())
- self.pkt_bld.set_pkt_payload("HelloWorld")
-
- self.pkt_bld.set_vm_ip_range(ip_layer_name="l3",
- ip_field="src",
- ip_init="5001::DB8:1:3333:1:1", ip_start="5001::DB8:1:3333:1:1", ip_end="5001::DB8:1:3333:1:F",
- add_value=1,
- operation="inc",
- ip_type="ipv6")
- print ''
- self.pp.pprint(self.pkt_bld.vm.dump())
- assert_equal(self.pkt_bld.vm.dump(),
- { 'instructions': [ { 'init_value': 65537,
- 'max_value': 65551,
- 'min_value': 65537,
- 'name': 'l3__src',
- 'op': 'inc',
- 'size': 4,
- 'type': 'flow_var'},
- { 'add_value': 1,
- 'is_big_endian': True,
- 'name': 'l3__src',
- 'pkt_offset': 34,
- 'type': 'write_flow_var'}],
- 'split_by_var': ''}
- )
-
- def test_set_vm_eth_range(self):
- pass
-
- def test_set_vm_custom_range(self):
- # set some mock packet
- ip = dpkt.ip.IP()
- self.pkt_bld.add_pkt_layer("l3", ip)
- self.pkt_bld.add_pkt_layer("l4_tcp", dpkt.tcp.TCP())
- self.pkt_bld.set_pkt_payload("HelloWorld")
-
- self.pkt_bld.set_vm_custom_range(layer_name="l3",
- hdr_field="tos",
- init_val=10, start_val=10, end_val=200, add_val=2, val_size=1,
- operation="inc")
- print ''
- self.pp.pprint(self.pkt_bld.vm.dump())
- assert_equal(self.pkt_bld.vm.dump(),
- { 'instructions': [ { 'init_value': 10,
- 'max_value': 200,
- 'min_value': 10,
- 'name': 'l3__tos',
- 'op': 'inc',
- 'size': 1,
- 'type': 'flow_var'},
- { 'add_value': 2,
- 'is_big_endian': True,
- 'name': 'l3__tos',
- 'pkt_offset': 15,
- 'type': 'write_flow_var'},
- { 'pkt_offset': 14, 'type': 'fix_checksum_ipv4'}],
- 'split_by_var': ''}
- )
-
- def test_various_ranges(self):
- # set some mock packet
- ip = dpkt.ip.IP()
- self.pkt_bld.add_pkt_layer("l3", ip)
- self.pkt_bld.add_pkt_layer("l4_tcp", dpkt.tcp.TCP())
- self.pkt_bld.set_pkt_payload("HelloWorld")
-
- self.pkt_bld.set_vm_ip_range("l3",
- "src",
- "10.0.0.1",
- "10.0.0.255",
- "inc")
- self.pkt_bld.set_vm_custom_range(layer_name="l3",
- hdr_field="tos",
- init_val=10, start_val=10, end_val=200, add_val=2, val_size=1,
- operation="inc")
- print ''
- self.pp.pprint(self.pkt_bld.vm.dump())
-
- assert_equal(self.pkt_bld.vm.dump(),
- {'instructions': [{'init_value': 167772161,
- 'max_value': 167772415,
- 'min_value': 167772161,
- 'name': 'l3__src',
- 'op': 'inc',
- 'size': 4,
- 'type': 'flow_var'},
- {'init_value': 10,
- 'max_value': 200,
- 'min_value': 10,
- 'name': 'l3__tos',
- 'op': 'inc',
- 'size': 1,
- 'type': 'flow_var'},
- {'add_value': 2,
- 'is_big_endian': True,
- 'name': 'l3__tos',
- 'pkt_offset': 15,
- 'type': 'write_flow_var'},
- {'add_value': 0,
- 'is_big_endian': True,
- 'name': 'l3__src',
- 'pkt_offset': 26,
- 'type': 'write_flow_var'},
- {'pkt_offset': 14, 'type': 'fix_checksum_ipv4'}],
- 'split_by_var': ''}
- )
-
- def tearDown(self):
- pass
-
-
-if __name__ == "__main__":
- pass
-
diff --git a/scripts/automation/regression/unit_tests/functional_tests/vm_test.py b/scripts/automation/regression/unit_tests/functional_tests/vm_test.py
deleted file mode 100755
index f3948d99..00000000
--- a/scripts/automation/regression/unit_tests/functional_tests/vm_test.py
+++ /dev/null
@@ -1,77 +0,0 @@
-#!/router/bin/python
-
-import pkt_bld_general_test
-from client_utils.packet_builder import *
-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 CTRexVM_Test(pkt_bld_general_test.CGeneralPktBld_Test):
-
- def setUp(self):
- self.vm = CTRexPktBuilder.CTRexVM()
-
- def test_add_flow_man_inst(self):
- self.vm.add_flow_man_inst("src_ip")
- assert_raises(CTRexPktBuilder.VMVarNameExistsError, self.vm.add_flow_man_inst, "src_ip")
- self.vm.add_flow_man_inst("dst_ip", size=8, operation="inc", init_value=5, max_value=100)
- assert_equal(self.vm.vm_variables["dst_ip"].dump(),
- {"type": "flow_var",
- "name": "dst_ip",
- "size": 8,
- # "big_endian": True,
- "op": "inc",
-# "split_by_core": False,
- "init_value": 5,
- "min_value": 1,
- "max_value": 100})
- assert_raises(CTRexPktBuilder.VMFieldNameError, self.vm.add_flow_man_inst, "src_mac", no_field=1)
-
- def test_load_flow_man(self):
- vm2 = CTRexPktBuilder.CTRexVM()
- vm2.add_flow_man_inst("dst_ip", size=8, operation="inc", init_value=5, max_value=100)
- self.vm.load_flow_man(vm2.vm_variables["dst_ip"])
- assert_equal(self.vm.vm_variables["dst_ip"].dump(),
- vm2.vm_variables["dst_ip"].dump())
-
- def test_set_vm_var_field(self):
- self.vm.add_flow_man_inst("src_ip")
- self.vm.set_vm_var_field("src_ip", "size", 8)
- assert_equal(self.vm.vm_variables["src_ip"].size, 8)
- assert_raises(KeyError, self.vm.set_vm_var_field, "no_var", "no_field", 10)
- assert_raises(CTRexPktBuilder.VMFieldNameError, self.vm.set_vm_var_field, "src_ip", "no_field", 10)
- assert_raises(CTRexPktBuilder.VMFieldValueError, self.vm.set_vm_var_field, "src_ip", "operation", "rand")
-
- def test_dump(self):
- self.vm.add_flow_man_inst("dst_ip", size=8, operation="inc", init_value=5, max_value=100)
- self.vm.add_flow_man_inst("src_ip", size=8, operation="dec", init_value=10, min_value=2, max_value=100)
- from pprint import pprint
- print ''
- pprint (self.vm.dump())
- assert_equal(self.vm.dump(),
- {'instructions': [{'init_value': 10,
- 'max_value': 100,
- 'min_value': 2,
- 'name': 'src_ip',
- 'op': 'dec',
- 'size': 8,
- 'type': 'flow_var'},
- {'init_value': 5,
- 'max_value': 100,
- 'min_value': 1,
- 'name': 'dst_ip',
- 'op': 'inc',
- 'size': 8,
- 'type': 'flow_var'}],
- 'split_by_var': ''}
- )
-
-
- def tearDown(self):
- pass
-
-
-if __name__ == "__main__":
- pass
-
diff --git a/scripts/automation/regression/unit_tests/functional_tests/vm_variable_test.py b/scripts/automation/regression/unit_tests/functional_tests/vm_variable_test.py
deleted file mode 100755
index 699f1d97..00000000
--- a/scripts/automation/regression/unit_tests/functional_tests/vm_variable_test.py
+++ /dev/null
@@ -1,63 +0,0 @@
-#!/router/bin/python
-
-import pkt_bld_general_test
-from client_utils.packet_builder import *
-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 CTRexVMVariable_Test(pkt_bld_general_test.CGeneralPktBld_Test):
-
- def setUp(self):
- self.vm_var = CTRexPktBuilder.CTRexVM.CTRexVMFlowVariable("test_var")
-
- def test_var_name(self):
- assert_equal(self.vm_var.name, "test_var")
-
-# @raises(CTRexPktBuilder.VMVarNameError)
- def test_set_field(self):
- assert_raises(CTRexPktBuilder.VMFieldNameError, self.vm_var.set_field, "no_field", 10)
- # test for 'size' field
- assert_raises(CTRexPktBuilder.VMFieldTypeError, self.vm_var.set_field, "size", "wrong_type")
- assert_raises(CTRexPktBuilder.VMFieldValueError, self.vm_var.set_field, "size", 10) # 10 is illegal size
- self.vm_var.set_field("size", 8)
- assert_equal(self.vm_var.size, 8)
- # test for 'init_value' field
- assert_raises(CTRexPktBuilder.VMFieldTypeError, self.vm_var.set_field, "init_value", '123') # 123 is wrong type, should be int
- self.vm_var.set_field("init_value", 5)
- assert_equal(self.vm_var.init_value, 5)
- # test for 'operation' field
- assert_raises(CTRexPktBuilder.VMFieldTypeError, self.vm_var.set_field, "operation", 1) # operation is field of type str
- assert_raises(CTRexPktBuilder.VMFieldValueError, self.vm_var.set_field, "operation", "rand") # "rand" is illegal option
- self.vm_var.set_field("operation", "inc")
- assert_equal(self.vm_var.operation, "inc")
- # test for 'split_by_core' field
-# self.vm_var.set_field("split_by_core", 5)
-# assert_equal(self.vm_var.split_by_core, True)
-
- def test_var_dump (self):
- # set VM variable options
- self.vm_var.set_field("size", 8)
- self.vm_var.set_field("init_value", 5)
- self.vm_var.set_field("operation", "inc")
-# self.vm_var.set_field("split_by_core", False)
- self.vm_var.set_field("max_value", 100)
- assert_equal(self.vm_var.dump(),
- {"type": "flow_var",
- "name": "test_var",
- "size": 8,
- # "big_endian": True,
- "op": "inc",
-# "split_by_core": False,
- "init_value": 5,
- "min_value": 1,
- "max_value": 100})
-
- def tearDown(self):
- pass
-
-
-if __name__ == "__main__":
- pass
-