@cdc/map 4.26.2 → 4.26.4

package/src/_stories/CdcMap.Defaults.smoke.stories.tsx

@@ -0,0 +1,76 @@
+import type { Meta, StoryObj } from '@storybook/react-vite'
+import { expect } from 'storybook/test'
+import CdcMap from '../CdcMap'
+import EqualNumberMap from './_mock/equal-number.json'
+import { editConfigKeys } from '@cdc/core/helpers/configHelpers'
+import { assertVisualizationRendered, waitForPresence } from '@cdc/core/helpers/testing'
+
+const meta: Meta<typeof CdcMap> = {
+  title: 'Components/Templates/Map/Defaults',
+  component: CdcMap
+}
+
+type Story = StoryObj<typeof CdcMap>
+
+const oldConfig = editConfigKeys(EqualNumberMap, [
+  { path: ['legend', 'style'], value: 'circles' },
+  { path: ['legend', 'position'], value: 'side' },
+  { path: ['legend', 'numberOfItems'], value: 3 },
+  { path: ['legend', 'hideBorder'], value: false }
+])
+
+const newConfig = editConfigKeys(EqualNumberMap, [
+  { path: ['legend', 'style'], value: 'gradient' },
+  { path: ['legend', 'position'], value: 'top' },
+  { path: ['legend', 'numberOfItems'], value: 5 },
+  { path: ['legend', 'hideBorder'], value: true }
+])
+
+export const OldConfig_Preserves_Legacy_Defaults: Story = {
+  args: {
+    config: oldConfig,
+    isEditor: false
+  },
+  play: async ({ canvasElement }) => {
+    await assertVisualizationRendered(canvasElement)
+
+    await waitForPresence('aside[aria-label="Legend"]', canvasElement)
+    await waitForPresence('.legend-container__li', canvasElement)
+    const legend = canvasElement.querySelector('aside[aria-label="Legend"]')
+    expect(legend).toBeInTheDocument()
+    expect(legend?.classList.contains('side')).toBe(true)
+
+    expect(legend?.classList.contains('no-border')).toBe(false)
+
+    const legendContainer = canvasElement.querySelector('.legend-container')
+    const legendItems = legendContainer?.querySelectorAll('.legend-container__li')
+    expect(legendItems?.length).toBeGreaterThan(0)
+
+    const linearGradient = legendContainer?.querySelector('linearGradient')
+    expect(linearGradient).toBeNull()
+  }
+}
+
+export const NewConfig_Gets_New_Defaults: Story = {
+  args: {
+    config: newConfig,
+    isEditor: false
+  },
+  play: async ({ canvasElement }) => {
+    await assertVisualizationRendered(canvasElement)
+
+    await waitForPresence('aside[aria-label="Legend"]', canvasElement)
+    const legend = canvasElement.querySelector('aside[aria-label="Legend"]')
+    expect(legend).toBeInTheDocument()
+    expect(legend?.classList.contains('top')).toBe(true)
+
+    expect(legend?.classList.contains('no-border')).toBe(true)
+
+    await waitForPresence('linearGradient', canvasElement)
+    const legendContainer = canvasElement.querySelector('.legend-container')
+    const linearGradient = legendContainer?.querySelector('linearGradient')
+    expect(linearGradient).toBeInTheDocument()
+  }
+}
+
+export default meta

package/src/_stories/CdcMap.Editor.ColumnsSectionTests.stories.tsx

