fss-link 1.0.40 → 1.0.45

package/dist/config/database.test.ts

@@ -0,0 +1,96 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import { describe, it, expect, beforeEach, afterEach } from 'vitest';
+import { getFSSLinkDatabase } from './database.js';
+import * as fs from 'fs';
+import * as path from 'path';
+import * as os from 'os';
+
+describe('FSS Link Database Initialization', () => {
+  let testDbPath: string;
+
+  beforeEach(async () => {
+    // Create temporary database for testing
+    testDbPath = path.join(os.tmpdir(), `test-fss-link-init-${Date.now()}.db`);
+    
+    // Override the database path for testing
+    process.env['FSS_LINK_TEST_DB'] = testDbPath;
+  });
+
+  afterEach(async () => {
+    // Clean up test database
+    if (fs.existsSync(testDbPath)) {
+      fs.unlinkSync(testDbPath);
+    }
+    delete process.env['FSS_LINK_TEST_DB'];
+  });
+
+  it('should create default configurations on first initialization', async () => {
+    const db = await getFSSLinkDatabase();
+    await db.initialize();
+
+    // Check that default configurations were created
+    const allConfigs = await db.getAllModels();
+    console.log('All configurations:', JSON.stringify(allConfigs, null, 2));
+    expect(allConfigs).toBeDefined();
+    expect(allConfigs.length).toBeGreaterThan(0);
+
+    // Should have default Ollama config
+    const ollamaConfig = allConfigs.find(c => c.authType === 'ollama');
+    expect(ollamaConfig).toBeDefined();
+    expect(ollamaConfig?.modelName).toBe('qwen3-coder:latest');
+    expect(ollamaConfig?.isActive).toBe(true);
+    expect(ollamaConfig?.isFavorite).toBe(true);
+
+    // Should have default Gemini config (inactive)
+    const geminiConfig = allConfigs.find(c => c.authType === 'gemini');
+    expect(geminiConfig).toBeDefined();
+    expect(geminiConfig?.modelName).toBe('gemini-2.5-flash-thinking');
+    expect(geminiConfig?.isActive).toBe(false);
+
+    // Should have default LM Studio config (active)
+    const lmStudioConfig = allConfigs.find(c => c.authType === 'lmstudio');
+    expect(lmStudioConfig).toBeDefined();
+    expect(lmStudioConfig?.modelName).toBe('qwen3-coder:latest');
+    expect(lmStudioConfig?.isActive).toBe(true);
+
+    // Should have default OpenAI config (inactive)
+    const openaiConfig = allConfigs.find(c => c.authType === 'openai');
+    expect(openaiConfig).toBeDefined();
+    expect(openaiConfig?.modelName).toBe('gpt-4o');
+    expect(openaiConfig?.endpointUrl).toBe('https://api.openai.com/v1');
+    expect(openaiConfig?.displayName).toBe('GPT-4o (Requires API Key)');
+    expect(openaiConfig?.isActive).toBe(false);
+  });
+
+  it('should return smart default provider', async () => {
+    const db = await getFSSLinkDatabase();
+    await db.initialize();
+
+    const defaultProvider = await db.getSmartDefaultProvider();
+    expect(defaultProvider).toBeDefined();
+    expect(defaultProvider?.authType).toBe('ollama');
+    expect(defaultProvider?.modelName).toBe('qwen3-coder:latest');
+    expect(defaultProvider?.isActive).toBe(true);
+    expect(defaultProvider?.isFavorite).toBe(true);
+  });
+
+  it('should not duplicate default configurations on subsequent initializations', async () => {
+    const db = await getFSSLinkDatabase();
+    
+    // First initialization
+    await db.initialize();
+    const firstCount = (await db.getAllModels()).length;
+    
+    // Second initialization should not add more defaults
+    await db.initialize();
+    const secondCount = (await db.getAllModels()).length;
+    
+    expect(firstCount).toBe(secondCount);
+    expect(firstCount).toBe(4); // Should have exactly 4 default configs: Ollama, Gemini, LM Studio, OpenAI
+  });
+});