cleaning up test case yaml refs (#101)

muchdogesec · Nov 25, 2024 · 0e27d53 · 0e27d53
1 parent c6d75d8
commit 0e27d53
Show file tree

Hide file tree

Showing 8 changed files with 4 additions and 753 deletions.
diff --git a/includes/test_cases.yaml → includes/tests/test_cases.yaml b/includes/test_cases.yaml → includes/tests/test_cases.yaml
diff --git a/tests/data/extraction_types/ai_mitre_attack_enterprise.txt b/tests/data/extraction_types/ai_mitre_attack_enterprise.txt
@@ -9,9 +9,7 @@ T1040
 TA0003
 Rundll32
 OS Credential Dumping
-Operation Sharpshooter
-Operation Spalax
 
 ====Bad====
 
-
+None
diff --git a/tests/data/extraction_types/ai_mitre_attack_ics.txt b/tests/data/extraction_types/ai_mitre_attack_ics.txt
@@ -3,9 +3,4 @@
 TA0111
 Scripting
 Program Upload
-Energetic Bear
-BROMINE
-
-====Bad====
-
 
diff --git a/tests/data/extraction_types/ai_mitre_attack_mobile.txt b/tests/data/extraction_types/ai_mitre_attack_mobile.txt
@@ -6,9 +6,7 @@ T1630.001
 TA0029
 Impair Defenses
 Call Log
-Storm-0875
-Octo Tempest
 
 ====Bad====
 
-
+None
diff --git a/tests/data/extraction_types/all_cases.txt b/tests/data/extraction_types/all_cases.txt
@@ -121,8 +121,6 @@ T1040
 TA0003
 Rundll32
 OS Credential Dumping
-Operation Sharpshooter
-Operation Spalax
 T1174
 TA0006
 TA0011
@@ -132,34 +130,24 @@ T1040
 TA0003
 Rundll32
 OS Credential Dumping
-Operation Sharpshooter
-Operation Spalax
 M1013
 S0505
 T1630.001
 TA0029
 Impair Defenses
 Call Log
-Storm-0875
-Octo Tempest
 M1013
 S0505
 T1630.001
 TA0029
 Impair Defenses
 Call Log
-Storm-0875
-Octo Tempest
 TA0111
 Scripting
 Program Upload
-Energetic Bear
-BROMINE
 TA0111
 Scripting
 Program Upload
-Energetic Bear
-BROMINE
 CAPEC-110
 Clickjacking
 Overflow Buffers

diff --git a/tests/scripts/generate_simple_extraction_test_cases_txt_files.py b/tests/scripts/generate_simple_extraction_test_cases_txt_files.py
@@ -50,7 +50,7 @@ def write_to_file(file_path, positive_data, negative_data):
             file.write(f"{item}\n")
 
 # Usage
-file_path = 'tests/test_cases.yaml'  # Update this to your file's location
+file_path = 'includes/test_cases.yaml'  # Update this to your file's location
 output_dir = 'tests/data/extraction_types/'  # Update this to your desired output directory
 create_test_case_files(file_path, output_dir)