Merge pull request #12 from bertyuan/fix-vitestv1.1

Fix vitest

1 files changed, 30 insertions, 0 deletions

diff --git a/src/components/tailwind-indicator.test.tsx b/src/components/tailwind-indicator.test.tsx
new file mode 100644
index 0000000..4244ef6
--- /dev/null
+++ b/src/components/tailwind-indicator.test.tsx
@@ -0,0 +1,30 @@
+import { render, screen } from '@testing-library/react';
+import { afterAll, beforeEach, describe, expect, test } from 'vitest';
+import { TailwindIndicator } from './tailwind-indicator';
+
+const originalEnv = process.env;
+
+describe('TailwindIndicator', () => {
+  beforeEach(() => {
+    process.env = { ...originalEnv };
+  });
+
+  afterAll(() => {
+    process.env = originalEnv;
+  });
+
+  test('renders in non-production environments', () => {
+    process.env.NODE_ENV = 'test';
+    render(<TailwindIndicator />);
+
+    expect(screen.getByText('xs')).toBeInTheDocument();
+  });
+
+  test('returns null in production', () => {
+    process.env.NODE_ENV = 'production';
+    const { container } = render(<TailwindIndicator />);
+
+    expect(container.firstChild).toBeNull();
+  });
+});
+