| @@ 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 | ||