From 4d49bbd3737786982b5aa75274bba4d42bd68e78 Mon Sep 17 00:00:00 2001 From: MSAdministrator Date: Thu, 9 Jun 2022 09:51:11 -0500 Subject: [PATCH] dev: Updated tests --- tests/test_art.py | 17 ++++++++--------- tests/test_config_parser.py | 12 ------------ 2 files changed, 8 insertions(+), 21 deletions(-) delete mode 100644 tests/test_config_parser.py diff --git a/tests/test_art.py b/tests/test_art.py index e580efb..3e740d7 100644 --- 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 deleted file mode 100644 index 33978cc..0000000 --- a/tests/test_config_parser.py +++ /dev/null @@ -1,12 +0,0 @@ -from atomic_operator.configparser import ConfigParser -from atomic_operator.base import Base - -def test_config_parser(): - config = ConfigParser( - config_file='config.example.yml', - techniques=Base().parse_input_lists("T1003,T1004"), - host_list=Base().parse_input_lists("123.123.123.123, 1.1.1.1"), - username='username', - password='password' - ) - assert isinstance(config.run_list, list)