Code Duplication    Length = 123-123 lines in 2 locations

tests/protocols/gmpv7/testcmds/test_create_port_range.py 1 location

@@ 26-148 (lines=123) @@
23
from gvm.protocols.gmpv7 import PortRangeType
24
25
26
class GmpCreatePortRangeTestCase:
27
    def test_create_port_range_missing_port_list_id(self):
28
        with self.assertRaises(RequiredArgument):
29
            self.gmp.create_port_range(
30
                port_list_id=None,
31
                start=1,
32
                end=1234,
33
                port_range_type=PortRangeType.TCP,
34
            )
35
36
        with self.assertRaises(RequiredArgument):
37
            self.gmp.create_port_range(
38
                port_list_id='',
39
                start=1,
40
                end=1234,
41
                port_range_type=PortRangeType.TCP,
42
            )
43
44
    def test_create_port_range_missing_start(self):
45
        with self.assertRaises(RequiredArgument):
46
            self.gmp.create_port_range(
47
                port_list_id='pl1',
48
                start=None,
49
                end=1234,
50
                port_range_type=PortRangeType.TCP,
51
            )
52
53
        with self.assertRaises(RequiredArgument):
54
            self.gmp.create_port_range(
55
                port_list_id='pl1',
56
                start='',
57
                end=1234,
58
                port_range_type=PortRangeType.TCP,
59
            )
60
61
    def test_create_port_range_missing_end(self):
62
        with self.assertRaises(RequiredArgument):
63
            self.gmp.create_port_range(
64
                port_list_id='pl1',
65
                start=1,
66
                end=None,
67
                port_range_type=PortRangeType.TCP,
68
            )
69
70
        with self.assertRaises(RequiredArgument):
71
            self.gmp.create_port_range(
72
                port_list_id='pl1',
73
                start=1,
74
                end='',
75
                port_range_type=PortRangeType.TCP,
76
            )
77
78
    def test_create_port_range_missing_port_range_type(self):
79
        with self.assertRaises(RequiredArgument):
80
            self.gmp.create_port_range(
81
                port_list_id='pl1', start=1, end=1234, port_range_type=None
82
            )
83
84
        with self.assertRaises(RequiredArgument):
85
            self.gmp.create_port_range(
86
                port_list_id='pl1', start=1, end=1234, port_range_type=''
87
            )
88
89
    def test_create_port_range(self):
90
        self.gmp.create_port_range(
91
            port_list_id='pl1',
92
            start=1,
93
            end=1234,
94
            port_range_type=PortRangeType.TCP,
95
        )
96
97
        self.connection.send.has_been_called_with(
98
            '<create_port_range>'
99
            '<port_list id="pl1"/>'
100
            '<start>1</start>'
101
            '<end>1234</end>'
102
            '<type>TCP</type>'
103
            '</create_port_range>'
104
        )
105
106
        self.gmp.create_port_range(
107
            port_list_id='pl1',
108
            start=1,
109
            end=1234,
110
            port_range_type=PortRangeType.UDP,
111
        )
112
113
        self.connection.send.has_been_called_with(
114
            '<create_port_range>'
115
            '<port_list id="pl1"/>'
116
            '<start>1</start>'
117
            '<end>1234</end>'
118
            '<type>UDP</type>'
119
            '</create_port_range>'
120
        )
121
122
        self.gmp.create_port_range(
123
            port_list_id='pl1',
124
            start='1',
125
            end='1234',
126
            port_range_type=PortRangeType.TCP,
127
        )
128
129
        self.connection.send.has_been_called_with(
130
            '<create_port_range>'
131
            '<port_list id="pl1"/>'
132
            '<start>1</start>'
133
            '<end>1234</end>'
134
            '<type>TCP</type>'
135
            '</create_port_range>'
136
        )
137
138
    def test_create_port_range_with_comment(self):
139
        self.gmp.create_port_range(
140
            port_list_id='pl1',
141
            start=1,
142
            end=1234,
143
            port_range_type=PortRangeType.TCP,
144
            comment='lorem',
145
        )
