def test_calculate_vts_collection_hash_no_params(self):
vts = Vts(storage=OrderedDict)
vts.add(
'id_1',
name='foo',
vt_modification_time='01234',
vt_params={
'0': {
'id': '0',
'name': 'timeout',
'default': '20',
},
'1': {
'id': '1',
'name': 'foo_pref:',
'default': 'bar_value',
},
},
)
vts.add('id_2', name='bar', vt_modification_time='56789')
vts.calculate_vts_collection_hash(include_vt_params=False)
vt_hash = sha256()
vt_hash.update("id_101234id_256789".encode('utf-8'))
hash_test = vt_hash.hexdigest()
self.assertEqual(hash_test, vts.sha256_hash)
def test_add_vt_with_invalid_id(self):
vts = Vts()
with self.assertRaises(OspdError):
vts.add('$$$_1', name='foo')
self.assertEqual(len(vts.vts), 0)
def test_keys(self):
vts = Vts()
vts.add('id_1', name='foo')
vts.add('id_2', name='bar')
self.assertEqual(vts.keys(), ['id_1', 'id_2'])
def __init__(
self,
*,
customvtfilter=None,
storage=None,
max_scans=0,
min_free_mem_scan_queue=0,
file_storage_dir='/var/run/ospd',
max_queued_scans=0,
**kwargs,
): # pylint: disable=unused-argument
"""Initializes the daemon's internal data."""
self.scan_collection = ScanCollection(file_storage_dir)
self.scan_processes = dict()
self.daemon_info = dict()
self.daemon_info['name'] = "OSPd"
self.daemon_info['version'] = __version__
self.daemon_info['description'] = "No description"
self.scanner_info = dict()
self.scanner_info['name'] = 'No name'
self.scanner_info['version'] = 'No version'
self.scanner_info['description'] = 'No description'
self.server_version = None # Set by the subclass.
self.initialized = None # Set after initialization finished
self.max_scans = max_scans
self.min_free_mem_scan_queue = min_free_mem_scan_queue
self.max_queued_scans = max_queued_scans
self.last_scan_start_time = 0
self.scaninfo_store_time = kwargs.get('scaninfo_store_time')
self.protocol_version = PROTOCOL_VERSION
self.commands = {}
for command_class in get_commands():
command = command_class(self)
self.commands[command.get_name()] = command
self.scanner_params = dict()
for name, params in BASE_SCANNER_PARAMS.items():
self.set_scanner_param(name, params)
self.vts = Vts(storage)
self.vts_version = None
self.feed_name = None
self.feed_vendor = None
self.feed_home = None
if customvtfilter:
self.vts_filter = customvtfilter
else:
self.vts_filter = VtsFilter()
def test_add_vt_with_empty_id(self):
vts = Vts()
with self.assertRaises(OspdError):
vts.add(None, name='foo')
with self.assertRaises(OspdError):
vts.add('', name='foo')
def test_add_duplicate_vt(self):
vts = Vts()
vts.add('id_1', name='foo')
with self.assertRaises(OspdError):
vts.add('id_1', name='bar')
self.assertEqual(len(vts.vts), 1)
def test_calculate_vts_collection_hash_empty(self):
vts = Vts()
logging.Logger.debug = Mock()
vts.calculate_vts_collection_hash()
self.assertEqual(vts.sha256_hash, None)
logging.Logger.debug.assert_called_with( # pylint: disable=no-member
"Error calculating VTs collection hash. Cache is empty")
def test_keys(self):
vts = Vts()
vts.add('id_1', name='foo')
vts.add('id_2', name='bar')
# use assertCountEqual for Python 3.5 because dict.keys order is
# undefined
self.assertCountEqual(vts.keys(), ['id_1', 'id_2'])
def test_get(self):
vts = Vts()
vts.add('id_1', name='foo')
vt = vts.get('id_1')
self.assertIsNotNone(vt)
self.assertEqual(vt['name'], 'foo')
self.assertIsNone(vt.get('bar'))
def test_getitem(self):
vts = Vts()
vts.add('id_1', name='foo')
vt = vts['id_1']
self.assertEqual(vt['name'], 'foo')
with self.assertRaises(KeyError):
vt = vts['foo']
def test_calculate_vts_collection_hash(self):
vts = Vts()
vts.add('id_2', name='bar', vt_modification_time='56789')
vts.add('id_1', name='foo', vt_modification_time='01234')
vts.calculate_vts_collection_hash()
h = sha256()
h.update("id_101234id_256789".encode('utf-8'))
hash_test = h.hexdigest()
self.assertEqual(hash_test, vts.sha256_hash)
def test_iterator(self):
vts = Vts()
vts.add('id_1', name='foo')
vts.add('id_2', name='bar')
it = iter(vts)
vt_id = next(it)
self.assertIn(vt_id, ['id_1', 'id_2'])
vt_id = next(it)
self.assertIn(vt_id, ['id_1', 'id_2'])
with self.assertRaises(StopIteration):
next(it)
def test_iterator(self):
vts = Vts()
vts.add('id_1', name='foo')
vts.add('id_2', name='bar')
it = iter(vts)
vt_id = next(it)
# Python 3.5 doesn't ensure the order of the returned keys
self.assertIn(vt_id, ['id_1', 'id_2'])
vt_id = next(it)
self.assertIn(vt_id, ['id_1', 'id_2'])
with self.assertRaises(StopIteration):
next(it)
def test_copy(self):
vts = Vts()
vts.add('id_1', name='foo')
vts.add('id_2', name='bar')
vts2 = vts.copy()
self.assertIsNot(vts, vts2)
self.assertIsNot(vts.vts, vts2.vts)
vta = vts.get('id_1')
vtb = vts2.get('id_1')
self.assertEqual(vta['name'], vtb['name'])
self.assertIsNot(vta, vtb)
vta = vts.get('id_2')
vtb = vts2.get('id_2')
self.assertEqual(vta['name'], vtb['name'])
self.assertIsNot(vta, vtb)
def __init__(self, *, customvtfilter=None, **kwargs): # pylint: disable=unused-argument
""" Initializes the daemon's internal data. """
self.scan_collection = ScanCollection()
self.scan_processes = dict()
self.daemon_info = dict()
self.daemon_info['name'] = "OSPd"
self.daemon_info['version'] = __version__
self.daemon_info['description'] = "No description"
self.scanner_info = dict()
self.scanner_info['name'] = 'No name'
self.scanner_info['version'] = 'No version'
self.scanner_info['description'] = 'No description'
self.server_version = None # Set by the subclass.
self.initialized = None # Set after initialization finished
self.scaninfo_store_time = kwargs.get('scaninfo_store_time')
self.protocol_version = PROTOCOL_VERSION
self.commands = {}
for command_class in get_commands():
command = command_class(self)
self.commands[command.get_name()] = command
self.scanner_params = dict()
for name, params in BASE_SCANNER_PARAMS.items():
self.set_scanner_param(name, params)
self.vts = Vts()
self.vts_version = None
if customvtfilter:
self.vts_filter = customvtfilter
else:
self.vts_filter = VtsFilter()
def test_contains(self):
vts = Vts()
vts.add('id_1', name='foo')
self.assertIn('id_1', vts)
def test_load_vts(self, ):
w = DummyDaemon()
w.load_vts()
self.assertIsInstance(w.vts, type(Vts()))
self.assertEqual(len(w.vts), len(w.VTS))
def test_add_vt(self):
vts = Vts()
vts.add('id_1', name='foo')
self.assertEqual(len(vts.vts), 1)
