From 669a597aa8f4a02aa6f645498917c5057f99c0e7 Mon Sep 17 00:00:00 2001
From: Artur Arseniev <artur.catch@hotmail.it>
Date: Thu, 23 Oct 2025 15:46:39 +0400
Subject: [PATCH] Setup relation tests

---
 packages/core/src/data_sources/types.ts       |  3 +-
 .../specs/data_sources/model/DataSource.ts    | 58 +++++++++++++++++++
 2 files changed, 60 insertions(+), 1 deletion(-)

diff --git a/packages/core/src/data_sources/types.ts b/packages/core/src/data_sources/types.ts
index cc504ed3b..691e000ee 100644
--- a/packages/core/src/data_sources/types.ts
+++ b/packages/core/src/data_sources/types.ts
@@ -113,10 +113,11 @@ export type DataFieldSchemas =
 
 export interface DataSourceType<DR extends DataRecordProps> extends BaseDataSource {
   records: DataRecords<DR>;
-  schema: Record<keyof DR, DataFieldSchemas | undefined>;
+  schema: Partial<Record<keyof DR, DataFieldSchemas | undefined>>;
 }
 export interface DataSourceProps<DR extends DataRecordProps> extends BaseDataSource {
   records?: DataRecords<DR> | DataRecord<DR>[] | DR[];
+  schema?: DataSourceType<DR>['schema'];
 }
 export type RecordPropsType<T> = T extends DataRecord<infer U> ? U : never;
 export interface DataSourceTransformers {
diff --git a/packages/core/test/specs/data_sources/model/DataSource.ts b/packages/core/test/specs/data_sources/model/DataSource.ts
index 19bb07475..f43e10ff8 100644
--- a/packages/core/test/specs/data_sources/model/DataSource.ts
+++ b/packages/core/test/specs/data_sources/model/DataSource.ts
@@ -14,6 +14,8 @@ interface TestRecord extends DataRecordProps {
   age?: number;
 }
 
+const serializeRecords = (records: any[]) => JSON.parse(JSON.stringify(records));
+
 describe('DataSource', () => {
   let em: Editor;
   let dsm: DataSourceManager;
@@ -80,5 +82,61 @@ describe('DataSource', () => {
       expect(ds.getSchemaField('age')).toEqual(schemaAge);
       expect(ds.getSchemaField('nonExistentField')).toBeUndefined();
     });
+
+    describe('Relations', () => {
+      const categoryRecords = [
+        { id: 'cat1', name: 'Category 1' },
+        { id: 'cat2', name: 'Category 2' },
+      ];
+      const userRecords = [
+        { id: 'user1', username: 'user_one' },
+        { id: 'user2', username: 'user_two' },
+      ];
+      const blogRecords = [
+        { id: 'blog1', title: 'First Blog', author: 'user1', categories: ['cat1'] },
+        { id: 'blog2', title: 'Second Blog', author: 'user2' },
+        { id: 'blog3', title: 'Third Blog', categories: ['cat1', 'cat2'] },
+      ];
+
+      beforeEach(() => {
+        dsm.add({
+          id: 'categories',
+          records: categoryRecords,
+        });
+        dsm.add({
+          id: 'users',
+          records: userRecords,
+        });
+        dsm.add({
+          id: 'blogs',
+          records: blogRecords,
+          schema: {
+            title: {
+              type: DataFieldPrimitiveType.string,
+            },
+            author: {
+              type: DataFieldPrimitiveType.relation,
+              target: 'users',
+              targetField: 'id',
+            },
+          },
+        });
+      });
+
+      test('return default values', () => {
+        const blogsDS = dsm.get('blogs');
+        expect(serializeRecords(blogsDS.getRecords())).toEqual(blogRecords);
+      });
+
+      test('return resolved values', () => {
+        const blogsDS = dsm.get('blogs');
+        const records = blogsDS.getRecords({ resolveRelations: true });
+        expect(records).toEqual([
+          { ...blogRecords[0], author: userRecords[0] },
+          { ...blogRecords[1], author: userRecords[1] },
+          blogRecords[2],
+        ]);
+      });
+    });
   });
 });