146
147
        self.connection.send.has_been_called_with(
148
            '<create_port_range>'
149
            '<port_list id="pl1"/>'
150
            '<start>1</start>'
151
            '<end>1234</end>'

tests/protocols/gmpv208/testcmds/test_create_port_range.py 1 location

@@ 26-148 (lines=123) @@
23
from gvm.protocols.gmpv208 import PortRangeType
24
25
26
class GmpCreatePortRangeTestCase:
27
    def test_create_port_range_missing_port_list_id(self):
28
        with self.assertRaises(RequiredArgument):
29
            self.gmp.create_port_range(
30
                port_list_id=None,
31
                start=1,
32
                end=1234,
33
                port_range_type=PortRangeType.TCP,
34
            )
35
36
        with self.assertRaises(RequiredArgument):
37
            self.gmp.create_port_range(
38
                port_list_id='',
39
                start=1,
40
                end=1234,
41
                port_range_type=PortRangeType.TCP,
42
            )
43
44
    def test_create_port_range_missing_start(self):
45
        with self.assertRaises(RequiredArgument):
46
            self.gmp.create_port_range(
47
                port_list_id='pl1',
48
                start=None,
49
                end=1234,
50
                port_range_type=PortRangeType.TCP,
51
            )
52
53
        with self.assertRaises(RequiredArgument):
54
            self.gmp.create_port_range(
55
                port_list_id='pl1',
56
                start='',
57
                end=1234,
58
                port_range_type=PortRangeType.TCP,
59
            )
60
61
    def test_create_port_range_missing_end(self):
62
        with self.assertRaises(RequiredArgument):
63
            self.gmp.create_port_range(
64
                port_list_id='pl1',
65
                start=1,
66
                end=None,
67
                port_range_type=PortRangeType.TCP,
68
            )
69
70
        with self.assertRaises(RequiredArgument):
71
            self.gmp.create_port_range(
72
                port_list_id='pl1',
73
                start=1,
74
                end='',
75
                port_range_type=PortRangeType.TCP,
76
            )
77
78
    def test_create_port_range_missing_port_range_type(self):
79
        with self.assertRaises(RequiredArgument):
80
            self.gmp.create_port_range(
81
                port_list_id='pl1', start=1, end=1234, port_range_type=None
82
            )
83
84
        with self.assertRaises(RequiredArgument):
85
            self.gmp.create_port_range(
86
                port_list_id='pl1', start=1, end=1234, port_range_type=''
87
            )
88
89
    def test_create_port_range(self):
90
        self.gmp.create_port_range(
91
            port_list_id='pl1',
92
            start=1,
93
            end=1234,
94
            port_range_type=PortRangeType.TCP,
95
        )
96
97
        self.connection.send.has_been_called_with(
98
            '<create_port_range>'
99
            '<port_list id="pl1"/>'
100
            '<start>1</start>'
101
            '<end>1234</end>'
102
            '<type>TCP</type>'
103
            '</create_port_range>'
104
        )
105
106
        self.gmp.create_port_range(
107
            port_list_id='pl1',
108
            start=1,
109
            end=1234,
110
            port_range_type=PortRangeType.UDP,
111
        )
112
113
        self.connection.send.has_been_called_with(
114
            '<create_port_range>'
115
            '<port_list id="pl1"/>'
116
            '<start>1</start>'
117
            '<end>1234</end>'
118
            '<type>UDP</type>'
119
            '</create_port_range>'
120
        )
121
122
        self.gmp.create_port_range(
123
            port_list_id='pl1',
124
            start='1',
125
            end='1234',
126
            port_range_type=PortRangeType.TCP,
127
        )
128
129
        self.connection.send.has_been_called_with(
130
            '<create_port_range>'
131
            '<port_list id="pl1"/>'
132
            '<start>1</start>'
133
            '<end>1234</end>'
134
            '<type>TCP</type>'
135
            '</create_port_range>'
136
        )
137
138
    def test_create_port_range_with_comment(self):
139
        self.gmp.create_port_range(
140
            port_list_id='pl1',
141
            start=1,
142
            end=1234,
143
            port_range_type=PortRangeType.TCP,
144
            comment='lorem',
145
        )
146
147
        self.connection.send.has_been_called_with(
148
            '<create_port_range>'
149
            '<port_list id="pl1"/>'
150
            '<start>1</start>'
151
            '<end>1234</end>'