From e56bd7a0c29b3a844f62b6b1616e86c0743b592e Mon Sep 17 00:00:00 2001
From: Miauwkeru <Miauwkeru@users.noreply.github.com>
Date: Thu, 16 Nov 2023 13:39:14 +0000
Subject: [PATCH] Fix tests

---
 tests/test_acquire_command.py | 3 ++-
 tests/test_utils.py           | 4 ++--
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/tests/test_acquire_command.py b/tests/test_acquire_command.py
index 9eb7bcee..46c505e0 100644
--- 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
index 69007f92..9321c119 100644
--- 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)