From cc8e1cc3432ab9f0c4eef8f12a9f15dedfa95a35 Mon Sep 17 00:00:00 2001
From: "Irina V. Kuzmina" <kuzmadom@yandex-team.ru>
Date: Tue, 2 Dec 2025 15:24:39 +0300
Subject: [PATCH] Add test for custom color palette in dataset fields

---
 tests/constants/test-entities/datasets.ts     |  2 +-
 .../pie/dataset-field-settings.test.ts        | 89 +++++++++++++++++++
 2 files changed, 90 insertions(+), 1 deletion(-)
 create mode 100644 tests/opensource-suites/wizard/visualizations/pie/dataset-field-settings.test.ts

diff --git a/tests/constants/test-entities/datasets.ts b/tests/constants/test-entities/datasets.ts
index e43adc5759..c8c2e4bcca 100644
--- a/tests/constants/test-entities/datasets.ts
+++ b/tests/constants/test-entities/datasets.ts
@@ -1,4 +1,4 @@
 export const DatasetsEntities = {
-    Basic: {url: '/v9ign8imt0qkk', name: 'Dataset'},
+    Basic: {id: 'v9ign8imt0qkk', url: '/v9ign8imt0qkk', name: 'Dataset'},
     Orders: {url: '/wiuimfkr0lful', name: 'Orders'},
 };
diff --git a/tests/opensource-suites/wizard/visualizations/pie/dataset-field-settings.test.ts b/tests/opensource-suites/wizard/visualizations/pie/dataset-field-settings.test.ts
new file mode 100644
index 0000000000..16732cd074
--- /dev/null
+++ b/tests/opensource-suites/wizard/visualizations/pie/dataset-field-settings.test.ts
@@ -0,0 +1,89 @@
+import {expect} from '@playwright/test';
+
+import datalensTest from '../../../../utils/playwright/globalTestDefinition';
+import {getUniqueTimestamp, openTestPage, slct} from '../../../../utils';
+import {
+    ChartKitQa,
+    DatasetActionQA,
+    DatasetFieldColorsDialogQa,
+    DatasetFieldSettingsDialogQa,
+    DatasetFieldsTabQa,
+    SelectQa,
+    WizardPageQa,
+    WizardVisualizationId,
+    WorkbookPageQa,
+} from '../../../../../src/shared';
+import WizardPage from '../../../../page-objects/wizard/WizardPage';
+import {PlaceholderName} from '../../../../page-objects/wizard/SectionVisualization';
+import {WorkbooksUrls} from '../../../../constants/constants';
+import {DatasetsEntities} from '../../../../constants/test-entities/datasets';
+import {Workbook} from '../../../../page-objects/workbook/Workbook';
+import {addCustomPalette} from '../../../utils';
+
+datalensTest.describe('Wizard', () => {
+    datalensTest.describe('Pie chart', () => {
+        datalensTest('Use color palette from dataset @screenshot', async ({page}, testInfo) => {
+            const customPaletteName = `${testInfo.title} (${getUniqueTimestamp()})`;
+            await addCustomPalette(page, {name: customPaletteName, colors: ['ff0000']});
+
+            await openTestPage(page, `${WorkbooksUrls.E2EWorkbook}/?tab=dataset`);
+
+            // copy dataset - so that changes in it do not affect other tests
+            const originalDatasetRow = page.locator(slct(WorkbookPageQa.ListItem), {
+                has: page.locator(slct(DatasetsEntities.Basic.id)),
+            });
+            await originalDatasetRow.locator(slct(WorkbookPageQa.MenuDropDownBtn)).click();
+            await page.locator(slct(WorkbookPageQa.MenuItemDuplicate)).click();
+
+            const workbookPO = new Workbook(page);
+            const newDatasetName = `${testInfo.title} (${getUniqueTimestamp()})`;
+            await workbookPO.dialogCreateEntry.createEntryWithName(newDatasetName);
+            await page.locator(slct(WorkbookPageQa.ListItem), {hasText: newDatasetName}).click();
+
+            // add field colors setting
+            const fieldName = 'segment';
+            const row = page.locator('.dataset-table__row', {
+                has: page
+                    .locator(slct(DatasetFieldsTabQa.FieldNameColumnInput))
+                    .locator(`[value=${fieldName}]`),
+            });
+            await row.hover();
+            await row.locator(slct(DatasetFieldsTabQa.FieldSettingsColumnIcon)).click();
+            const dialog = page.locator(slct(DatasetFieldSettingsDialogQa.Dialog));
+            await dialog.locator(slct(DatasetFieldSettingsDialogQa.ColorSettingsButton)).click();
+            // select custom palette
+            const colorsDialog = page.locator(slct(DatasetFieldColorsDialogQa.Dialog));
+            await colorsDialog.locator('.color-palette-select').click();
+            await colorsDialog.locator(slct(SelectQa.POPUP)).getByText(customPaletteName).click();
+            // set color from palette for field value
+            const paletteItem = colorsDialog
+                .locator(slct(DatasetFieldColorsDialogQa.PaleteItem))
+                .first();
+            await paletteItem.click();
+            await colorsDialog.locator('button', {hasText: 'Apply'}).click();
+            await dialog.locator('button', {hasText: 'Save'}).click();
+            await page.locator(slct(DatasetActionQA.CreateButton)).click();
+
+            const newPagePromise = page.context().waitForEvent('page');
+            await page.locator('button', {hasText: 'Create chart'}).click();
+
+            // create pie chart
+            const newPage = await newPagePromise;
+            const wizardPage = new WizardPage({page: newPage});
+            await wizardPage.setVisualization(WizardVisualizationId.PieD3);
+
+            const chartContainer = wizardPage.page.locator(slct(WizardPageQa.SectionPreview));
+            const chart = chartContainer.locator('.chartkit-graph,.gcharts-chart');
+            const previewLoader = chartContainer.locator(slct(ChartKitQa.Loader));
+
+            await wizardPage.sectionVisualization.addFieldByClick(PlaceholderName.Colors, 'region');
+            await wizardPage.sectionVisualization.addFieldByClick(
+                PlaceholderName.Measures,
+                'Sales',
+            );
+
+            await expect(previewLoader).not.toBeVisible();
+            await expect(chart).toHaveScreenshot();
+        });
+    });
+});