restore custom icon control in storybook

WordPress · May 9, 2023 · 9b920e0 · 9b920e0
1 parent 5d317f9
commit 9b920e0
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/packages/components/src/dropdown-menu/stories/index.tsx b/packages/components/src/dropdown-menu/stories/index.tsx
@@ -11,7 +11,14 @@ import { MenuGroup, MenuItem } from '../..';
 /**
  * WordPress dependencies
  */
-import { menu, arrowUp, arrowDown, more, trash } from '@wordpress/icons';
+import {
+	menu,
+	arrowUp,
+	arrowDown,
+	chevronDown,
+	more,
+	trash,
+} from '@wordpress/icons';
 
 const meta: ComponentMeta< typeof DropdownMenu > = {
 	title: 'Components/DropdownMenu',
@@ -20,6 +27,13 @@ const meta: ComponentMeta< typeof DropdownMenu > = {
 		controls: { expanded: true },
 		docs: { source: { state: 'open' } },
 	},
+	argTypes: {
+		icon: {
+			options: [ 'menu', 'chevronDown', 'more' ],
+			mapping: { menu, chevronDown, more },
+			control: { type: 'select' },
+		},
+	},
 };
 export default meta;