DRY story and make sure it uses all props setup in the original template

Co-authored-by: Lena Morita <[email protected]>
WordPress · May 28, 2024 · e75e941 · e75e941
1 parent a8ad3e7
commit e75e941
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/packages/components/src/progress-bar/stories/index.story.tsx b/packages/components/src/progress-bar/stories/index.story.tsx
@@ -51,11 +51,15 @@ const withCustomWidthCustomCSS = `
  * }
  * ```
  */
-export const WithCustomWidth: StoryFn< typeof ProgressBar > = () => {
-	return (
+export const WithCustomWidth = Template.bind( {} );
+WithCustomWidth.args = {
+	className: 'custom-progress-bar',
+};
+WithCustomWidth.decorators = [
+	( Story ) => (
 		<>
 			<style>{ withCustomWidthCustomCSS }</style>
-			<ProgressBar className="custom-progress-bar" />
+			<Story />
 		</>
-	);
-};
+	),
+];