feat: add new property shape and size

juntossomosmais · Feb 25, 2025 · cc99782 · cc99782
1 parent 5c49ce5
commit cc99782
Showing 1 changed file with 20 additions and 4 deletions.
diff --git a/packages/core/src/components/badge/stories/badge.react.stories.tsx b/packages/core/src/components/badge/stories/badge.react.stories.tsx
@@ -2,7 +2,7 @@ import { AtomBadge } from '@juntossomosmais/atomium/react'
 import { Meta, StoryObj } from '@storybook/react'
 import React from 'react'
 
-import { BadgeStoryArgs } from './badge.args'
+import { BadgeComponentArgs, BadgeStoryArgs } from './badge.args'
 
 export default {
   title: 'Components/Badge',
@@ -11,14 +11,20 @@ export default {
 } as Meta
 
 const createBadge = (args) => (
-  <AtomBadge color={args.color}>{args.label}</AtomBadge>
+  <AtomBadge
+    color={args.color}
+    shape={args.shape}
+    size={args.size}
+    type={args.type}
+  >
+    {args.label}
+  </AtomBadge>
 )
 
 export const Primary: StoryObj = {
   render: (args) => createBadge(args),
   args: {
-    type: 'primary',
-    label: 'Badge',
+    ...BadgeComponentArgs,
   },
 }
 
@@ -77,3 +83,13 @@ export const Info: StoryObj = {
     type: 'info',
   },
 }
+
+export const RectangleButton: StoryObj = {
+  render: (args) => createBadge(args),
+  args: {
+    ...Primary.args,
+    color: 'secondary',
+    size: 'default',
+    shape: 'rectangle',
+  },
+}