From c9f3ebfbb16de3b5e9b009046aeac54f651bd723 Mon Sep 17 00:00:00 2001 From: Juan Jose Nicola Date: Wed, 20 Oct 2021 03:47:05 -0500 Subject: [PATCH] Update tests --- tests/test_nvti_cache.py | 15 +++++---------- 1 file changed, 5 insertions(+), 10 deletions(-) diff --git a/tests/test_nvti_cache.py b/tests/test_nvti_cache.py index beef386b..aff8345e 100644 --- a/tests/test_nvti_cache.py +++ b/tests/test_nvti_cache.py @@ -174,7 +174,6 @@ def test_get_nvt_metadata(self, MockOpenvasDB): '', 'URL:http://www.mantisbt.org/', '3', - '10', 'Product detection', 'Mantis Detection', ] @@ -204,7 +203,7 @@ def test_get_nvt_metadata(self, MockOpenvasDB): 'id': '0', 'type': 'entry', 'name': 'timeout', - 'description': 'Script Timeout', + 'description': 'Description', 'default': '10', }, '1': { @@ -217,7 +216,10 @@ def test_get_nvt_metadata(self, MockOpenvasDB): }, } - prefs1 = ['1|||dns-fuzz.timelimit|||entry|||default'] + prefs1 = [ + '0|||timeout|||entry|||10', + '1|||dns-fuzz.timelimit|||entry|||default', + ] MockOpenvasDB.get_list_item.side_effect = [metadata, prefs1] resp = self.nvti.get_nvt_metadata('1.2.3.4') @@ -261,13 +263,6 @@ def test_get_nvt_prefs(self, MockOpenvasDB): self.assertEqual(resp, prefs) - def test_get_nvt_timeout(self, MockOpenvasDB): - MockOpenvasDB.get_single_item.return_value = '300' - - resp = self.nvti.get_nvt_timeout('1.2.3.4') - - self.assertEqual(resp, '300') - def test_get_nvt_tags(self, MockOpenvasDB): tag = ( 'last_modification=1533906565'