[client]: unify empty state handling

Author: pdamoune

pycti/connector/opencti_connector_helper.py

@@ -783,9 +783,10 @@ def run(self) -> None:  # pylint: disable=too-many-branches
                         # state can be None if reset from the UI
                         # In this case, default parameters will be used but SSE Client needs to be restarted
                         if state is None:
-                            self.exit = True
-                        state["start_from"] = str(msg.id)
-                        self.helper.set_state(state)
+                            self.exit_event.set()
+                        else:
+                            state["start_from"] = str(msg.id)
+                            self.helper.set_state(state)
         except Exception as ex:
             self.helper.connector_logger.error(
                 "Error in ListenStream loop, exit.", {"reason": str(ex)}
@@ -1812,9 +1813,9 @@ def send_stix2_bundle(self, bundle: str, **kwargs) -> list:
                             )
                         )
                     else:
-                        octi_extensions["granted_refs"] = (
-                            self.enrichment_shared_organizations
-                        )
+                        octi_extensions[
+                            "granted_refs"
+                        ] = self.enrichment_shared_organizations
                 else:
                     if item.get("x_opencti_granted_refs") is not None:
                         item["x_opencti_granted_refs"] = list(
@@ -1824,9 +1825,9 @@ def send_stix2_bundle(self, bundle: str, **kwargs) -> list:
                             )
                         )
                     else:
-                        item["x_opencti_granted_refs"] = (
-                            self.enrichment_shared_organizations
-                        )
+                        item[
+                            "x_opencti_granted_refs"
+                        ] = self.enrichment_shared_organizations
             bundle = json.dumps(bundle_data)
 
         # If execution in playbook, callback the api
@@ -1919,13 +1920,15 @@ def send_stix2_bundle(self, bundle: str, **kwargs) -> list:
             os.rename(write_file, final_write_file)
 
         stix2_splitter = OpenCTIStix2Splitter()
-        (expectations_number, _, bundles) = (
-            stix2_splitter.split_bundle_with_expectations(
-                bundle=bundle,
-                use_json=True,
-                event_version=event_version,
-                cleanup_inconsistent_bundle=cleanup_inconsistent_bundle,
-            )
+        (
+            expectations_number,
+            _,
+            bundles,
+        ) = stix2_splitter.split_bundle_with_expectations(
+            bundle=bundle,
+            use_json=True,
+            event_version=event_version,
+            cleanup_inconsistent_bundle=cleanup_inconsistent_bundle,
         )
 
         if len(bundles) == 0: