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
|
#!/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
|
