1
1
import pytest
2
+
2
3
from pyshark .packet .fields import LayerField , LayerFieldsContainer
3
4
4
5
# Unit Tests - Layer Fields
5
- # **************************************************************************
6
6
def test_layer_field_hide ():
7
7
hide_yes = LayerField (hide = "yes" )
8
8
hide_not_yes = LayerField (hide = "not yes" )
9
- assert ( hide_yes .hide == True and hide_not_yes .hide == False )
9
+ assert hide_yes .hide == True and hide_not_yes .hide == False
10
10
11
11
def test_layer_field_print_format ():
12
12
name = "test_name"
13
13
show = "test"
14
14
test_layer_field = LayerField (name = name , show = show )
15
15
str_layer_field = test_layer_field .__repr__ ()
16
- assert ( str_layer_field == f'<LayerField { name } : { show } >' )
16
+ assert str_layer_field == f'<LayerField { name } : { show } >'
17
17
18
18
def test_layer_field_default_value_show ():
19
19
show = "test"
20
20
test_layer_field = LayerField (show = show )
21
21
default_value = test_layer_field .get_default_value ()
22
- assert ( default_value == show )
22
+ assert default_value == show
23
23
24
24
def test_layer_field_default_value_value ():
25
25
value = "test"
26
26
test_layer_field = LayerField (value = value )
27
27
default_value = test_layer_field .get_default_value ()
28
- assert ( default_value == value )
28
+ assert default_value == value
29
29
30
30
def test_layer_field_default_value_showname ():
31
31
showname = "test"
32
32
test_layer_field = LayerField (showname = showname )
33
33
default_value = test_layer_field .get_default_value ()
34
- assert ( default_value == showname )
34
+ assert default_value == showname
35
35
36
36
def test_layer_field_showname_value ():
37
37
value = "test_value"
38
38
showname = f"test: { value } "
39
39
test_layer_field = LayerField (showname = showname )
40
- assert ( value == test_layer_field .showname_value )
40
+ assert value == test_layer_field .showname_value
41
41
42
42
def test_layer_field_showname_key ():
43
43
key = "test_key"
44
44
showname = f"{ key } : value"
45
45
test_layer_field = LayerField (showname = showname )
46
- assert ( key == test_layer_field .showname_key )
46
+ assert key == test_layer_field .showname_key
47
47
48
48
def test_layer_field_showname_binary ():
49
49
binary = b"\x12 4"
50
50
test_layer_field = LayerField (value = "1234" )
51
- assert ( binary == test_layer_field .binary_value )
51
+ assert binary == test_layer_field .binary_value
52
52
53
53
def test_layer_field_str_int_value ():
54
54
str_int_value = "10"
55
55
int_value = 10
56
56
test_layer_field = LayerField (value = str_int_value )
57
- assert ( test_layer_field .int_value == int_value )
57
+ assert test_layer_field .int_value == int_value
58
58
59
59
def test_layer_field_hex_value ():
60
60
test_int_value = "0x75BCD15"
61
61
expected_value = 123456789
62
62
test_layer_field = LayerField (value = test_int_value )
63
- assert (test_layer_field .hex_value == expected_value )
64
- # ---------------------------------------------------------------------------
63
+ asserttest_layer_field .hex_value == expected_value
65
64
66
65
# Test Data - Layer Fields Container
67
- # **************************************************************************
68
66
test_layer_field = LayerField (name = "test_field" , value = {"test" : "value" })
69
67
70
68
@pytest .fixture
71
69
def layer_fields_container ():
72
70
return LayerFieldsContainer (test_layer_field )
73
- # ---------------------------------------------------------------------------
74
71
75
72
# Unit Tests - Layer Fields Container
76
- # **************************************************************************
77
73
def test_layer_fields_container_adds_single_field (layer_fields_container ):
78
74
new_field = LayerField (name = "new_field" , value = {"new_test" : "new_test_value" })
79
75
layer_fields_container .add_field (new_field )
80
76
fields = layer_fields_container .all_fields
81
- assert ( fields == [test_layer_field , new_field ])
77
+ assert fields == [test_layer_field , new_field ]
82
78
83
79
def test_layer_fields_container_returns_all_fields (layer_fields_container ):
84
80
fields = layer_fields_container .all_fields
85
- assert ( fields == [test_layer_field ])
81
+ assert fields == [test_layer_field ]
86
82
87
83
def test_layer_fields_container_gets_main_field (layer_fields_container ):
88
84
main_field = layer_fields_container .main_field
89
- assert ( main_field == test_layer_field )
85
+ assert main_field == test_layer_field
90
86
91
87
def test_layer_fields_container_gets_alternate_fields (layer_fields_container ):
92
88
alternate_field_1 = LayerField (name = "alt_field_1" , value = {"alt_field_1" : "alt_value_1" })
93
89
alternate_field_2 = LayerField (name = "alt_field_2" , value = {"alt_field_2" : "alt_value_2" })
94
90
layer_fields_container .add_field (alternate_field_1 )
95
91
layer_fields_container .add_field (alternate_field_2 )
96
92
alternate_fields = layer_fields_container .alternate_fields
97
- assert (alternate_fields == [alternate_field_1 , alternate_field_2 ])
98
- # ---------------------------------------------------------------------------
93
+ assert alternate_fields == [alternate_field_1 , alternate_field_2 ]
0 commit comments