fix(Waterfall): fix calculation of yValue for total item (#297)

Co-authored-by: Denis Ryabko <[email protected]>

Author: arteria32

src/__snapshots__/waterfall.visual.test.tsx-snapshots/Waterfall-series-With-multiple-total-columns-1-chromium-linux.png

@@ -9,6 +9,7 @@ import {
 } from 'src/__stories__/__data__';
 
 import {ChartTestStory} from '../../playwright/components/ChartTestStory';
+import type {ChartData, WaterfallSeriesData} from '../types';
 
 test.describe('Waterfall series', () => {
     test('Basic', async ({mount}) => {
@@ -36,4 +37,33 @@ test.describe('Waterfall series', () => {
         const component = await mount(<ChartTestStory data={waterfallNullModeZeroData} />);
         await expect(component.locator('svg')).toHaveScreenshot();
     });
+
+    test('With multiple total columns', async ({mount}) => {
+        const dataWithTotals: WaterfallSeriesData[] = [
+            {x: 'A', y: 10},
+            {x: 'Total1', y: 0, total: true},
+            {x: 'D', y: 25},
+            {x: 'Total2', y: 0, total: true},
+        ];
+
+        const waterfallDataWithTotals: ChartData = {
+            series: {
+                data: [
+                    {
+                        type: 'waterfall',
+                        name: 'Series',
+                        data: dataWithTotals,
+                        nullMode: 'skip',
+                    },
+                ],
+            },
+            xAxis: {
+                type: 'category',
+                categories: dataWithTotals.map((d) => d.x) as string[],
+            },
+        };
+
+        const component = await mount(<ChartTestStory data={waterfallDataWithTotals} />);
+        await expect(component.locator('svg')).toHaveScreenshot();
+    });
 });

src/__tests__/waterfall.visual.test.tsx

@@ -139,7 +139,9 @@ export const prepareWaterfallData = async (args: {
             continue;
         }
 
-        if (!item.data.total) {
+        if (item.data.total) {
+            item.data.y = totalValue;
+        } else {
             totalValue += Number(item.data.y);
         }