txt2stix 1.0.3__tar.gz → 1.0.4__tar.gz

{txt2stix-1.0.3 → txt2stix-1.0.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: txt2stix
-Version: 1.0.3
+Version: 1.0.4
 Summary: txt2stix is a Python script that is designed to identify and extract IoCs and TTPs from text files, identify the relationships between them, convert them to STIX 2.1 objects, and output as a STIX 2.1 bundle.
 Project-URL: Homepage, https://github.com/muchdogesec/txt2stix
 Project-URL: Issues, https://github.com/muchdogesec/txt2stix/issues

{txt2stix-1.0.3 → txt2stix-1.0.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "txt2stix"
-version = "1.0.3"
+version = "1.0.4"
 authors = [{ name = "dogesec" }]
 maintainers = [{ name = "dogesec" }]
 description = "txt2stix is a Python script that is designed to identify and extract IoCs and TTPs from text files, identify the relationships between them, convert them to STIX 2.1 objects, and output as a STIX 2.1 bundle."

{txt2stix-1.0.3 → txt2stix-1.0.4}/tests/src/test_attack_flow.py

@@ -127,7 +127,7 @@ def test_extract_attack_flow_and_navigator(dummy_objects, dummy_report):
             "txt2stix.attack_flow.create_navigator_layer"
         ) as mock_create_navigator_layer,
     ):
-        ## Both flow and navigator
+        # ================= Both flow and navigator ===================
         flow, nav = extract_attack_flow_and_navigator(
             bundler, text, True, True, ai_extractor
         )
@@ -150,7 +150,7 @@ def test_extract_attack_flow_and_navigator(dummy_objects, dummy_report):
         mock_create_navigator_layer.reset_mock()
         mock_extract_flow.reset_mock()
 
-        ## only flow
+        # ================= only flow ===================
         flow, nav = extract_attack_flow_and_navigator(
             bundler, text, True, False, ai_extractor
         )
@@ -168,7 +168,7 @@ def test_extract_attack_flow_and_navigator(dummy_objects, dummy_report):
         mock_create_navigator_layer.reset_mock()
         mock_extract_flow.reset_mock()
 
-        ## only navigator
+        # ================= only navigator ===================
         flow, nav = extract_attack_flow_and_navigator(
             bundler, text, False, True, ai_extractor
         )
@@ -184,6 +184,21 @@ def test_extract_attack_flow_and_navigator(dummy_objects, dummy_report):
             bundler.report, bundler.summary, mock_extract_flow.return_value, techniques
         )
 
+        ### reset mocks
+        mock_parse_flow.reset_mock()
+        mock_create_navigator_layer.reset_mock()
+        mock_extract_flow.reset_mock()
+        # ============ no technique object ============
+        bundler.bundle.objects = []
+        flow, nav = extract_attack_flow_and_navigator(
+            bundler, text, True, True, ai_extractor
+        )
+        mock_extract_flow.assert_not_called()
+        assert (flow, nav) == (None, None)
+        mock_parse_flow.assert_not_called()
+
+        mock_create_navigator_layer.assert_not_called()
+
 
 def test_create_navigator_layer(dummy_report):
     summary = "this is a summary"
@@ -252,7 +267,6 @@ def test_create_navigator_layer(dummy_report):
 
     retval = create_navigator_layer(dummy_report, summary, flow, techniques)
     assert len(retval) == 3
-    print(retval)
     assert retval == [
         {
             "version": "4.5",

{txt2stix-1.0.3 → txt2stix-1.0.4}/txt2stix/attack_flow.py

@@ -204,6 +204,9 @@ def extract_attack_flow_and_navigator(
     ex: BaseAIExtractor = ai_settings_relationships
     tactics = get_all_tactics()
     techniques = get_techniques_from_extracted_objects(bundler.bundle.objects, tactics)
+    if not techniques:
+        return None, None
+    
     logged_techniques = [
         {k: v for k, v in t.items() if k != "stix_obj"} for t in techniques.values()
     ]