Fix tests

fox-it · Nov 16, 2023 · e56bd7a · e56bd7a
1 parent 4013015
commit e56bd7a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/tests/test_acquire_command.py b/tests/test_acquire_command.py
@@ -8,6 +8,7 @@
     CONFIG,
     MODULES,
     PROFILES,
+    VOLATILE,
     create_argument_parser,
     parse_acquire_args,
 )
@@ -17,7 +18,7 @@
 def acquire_parser_args(config: List, argument_list: List) -> Namespace:
     CONFIG["arguments"] = config
     with patch("argparse._sys.argv", [""] + argument_list):
-        return parse_acquire_args(create_argument_parser(PROFILES, MODULES), config=CONFIG)
+        return parse_acquire_args(create_argument_parser(PROFILES, VOLATILE, MODULES), config=CONFIG)
 
 
 @pytest.mark.parametrize("config, argument_list", [([], [])])

diff --git a/tests/test_utils.py b/tests/test_utils.py
@@ -6,7 +6,7 @@
 import pytest
 from dissect.target import Target
 
-from acquire.acquire import MODULES, PROFILES
+from acquire.acquire import MODULES, PROFILES, VOLATILE
 from acquire.utils import (
     check_and_set_acquire_args,
     check_and_set_log_args,
@@ -26,7 +26,7 @@ def get_args(**kwargs):
         "public_key": None,
     }
 
-    parser = create_argument_parser(PROFILES, MODULES)
+    parser = create_argument_parser(PROFILES, VOLATILE, MODULES)
     default_args = dict(parser.parse_args(args=[])._get_kwargs())
     default_args.update(kwargs)