update

secator/__init__.py

@@ -7,7 +7,7 @@
 import requests
 import yaml
 from dotmap import DotMap
-from pydantic import AfterValidator, BaseModel, model_validator, ValidationError, Extra
+from pydantic import AfterValidator, BaseModel, model_validator, ValidationError
 
 from secator.rich import console, console_stdout
 
@@ -19,7 +19,7 @@
 CONFIGS_FOLDER = LIB_FOLDER / 'configs'
 
 
-class StrictModel(BaseModel, extra=Extra.forbid):
+class StrictModel(BaseModel, extra='forbid'):
 	pass
 
 

tests/unit/test_offline.py

@@ -1,9 +1,17 @@
 import os
+import sys
 import unittest
 
 
 class TestOffline(unittest.TestCase):
 	def setUp(self):
+		try:
+			# This allows to drop the secator module loaded from other tests in order to reload the config with modified
+   			# environment variables.
+			# See https://stackoverflow.com/questions/7460363/re-import-module-under-test-to-lose-context for context.
+			del sys.modules['secator']
+		except KeyError:
+			pass
 		os.environ['SECATOR_OFFLINE_MODE'] = '1'
 
 	def test_offline_cve_lookup(self):