Add tests

greenbone · Apr 26, 2021 · 8a4f2c5 · 8a4f2c5
1 parent 5369e05
commit 8a4f2c5
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/tests/test_preferencehandler.py b/tests/test_preferencehandler.py
@@ -280,6 +280,17 @@ def test_set_ports(self, mock_kb):
             ['port_range|||80,443'],
         )
 
+    @patch('ospd_openvas.db.KbDB')
+    def test_set_ports_invalid(self, mock_kb):
+        w = DummyDaemon()
+
+        w.scan_collection.get_ports = MagicMock(return_value='2,-9,4')
+
+        p = PreferenceHandler('1234-1234', mock_kb, w.scan_collection, None)
+        p.scan_id = '456-789'
+        p.kbdb.add_scan_preferences = MagicMock()
+        self.assertFalse(p.prepare_ports_for_openvas())
+
     @patch('ospd_openvas.db.KbDB')
     def test_set_main_kbindex(self, mock_kb):
         w = DummyDaemon()