test

Author: pashidlos

src/compare/compare.service.spec.ts

@@ -1,9 +1,12 @@
 import { Test, TestingModule } from '@nestjs/testing';
+import { ConfigService } from '@nestjs/config';
 import { PrismaService } from '../prisma/prisma.service';
 import { CompareService } from './compare.service';
 import { LookSameService } from './libs/looks-same/looks-same.service';
 import { OdiffService } from './libs/odiff/odiff.service';
 import { PixelmatchService } from './libs/pixelmatch/pixelmatch.service';
+import { VlmService } from './libs/vlm/vlm.service';
+import { OllamaService } from './libs/vlm/ollama.service';
 import { StaticModule } from '../static/static.module';
 import { ImageComparison } from '@prisma/client';
 import * as utils from '../static/utils';
@@ -16,7 +19,21 @@ describe('CompareService', () => {
 
   beforeEach(async () => {
     const module: TestingModule = await Test.createTestingModule({
-      providers: [CompareService, OdiffService, PixelmatchService, LookSameService, PrismaService],
+      providers: [
+        CompareService,
+        OdiffService,
+        PixelmatchService,
+        LookSameService,
+        VlmService,
+        OllamaService,
+        PrismaService,
+        {
+          provide: ConfigService,
+          useValue: {
+            getOrThrow: jest.fn().mockReturnValue('http://localhost:11434'),
+          },
+        },
+      ],
       imports: [StaticModule],
     }).compile();
 

src/compare/libs/vlm/vlm.service.spec.ts

@@ -139,12 +139,21 @@ describe('VlmService', () => {
     expect(result.diffName).toBeNull();
   });
 
-  it('should parse config with defaults for invalid input', async () => {
+  it.each([
+    ['empty string', '', DEFAULT_CONFIG],
+    ['invalid JSON', 'invalid', DEFAULT_CONFIG],
+    ['partial config', '{"model":"llava:7b"}', { model: 'llava:7b' }],
+    [
+      'full config',
+      '{"model":"llava:13b","prompt":"Custom prompt","temperature":0.2}',
+      {
+        model: 'llava:13b',
+        prompt: 'Custom prompt',
+        temperature: 0.2,
+      },
+    ],
+  ])('should parse config: %s', async (_, configJson, expected) => {
     const service = await initService({});
-
-    expect(service.parseConfig('')).toStrictEqual(DEFAULT_CONFIG);
-    expect(service.parseConfig('invalid')).toStrictEqual(DEFAULT_CONFIG);
-    expect(service.parseConfig('{"model":"llava:7b"}').model).toBe('llava:7b');
-    expect(service.parseConfig('{"model":"llava:7b"}').prompt).toBe(DEFAULT_CONFIG.prompt);
+    expect(service.parseConfig(configJson)).toEqual(expected);
   });
 });