From 9e68c92148130848b67cc2e67411c8046a8497c1 Mon Sep 17 00:00:00 2001
From: mohamedsalem401 <salemmohamed780@gmail.com>
Date: Tue, 9 Sep 2025 20:28:18 +0300
Subject: [PATCH] update component wrapper tests

---
 .../dom_components/model/ComponentWrapper.ts  | 30 +++++--------------
 1 file changed, 7 insertions(+), 23 deletions(-)

diff --git a/packages/core/test/specs/dom_components/model/ComponentWrapper.ts b/packages/core/test/specs/dom_components/model/ComponentWrapper.ts
index dfa0ed539..b719b0204 100644
--- a/packages/core/test/specs/dom_components/model/ComponentWrapper.ts
+++ b/packages/core/test/specs/dom_components/model/ComponentWrapper.ts
@@ -44,13 +44,13 @@ describe('ComponentWrapper', () => {
     let em: EditorModel;
     let dsm: DataSourceManager;
     let pagesDataSource: DataSource;
-    let flatPagesDataSource: DataSource;
     let wrapper: ComponentWrapper;
+    let firstRecord: DataRecord;
 
     const pagesData = [
-      { id: 'page1', page: 'page1', title: 'Title1', content: 'content 1' },
-      { id: 'page2', page: 'page2', title: 'Title2', content: 'content 2' },
-      { id: 'page3', page: 'page3', title: 'Title3', content: 'content 3' },
+      { id: 'page1', title: 'Title1' },
+      { id: 'page2', title: 'Title2' },
+      { id: 'page3', title: 'Title3' },
     ];
 
     beforeEach(() => {
@@ -62,10 +62,7 @@ describe('ComponentWrapper', () => {
         records: [{ id: 'pages', data: pagesData }],
       });
 
-      flatPagesDataSource = dsm.add({
-        id: 'flatPagesDataSource',
-        records: pagesData,
-      });
+      firstRecord = em.DataSources.get('pagesDataSource').getRecord('pages')!;
     });
 
     afterEach(() => {
@@ -104,22 +101,9 @@ describe('ComponentWrapper', () => {
       setResolver('pagesDataSource.pages.data');
       const child = appendChildWithTitle();
 
-      expect(child.collectionsStateMap).toEqual(wrapper.collectionsStateMap);
-      expect(child.collectionsStateMap).toEqual({
-        [keyRootData]: pagesData[0],
-      });
-      expect(child.get('title')).toBe(pagesData[0].title);
-    });
-
-    test('updating record propagates to children2', () => {
-      setResolver('flatPagesDataSource');
-      const child = appendChildWithTitle();
-
-      expect(child.collectionsStateMap).toEqual(wrapper.collectionsStateMap);
-      expect(child.collectionsStateMap).toEqual({
-        [keyRootData]: pagesData[0],
-      });
       expect(child.get('title')).toBe(pagesData[0].title);
+      firstRecord.set('data', [{ id: 'page1', title: 'new_title' }]);
+      expect(child.get('title')).toBe('new_title');
     });
   });
 });