[client] add reliability in Report (#opencti/2362)

Author: SouadHadjiat

pycti/entities/opencti_identity.py

@@ -33,6 +33,7 @@ def __init__(self, opencti):
                     roles
                     contact_information
                     x_opencti_aliases
+                    x_opencti_reliability
                     created
                     modified
                     objectLabel {
@@ -47,7 +48,6 @@ def __init__(self, opencti):
                 }
                 ... on Organization {
                     x_opencti_organization_type
-                    x_opencti_reliability
                 }
                 ... on Individual {
                     x_opencti_firstname
@@ -115,14 +115,14 @@ def __init__(self, opencti):
             name
             description
             x_opencti_aliases
+            x_opencti_reliability
             contact_information
             ... on Individual {
                 x_opencti_firstname
                 x_opencti_lastname
             }
             ... on Organization {
                 x_opencti_organization_type
-                x_opencti_reliability
             }
             importFiles {
                 edges {
@@ -355,7 +355,23 @@ def create(self, **kwargs):
                 """
                 input_variables["x_opencti_firstname"] = x_opencti_firstname
                 input_variables["x_opencti_lastname"] = x_opencti_lastname
+                input_variables["x_opencti_reliability"] = x_opencti_reliability
                 result_data_field = "individualAdd"
+            elif type == IdentityTypes.SYSTEM.value:
+                query = """
+                    mutation SystemAdd($input: SystemAddInput!) {
+                        systemAdd(input: $input) {
+                            id
+                            standard_id
+                            entity_type
+                            parent_types
+                        }
+                    }
+                """
+                input_variables["x_opencti_firstname"] = x_opencti_firstname
+                input_variables["x_opencti_lastname"] = x_opencti_lastname
+                input_variables["x_opencti_reliability"] = x_opencti_reliability
+                result_data_field = "systemAdd"
             else:
                 query = """
                     mutation IdentityAdd($input: IdentityAddInput!) {

pycti/entities/opencti_report.py

@@ -34,6 +34,7 @@ def __init__(self, opencti):
                     roles
                     contact_information
                     x_opencti_aliases
+                    x_opencti_reliability
                     created
                     modified
                     objectLabel {
@@ -48,7 +49,6 @@ def __init__(self, opencti):
                 }
                 ... on Organization {
                     x_opencti_organization_type
-                    x_opencti_reliability
                 }
                 ... on Individual {
                     x_opencti_firstname
@@ -109,6 +109,7 @@ def __init__(self, opencti):
                 }
             }
             revoked
+            x_opencti_reliability
             confidence
             created
             modified
@@ -465,6 +466,7 @@ def create(self, **kwargs):
         description = kwargs.get("description", None)
         report_types = kwargs.get("report_types", None)
         published = kwargs.get("published", None)
+        x_opencti_reliability = kwargs.get("x_opencti_reliability", None)
         x_opencti_stix_ids = kwargs.get("x_opencti_stix_ids", None)
         granted_refs = kwargs.get("objectOrganization", None)
         update = kwargs.get("update", False)
@@ -501,6 +503,7 @@ def create(self, **kwargs):
                         "description": description,
                         "report_types": report_types,
                         "published": published,
+                        "x_opencti_reliability": x_opencti_reliability,
                         "x_opencti_stix_ids": x_opencti_stix_ids,
                         "update": update,
                     }
@@ -620,6 +623,10 @@ def import_from_stix2(self, **kwargs):
                 stix_object["granted_refs"] = self.opencti.get_attribute_in_extension(
                     "granted_refs", stix_object
                 )
+            if "x_opencti_reliability" not in stix_object:
+                stix_object[
+                    "x_opencti_reliability"
+                ] = self.opencti.get_attribute_in_extension("reliability", stix_object)
 
             return self.create(
                 stix_id=stix_object["id"],
@@ -658,6 +665,9 @@ def import_from_stix2(self, **kwargs):
                 x_opencti_stix_ids=stix_object["x_opencti_stix_ids"]
                 if "x_opencti_stix_ids" in stix_object
                 else None,
+                x_opencti_reliability=stix_object["x_opencti_reliability"]
+                if "x_opencti_reliability" in stix_object
+                else None,
                 objectOrganization=stix_object["x_opencti_granted_refs"]
                 if "x_opencti_granted_refs" in stix_object
                 else None,

tests/cases/entities.py

@@ -35,6 +35,10 @@ def case_identity_individual(api_client):
     def case_identity_sector(api_client):
         return IdentitySectorTest(api_client)
 
+    @staticmethod
+    def case_identity_system(api_client):
+        return IdentitySystemTest(api_client)
+
     @staticmethod
     def case_incident(api_client):
         return IncidentTest(api_client)
@@ -238,6 +242,15 @@ def data(self) -> Dict:
         }
 
 
+class IdentitySystemTest(IdentityTest):
+    def data(self) -> Dict:
+        return {
+            "type": IdentityTypes.SYSTEM.value,
+            "name": "System A",
+            "description": "The system A",
+        }
+
+
 class IndicatorTest(EntityTest):
     def setup(self):
         self.marking_definition_green = self.api_client.marking_definition.read(