test: knob for partition layout selector (opensearch-project#1120)

AMoo-Miki · Apr 14, 2021 · d0cf0d9 · d0cf0d9
1 parent a0020ba
commit d0cf0d9
Showing 1 changed file with 14 additions and 3 deletions.
diff --git a/packages/osd-charts/stories/mosaic/10_mosaic_simple.tsx b/packages/osd-charts/stories/mosaic/10_mosaic_simple.tsx
@@ -17,7 +17,7 @@
  * under the License.
  */
 
-import { boolean } from '@storybook/addon-knobs';
+import { boolean, radios } from '@storybook/addon-knobs';
 import React from 'react';
 
 import {
@@ -62,6 +62,15 @@ const productToColor = new Map(
 );
 
 export const Example = () => {
+  const partitionLayout = radios(
+    'Partition layout',
+    {
+      [PartitionLayout.mosaic]: PartitionLayout.mosaic,
+      [PartitionLayout.treemap]: PartitionLayout.treemap,
+      [PartitionLayout.sunburst]: PartitionLayout.sunburst,
+    },
+    PartitionLayout.mosaic,
+  );
   return (
     <Chart className="story-chart">
       <Settings showLegend={boolean('Show legend', true)} showLegendExtra={boolean('Show legend values', true)} />
@@ -78,7 +87,7 @@ export const Example = () => {
               fontWeight: 400,
             },
             shape: {
-              fillColor: () => 'white',
+              fillColor: partitionLayout === PartitionLayout.sunburst ? 'lightgrey' : 'white',
             },
           },
           {
@@ -103,7 +112,9 @@ export const Example = () => {
           },
         ]}
         config={{
-          partitionLayout: PartitionLayout.mosaic,
+          partitionLayout,
+          linkLabel: { maxCount: 0 }, // relevant for sunburst only
+          outerSizeRatio: 0.9, // relevant for sunburst only
         }}
       />
     </Chart>