@@ -0,0 +1,601 @@
+import type { Meta, StoryObj } from '@storybook/react-vite'
+import { within, userEvent, expect } from 'storybook/test'
+import CdcMap from '../CdcMap'
+import usaStateGradientConfig from './_mock/usa-state-gradient.json'
+import multiCountryConfig from './_mock/multi-country.json'
+import wastewaterMapSmallMultiplesConfig from './_mock/small_multiples/wastewater-map-small-multiples.json'
+import { performAndAssert, waitForEditor, waitForPresence, openAccordion } from '@cdc/core/helpers/testing'
+
+type Story = StoryObj<typeof CdcMap>
+
+const mapMeta: Meta<typeof CdcMap> = {
+  title: 'Components/Templates/Map/Editor Tests',
+  component: CdcMap,
+  parameters: {
+    layout: 'fullscreen'
+  }
+}
+
+export default mapMeta
+
+const DEFAULT_ARGS = {
+  isEditor: true,
+  config: usaStateGradientConfig
+}
+
+export const ColumnsSectionTests: Story = {
+  args: {
+    ...DEFAULT_ARGS
+  },
+  play: async ({ canvasElement }) => {
+    const setTextFieldValue = (input: HTMLInputElement | HTMLTextAreaElement, value: string) => {
+      const prototype =
+        input instanceof HTMLTextAreaElement ? HTMLTextAreaElement.prototype : HTMLInputElement.prototype
+      const valueSetter = Object.getOwnPropertyDescriptor(prototype, 'value')?.set
+      valueSetter?.call(input, value)
+      input.dispatchEvent(new Event('input', { bubbles: true }))
+      input.dispatchEvent(new Event('change', { bubbles: true }))
+    }
+    const canvas = within(canvasElement)
+
+    await waitForEditor(canvas)
+    await waitForPresence('.map-container', canvasElement)
+
+    await openAccordion(canvas, 'Columns')
+
+    // ==========================================================================
+    // TEST: Geography column select
+    // Verifies: Changing geo column to invalid data makes map gray and legend shows "No data"
+    // ==========================================================================
+    const columnsAccordionButton = canvas
+      .getAllByRole('button', { name: /^Columns$/i })
+      .find(button => button.closest('[data-accordion-component="AccordionItem"], .accordion__item'))
+    const columnsAccordion = columnsAccordionButton?.closest(
+      '[data-accordion-component="AccordionItem"], .accordion__item'
+    )
+    expect(columnsAccordion).toBeTruthy()
+    const geoColumnLabel = Array.from(columnsAccordion?.querySelectorAll('label') || []).find(label => {
+      const columnHeading = label.querySelector('.edit-label.column-heading')
+      return columnHeading?.textContent?.includes('Geography')
+    })
+    const geoSelect = geoColumnLabel?.querySelector('select') as HTMLSelectElement
+    expect(geoSelect).toBeTruthy()
+
+    const getMapAndLegendState = () => {
+      const legendContainer = canvasElement.querySelector('.legend-container')
+      const legendSvg = legendContainer?.querySelector('svg')
+      const gradientStops = legendSvg?.querySelectorAll('linearGradient stop')
+
+      // Gradient legend has color stops when data is valid
+      const hasGradientStops = (gradientStops?.length || 0) > 0
+
+      return {
+        legendHTML: legendContainer?.innerHTML || '',
+        hasGradientStops: hasGradientStops,
+        gradientStopCount: gradientStops?.length || 0
+      }
+    }
+
+    // Change to a column that's NOT valid geography data (Rate instead of STATE)
+    await performAndAssert(
+      'Geography Column → Change to invalid column',
+      getMapAndLegendState,
+      async () => {
+        await userEvent.selectOptions(geoSelect, 'Rate')
+      },
+      (before, after) => {
+        // Legend gradient should lose color stops when geo column is invalid
+        return before.hasGradientStops && !after.hasGradientStops
+      }
+    )
+
+    // Change back to valid geography column
+    await performAndAssert(
+      'Geography Column → Change back to valid column',
+      getMapAndLegendState,
+      async () => {
+        await userEvent.selectOptions(geoSelect, 'STATE')
+      },
+      (before, after) => {
+        // Legend gradient should regain color stops when geo column is valid
+        return !before.hasGradientStops && after.hasGradientStops
+      }
+    )
+
+    // ==========================================================================
+    // TEST: Hide Geography Column Name in Tooltip checkbox
+    // Verifies: Tooltip content changes when geography label is hidden/shown
+    // ==========================================================================
+    const hideGeoLabel = Array.from(columnsAccordion?.querySelectorAll('label') || []).find(label =>
+      label.textContent?.includes('Hide Geography Column Name in Tooltip')
+    )
+    const hideGeoCheckbox = hideGeoLabel?.querySelector('input[type="checkbox"]') as HTMLInputElement
+    expect(hideGeoCheckbox).toBeTruthy()
+
+    const getTooltipContent = () => {
+      const geoGroups = Array.from(canvasElement.querySelectorAll('g.geo-group')) as SVGGElement[]
+      const tooltipHtml =
+        geoGroups.map(group => group.getAttribute('data-tooltip-html') || '').find(html => html.trim().length > 0) || ''
+
+      return {
+        tooltipContent: tooltipHtml,
+        hasTooltipHeading: tooltipHtml.includes('tooltip-heading'),
+        hasStatePrefix: tooltipHtml.includes('State:'),
+        usesStrongGeoName: tooltipHtml.startsWith('<strong>')
+      }
+    }
+
+    await performAndAssert(
+      'Wait for geography tooltip to populate',
+      getTooltipContent,
+      async () => {},
+      (_before, after) => after.tooltipContent.length > 0
+    )
+
+    await performAndAssert(
+      'Hide Geography Column Name → Enable',
+      getTooltipContent,
+      async () => {
+        await userEvent.click(hideGeoCheckbox)
+      },
+      (before, after) => {
+        // Hidden geography labels now render as a standalone <strong> name instead of a tooltip heading.
+        return before.tooltipContent.length > 0 && after.usesStrongGeoName && !after.hasTooltipHeading
+      }
+    )
+
+    await performAndAssert(
+      'Hide Geography Column Name → Disable',
+      getTooltipContent,
+      async () => {
+        await userEvent.click(hideGeoCheckbox)
+      },
+      (before, after) => {
+        // Re-enabling restores the heading-based tooltip format.
+        return (
+          before.usesStrongGeoName &&
+          after.hasTooltipHeading &&
+          (after.hasStatePrefix || after.tooltipContent.length > 0)
+        )
+      }
+    )
+
+    // ==========================================================================
+    // TEST: Geography Label field
+    // Verifies: Custom geography label appears in tooltip instead of default "State:"
+    // ==========================================================================
+    const geoLabelInput = Array.from(columnsAccordion?.querySelectorAll('input[type="text"]') || []).find(input => {
+      const label = input.closest('label')
+      return label?.textContent?.includes('Geography Label')
+    }) as HTMLInputElement
+    expect(geoLabelInput).toBeTruthy()
+
+    await performAndAssert(
+      'Geography Label → Add custom label',
+      getTooltipContent,
+      async () => {
+        geoLabelInput.scrollIntoView({ block: 'center' })
+        setTextFieldValue(geoLabelInput, 'Region')
+      },
+      (before, after) => {
+        // Custom label "Region:" should appear in tooltip
+        return !before.tooltipContent.includes('Region:') && after.tooltipContent.includes('Region:')
+      }
+    )
+
+    // ==========================================================================
+    // TEST: Data Column select
+    // Verifies: Changing data column changes tooltip data values
+    // ==========================================================================
+    const dataColumnLabel = Array.from(columnsAccordion?.querySelectorAll('label') || []).find(label =>
+      label.textContent?.includes('Data Column')
+    )
+    const dataColumnSelect = dataColumnLabel?.querySelector('select') as HTMLSelectElement
+    expect(dataColumnSelect).toBeTruthy()
+
+    const getDataTooltipContent = () => {
+      const geoGroup = canvasElement.querySelector('g.geo-group') as SVGGElement
+      const tooltipHtml = geoGroup?.getAttribute('data-tooltip-html') || ''
+      return {
+        tooltipContent: tooltipHtml
+      }
+    }
+
+    await performAndAssert(
+      'Data Column → Change to STATE column',
+      getDataTooltipContent,
+      async () => {
+        await userEvent.selectOptions(dataColumnSelect, 'STATE')
+      },
+      (before, after) => {
+        // Tooltip should show STATE values (e.g., state names) instead of Rate values (numbers)
+        const beforeHasNumbers = /\d+/.test(before.tooltipContent)
+        const afterHasStateNames = /Alabama|Alaska|Arizona|Arkansas|California/.test(after.tooltipContent)
+        return beforeHasNumbers && afterHasStateNames
+      }
+    )
+
+    // Change back to Rate
+    await performAndAssert(
+      'Data Column → Change back to Rate',
+      getDataTooltipContent,
+      async () => {
+        await userEvent.selectOptions(dataColumnSelect, 'Rate')
+      },
+      (before, after) => {
+        // Tooltip should show numeric Rate values again
+        const beforeHasStateNames = /Alabama|Alaska|Arizona|Arkansas|California/.test(before.tooltipContent)
+        const afterHasNumbers = /\d+/.test(after.tooltipContent)
+        return beforeHasStateNames && afterHasNumbers
+      }
+    )
+
+    // ==========================================================================
+    // TEST: Hide Data Column Name in Tooltip
+    // Verifies: Data column label disappears from tooltip when hidden
+    // ==========================================================================
+    const hideDataLabel = Array.from(columnsAccordion?.querySelectorAll('label') || []).find(label =>
+      label.textContent?.includes('Hide Data Column Name in Tooltip')
+    )
+    const hideDataCheckbox = hideDataLabel?.querySelector('input[type="checkbox"]') as HTMLInputElement
+    expect(hideDataCheckbox).toBeTruthy()
+
+    await performAndAssert(
+      'Hide Data Column Name → Enable',
+      getDataTooltipContent,
+      async () => {
+        await userEvent.click(hideDataCheckbox)
+      },
+      (before, after) => {
+        // The label prefix (e.g., "Rate:" or similar) should be removed
+        // Before should have a colon pattern indicating a label, after should have less structure
+        const beforeHasLabelStructure = before.tooltipContent.includes(':</li>') || before.tooltipContent.includes(': ')
+        const afterHasLessStructure = after.tooltipContent.length < before.tooltipContent.length
+        return beforeHasLabelStructure && afterHasLessStructure
+      }
+    )
+
+    await performAndAssert(
+      'Hide Data Column Name → Disable',
+      getDataTooltipContent,
+      async () => {
+        await userEvent.click(hideDataCheckbox)
+      },
+      (before, after) => {
+        // The label prefix should reappear
+        const beforeHasLessStructure = before.tooltipContent.length < after.tooltipContent.length
+        const afterHasLabelStructure = after.tooltipContent.includes(':</li>') || after.tooltipContent.includes(': ')
+        return beforeHasLessStructure && afterHasLabelStructure
+      }
+    )
+
+    // ==========================================================================
+    // TEST: Data Label field
+    // Verifies: Custom data label appears in tooltip
+    // ==========================================================================
+    const dataLabelInputs = Array.from(columnsAccordion?.querySelectorAll('input') || [])
+    const dataLabelInput = dataLabelInputs.find(input => {
+      const label = input.closest('label')
+      return label?.textContent?.includes('Data Label')
+    }) as HTMLInputElement
+    expect(dataLabelInput).toBeTruthy()
+
+    await performAndAssert(
+      'Data Label → Add custom label',
+      getDataTooltipContent,
+      async () => {
+        dataLabelInput.scrollIntoView({ block: 'center' })
+        setTextFieldValue(dataLabelInput, 'Value')
+      },
+      (before, after) => {
+        // Custom label "Value:" should appear in tooltip
+        return !before.tooltipContent.includes('Value:') && after.tooltipContent.includes('Value:')
+      }
+    )
+
+    // ==========================================================================
+    // TEST: Prefix field
+    // Verifies: Prefix appears before data values in both tooltip and legend
+    // ==========================================================================
+    const prefixInputs = Array.from(columnsAccordion?.querySelectorAll('input') || [])
+    const prefixInput = prefixInputs.find(input => {
+      const label = input.closest('label')
+      return label?.textContent?.includes('Prefix')
+    }) as HTMLInputElement
+    expect(prefixInput).toBeTruthy()
+
+    const getPrefixVisualization = () => {
+      const geoGroup = canvasElement.querySelector('g.geo-group') as SVGGElement
+      const tooltipHtml = geoGroup?.getAttribute('data-tooltip-html') || ''
+      const legendContainer = canvasElement.querySelector('.legend-container')
+      const legendSvg = legendContainer?.querySelector('svg')
+      const legendText = legendSvg?.textContent || ''
+
+      return {
+        tooltipHasPrefix: tooltipHtml.includes('$'),
+        legendHasPrefix: legendText.includes('$'),
+        tooltipContent: tooltipHtml,
+        legendContent: legendText
+      }
+    }
+
+    await performAndAssert(
+      'Prefix → Add dollar sign',
+      getPrefixVisualization,
+      async () => {
+        prefixInput.scrollIntoView({ block: 'center' })
+        setTextFieldValue(prefixInput, '$')
+      },
+      (before, after) => {
+        // Dollar sign should appear in both tooltip and legend
+        return !before.tooltipHasPrefix && after.tooltipHasPrefix && !before.legendHasPrefix && after.legendHasPrefix
+      }
+    )
+
+    // ==========================================================================
+    // TEST: Suffix field
+    // Verifies: Suffix appears after data values in both tooltip and legend
+    // ==========================================================================
+    const suffixInputs = Array.from(columnsAccordion?.querySelectorAll('input') || [])
+    const suffixInput = suffixInputs.find(input => {
+      const label = input.closest('label')
+      return label?.textContent?.includes('Suffix')
+    }) as HTMLInputElement
+    expect(suffixInput).toBeTruthy()
+
+    const getSuffixVisualization = () => {
+      const geoGroup = canvasElement.querySelector('g.geo-group') as SVGGElement
+      const tooltipHtml = geoGroup?.getAttribute('data-tooltip-html') || ''
+      const legendContainer = canvasElement.querySelector('.legend-container')
+      const legendSvg = legendContainer?.querySelector('svg')
+      const legendText = legendSvg?.textContent || ''
+
+      return {
+        tooltipHasSuffix: tooltipHtml.includes('%'),
+        legendHasSuffix: legendText.includes('%'),
+        tooltipContent: tooltipHtml,
+        legendContent: legendText
+      }
+    }
+
+    await performAndAssert(
+      'Suffix → Add percent sign',
+      getSuffixVisualization,
+      async () => {
+        suffixInput.scrollIntoView({ block: 'center' })
+        setTextFieldValue(suffixInput, '%')
+      },
+      (before, after) => {
+        // Percent sign should appear in both tooltip and legend
+        return !before.tooltipHasSuffix && after.tooltipHasSuffix && !before.legendHasSuffix && after.legendHasSuffix
+      }
+    )
+
+    // ==========================================================================
+    // TEST: Round field
+    // Verifies: Changing rounding adds decimal places in both tooltip and legend
+    // ==========================================================================
+    const roundInputs = Array.from(columnsAccordion?.querySelectorAll('input') || [])
+    const roundInput = roundInputs.find(input => {
+      const label = input.closest('label')
+      return label?.textContent?.includes('Round')
+    }) as HTMLInputElement
+    expect(roundInput).toBeTruthy()
+
+    const getRoundingVisualization = () => {
+      const geoGroup = canvasElement.querySelector('g.geo-group') as SVGGElement
+      const tooltipHtml = geoGroup?.getAttribute('data-tooltip-html') || ''
+      const legendContainer = canvasElement.querySelector('.legend-container')
+      const legendSvg = legendContainer?.querySelector('svg')
+      const legendText = legendSvg?.textContent || ''
+
+      // Check for 2 decimal places pattern (e.g., "12.34", "0.50", etc.)
+      const twoDecimalPattern = /\d+\.\d{2}/
+
+      return {
+        tooltipHasTwoDecimals: twoDecimalPattern.test(tooltipHtml),
+        legendHasTwoDecimals: twoDecimalPattern.test(legendText),
+        tooltipContent: tooltipHtml,
+        legendContent: legendText
+      }
+    }
+
+    await performAndAssert(
+      'Round → Set to 2 decimal places',
+      getRoundingVisualization,
+      async () => {
+        roundInput.scrollIntoView({ block: 'center' })
+        setTextFieldValue(roundInput, '2')
+      },
+      (before, after) => {
+        // Numbers should now show 2 decimal places in both tooltip and legend
+        return (
+          !before.tooltipHasTwoDecimals &&
+          after.tooltipHasTwoDecimals &&
+          !before.legendHasTwoDecimals &&
+          after.legendHasTwoDecimals
+        )
+      }
+    )
+
+    // ==========================================================================
+    // TEST: Show in Data Table checkbox
+    // Verifies: Primary column appears/disappears in the data table
+    // ==========================================================================
+    const showInTableLabel = Array.from(columnsAccordion?.querySelectorAll('label') || []).find(label =>
+      label.textContent?.includes('Show in Data Table')
+    )
+    const showInTableCheckbox = showInTableLabel?.querySelector('input[type="checkbox"]') as HTMLInputElement
+    expect(showInTableCheckbox).toBeTruthy()
+
+    const getDataTableState = () => {
+      const dataTable = canvasElement.querySelector('.data-table-container, table')
+      const tableHeaders = Array.from(dataTable?.querySelectorAll('th') || [])
+      const tableHeaderText = tableHeaders.map(th => th.textContent?.trim() || '')
+
+      // Check if "Rate" column header appears in data table
+      const hasRateColumn = tableHeaderText.some(text => text.includes('Rate') || text.includes('Value'))
+
+      return {
+        hasDataTable: Boolean(dataTable),
+        tableHeaders: tableHeaderText,
+        hasRateColumn: hasRateColumn
+      }
+    }
+
+    // Test config has dataTable: true for primary column
+    await performAndAssert(
+      'Show in Data Table → Disable',
+      getDataTableState,
+      async () => {
+        await userEvent.click(showInTableCheckbox)
+      },
+      (before, after) => {
+        // Rate column should disappear from data table headers
+        return before.hasRateColumn && !after.hasRateColumn
+      }
+    )
+
+    await performAndAssert(
+      'Show in Data Table → Enable',
+      getDataTableState,
+      async () => {
+        await userEvent.click(showInTableCheckbox)
+      },
+      (before, after) => {
+        // Rate column should reappear in data table headers
+        return !before.hasRateColumn && after.hasRateColumn
+      }
+    )
+
+    // ==========================================================================
+    // TEST: Show in Tooltips checkbox
+    // Verifies: Primary column data appears/disappears in tooltips
+    // ==========================================================================
+    const showInTooltipsLabel = Array.from(columnsAccordion?.querySelectorAll('label') || []).find(label =>
+      label.textContent?.includes('Show in Tooltips')
+    )
+    const showInTooltipsCheckbox = showInTooltipsLabel?.querySelector('input[type="checkbox"]') as HTMLInputElement
+    expect(showInTooltipsCheckbox).toBeTruthy()
+
+    const getTooltipDataState = () => {
+      const geoGroup = canvasElement.querySelector('g.geo-group') as SVGGElement
+      const tooltipHtml = geoGroup?.getAttribute('data-tooltip-html') || ''
+
+      // Check if tooltip contains data values (numbers with potential $ and % from earlier tests)
+      // Look for patterns like list items with numeric values
+      const hasDataValues = tooltipHtml.includes('<li') && /\$?\d+\.\d{2}%?/.test(tooltipHtml)
+
+      return {
+        tooltipContent: tooltipHtml,
+        hasDataValues: hasDataValues,
+        tooltipLength: tooltipHtml.length
+      }
+    }
+
+    // Test config has tooltip: true for primary column
+    await performAndAssert(
+      'Show in Tooltips → Disable',
+      getTooltipDataState,
+      async () => {
+        await userEvent.click(showInTooltipsCheckbox)
+      },
+      (before, after) => {
+        // Tooltip should no longer contain data values, should be shorter
+        return before.hasDataValues && !after.hasDataValues && after.tooltipLength < before.tooltipLength
+      }
+    )
+
+    await performAndAssert(
+      'Show in Tooltips → Enable',
+      getTooltipDataState,
+      async () => {
+        await userEvent.click(showInTooltipsCheckbox)
+      },
+      (before, after) => {
+        // Tooltip should contain data values again, should be longer
+        return !before.hasDataValues && after.hasDataValues && after.tooltipLength > before.tooltipLength
+      }
+    )
+
+    // ==========================================================================
+    // TEST: Add Special Class button
+    // Verifies: New special class item appears in legend
+    // ==========================================================================
+    const addSpecialClassButton = Array.from(columnsAccordion?.querySelectorAll('button') || []).find(button =>
+      button.textContent?.includes('Add Special Class')
+    ) as HTMLButtonElement
+    expect(addSpecialClassButton).toBeTruthy()
+
+    const getSpecialClassVisualization = () => {
+      const legendContainer = canvasElement.querySelector('.legend-container')
+      const legendSvg = legendContainer?.querySelector('svg')
+      const gradientStops = legendSvg?.querySelectorAll('linearGradient stop')
+      const gradientRects = legendSvg?.querySelectorAll('g.visx-group rect')
+
+      // Get all stop colors to detect changes
+      const stopColors = Array.from(gradientStops || []).map(stop => (stop as SVGStopElement).getAttribute('style'))
+
+      // Get all rect fills to detect new colors
+      const rectFills = Array.from(gradientRects || []).map(rect => (rect as SVGRectElement).getAttribute('fill'))
+
+      return {
+        gradientStopCount: gradientStops?.length || 0,
+        gradientRectCount: gradientRects?.length || 0,
+        stopColors: stopColors,
+        rectFills: rectFills,
+        legendHTML: legendContainer?.innerHTML || ''
+      }
+    }
+
+    await performAndAssert(
+      'Add Special Class → Configure with STATE=Alabama',
+      getSpecialClassVisualization,
+      async () => {
+        // Click the Add Special Class button
+        await userEvent.click(addSpecialClassButton)
+
+        // Find the newly added special class fields
+        const specialClassSection = Array.from(columnsAccordion?.querySelectorAll('.edit-block') || []).find(block =>
+          block.textContent?.includes('Special Class 1')
+        )
+
+        // Get all selects in the special class section
+        // First select is Data Key, second is Value
+        const selects = Array.from(specialClassSection?.querySelectorAll('select') || [])
+        const dataKeySelect = selects[0] as HTMLSelectElement
+        const valueSelect = selects[1] as HTMLSelectElement
+
+        // Set Data Key to STATE
+        await userEvent.selectOptions(dataKeySelect, 'STATE')
+
+        // Wait for the Value select to populate with options based on STATE column
+        await new Promise(resolve => setTimeout(resolve, 200))
+
+        // Select "Alabama" from the Value select
+        await userEvent.selectOptions(valueSelect, 'Alabama')
+
+        // Find and fill in the Label field
+        const labelInput = Array.from(specialClassSection?.querySelectorAll('input') || []).find(input => {
+          const label = input.closest('label')
+          return label?.textContent?.includes('Label')
+        }) as HTMLInputElement
+
+        labelInput.scrollIntoView({ block: 'center' })
+        setTextFieldValue(labelInput, 'Alabama')
+      },
+      (before, after) => {
+        // Check that "Alabama" text appears in the legend
+        const alabamaInLegend = after.legendHTML.includes('Alabama')
+
+        // Check that the special class gray color appears
+        const hasGrayColor = after.rectFills.includes('#b4b4b4')
+
+        // Legend should change
+        const legendChanged = after.legendHTML !== before.legendHTML
+
+        return alabamaInLegend && hasGrayColor && legendChanged
+      }
+    )
+  }
+}