Added VAR_STATE_PATH check.

certtools · Aug 24, 2023 · 0660a89 · 0660a89
1 parent a045bee
commit 0660a89
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 6 deletions.
diff --git a/intelmq/bots/parsers/shadowserver/_config.py b/intelmq/bots/parsers/shadowserver/_config.py
@@ -117,6 +117,7 @@ def enable_test_mode(enable):
     else:
         __config.schema_active = __config.schema_file
 
+
 def enable_auto_update(enable):
     """ Enable automatic schema update. """
     __config.auto_update = enable

diff --git a/intelmq/tests/bots/parsers/shadowserver/test_download_schema.py b/intelmq/tests/bots/parsers/shadowserver/test_download_schema.py
@@ -20,9 +20,10 @@
 class TestShadowserverSchemaDownload(unittest.TestCase):
 
     def test_download(self):
-        schema_file = os.path.join(VAR_STATE_PATH, 'shadowserver-schema.json')
-        config.set_logger(utils.log('test-bot', log_path=None))
-        if os.path.exists(schema_file):
-            os.unlink(schema_file)
-        self.assertEqual(True, config.update_schema())
-        self.assertEqual(True, os.path.exists(schema_file))
+        if os.path.isdir(VAR_STATE_PATH):
+            schema_file = os.path.join(VAR_STATE_PATH, 'shadowserver-schema.json')
+            config.set_logger(utils.log('test-bot', log_path=None))
+            if os.path.exists(schema_file):
+                os.unlink(schema_file)
+            self.assertEqual(True, config.update_schema())
+            self.assertEqual(True, os.path.exists(schema_file))