dev: Updated tests

swimlane · Jun 9, 2022 · 4d49bbd · 4d49bbd
1 parent 995ff52
commit 4d49bbd
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 21 deletions.
diff --git a/tests/test_art.py b/tests/test_art.py
@@ -1,11 +1,12 @@
 import os
 from atomic_operator.frameworks.loader import Loader
-from atomic_operator.configparser import ConfigParser
+from atomic_operator.runlist import RunList
+
 
 
 def test_download_of_atomic_red_team_repo():
     from atomic_operator import AtomicOperator
-    AtomicOperator().art.get_content(desintation=os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+    AtomicOperator().art.get_content(destination=os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
     for item in os.listdir(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))):
         if 'redcanaryco-atomic-red-team' in item and os.path.isdir(item):
             assert True
@@ -14,10 +15,8 @@ def test_loading_of_technique():
     assert Loader().load_yaml(os.path.join(os.path.dirname(os.path.abspath(__file__)), 'data', 'atomic_operator_config.yml'))
 
 def test_parsing_of_config():
-    config_parser = ConfigParser(config_file=os.path.join(os.path.dirname(os.path.abspath(__file__)), 'data', 'atomic_operator_config.yml'))
-    assert config_parser.is_defined('f7e6ec05-c19e-4a80-a7e7-241027992fdb')
-    inputs = config_parser.get_inputs('f7e6ec05-c19e-4a80-a7e7-241027992fdb')
-    assert inputs.get('output_file')
-    assert inputs.get('input_file')
-    assert config_parser.is_defined('3ff64f0b-3af2-3866-339d-38d9791407c3')
-    assert config_parser.is_defined('32f90516-4bc9-43bd-b18d-2cbe0b7ca9b2')
+    run_list = RunList(
+        adversary='apt29',
+        config_file=os.path.join(os.path.dirname(os.path.abspath(__file__)), 'data', 'atomic_operator_config.yml')
+    )
+    assert run_list
diff --git a/tests/test_config_parser.py b/tests/test_config_parser.py