add more tests

1 files changed, 28 insertions, 0 deletions

diff --git a/src/components/ui/button-variants.test.ts b/src/components/ui/button-variants.test.ts
new file mode 100644
index 0000000..5090703
--- /dev/null
+++ b/src/components/ui/button-variants.test.ts
@@ -0,0 +1,28 @@
+import { describe, expect, test } from 'vitest';
+import { buttonVariants } from './button';
+
+describe('buttonVariants', () => {
+  test('returns default variant and size classes when no args are provided', () => {
+    const classes = buttonVariants();
+
+    expect(classes).toContain('bg-primary');
+    expect(classes).toContain('h-9');
+  });
+
+  test('returns destructive and icon classes for explicit options', () => {
+    const classes = buttonVariants({
+      variant: 'destructive',
+      size: 'icon',
+    });
+
+    expect(classes).toContain('bg-destructive');
+    expect(classes).toContain('size-9');
+  });
+
+  test('includes caller-provided className', () => {
+    const classes = buttonVariants({ className: 'w-full' });
+
+    expect(classes).toContain('w-full');
+  });
+});
+