Feature/lmstudio provider (#30)
* chore: just a plan update * Add LM Studio as local AI provider (OpenAI-compatible, like Ollama) * Convert WebP thumbnails to JPEG before image analysis for LM Studio compatibility * Strengthen language enforcement in image analysis prompt for local models * Use i18n localized prompts for image analysis instead of English instructions * Add airplane mode (Flugmodus) with status bar toggle and offline model preferences * Fix flightmode: persist model IDs, skip network when offline, airplane icon * Auto-fallback to offline models in airplane mode for chat, title, and image analysis * Auto-select first local model as offline fallback when no explicit offline model configured * Block git fetch/pull/push and site upload in airplane mode * fix: thumbnails optimized for AI * fix: error handling in airplane mode --------- Co-authored-by: hugo <hugoms@me.com>
This commit is contained in:
Georg Bauer
@@ -140,13 +140,15 @@ describe('ProviderRegistry', () => {
|
||||
});
|
||||
|
||||
it('getProviderStatus() reports all providers', () => {
|
||||
expect(registry.getProviderStatus()).toEqual({ opencode: false, mistral: false, ollama: false });
|
||||
expect(registry.getProviderStatus()).toEqual({ opencode: false, mistral: false, ollama: false, lmstudio: false, offlineMode: false });
|
||||
registry.setOpencodeKey('test');
|
||||
expect(registry.getProviderStatus()).toEqual({ opencode: true, mistral: false, ollama: false });
|
||||
expect(registry.getProviderStatus()).toEqual({ opencode: true, mistral: false, ollama: false, lmstudio: false, offlineMode: false });
|
||||
registry.setMistralKey('test2');
|
||||
expect(registry.getProviderStatus()).toEqual({ opencode: true, mistral: true, ollama: false });
|
||||
expect(registry.getProviderStatus()).toEqual({ opencode: true, mistral: true, ollama: false, lmstudio: false, offlineMode: false });
|
||||
registry.setOllamaEnabled(true);
|
||||
expect(registry.getProviderStatus()).toEqual({ opencode: true, mistral: true, ollama: true });
|
||||
expect(registry.getProviderStatus()).toEqual({ opencode: true, mistral: true, ollama: true, lmstudio: false, offlineMode: false });
|
||||
registry.setLmstudioEnabled(true);
|
||||
expect(registry.getProviderStatus()).toEqual({ opencode: true, mistral: true, ollama: true, lmstudio: true, offlineMode: false });
|
||||
});
|
||||
|
||||
it('isProviderKeySet() checks per-provider', () => {
|
||||
@@ -444,6 +446,120 @@ describe('OneShotTasks', () => {
|
||||
expect(result.error).toContain('thumbnail');
|
||||
});
|
||||
|
||||
it('uses pre-generated AI JPEG thumbnail without sharp conversion', async () => {
|
||||
registry.setOpencodeKey('test-key');
|
||||
chatEngine.getSetting.mockResolvedValue('claude-sonnet-4');
|
||||
mediaEngine.getMedia.mockResolvedValue({
|
||||
id: 'media-1',
|
||||
mimeType: 'image/jpeg',
|
||||
filename: 'photo.jpg',
|
||||
});
|
||||
// Tiny valid JPEG — simulates the pre-generated 'ai' thumbnail
|
||||
const jpegBase64 = '/9j/2wBDAAYEBQYFBAYGBQYHBwYIChAKCgkJChQODwwQFxQYGBcUFhYaHSUfGhsjHBYWICwgIyYnKSopGR8tMC0oMCUoKSj/2wBDAQcHBwoIChMKChMoGhYaKCgoKCgoKCgoKCgoKCgoKCgoKCgoKCgoKCgoKCgoKCgoKCgoKCgoKCgoKCgoKCgoKCj/wAARCAACAAIDASIAAhEBAxEB/8QAFAABAAAAAAAAAAAAAAAAAAAAAP/EABQQAQAAAAAAAAAAAAAAAAAAAAD/xAAUAQEAAAAAAAAAAAAAAAAAAAAA/8QAFBEBAAAAAAAAAAAAAAAAAAAAAP/aAAwDAQACEQMRAD8AAA//2Q==';
|
||||
// Return JPEG for 'ai' size, null for others
|
||||
mediaEngine.getThumbnailDataUrl.mockImplementation(async (_id: string, size: string) => {
|
||||
if (size === 'ai') return `data:image/jpeg;base64,${jpegBase64}`;
|
||||
return null;
|
||||
});
|
||||
|
||||
const originalFetch = globalThis.fetch;
|
||||
let capturedBody: any = null;
|
||||
globalThis.fetch = vi.fn().mockImplementation(async (url: string, init: any) => {
|
||||
if (init?.body) {
|
||||
capturedBody = JSON.parse(init.body);
|
||||
}
|
||||
return new Response(JSON.stringify({
|
||||
id: 'msg_test',
|
||||
type: 'message',
|
||||
role: 'assistant',
|
||||
content: [{ type: 'text', text: '{"title": "Test", "alt": "Test image", "caption": "A test"}' }],
|
||||
model: 'claude-sonnet-4',
|
||||
stop_reason: 'end_turn',
|
||||
usage: { input_tokens: 100, output_tokens: 30, cache_creation_input_tokens: 0, cache_read_input_tokens: 0 },
|
||||
}), { status: 200, headers: { 'Content-Type': 'application/json' } });
|
||||
});
|
||||
|
||||
try {
|
||||
const result = await tasks.analyzeMediaImage('media-1', 'en');
|
||||
// Check the image was sent as JPEG, not WebP
|
||||
if (capturedBody?.messages) {
|
||||
const userMsg = capturedBody.messages.find((m: any) => m.role === 'user');
|
||||
if (userMsg?.content) {
|
||||
const imagePart = userMsg.content.find((p: any) => p.type === 'image_url');
|
||||
if (imagePart?.image_url?.url) {
|
||||
expect(imagePart.image_url.url).toMatch(/^data:image\/jpeg;base64,/);
|
||||
expect(imagePart.image_url.url).not.toMatch(/^data:image\/webp;base64,/);
|
||||
}
|
||||
}
|
||||
}
|
||||
// Also verify it succeeded (may fail on response parsing but the format check is key)
|
||||
if (result.success) {
|
||||
expect(result.title).toBe('Test');
|
||||
}
|
||||
} finally {
|
||||
globalThis.fetch = originalFetch;
|
||||
}
|
||||
});
|
||||
|
||||
it('sends localized prompts based on project language', async () => {
|
||||
registry.setOpencodeKey('test-key');
|
||||
chatEngine.getSetting.mockResolvedValue('claude-sonnet-4');
|
||||
mediaEngine.getMedia.mockResolvedValue({
|
||||
id: 'media-1',
|
||||
mimeType: 'image/jpeg',
|
||||
filename: 'photo.jpg',
|
||||
});
|
||||
// Tiny valid JPEG — simulates the pre-generated 'ai' thumbnail
|
||||
const jpegBase64 = '/9j/2wBDAAYEBQYFBAYGBQYHBwYIChAKCgkJChQODwwQFxQYGBcUFhYaHSUfGhsjHBYWICwgIyYnKSopGR8tMC0oMCUoKSj/2wBDAQcHBwoIChMKChMoGhYaKCgoKCgoKCgoKCgoKCgoKCgoKCgoKCgoKCgoKCgoKCgoKCgoKCgoKCgoKCgoKCgoKCj/wAARCAACAAIDASIAAhEBAxEB/8QAFAABAAAAAAAAAAAAAAAAAAAAAP/EABQQAQAAAAAAAAAAAAAAAAAAAAD/xAAUAQEAAAAAAAAAAAAAAAAAAAAA/8QAFBEBAAAAAAAAAAAAAAAAAAAAAP/aAAwDAQACEQMRAD8AAA//2Q==';
|
||||
mediaEngine.getThumbnailDataUrl.mockImplementation(async (_id: string, size: string) => {
|
||||
if (size === 'ai') return `data:image/jpeg;base64,${jpegBase64}`;
|
||||
return null;
|
||||
});
|
||||
|
||||
const originalFetch = globalThis.fetch;
|
||||
let capturedBody: any = null;
|
||||
globalThis.fetch = vi.fn().mockImplementation(async (_url: string, init: any) => {
|
||||
if (init?.body) {
|
||||
capturedBody = JSON.parse(init.body);
|
||||
}
|
||||
return new Response(JSON.stringify({
|
||||
id: 'msg_test',
|
||||
type: 'message',
|
||||
role: 'assistant',
|
||||
content: [{ type: 'text', text: '{"title": "Testbild", "alt": "Rotes Quadrat", "caption": "Ein Test"}' }],
|
||||
model: 'claude-sonnet-4',
|
||||
stop_reason: 'end_turn',
|
||||
usage: { input_tokens: 100, output_tokens: 30, cache_creation_input_tokens: 0, cache_read_input_tokens: 0 },
|
||||
}), { status: 200, headers: { 'Content-Type': 'application/json' } });
|
||||
});
|
||||
|
||||
try {
|
||||
await tasks.analyzeMediaImage('media-1', 'de');
|
||||
// System prompt should be in German (from i18n), not contain English instructions
|
||||
if (capturedBody) {
|
||||
const systemMsg = capturedBody.messages?.find((m: any) => m.role === 'system')
|
||||
?? capturedBody.system;
|
||||
const systemText = typeof systemMsg === 'string' ? systemMsg
|
||||
: Array.isArray(systemMsg) ? systemMsg.map((p: any) => p.text).join('')
|
||||
: systemMsg?.content ?? '';
|
||||
expect(systemText).toContain('Deutsch');
|
||||
expect(systemText).not.toContain('English');
|
||||
// User message should also be in German
|
||||
const userMsg = capturedBody.messages?.find((m: any) => m.role === 'user');
|
||||
if (userMsg?.content) {
|
||||
const textPart = Array.isArray(userMsg.content)
|
||||
? userMsg.content.find((p: any) => p.type === 'text')
|
||||
: null;
|
||||
if (textPart?.text) {
|
||||
expect(textPart.text).toContain('Deutsch');
|
||||
}
|
||||
}
|
||||
}
|
||||
} finally {
|
||||
globalThis.fetch = originalFetch;
|
||||
}
|
||||
});
|
||||
|
||||
it('falls back to claude-sonnet-4-5 when no image analysis model is configured', async () => {
|
||||
registry.setOpencodeKey('test-key');
|
||||
chatEngine.getSetting.mockResolvedValue(null);
|
||||
@@ -452,7 +568,12 @@ describe('OneShotTasks', () => {
|
||||
mimeType: 'image/jpeg',
|
||||
filename: 'photo.jpg',
|
||||
});
|
||||
mediaEngine.getThumbnailDataUrl.mockResolvedValue('data:image/webp;base64,abc123');
|
||||
// Tiny valid JPEG — simulates the pre-generated 'ai' thumbnail
|
||||
const jpegBase64 = '/9j/2wBDAAYEBQYFBAYGBQYHBwYIChAKCgkJChQODwwQFxQYGBcUFhYaHSUfGhsjHBYWICwgIyYnKSopGR8tMC0oMCUoKSj/2wBDAQcHBwoIChMKChMoGhYaKCgoKCgoKCgoKCgoKCgoKCgoKCgoKCgoKCgoKCgoKCgoKCgoKCgoKCgoKCgoKCgoKCj/wAARCAACAAIDASIAAhEBAxEB/8QAFAABAAAAAAAAAAAAAAAAAAAAAP/EABQQAQAAAAAAAAAAAAAAAAAAAAD/xAAUAQEAAAAAAAAAAAAAAAAAAAAA/8QAFBEBAAAAAAAAAAAAAAAAAAAAAP/aAAwDAQACEQMRAD8AAA//2Q==';
|
||||
mediaEngine.getThumbnailDataUrl.mockImplementation(async (_id: string, size: string) => {
|
||||
if (size === 'ai') return `data:image/jpeg;base64,${jpegBase64}`;
|
||||
return null;
|
||||
});
|
||||
|
||||
// Verify the method selects the right model by checking it attempts
|
||||
// to call the resolver (which hits the network). We mock fetch to
|
||||
|
||||
319
tests/engine/lmstudio-provider.test.ts
Normal file
319
tests/engine/lmstudio-provider.test.ts
Normal file
@@ -0,0 +1,319 @@
|
||||
/**
|
||||
* Tests for LM Studio provider integration in ProviderRegistry.
|
||||
*
|
||||
* LM Studio provides an OpenAI-compatible API at http://localhost:1234/v1
|
||||
* with a standard /v1/models endpoint for model listing.
|
||||
*/
|
||||
|
||||
import { describe, it, expect, vi, beforeEach } from 'vitest';
|
||||
import { ProviderRegistry, LMSTUDIO_BASE_URL, LMSTUDIO_MODELS_URL } from '../../src/main/engine/ai/providers';
|
||||
|
||||
// Mock ModelCatalogEngine — no DB in unit tests
|
||||
vi.mock('../../src/main/engine/ModelCatalogEngine', () => ({
|
||||
ModelCatalogEngine: class {
|
||||
getAll = vi.fn().mockResolvedValue([]);
|
||||
getContextWindow = vi.fn().mockResolvedValue(null);
|
||||
},
|
||||
}));
|
||||
|
||||
describe('LM Studio provider support', () => {
|
||||
let registry: ProviderRegistry;
|
||||
|
||||
beforeEach(() => {
|
||||
registry = new ProviderRegistry();
|
||||
});
|
||||
|
||||
// ---- Constants ----
|
||||
|
||||
it('exports LM Studio URL constants', () => {
|
||||
expect(LMSTUDIO_BASE_URL).toBe('http://localhost:1234/v1');
|
||||
expect(LMSTUDIO_MODELS_URL).toBe('http://localhost:1234/v1/models');
|
||||
});
|
||||
|
||||
// ---- LM Studio enable/disable ----
|
||||
|
||||
it('is not enabled by default', () => {
|
||||
expect(registry.isLmstudioEnabled()).toBe(false);
|
||||
});
|
||||
|
||||
it('can be enabled and disabled', () => {
|
||||
registry.setLmstudioEnabled(true);
|
||||
expect(registry.isLmstudioEnabled()).toBe(true);
|
||||
registry.setLmstudioEnabled(false);
|
||||
expect(registry.isLmstudioEnabled()).toBe(false);
|
||||
});
|
||||
|
||||
it('enabling LM Studio invalidates model cache', () => {
|
||||
// Populate cache
|
||||
registry['cachedModels'] = [{ id: 'test', name: 'test', provider: 'other' }];
|
||||
registry['cachedModelsAt'] = Date.now();
|
||||
|
||||
registry.setLmstudioEnabled(true);
|
||||
|
||||
expect(registry['cachedModels']).toBeNull();
|
||||
expect(registry['cachedModelsAt']).toBe(0);
|
||||
});
|
||||
|
||||
// ---- Provider status ----
|
||||
|
||||
it('getProviderStatus includes lmstudio field', () => {
|
||||
const status = registry.getProviderStatus();
|
||||
expect(status).toHaveProperty('lmstudio');
|
||||
expect(status.lmstudio).toBe(false);
|
||||
|
||||
registry.setLmstudioEnabled(true);
|
||||
expect(registry.getProviderStatus().lmstudio).toBe(true);
|
||||
});
|
||||
|
||||
// ---- isReady includes lmstudio ----
|
||||
|
||||
it('isReady returns true when only LM Studio is enabled', () => {
|
||||
expect(registry.isReady()).toBe(false);
|
||||
registry.setLmstudioEnabled(true);
|
||||
expect(registry.isReady()).toBe(true);
|
||||
});
|
||||
|
||||
// ---- isProviderKeySet for lmstudio ----
|
||||
|
||||
it('isProviderKeySet returns lmstudio enabled state for provider "lmstudio"', () => {
|
||||
expect(registry.isProviderKeySet('lmstudio')).toBe(false);
|
||||
registry.setLmstudioEnabled(true);
|
||||
expect(registry.isProviderKeySet('lmstudio')).toBe(true);
|
||||
});
|
||||
|
||||
// ---- resolveModel for lmstudio ----
|
||||
|
||||
it('resolveModel creates an OpenAI-compatible model for LM Studio models', () => {
|
||||
registry.setLmstudioEnabled(true);
|
||||
registry.registerLmstudioModel('lmstudio-community/Meta-Llama-3-8B');
|
||||
|
||||
const model = registry.resolveModel('lmstudio-community/Meta-Llama-3-8B');
|
||||
expect(model).toBeDefined();
|
||||
expect(model.modelId).toBe('lmstudio-community/Meta-Llama-3-8B');
|
||||
});
|
||||
|
||||
it('resolveModel throws when LM Studio is disabled', () => {
|
||||
registry.registerLmstudioModel('lmstudio-community/Meta-Llama-3-8B');
|
||||
expect(() => registry.resolveModel('lmstudio-community/Meta-Llama-3-8B')).toThrow(/not configured/i);
|
||||
});
|
||||
|
||||
// ---- LM Studio model registration ----
|
||||
|
||||
it('tracks registered LM Studio model IDs', () => {
|
||||
expect(registry.isLmstudioModel('some-model')).toBe(false);
|
||||
registry.registerLmstudioModel('some-model');
|
||||
expect(registry.isLmstudioModel('some-model')).toBe(true);
|
||||
});
|
||||
|
||||
it('clearLmstudioModels removes all registered models', () => {
|
||||
registry.registerLmstudioModel('model-a');
|
||||
registry.registerLmstudioModel('model-b');
|
||||
registry.clearLmstudioModels();
|
||||
expect(registry.isLmstudioModel('model-a')).toBe(false);
|
||||
expect(registry.isLmstudioModel('model-b')).toBe(false);
|
||||
});
|
||||
|
||||
// ---- detectModelProvider ----
|
||||
|
||||
it('detectModelProvider returns "lmstudio" for registered LM Studio models', () => {
|
||||
registry.registerLmstudioModel('some-local-model');
|
||||
expect(registry.detectModelProvider('some-local-model')).toBe('lmstudio');
|
||||
});
|
||||
|
||||
it('detectModelProvider returns "ollama" for registered Ollama models (not lmstudio)', () => {
|
||||
registry.registerOllamaModel('llama3:latest');
|
||||
registry.registerLmstudioModel('some-model');
|
||||
expect(registry.detectModelProvider('llama3:latest')).toBe('ollama');
|
||||
expect(registry.detectModelProvider('some-model')).toBe('lmstudio');
|
||||
});
|
||||
|
||||
// ---- fetchLmstudioModels ----
|
||||
|
||||
it('fetchLmstudioModels calls the LM Studio models endpoint', async () => {
|
||||
const mockFetch = vi.fn().mockResolvedValue({
|
||||
ok: true,
|
||||
json: async () => ({
|
||||
data: [
|
||||
{ id: 'lmstudio-community/Meta-Llama-3-8B' },
|
||||
{ id: 'TheBloke/Mistral-7B-v0.1-GGUF' },
|
||||
],
|
||||
}),
|
||||
});
|
||||
const originalFetch = globalThis.fetch;
|
||||
globalThis.fetch = mockFetch;
|
||||
|
||||
try {
|
||||
const models = await registry.fetchLmstudioModels();
|
||||
expect(mockFetch).toHaveBeenCalledWith(
|
||||
LMSTUDIO_MODELS_URL,
|
||||
expect.objectContaining({ method: 'GET', signal: expect.any(AbortSignal) }),
|
||||
);
|
||||
expect(models).toHaveLength(2);
|
||||
expect(models[0]).toMatchObject({ id: 'lmstudio-community/Meta-Llama-3-8B', provider: 'lmstudio' });
|
||||
expect(models[1]).toMatchObject({ id: 'TheBloke/Mistral-7B-v0.1-GGUF', provider: 'lmstudio' });
|
||||
} finally {
|
||||
globalThis.fetch = originalFetch;
|
||||
}
|
||||
});
|
||||
|
||||
it('fetchLmstudioModels returns empty array on network error', async () => {
|
||||
const mockFetch = vi.fn().mockRejectedValue(new Error('ECONNREFUSED'));
|
||||
const originalFetch = globalThis.fetch;
|
||||
globalThis.fetch = mockFetch;
|
||||
|
||||
try {
|
||||
const models = await registry.fetchLmstudioModels();
|
||||
expect(models).toEqual([]);
|
||||
} finally {
|
||||
globalThis.fetch = originalFetch;
|
||||
}
|
||||
});
|
||||
|
||||
it('fetchLmstudioModels registers returned models', async () => {
|
||||
const mockFetch = vi.fn().mockResolvedValue({
|
||||
ok: true,
|
||||
json: async () => ({
|
||||
data: [{ id: 'my-local-model' }],
|
||||
}),
|
||||
});
|
||||
const originalFetch = globalThis.fetch;
|
||||
globalThis.fetch = mockFetch;
|
||||
|
||||
try {
|
||||
await registry.fetchLmstudioModels();
|
||||
expect(registry.isLmstudioModel('my-local-model')).toBe(true);
|
||||
} finally {
|
||||
globalThis.fetch = originalFetch;
|
||||
}
|
||||
});
|
||||
|
||||
// ---- getAvailableModels includes LM Studio when enabled ----
|
||||
|
||||
it('getAvailableModels includes LM Studio models when enabled', async () => {
|
||||
registry.setLmstudioEnabled(true);
|
||||
|
||||
const mockFetch = vi.fn().mockResolvedValue({
|
||||
ok: true,
|
||||
json: async () => ({
|
||||
data: [{ id: 'my-local-model' }],
|
||||
}),
|
||||
});
|
||||
const originalFetch = globalThis.fetch;
|
||||
globalThis.fetch = mockFetch;
|
||||
|
||||
try {
|
||||
const models = await registry.getAvailableModels();
|
||||
const lmModels = models.filter(m => m.provider === 'lmstudio');
|
||||
expect(lmModels.length).toBeGreaterThanOrEqual(1);
|
||||
expect(lmModels[0].id).toBe('my-local-model');
|
||||
} finally {
|
||||
globalThis.fetch = originalFetch;
|
||||
}
|
||||
});
|
||||
|
||||
it('getAvailableModels excludes LM Studio models when disabled', async () => {
|
||||
registry.setLmstudioEnabled(false);
|
||||
|
||||
const mockFetch = vi.fn().mockResolvedValue({
|
||||
ok: true,
|
||||
json: async () => ({ data: [{ id: 'my-local-model' }] }),
|
||||
});
|
||||
const originalFetch = globalThis.fetch;
|
||||
globalThis.fetch = mockFetch;
|
||||
|
||||
try {
|
||||
const models = await registry.getAvailableModels();
|
||||
const lmModels = models.filter(m => m.provider === 'lmstudio');
|
||||
expect(lmModels).toHaveLength(0);
|
||||
} finally {
|
||||
globalThis.fetch = originalFetch;
|
||||
}
|
||||
});
|
||||
|
||||
// ---- LM Studio model capability overrides ----
|
||||
|
||||
describe('model capability overrides', () => {
|
||||
it('returns default capabilities (tools=false, vision=false) for unknown model', () => {
|
||||
const caps = registry.getLmstudioModelCapabilities('unknown-model');
|
||||
expect(caps).toEqual({ tools: false, vision: false });
|
||||
});
|
||||
|
||||
it('stores and retrieves capability overrides for a model', () => {
|
||||
registry.setLmstudioModelCapabilities('my-model', { tools: true, vision: false });
|
||||
expect(registry.getLmstudioModelCapabilities('my-model')).toEqual({ tools: true, vision: false });
|
||||
});
|
||||
|
||||
it('stores vision capability override', () => {
|
||||
registry.setLmstudioModelCapabilities('vision-model', { tools: false, vision: true });
|
||||
expect(registry.getLmstudioModelCapabilities('vision-model')).toEqual({ tools: false, vision: true });
|
||||
});
|
||||
|
||||
it('supports both capabilities enabled', () => {
|
||||
registry.setLmstudioModelCapabilities('full-model', { tools: true, vision: true });
|
||||
expect(registry.getLmstudioModelCapabilities('full-model')).toEqual({ tools: true, vision: true });
|
||||
});
|
||||
|
||||
it('getAllLmstudioModelCapabilities returns all stored overrides', () => {
|
||||
registry.setLmstudioModelCapabilities('model-a', { tools: true, vision: false });
|
||||
registry.setLmstudioModelCapabilities('model-b', { tools: false, vision: true });
|
||||
const all = registry.getAllLmstudioModelCapabilities();
|
||||
expect(all).toEqual({
|
||||
'model-a': { tools: true, vision: false },
|
||||
'model-b': { tools: false, vision: true },
|
||||
});
|
||||
});
|
||||
|
||||
it('getAllLmstudioModelCapabilities returns empty object when no overrides', () => {
|
||||
expect(registry.getAllLmstudioModelCapabilities()).toEqual({});
|
||||
});
|
||||
|
||||
it('loadLmstudioModelCapabilities restores from serialized JSON', () => {
|
||||
const data = { 'my-model': { tools: true, vision: false } };
|
||||
registry.loadLmstudioModelCapabilities(data);
|
||||
expect(registry.getLmstudioModelCapabilities('my-model')).toEqual({ tools: true, vision: false });
|
||||
});
|
||||
|
||||
it('lmstudioModelSupportsTools returns false by default', () => {
|
||||
expect(registry.lmstudioModelSupportsTools('unknown')).toBe(false);
|
||||
});
|
||||
|
||||
it('lmstudioModelSupportsTools returns true when override is set', () => {
|
||||
registry.setLmstudioModelCapabilities('my-model', { tools: true, vision: false });
|
||||
expect(registry.lmstudioModelSupportsTools('my-model')).toBe(true);
|
||||
});
|
||||
|
||||
it('lmstudioModelSupportsVision returns false by default', () => {
|
||||
expect(registry.lmstudioModelSupportsVision('unknown')).toBe(false);
|
||||
});
|
||||
|
||||
it('lmstudioModelSupportsVision returns true when override is set', () => {
|
||||
registry.setLmstudioModelCapabilities('vision-model', { tools: false, vision: true });
|
||||
expect(registry.lmstudioModelSupportsVision('vision-model')).toBe(true);
|
||||
});
|
||||
|
||||
it('fetchLmstudioModels applies vision overrides to returned models', async () => {
|
||||
registry.setLmstudioModelCapabilities('vision-model', { tools: false, vision: true });
|
||||
|
||||
const mockFetch = vi.fn().mockResolvedValue({
|
||||
ok: true,
|
||||
json: async () => ({
|
||||
data: [
|
||||
{ id: 'text-model' },
|
||||
{ id: 'vision-model' },
|
||||
],
|
||||
}),
|
||||
});
|
||||
const originalFetch = globalThis.fetch;
|
||||
globalThis.fetch = mockFetch;
|
||||
|
||||
try {
|
||||
const models = await registry.fetchLmstudioModels();
|
||||
expect(models).toHaveLength(2);
|
||||
expect(models.find(m => m.id === 'text-model')?.vision).toBe(false);
|
||||
expect(models.find(m => m.id === 'vision-model')?.vision).toBe(true);
|
||||
} finally {
|
||||
globalThis.fetch = originalFetch;
|
||||
}
|
||||
});
|
||||
});
|
||||
});
|
||||
191
tests/engine/offline-mode.test.ts
Normal file
191
tests/engine/offline-mode.test.ts
Normal file
@@ -0,0 +1,191 @@
|
||||
/**
|
||||
* Offline / airplane mode tests.
|
||||
*
|
||||
* Verifies ProviderRegistry offline mode behavior:
|
||||
* - toggles on/off
|
||||
* - restricts model resolution to local providers only
|
||||
* - restricts available models to local only
|
||||
* - isReady reflects local provider state
|
||||
* - getProviderStatus includes offline state
|
||||
*/
|
||||
|
||||
import { describe, it, expect, vi, beforeEach } from 'vitest';
|
||||
import { ProviderRegistry } from '../../src/main/engine/ai/providers';
|
||||
|
||||
describe('ProviderRegistry offline mode', () => {
|
||||
let registry: ProviderRegistry;
|
||||
|
||||
beforeEach(() => {
|
||||
registry = new ProviderRegistry();
|
||||
});
|
||||
|
||||
// ---------- toggle ----------
|
||||
|
||||
it('starts with offline mode disabled', () => {
|
||||
expect(registry.isOfflineMode()).toBe(false);
|
||||
});
|
||||
|
||||
it('can enable and disable offline mode', () => {
|
||||
registry.setOfflineMode(true);
|
||||
expect(registry.isOfflineMode()).toBe(true);
|
||||
registry.setOfflineMode(false);
|
||||
expect(registry.isOfflineMode()).toBe(false);
|
||||
});
|
||||
|
||||
// ---------- isReady in offline mode ----------
|
||||
|
||||
it('isReady returns false in offline mode when no local provider enabled', () => {
|
||||
registry.setOpencodeKey('test-key');
|
||||
registry.setOfflineMode(true);
|
||||
expect(registry.isReady()).toBe(false);
|
||||
});
|
||||
|
||||
it('isReady returns true in offline mode when Ollama enabled', () => {
|
||||
registry.setOfflineMode(true);
|
||||
registry.setOllamaEnabled(true);
|
||||
expect(registry.isReady()).toBe(true);
|
||||
});
|
||||
|
||||
it('isReady returns true in offline mode when LM Studio enabled', () => {
|
||||
registry.setOfflineMode(true);
|
||||
registry.setLmstudioEnabled(true);
|
||||
expect(registry.isReady()).toBe(true);
|
||||
});
|
||||
|
||||
// ---------- resolveModel in offline mode ----------
|
||||
|
||||
it('resolveModel throws for cloud models when offline', () => {
|
||||
registry.setOpencodeKey('test-key');
|
||||
registry.setOfflineMode(true);
|
||||
expect(() => registry.resolveModel('claude-sonnet-4')).toThrow('offline');
|
||||
});
|
||||
|
||||
it('resolveModel throws for mistral models when offline', () => {
|
||||
registry.setMistralKey('test-key');
|
||||
registry.setOfflineMode(true);
|
||||
expect(() => registry.resolveModel('mistral-large-latest')).toThrow('offline');
|
||||
});
|
||||
|
||||
it('resolveModel succeeds for Ollama model when offline and Ollama enabled', () => {
|
||||
registry.setOllamaEnabled(true);
|
||||
registry.registerOllamaModel('llama3');
|
||||
registry.setOfflineMode(true);
|
||||
const model = registry.resolveModel('llama3');
|
||||
expect(model).toBeDefined();
|
||||
});
|
||||
|
||||
it('resolveModel succeeds for LM Studio model when offline and LM Studio enabled', () => {
|
||||
registry.setLmstudioEnabled(true);
|
||||
registry.registerLmstudioModel('gemma-3-12b-it');
|
||||
registry.setOfflineMode(true);
|
||||
const model = registry.resolveModel('gemma-3-12b-it');
|
||||
expect(model).toBeDefined();
|
||||
});
|
||||
|
||||
// ---------- getAvailableModels filtering ----------
|
||||
|
||||
it('getAvailableModels returns only local models when offline', async () => {
|
||||
registry.setOpencodeKey('test-key');
|
||||
registry.setOllamaEnabled(true);
|
||||
registry.registerOllamaModel('llama3:latest');
|
||||
registry.setOfflineMode(true);
|
||||
|
||||
// No fetch should happen at all in offline mode
|
||||
const originalFetch = globalThis.fetch;
|
||||
globalThis.fetch = vi.fn().mockImplementation(async (url: string) => {
|
||||
throw new Error(`Unexpected fetch to ${url} in offline mode`);
|
||||
});
|
||||
|
||||
try {
|
||||
const models = await registry.getAvailableModels();
|
||||
// Only local model should appear
|
||||
expect(models.length).toBe(1);
|
||||
expect(models.every(m => m.provider === 'ollama' || m.provider === 'lmstudio')).toBe(true);
|
||||
// fetch should NOT have been called
|
||||
expect(globalThis.fetch).not.toHaveBeenCalled();
|
||||
} finally {
|
||||
globalThis.fetch = originalFetch;
|
||||
}
|
||||
});
|
||||
|
||||
// ---------- getKnownLocalModels ----------
|
||||
|
||||
it('getKnownLocalModels returns registered Ollama and LM Studio models', () => {
|
||||
registry.setOllamaEnabled(true);
|
||||
registry.setLmstudioEnabled(true);
|
||||
registry.registerOllamaModel('llama3:latest');
|
||||
registry.registerLmstudioModel('gemma-3-12b-it');
|
||||
|
||||
const models = registry.getKnownLocalModels();
|
||||
expect(models.length).toBe(2);
|
||||
expect(models.find(m => m.id === 'llama3:latest')?.provider).toBe('ollama');
|
||||
expect(models.find(m => m.id === 'gemma-3-12b-it')?.provider).toBe('lmstudio');
|
||||
});
|
||||
|
||||
it('getKnownLocalModels returns empty when no models registered', () => {
|
||||
const models = registry.getKnownLocalModels();
|
||||
expect(models.length).toBe(0);
|
||||
});
|
||||
|
||||
// ---------- getFirstKnownLocalModelId ----------
|
||||
|
||||
it('getFirstKnownLocalModelId returns first registered model', () => {
|
||||
registry.setOllamaEnabled(true);
|
||||
registry.registerOllamaModel('llama3:latest');
|
||||
registry.registerOllamaModel('mistral:latest');
|
||||
expect(registry.getFirstKnownLocalModelId()).toBe('llama3:latest');
|
||||
});
|
||||
|
||||
it('getFirstKnownLocalModelId returns null when no models registered', () => {
|
||||
expect(registry.getFirstKnownLocalModelId()).toBeNull();
|
||||
});
|
||||
|
||||
it('getFirstKnownLocalModelId falls back to LM Studio when no Ollama models', () => {
|
||||
registry.setLmstudioEnabled(true);
|
||||
registry.registerLmstudioModel('gemma-3-12b-it');
|
||||
expect(registry.getFirstKnownLocalModelId()).toBe('gemma-3-12b-it');
|
||||
});
|
||||
|
||||
// ---------- getProviderStatus includes offline ----------
|
||||
|
||||
it('getProviderStatus includes offlineMode field', () => {
|
||||
registry.setOfflineMode(true);
|
||||
const status = registry.getProviderStatus();
|
||||
expect(status.offlineMode).toBe(true);
|
||||
|
||||
registry.setOfflineMode(false);
|
||||
expect(registry.getProviderStatus().offlineMode).toBe(false);
|
||||
});
|
||||
|
||||
// ---------- isProviderKeySet in offline mode ----------
|
||||
|
||||
it('isProviderKeySet returns false for cloud providers when offline', () => {
|
||||
registry.setOpencodeKey('test-key');
|
||||
registry.setMistralKey('test-key');
|
||||
registry.setOfflineMode(true);
|
||||
expect(registry.isProviderKeySet('anthropic')).toBe(false);
|
||||
expect(registry.isProviderKeySet('openai')).toBe(false);
|
||||
expect(registry.isProviderKeySet('mistral')).toBe(false);
|
||||
});
|
||||
|
||||
it('isProviderKeySet returns true for local providers when offline and enabled', () => {
|
||||
registry.setOllamaEnabled(true);
|
||||
registry.setLmstudioEnabled(true);
|
||||
registry.setOfflineMode(true);
|
||||
expect(registry.isProviderKeySet('ollama')).toBe(true);
|
||||
expect(registry.isProviderKeySet('lmstudio')).toBe(true);
|
||||
});
|
||||
|
||||
// ---------- model cache invalidation on toggle ----------
|
||||
|
||||
it('invalidates model cache on offline mode toggle', () => {
|
||||
// Pre-populate cache
|
||||
registry.setOpencodeKey('test-key');
|
||||
registry.setOfflineMode(true);
|
||||
// Toggling should have invalidated
|
||||
expect(registry.isOfflineMode()).toBe(true);
|
||||
// Toggle back
|
||||
registry.setOfflineMode(false);
|
||||
expect(registry.isOfflineMode()).toBe(false);
|
||||
});
|
||||
});
|
||||
323
tests/engine/offline-model-fallback.test.ts
Normal file
323
tests/engine/offline-model-fallback.test.ts
Normal file
@@ -0,0 +1,323 @@
|
||||
/**
|
||||
* Offline model fallback tests.
|
||||
*
|
||||
* Verifies that OneShotTasks.analyzeMediaImage(), ChatService.sendMessage(),
|
||||
* and ChatService.generateConversationTitle() automatically fall back to
|
||||
* the configured offline model when airplane mode is active.
|
||||
*
|
||||
* Strategy: spy on resolveModel to capture which model ID is passed,
|
||||
* then let it throw to short-circuit the actual AI call — the engine's
|
||||
* try/catch returns { success: false } which is fine for our assertions.
|
||||
*/
|
||||
|
||||
import { describe, it, expect, vi, beforeEach } from 'vitest';
|
||||
import { OneShotTasks } from '../../src/main/engine/ai/tasks';
|
||||
import { ChatService } from '../../src/main/engine/ai/chat';
|
||||
import { ProviderRegistry } from '../../src/main/engine/ai/providers';
|
||||
|
||||
// Tiny valid 2x2 JPEG (base64) — avoids sharp "corrupt header" error
|
||||
// eslint-disable-next-line max-len
|
||||
const TINY_JPEG_B64 = '/9j/2wBDAAYEBQYFBAYGBQYHBwYIChAKCgkJChQODwwQFxQYGBcUFhYaHSUfGhsjHBYWICwgIyYnKSopGR8tMC0oMCUoKSj/2wBDAQcHBwoIChMKChMoGhYaKCgoKCgoKCgoKCgoKCgoKCgoKCgoKCgoKCgoKCgoKCgoKCgoKCgoKCgoKCgoKCgoKCj/wAARCAACAAIDASIAAhEBAxEB/8QAFAABAAAAAAAAAAAAAAAAAAAAAP/EABQQAQAAAAAAAAAAAAAAAAAAAAD/xAAUAQEAAAAAAAAAAAAAAAAAAAAA/8QAFBEBAAAAAAAAAAAAAAAAAAAAAP/aAAwDAQACEQMRAD8AAA//2Q==';
|
||||
|
||||
// ---------------------------------------------------------------------------
|
||||
// Shared mock helpers
|
||||
// ---------------------------------------------------------------------------
|
||||
|
||||
function createMockChatEngine(settings: Record<string, string | null> = {}) {
|
||||
return {
|
||||
getSetting: vi.fn(async (key: string) => settings[key] ?? null),
|
||||
getConversation: vi.fn(),
|
||||
addMessage: vi.fn(async (msg: unknown) => ({ id: 'msg-1', ...msg as Record<string, unknown> })),
|
||||
getDefaultSystemPrompt: vi.fn(async () => 'You are a helpful assistant'),
|
||||
updateConversation: vi.fn(),
|
||||
} as unknown as InstanceType<typeof import('../../src/main/engine/ChatEngine').ChatEngine>;
|
||||
}
|
||||
|
||||
function createMockMediaEngine() {
|
||||
return {
|
||||
getMedia: vi.fn(async () => ({
|
||||
id: 'media-1',
|
||||
mimeType: 'image/jpeg',
|
||||
filename: 'test.jpg',
|
||||
})),
|
||||
getThumbnailDataUrl: vi.fn(async () => `data:image/jpeg;base64,${TINY_JPEG_B64}`),
|
||||
} as unknown as InstanceType<typeof import('../../src/main/engine/MediaEngine').MediaEngine>;
|
||||
}
|
||||
|
||||
/** Mock the ModelCatalogEngine returned by ProviderRegistry */
|
||||
function mockModelCatalog(registry: ProviderRegistry): void {
|
||||
vi.spyOn(registry, 'getModelCatalogEngine').mockReturnValue({
|
||||
getContextWindow: vi.fn(async () => 8192),
|
||||
} as never);
|
||||
}
|
||||
|
||||
// ---------------------------------------------------------------------------
|
||||
// OneShotTasks — analyzeMediaImage offline fallback
|
||||
// ---------------------------------------------------------------------------
|
||||
|
||||
describe('OneShotTasks offline model fallback', () => {
|
||||
let registry: ProviderRegistry;
|
||||
|
||||
beforeEach(() => {
|
||||
registry = new ProviderRegistry();
|
||||
registry.setOllamaEnabled(true);
|
||||
registry.registerOllamaModel('llava:latest');
|
||||
});
|
||||
|
||||
it('analyzeMediaImage uses offline_image_analysis_model when airplane mode is on', async () => {
|
||||
registry.setOpencodeKey('test-key');
|
||||
registry.setOfflineMode(true);
|
||||
|
||||
const chatEngine = createMockChatEngine({
|
||||
chat_image_analysis_model: 'claude-sonnet-4-5', // cloud model
|
||||
offline_image_analysis_model: 'llava:latest', // local model
|
||||
});
|
||||
const tasks = new OneShotTasks(registry, chatEngine, createMockMediaEngine());
|
||||
|
||||
// resolveModel spy — let it throw to short-circuit the generateText call
|
||||
const resolveModelSpy = vi.spyOn(registry, 'resolveModel')
|
||||
.mockImplementation(() => { throw new Error('mock-stop'); });
|
||||
|
||||
await tasks.analyzeMediaImage('media-1', 'en');
|
||||
|
||||
expect(resolveModelSpy).toHaveBeenCalledWith('llava:latest');
|
||||
});
|
||||
|
||||
it('analyzeMediaImage auto-falls back to first local model when no offline model configured', async () => {
|
||||
registry.setOpencodeKey('test-key');
|
||||
registry.setOfflineMode(true);
|
||||
|
||||
const chatEngine = createMockChatEngine({
|
||||
chat_image_analysis_model: 'claude-sonnet-4-5',
|
||||
// No offline_image_analysis_model set — should auto-pick llava:latest
|
||||
});
|
||||
const tasks = new OneShotTasks(registry, chatEngine, createMockMediaEngine());
|
||||
|
||||
const resolveModelSpy = vi.spyOn(registry, 'resolveModel')
|
||||
.mockImplementation(() => { throw new Error('mock-stop'); });
|
||||
|
||||
await tasks.analyzeMediaImage('media-1', 'en');
|
||||
|
||||
expect(resolveModelSpy).toHaveBeenCalledWith('llava:latest');
|
||||
});
|
||||
|
||||
it('analyzeMediaImage returns error when offline with no local models at all', async () => {
|
||||
const emptyRegistry = new ProviderRegistry();
|
||||
emptyRegistry.setOpencodeKey('test-key');
|
||||
emptyRegistry.setOfflineMode(true);
|
||||
|
||||
const chatEngine = createMockChatEngine({
|
||||
chat_image_analysis_model: 'claude-sonnet-4-5',
|
||||
});
|
||||
const tasks = new OneShotTasks(emptyRegistry, chatEngine, createMockMediaEngine());
|
||||
|
||||
const result = await tasks.analyzeMediaImage('media-1', 'en');
|
||||
|
||||
expect(result.success).toBe(false);
|
||||
expect(result.error).toContain('offline');
|
||||
});
|
||||
|
||||
it('analyzeMediaImage uses default model when NOT offline', async () => {
|
||||
registry.setOpencodeKey('test-key');
|
||||
// Offline mode is OFF
|
||||
|
||||
const chatEngine = createMockChatEngine({
|
||||
chat_image_analysis_model: 'claude-sonnet-4-5',
|
||||
offline_image_analysis_model: 'llava:latest',
|
||||
});
|
||||
const tasks = new OneShotTasks(registry, chatEngine, createMockMediaEngine());
|
||||
|
||||
const resolveModelSpy = vi.spyOn(registry, 'resolveModel')
|
||||
.mockImplementation(() => { throw new Error('mock-stop'); });
|
||||
|
||||
await tasks.analyzeMediaImage('media-1', 'en');
|
||||
|
||||
// Should use the regular model, NOT the offline one
|
||||
expect(resolveModelSpy).toHaveBeenCalledWith('claude-sonnet-4-5');
|
||||
});
|
||||
});
|
||||
|
||||
// ---------------------------------------------------------------------------
|
||||
// ChatService — sendMessage offline fallback
|
||||
// ---------------------------------------------------------------------------
|
||||
|
||||
describe('ChatService offline model fallback', () => {
|
||||
let registry: ProviderRegistry;
|
||||
|
||||
beforeEach(() => {
|
||||
registry = new ProviderRegistry();
|
||||
registry.setOllamaEnabled(true);
|
||||
registry.registerOllamaModel('llama3:latest');
|
||||
});
|
||||
|
||||
it('sendMessage swaps cloud model for offline_chat_model when airplane mode is on', async () => {
|
||||
registry.setOpencodeKey('test-key');
|
||||
registry.setOfflineMode(true);
|
||||
|
||||
const chatEngine = createMockChatEngine({
|
||||
offline_chat_model: 'llama3:latest',
|
||||
});
|
||||
chatEngine.getConversation = vi.fn(async () => ({
|
||||
id: 'conv-1',
|
||||
title: 'Test',
|
||||
model: 'claude-sonnet-4', // cloud model on conversation
|
||||
createdAt: new Date(),
|
||||
messages: [],
|
||||
}));
|
||||
|
||||
const service = new ChatService(
|
||||
chatEngine,
|
||||
registry,
|
||||
{} as never,
|
||||
() => null,
|
||||
);
|
||||
mockModelCatalog(registry);
|
||||
|
||||
// resolveModel spy — let it throw to short-circuit
|
||||
const resolveModelSpy = vi.spyOn(registry, 'resolveModel')
|
||||
.mockImplementation(() => { throw new Error('mock-stop'); });
|
||||
|
||||
const result = await service.sendMessage('conv-1', 'Hello', {});
|
||||
|
||||
// Model swap should have happened before resolveModel was called
|
||||
expect(resolveModelSpy).toHaveBeenCalledWith('llama3:latest');
|
||||
expect(result.success).toBe(false); // throws mock-stop in try/catch
|
||||
});
|
||||
|
||||
it('sendMessage auto-falls back to first local model when no offline_chat_model configured', async () => {
|
||||
registry.setOpencodeKey('test-key');
|
||||
registry.setOfflineMode(true);
|
||||
|
||||
const chatEngine = createMockChatEngine({
|
||||
// No offline_chat_model — should auto-pick llama3:latest
|
||||
});
|
||||
chatEngine.getConversation = vi.fn(async () => ({
|
||||
id: 'conv-1',
|
||||
title: 'Test',
|
||||
model: 'claude-sonnet-4',
|
||||
createdAt: new Date(),
|
||||
messages: [],
|
||||
}));
|
||||
|
||||
const service = new ChatService(
|
||||
chatEngine,
|
||||
registry,
|
||||
{} as never,
|
||||
() => null,
|
||||
);
|
||||
mockModelCatalog(registry);
|
||||
|
||||
const resolveModelSpy = vi.spyOn(registry, 'resolveModel')
|
||||
.mockImplementation(() => { throw new Error('mock-stop'); });
|
||||
|
||||
await service.sendMessage('conv-1', 'Hello', {});
|
||||
|
||||
expect(resolveModelSpy).toHaveBeenCalledWith('llama3:latest');
|
||||
});
|
||||
|
||||
it('sendMessage returns error when offline with no local models at all', async () => {
|
||||
const emptyRegistry = new ProviderRegistry();
|
||||
emptyRegistry.setOpencodeKey('test-key');
|
||||
emptyRegistry.setOfflineMode(true);
|
||||
|
||||
const chatEngine = createMockChatEngine({});
|
||||
chatEngine.getConversation = vi.fn(async () => ({
|
||||
id: 'conv-1',
|
||||
title: 'Test',
|
||||
model: 'claude-sonnet-4',
|
||||
createdAt: new Date(),
|
||||
messages: [],
|
||||
}));
|
||||
|
||||
const service = new ChatService(
|
||||
chatEngine,
|
||||
emptyRegistry,
|
||||
{} as never,
|
||||
() => null,
|
||||
);
|
||||
|
||||
const result = await service.sendMessage('conv-1', 'Hello', {});
|
||||
|
||||
// With no local providers enabled, isReady() returns false
|
||||
expect(result.success).toBe(false);
|
||||
});
|
||||
|
||||
it('sendMessage keeps local model when conversation already uses local model and offline', async () => {
|
||||
registry.setOfflineMode(true);
|
||||
|
||||
const chatEngine = createMockChatEngine({});
|
||||
chatEngine.getConversation = vi.fn(async () => ({
|
||||
id: 'conv-1',
|
||||
title: 'Test',
|
||||
model: 'llama3:latest', // already a local model
|
||||
createdAt: new Date(),
|
||||
messages: [],
|
||||
}));
|
||||
|
||||
const service = new ChatService(
|
||||
chatEngine,
|
||||
registry,
|
||||
{} as never,
|
||||
() => null,
|
||||
);
|
||||
mockModelCatalog(registry);
|
||||
|
||||
const resolveModelSpy = vi.spyOn(registry, 'resolveModel')
|
||||
.mockImplementation(() => { throw new Error('mock-stop'); });
|
||||
|
||||
await service.sendMessage('conv-1', 'Hello', {});
|
||||
|
||||
// Should use the local model directly, no swap needed
|
||||
expect(resolveModelSpy).toHaveBeenCalledWith('llama3:latest');
|
||||
});
|
||||
});
|
||||
|
||||
// ---------------------------------------------------------------------------
|
||||
// ChatService — generateConversationTitle offline fallback
|
||||
// ---------------------------------------------------------------------------
|
||||
|
||||
describe('ChatService title generation offline fallback', () => {
|
||||
let registry: ProviderRegistry;
|
||||
|
||||
beforeEach(() => {
|
||||
registry = new ProviderRegistry();
|
||||
registry.setOllamaEnabled(true);
|
||||
registry.registerOllamaModel('llama3:latest');
|
||||
});
|
||||
|
||||
it('title generation silently skips when offline with no offline_title_model', async () => {
|
||||
registry.setOpencodeKey('test-key');
|
||||
registry.setOfflineMode(true);
|
||||
|
||||
const chatEngine = createMockChatEngine({
|
||||
chat_title_model: 'claude-haiku-4-5', // cloud model
|
||||
// No offline_title_model set
|
||||
});
|
||||
chatEngine.getConversation = vi.fn(async () => ({
|
||||
id: 'conv-1',
|
||||
title: 'Test',
|
||||
model: 'llama3:latest', // local model for chat
|
||||
createdAt: new Date(),
|
||||
messages: [],
|
||||
}));
|
||||
|
||||
const service = new ChatService(
|
||||
chatEngine,
|
||||
registry,
|
||||
{} as never,
|
||||
() => null,
|
||||
);
|
||||
mockModelCatalog(registry);
|
||||
|
||||
// resolveModel used for chat model; title generation should be skipped
|
||||
const resolveModelSpy = vi.spyOn(registry, 'resolveModel')
|
||||
.mockImplementation(() => { throw new Error('mock-stop'); });
|
||||
|
||||
await service.sendMessage('conv-1', 'Hello', {});
|
||||
|
||||
// resolveModel should only be called once — for the chat model, not for title
|
||||
// (title generation is skipped silently when offline with no offline_title_model)
|
||||
const calls = resolveModelSpy.mock.calls;
|
||||
expect(calls.some(c => c[0] === 'claude-haiku-4-5')).toBe(false);
|
||||
});
|
||||
});
|
||||
@@ -333,6 +333,11 @@ vi.mock('fs/promises', () => ({
|
||||
unlink: vi.fn(),
|
||||
}));
|
||||
|
||||
let mockOfflineMode = false;
|
||||
vi.mock('../../src/main/ipc/chatHandlers', () => ({
|
||||
isOfflineModeActive: vi.fn(() => mockOfflineMode),
|
||||
}));
|
||||
|
||||
// Helper to invoke a registered handler
|
||||
async function invokeHandler(channel: string, ...args: any[]): Promise<any> {
|
||||
const handler = registeredHandlers.get(channel);
|
||||
@@ -383,6 +388,7 @@ describe('IPC Handlers', () => {
|
||||
registeredHandlers.clear();
|
||||
mockGeneratedFileHashStore.clear();
|
||||
resetMockCounters();
|
||||
mockOfflineMode = false;
|
||||
|
||||
// Create a real BlogGenerationEngine with mock engines for blog handler tests
|
||||
const { BlogGenerationEngine } = await import('../../src/main/engine/BlogGenerationEngine');
|
||||
@@ -513,6 +519,64 @@ describe('IPC Handlers', () => {
|
||||
behind: 1,
|
||||
});
|
||||
});
|
||||
|
||||
it('should return zeroed state when offline mode is active', async () => {
|
||||
mockOfflineMode = true;
|
||||
|
||||
const result = await invokeHandler('git:remoteState', '/repo');
|
||||
|
||||
expect(mockGitEngine.getRemoteState).not.toHaveBeenCalled();
|
||||
expect(result).toEqual({ ahead: 0, behind: 0 });
|
||||
});
|
||||
});
|
||||
|
||||
describe('offline mode blocks network git operations', () => {
|
||||
it('should block git:fetch when offline mode is active', async () => {
|
||||
mockOfflineMode = true;
|
||||
|
||||
const result = await invokeHandler('git:fetch', '/repo');
|
||||
|
||||
expect(mockGitEngine.fetch).not.toHaveBeenCalled();
|
||||
expect(result).toEqual({ success: false, code: 'offline' });
|
||||
});
|
||||
|
||||
it('should block git:pull when offline mode is active', async () => {
|
||||
mockOfflineMode = true;
|
||||
|
||||
const result = await invokeHandler('git:pull', '/repo');
|
||||
|
||||
expect(mockGitEngine.pull).not.toHaveBeenCalled();
|
||||
expect(result).toEqual({ success: false, code: 'offline' });
|
||||
});
|
||||
|
||||
it('should block git:push when offline mode is active', async () => {
|
||||
mockOfflineMode = true;
|
||||
|
||||
const result = await invokeHandler('git:push', '/repo');
|
||||
|
||||
expect(mockGitEngine.push).not.toHaveBeenCalled();
|
||||
expect(result).toEqual({ success: false, code: 'offline' });
|
||||
});
|
||||
|
||||
it('should allow git:fetch when offline mode is inactive', async () => {
|
||||
mockOfflineMode = false;
|
||||
mockGitEngine.fetch.mockResolvedValue({ success: true });
|
||||
|
||||
const result = await invokeHandler('git:fetch', '/repo');
|
||||
|
||||
expect(mockGitEngine.fetch).toHaveBeenCalledWith('/repo');
|
||||
expect(result).toEqual({ success: true });
|
||||
});
|
||||
|
||||
it('should allow git:commitAll regardless of offline mode', async () => {
|
||||
mockOfflineMode = true;
|
||||
mockGitEngine.commitAll.mockResolvedValue({ success: true });
|
||||
|
||||
const result = await invokeHandler('git:commitAll', '/repo', 'test commit');
|
||||
|
||||
expect(mockGitEngine.commitAll).toHaveBeenCalledWith('/repo', 'test commit');
|
||||
expect(result).toEqual({ success: true });
|
||||
});
|
||||
});
|
||||
|
||||
describe('git:diffContent', () => {
|
||||
@@ -738,6 +802,21 @@ describe('IPC Handlers', () => {
|
||||
});
|
||||
});
|
||||
|
||||
// ============ Publish Handlers ============
|
||||
describe('Publish Handlers', () => {
|
||||
describe('publish:uploadSite offline guard', () => {
|
||||
it('should throw when offline mode is active', async () => {
|
||||
mockOfflineMode = true;
|
||||
|
||||
await expect(invokeHandler('publish:uploadSite', {
|
||||
sshHost: 'example.com',
|
||||
sshUser: 'deploy',
|
||||
sshRemotePath: '/var/www',
|
||||
})).rejects.toThrow('Airplane mode');
|
||||
});
|
||||
});
|
||||
});
|
||||
|
||||
// ============ Project Handlers ============
|
||||
describe('Project Handlers', () => {
|
||||
describe('projects:create', () => {
|
||||
|
||||
@@ -1098,4 +1098,30 @@ describe('GitSidebar', () => {
|
||||
vi.useRealTimers();
|
||||
}
|
||||
});
|
||||
|
||||
it.each(['fetch', 'pull', 'push'] as const)('shows error modal instead of inline error when %s returns offline code', async (action) => {
|
||||
(window as any).electronAPI.git.getRepoState = vi.fn().mockResolvedValue({
|
||||
isRepo: true,
|
||||
rootPath: '/repo/path',
|
||||
currentBranch: 'main',
|
||||
hasRemote: true,
|
||||
});
|
||||
(window as any).electronAPI.git[action] = vi.fn().mockResolvedValue({ success: false, code: 'offline' });
|
||||
|
||||
render(<GitSidebar />);
|
||||
|
||||
const button = await screen.findByRole('button', { name: new RegExp(`^${action}$`, 'i') });
|
||||
|
||||
await act(async () => {
|
||||
fireEvent.click(button);
|
||||
});
|
||||
|
||||
// Should set errorModal in the store
|
||||
const store = getStore();
|
||||
expect(store.errorModal).not.toBeNull();
|
||||
expect(store.errorModal!.message).toBe('This action is blocked while airplane mode is active.');
|
||||
|
||||
// Should NOT show inline error in the sidebar
|
||||
expect(screen.queryByText('This action is blocked while airplane mode is active.')).not.toBeInTheDocument();
|
||||
});
|
||||
});
|
||||
|
||||
Reference in New Issue
Block a user