From 1b688ec35f20578511201fe788231e9435fa981b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?C=C3=A9line=20S=C3=A8be?=
 <87119259+CelineSebe@users.noreply.github.com>
Date: Mon, 8 Jul 2024 18:21:29 +0200
Subject: [PATCH] [backend] Fix the ability to add relationship references in
 CSV mapper(#7565)

---
 .../opencti-graphql/src/database/stix-ref.ts          | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/opencti-platform/opencti-graphql/src/database/stix-ref.ts b/opencti-platform/opencti-graphql/src/database/stix-ref.ts
index 670044ba6dda..6d22e85c17bb 100644
--- a/opencti-platform/opencti-graphql/src/database/stix-ref.ts
+++ b/opencti-platform/opencti-graphql/src/database/stix-ref.ts
@@ -1,7 +1,16 @@
 import { schemaRelationsRefDefinition } from '../schema/schema-relationsRef';
+import { STIX_CORE_RELATIONSHIPS } from '../schema/stixCoreRelationship';
+import type { RefAttribute } from '../schema/attribute-definition';
 
 export const schemaRelationsRefTypesMapping = () => {
-  return Array.from(schemaRelationsRefDefinition.relationsRefCacheArray.entries()).map(([key, refs]) => {
+  const relationshipsRefs = new Map<string, RefAttribute[]>();
+  STIX_CORE_RELATIONSHIPS.forEach((relation) => {
+    relationshipsRefs.set(relation, schemaRelationsRefDefinition.getRelationsRef(relation));
+  });
+  return [
+    ...Array.from(relationshipsRefs.entries()),
+    ...Array.from(schemaRelationsRefDefinition.relationsRefCacheArray.entries())
+  ].map(([key, refs]) => {
     return {
       key,
       values: refs.map((ref) => ({