test: Update entity-mock.ts functions

Vizzuality · Nov 8, 2024 · 3caf3f8 · 3caf3f8
1 parent 6175a9a
commit 3caf3f8
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 6 deletions.
diff --git a/api/test/utils/test-manager.ts b/api/test/utils/test-manager.ts
@@ -118,9 +118,9 @@ export class TestManager<FixtureType> {
       createSection: (additionalData: DeepPartial<Section>) =>
         createSection(this.getDataSource(), additionalData),
       createCustomWidget: (
-        additionalData?: DeepPartial<CustomWidget>,
+        data?: DeepPartial<CustomWidget>,
       ): Promise<CustomWidget> =>
-        createCustomWidget(this.getDataSource(), additionalData),
+        createCustomWidget(this.getDataSource(), data),
     };
   }
 }
diff --git a/shared/lib/e2e-test-manager.ts b/shared/lib/e2e-test-manager.ts
@@ -88,9 +88,9 @@ export class E2eTestManager {
         data: DeepPartial<BaseWidget> & { indicator: string },
       ): Promise<BaseWidget> => createBaseWidget(this.getDataSource(), data),
       createCustomWidget: (
-        additionalData?: DeepPartial<CustomWidget>,
+        data?: DeepPartial<CustomWidget>,
       ): Promise<CustomWidget> =>
-        createCustomWidget(this.getDataSource(), additionalData),
+        createCustomWidget(this.getDataSource(), data),
       ensureQuestionIndicatorMapExists: async (
         dataSource: DataSource,
         questionIndicatorMap: { indicator: string; question: string },

diff --git a/shared/lib/entity-mocks.ts b/shared/lib/entity-mocks.ts
@@ -103,15 +103,20 @@ export const createSection = async (
 
 export const createCustomWidget = async (
   dataSource: DataSource,
-  additionalData?: DeepPartial<CustomWidget>,
+  data?: DeepPartial<CustomWidget>,
 ) => {
   const baseWidgetsRepository = dataSource.getRepository(CustomWidget);
 
   const defaults: Partial<CustomWidget> = {
     name: 'custom-widget',
     defaultVisualization: WIDGET_VISUALIZATIONS.AREA_GRAPH,
     filters: {},
+    widget: {} as BaseWidget,
   };
 
-  return baseWidgetsRepository.save({ ...defaults, ...additionalData });
+  const customWidget = { ...defaults, ...data };
+  customWidget.widget!.indicator ??= new Date().toISOString();
+
+  await createBaseWidget(dataSource, customWidget.widget! as BaseWidget);
+  return baseWidgetsRepository.save(customWidget);
 };