fix unit test

datalens-tech · Oct 22, 2024 · 4cfc209 · 4cfc209
1 parent 22c3372
commit 4cfc209
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/src/ui/constants/visualizations/index.ts b/src/ui/constants/visualizations/index.ts
@@ -47,7 +47,7 @@ import {PIE_D3_VISUALIZATION, PIE_VISUALIZATION} from './pie';
 import {PIVOT_TABLE_VISUALIZATION} from './pivotTable';
 import {POLYLINE_VISUALIZATION} from './polyline';
 import {SCATTER_D3_VISUALIZATION, SCATTER_VISUALIZATION} from './scatter';
-import {TREEMAP_VISUALIZATION} from './treemap';
+import {TREEMAP_D3_VISUALIZATION, TREEMAP_VISUALIZATION} from './treemap';
 
 export const SETTINGS = {
     AXIS_FORMAT_MODE: {
@@ -200,6 +200,7 @@ export const VISUALIZATION_IDS = {
     PIE_D3: PIE_D3_VISUALIZATION.id,
     BAR_X_D3: BAR_X_D3_VISUALIZATION.id,
     DONUT_D3: DONUT_D3_VISUALIZATION.id,
+    TREEMAP_D3: TREEMAP_D3_VISUALIZATION.id,
 };
 
 export const VISUALIZATIONS_BY_ID = {

diff --git a/src/ui/units/wizard/reducers/__tests__/preview.test.ts b/src/ui/units/wizard/reducers/__tests__/preview.test.ts
@@ -42,6 +42,7 @@ describe('Preview reducer', () => {
             [VISUALIZATION_IDS.SCATTER_D3]: WizardType.D3WizardNode,
             [VISUALIZATION_IDS.COMBINED_CHART]: WizardType.GraphWizardNode,
             [VISUALIZATION_IDS.GEOPOINT_WITH_CLUSTER]: WizardType.GraphWizardNode,
+            [VISUALIZATION_IDS.TREEMAP_D3]: WizardType.D3WizardNode,
         };
 
         it('Should return same state when calls with empty args', () => {