Udpate tests for Diffirent Collection variable types

GrapesJS · Jan 10, 2025 · 697c11a · 697c11a
1 parent b7d2793
commit 697c11a
Showing 1 changed file with 26 additions and 0 deletions.
diff --git a/packages/core/test/specs/data_sources/model/collection_component/CollectionComponent.ts b/packages/core/test/specs/data_sources/model/collection_component/CollectionComponent.ts
@@ -590,6 +590,29 @@ describe('Collection component', () => {
                 type: CollectionVariableType,
                 variable_type: variableType,
               },
+              attributes: {
+                custom_attribute: {
+                  type: CollectionVariableType,
+                  variable_type: variableType,
+                },
+              },
+              traits: [
+                {
+                  name: 'attribute_trait',
+                  value: {
+                    type: CollectionVariableType,
+                    variable_type: variableType,
+                  },
+                },
+                {
+                  name: 'property_trait',
+                  changeProp: true,
+                  value: {
+                    type: CollectionVariableType,
+                    variable_type: variableType,
+                  },
+                },
+              ],
             },
             config: {
               dataSource: {
@@ -605,6 +628,9 @@ describe('Collection component', () => {
 
         children.each((child, index) => {
           expect(child.get('content')).toBe(expectedValues[index]);
+          expect(child.get('property_trait')).toBe(expectedValues[index]);
+          expect(child.getAttributes()['custom_attribute']).toBe(expectedValues[index]);
+          expect(child.getAttributes()['attribute_trait']).toBe(expectedValues[index]);
         });
       });
     });