Code Duplication    Length = 57-60 lines in 2 locations

tests/test_vthelper.py 2 locations

@@ 97-156 (lines=60) @@
94
        for _, values in vthelper.get_vt_iterator(vt_selection=vt):
95
            self.assertIs(values, None)
96
97
    def test_get_single_vt_severity_cvssv3(self):
98
        w = DummyDaemon()
99
        w.nvti.get_nvt_metadata.return_value = {
100
            'category': '3',
101
            'creation_date': '1237458156',
102
            'cvss_base_vector': 'AV:N/AC:L/Au:N/C:N/I:N/A:N',
103
            'severity_vector': 'CVSS:3.0/AV:L/AC:H/PR:H/UI:R/S:U/C:N/I:L/A:L',
104
            'severity_date': '1237458156',
105
            'severity_origin': 'Greenbone',
106
            'excluded_keys': 'Settings/disable_cgi_scanning',
107
            'family': 'Product detection',
108
            'filename': 'mantis_detect.nasl',
109
            'last_modification': ('1533906565'),
110
            'name': 'Mantis Detection',
111
            'qod_type': 'remote_banner',
112
            'required_ports': 'Services/www, 80',
113
            'solution': 'some solution',
114
            'solution_type': 'WillNotFix',
115
            'solution_method': 'DebianAPTUpgrade',
116
            'impact': 'some impact',
117
            'insight': 'some insight',
118
            'summary': ('some summary'),
119
            'affected': 'some affection',
120
            'timeout': '0',
121
            'vt_params': {
122
                '1': {
123
                    'id': '1',
124
                    'default': '',
125
                    'description': 'Description',
126
                    'name': 'Data length :',
127
                    'type': 'entry',
128
                },
129
                '2': {
130
                    'id': '2',
131
                    'default': 'no',
132
                    'description': 'Description',
133
                    'name': 'Do not randomize the  order  in  which ports are scanned',  # pylint: disable=line-too-long
134
                    'type': 'checkbox',
135
                },
136
            },
137
            'refs': {
138
                'bid': [''],
139
                'cve': [''],
140
                'xref': ['URL:http://www.mantisbt.org/'],
141
            },
142
        }
143
144
        vthelper = VtHelper(w.nvti)
145
146
        res = vthelper.get_single_vt("1.3.6.1.4.1.25623.1.0.100061")
147
        assert_called_once(w.nvti.get_nvt_metadata)
148
149
        severities = res.get('severities')
150
        self.assertEqual(
151
            "CVSS:3.0/AV:L/AC:H/PR:H/UI:R/S:U/C:N/I:L/A:L",
152
            severities.get('severity_base_vector'),
153
        )
154
        self.assertEqual("cvss_base_v3", severities.get('severity_type'))
155
        self.assertEqual("Greenbone", severities.get('severity_origin'))
156
        self.assertEqual("1237458156", severities.get('severity_date'))
157
158
    def test_get_single_vt_severity_cvssv2(self):
159
        w = DummyDaemon()
@@ 158-214 (lines=57) @@
155
        self.assertEqual("Greenbone", severities.get('severity_origin'))
156
        self.assertEqual("1237458156", severities.get('severity_date'))
157
158
    def test_get_single_vt_severity_cvssv2(self):
159
        w = DummyDaemon()
160
        w.nvti.get_nvt_metadata.return_value = {
161
            'category': '3',
162
            'creation_date': '1237458156',
163
            'cvss_base_vector': 'AV:N/AC:L/Au:N/C:N/I:N/A:N',
164
            'excluded_keys': 'Settings/disable_cgi_scanning',
165
            'family': 'Product detection',
166
            'filename': 'mantis_detect.nasl',
167
            'last_modification': ('1533906565'),
168
            'name': 'Mantis Detection',
169
            'qod_type': 'remote_banner',
170
            'required_ports': 'Services/www, 80',
171
            'solution': 'some solution',
172
            'solution_type': 'WillNotFix',
173
            'solution_method': 'DebianAPTUpgrade',
174
            'impact': 'some impact',
175
            'insight': 'some insight',
176
            'summary': ('some summary'),
177
            'affected': 'some affection',
178
            'timeout': '0',
179
            'vt_params': {
180
                '1': {
181
                    'id': '1',
182
                    'default': '',
183
                    'description': 'Description',
184
                    'name': 'Data length :',
185
                    'type': 'entry',
186
                },
187
                '2': {
188
                    'id': '2',
189
                    'default': 'no',
190
                    'description': 'Description',
191
                    'name': 'Do not randomize the  order  in  which ports are scanned',  # pylint: disable=line-too-long
192
                    'type': 'checkbox',
193
                },
194
            },
195
            'refs': {
196
                'bid': [''],
197
                'cve': [''],
198
                'xref': ['URL:http://www.mantisbt.org/'],
199
            },
200
        }
201
202
        vthelper = VtHelper(w.nvti)
203
204
        res = vthelper.get_single_vt("1.3.6.1.4.1.25623.1.0.100061")
205
        assert_called_once(w.nvti.get_nvt_metadata)
206
207
        severities = res.get('severities')
208
        self.assertEqual(
209
            "AV:N/AC:L/Au:N/C:N/I:N/A:N",
210
            severities.get('severity_base_vector'),
211
        )
212
        self.assertEqual("cvss_base_v2", severities.get('severity_type'))
213
        self.assertEqual(None, severities.get('severity_origin'))
214
        self.assertEqual("1237458156", severities.get('severity_date'))
215