feat(pie): add tests for gradients attribute

plouc · May 13, 2023 · b37fc17 · b37fc17
1 parent ce598c0
commit b37fc17
Showing 1 changed file with 62 additions and 2 deletions.
diff --git a/packages/pie/tests/Pie.test.tsx b/packages/pie/tests/Pie.test.tsx
@@ -1,5 +1,5 @@
 import { mount } from 'enzyme'
-import { radiansToDegrees } from '@nivo/core'
+import { linearGradientDef, radiansToDegrees } from '@nivo/core'
 import {
     ArcShape,
     ArcLabelsLayer,
@@ -29,6 +29,42 @@ const sampleData = [
     },
 ]
 
+const sampleGradientData = [
+    linearGradientDef('gradientA', [
+        {
+            offset: 0,
+            color: '#e8c1a0',
+        },
+        {
+            offset: 100,
+            color: '#ff5500',
+            opacity: 0,
+        },
+    ]),
+    linearGradientDef('gradientB', [
+        {
+            offset: 0,
+            color: '#f47560',
+        },
+        {
+            offset: 100,
+            color: '#f47560',
+            opacity: 0,
+        },
+    ]),
+    linearGradientDef('gradientC', [
+        {
+            offset: 0,
+            color: '#f1e15b',
+        },
+        {
+            offset: 100,
+            color: '#f1e15b',
+            opacity: 0,
+        },
+    ]),
+]
+
 sampleData.forEach(datum => {
     datum.nested = {
         color: datum.color,
@@ -331,7 +367,31 @@ describe('Pie', () => {
         })
 
         it('should support gradients', () => {
-            // @todo
+            const wrapper = mount(
+                <Pie
+                    width={400}
+                    height={400}
+                    data={sampleData}
+                    defs={sampleGradientData}
+                    fill={[
+                        { match: { id: 'A' }, id: 'gradientA' },
+                        { match: { id: 'B' }, id: 'gradientB' },
+                        { match: { id: 'C' }, id: 'gradientC' },
+                    ]}
+                    animate={false}
+                />
+            )
+            const slices = wrapper.find(ArcShape)
+            expect(slices).toHaveLength(sampleData.length)
+
+            expect(slices.at(0).prop('datum').id).toEqual('A')
+            expect(slices.at(0).prop('datum').fill).toEqual('url(#gradientA)')
+
+            expect(slices.at(1).prop('datum').id).toEqual('B')
+            expect(slices.at(1).prop('datum').fill).toEqual('url(#gradientB)')
+
+            expect(slices.at(2).prop('datum').id).toEqual('C')
+            expect(slices.at(2).prop('datum').fill).toEqual('url(#gradientC)')
         })
     })