test: add config.test

Author: xiaoyu2er

packages/translate/tests/unit/config.test.ts

@@ -0,0 +1,140 @@
+import { cosmiconfig } from 'cosmiconfig';
+import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
+import { getConfig } from '../../src/config';
+import type { MainConfig } from '../../src/types';
+
+// Mock the cosmiconfig module
+vi.mock('cosmiconfig', () => ({
+  cosmiconfig: vi.fn(),
+}));
+
+// Mock the logger
+vi.mock('../../src/logger', () => ({
+  logger: {
+    debug: vi.fn(),
+    warn: vi.fn(),
+    error: vi.fn(),
+    info: vi.fn(),
+    success: vi.fn(),
+    progress: vi.fn(),
+    divider: vi.fn(),
+  },
+}));
+
+describe('config', () => {
+  // Common mocks
+  const mockLoad = vi.fn();
+  const mockSearch = vi.fn();
+  const mockExplorer = { load: mockLoad, search: mockSearch };
+
+  beforeEach(() => {
+    // Reset mocks before each test
+    vi.clearAllMocks();
+
+    // Setup the default cosmiconfig mock
+    (cosmiconfig as unknown as ReturnType<typeof vi.fn>).mockReturnValue(
+      mockExplorer,
+    );
+  });
+
+  describe('getConfig', () => {
+    it('should load config from specific file when provided', async () => {
+      // Setup the mock to return a valid config
+      const configFile = 'custom.config.mjs';
+      const mockConfig: MainConfig = {
+        langs: {
+          'en-US': { name: 'English' },
+          fr: { name: 'French' },
+        },
+        docsRoot: 'content/en',
+      };
+      mockLoad.mockResolvedValue({ config: mockConfig, filepath: configFile });
+
+      // Call the function with a specific config file
+      const result = await getConfig({ config: configFile });
+
+      // Assertions
+      expect(cosmiconfig).toHaveBeenCalledWith(
+        'translation',
+        expect.any(Object),
+      );
+      expect(mockLoad).toHaveBeenCalledWith(configFile);
+      expect(result).toEqual(mockConfig);
+    });
+
+    it('should search for config when no specific file is provided', async () => {
+      // Setup the mock to return a valid config
+      const mockConfig: MainConfig = {
+        langs: {
+          'en-US': { name: 'English' },
+          fr: { name: 'French' },
+        },
+        docsRoot: 'content/en',
+      };
+      mockSearch.mockResolvedValue({
+        config: mockConfig,
+        filepath: 'translation.config.mjs',
+      });
+
+      // Call the function without a specific config file
+      const result = await getConfig({});
+
+      // Assertions
+      expect(cosmiconfig).toHaveBeenCalledWith(
+        'translation',
+        expect.any(Object),
+      );
+      expect(mockSearch).toHaveBeenCalled();
+      expect(result).toEqual(mockConfig);
+    });
+
+    it('should return empty config when no config file is found', async () => {
+      // Setup the mock to return null (no config found)
+      mockSearch.mockResolvedValue(null);
+
+      // Call the function
+      const result = await getConfig({});
+
+      // Assertions
+      expect(result).toEqual({});
+    });
+
+    it('should handle errors when loading config file', async () => {
+      // Setup the mock to throw an error
+      const configFile = 'broken.config.mjs';
+      const error = new Error('Config file not found');
+      mockLoad.mockRejectedValue(error);
+
+      // Mock process.exit to prevent the test from actually exiting
+      const mockExit = vi
+        .spyOn(process, 'exit')
+        .mockImplementation(() => undefined as never);
+
+      // Call the function with a specific config file that will error
+      await getConfig({ config: configFile });
+
+      // Assertions
+      expect(mockExit).toHaveBeenCalledWith(1);
+    });
+
+    it('should handle targetLanguage option correctly', async () => {
+      const mockConfig: MainConfig = {
+        langs: {
+          'en-US': { name: 'English' },
+          fr: { name: 'French' },
+        },
+        docsRoot: 'content/en',
+      };
+      mockSearch.mockResolvedValue({
+        config: mockConfig,
+        filepath: 'translation.config.mjs',
+      });
+
+      // Call with targetLanguage option
+      const result = await getConfig({ targetLanguage: 'fr' });
+
+      // Assertions
+      expect(result).toEqual(mockConfig);
+    });
+  });
+});