[client] Add Data Component and Data Sourcer entities (#304)

Co-authored-by: Julien Richard <[email protected]>

Author: RomuDeuxfois

README.md

@@ -54,6 +54,35 @@ To learn about how to use the OpenCTI Python client and read some examples and c
 
 To learn about the methods available for executing queries and retrieving their answers, refer to [the client API Reference](https://opencti-client-for-python.readthedocs.io/en/latest/pycti/pycti.html).
 
+## Tests
+
+### Install dependencies
+
+```bash
+$ pip install -r ./test-requirements.txt
+```
+
+[pytest](https://docs.pytest.org/en/7.2.x/) is used to launch the tests.
+
+### Launch tests
+
+#### Prerequisite
+
+Your OpenCTI API should be running.
+Your conftest.py should be configured with your API url and your token.
+
+#### Launching
+
+Unit tests
+```bash
+$ pytest ./tests/01-unit/
+```
+
+Integration testing
+```bash
+$ pytest ./tests/02-integration/
+```
+
 ## About
 
 OpenCTI is a product powered by the collaboration of the private company [Filigran](https://www.filigran.io), the [French national cybersecurity agency (ANSSI)](https://ssi.gouv.fr), the [CERT-EU](https://cert.europa.eu) and the [Luatix](https://www.luatix.org) non-profit organization.

pycti/__init__.py

@@ -12,6 +12,8 @@
 from .entities.opencti_attack_pattern import AttackPattern
 from .entities.opencti_campaign import Campaign
 from .entities.opencti_course_of_action import CourseOfAction
+from .entities.opencti_data_component import DataComponent
+from .entities.opencti_data_source import DataSource
 from .entities.opencti_external_reference import ExternalReference
 from .entities.opencti_grouping import Grouping
 from .entities.opencti_identity import Identity
@@ -56,6 +58,8 @@
     "Campaign",
     "ConnectorType",
     "CourseOfAction",
+    "DataComponent",
+    "DataSource",
     "ExternalReference",
     "Grouping",
     "Identity",

pycti/api/opencti_api_client.py

@@ -17,6 +17,8 @@
 from pycti.entities.opencti_campaign import Campaign
 from pycti.entities.opencti_channel import Channel
 from pycti.entities.opencti_course_of_action import CourseOfAction
+from pycti.entities.opencti_data_component import DataComponent
+from pycti.entities.opencti_data_source import DataSource
 from pycti.entities.opencti_event import Event
 from pycti.entities.opencti_external_reference import ExternalReference
 from pycti.entities.opencti_grouping import Grouping
@@ -52,6 +54,7 @@
 from pycti.entities.opencti_tool import Tool
 from pycti.entities.opencti_vulnerability import Vulnerability
 from pycti.utils.opencti_stix2 import OpenCTIStix2
+from pycti.utils.opencti_stix2_utils import OpenCTIStix2Utils
 
 urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
 
@@ -177,6 +180,8 @@ def __init__(
         self.vulnerability = Vulnerability(self)
         self.attack_pattern = AttackPattern(self)
         self.course_of_action = CourseOfAction(self)
+        self.data_component = DataComponent(self)
+        self.data_source = DataSource(self)
         self.report = Report(self)
         self.note = Note(self)
         self.observed_data = ObservedData(self)
@@ -497,6 +502,13 @@ def process_multiple_fields(self, data):
         :rtype: dict
         """
 
+        # Handle process_multiple_fields specific case
+        attribute = OpenCTIStix2Utils.retrieveClassForMethod(
+            self, data, "entity_type", "process_multiple_fields"
+        )
+        if attribute is not None:
+            data = attribute.process_multiple_fields(data)
+
         if data is None:
             return data
         if "createdBy" in data and data["createdBy"] is not None: