diff --git a/web-app/src/hooks/__tests__/useModelSources.test.ts b/web-app/src/hooks/__tests__/useModelSources.test.ts index a14a8107f..d010f7f8a 100644 --- a/web-app/src/hooks/__tests__/useModelSources.test.ts +++ b/web-app/src/hooks/__tests__/useModelSources.test.ts @@ -49,7 +49,6 @@ describe('useModelSources', () => { expect(result.current.error).toBe(null) expect(result.current.loading).toBe(false) expect(typeof result.current.fetchSources).toBe('function') - expect(typeof result.current.addSource).toBe('function') }) describe('fetchSources', () => { @@ -225,153 +224,6 @@ describe('useModelSources', () => { }) }) - describe('addSource', () => { - it('should add a new source to the store', () => { - const { result } = renderHook(() => useModelSources()) - - const testModel: CatalogModel = { - model_name: 'test-model', - description: 'Test model description', - developer: 'test-developer', - downloads: 100, - num_quants: 2, - quants: [ - { - model_id: 'test-model-q4', - path: 'https://example.com/test-model-q4.gguf', - file_size: '2.0 GB', - }, - ], - created_at: '2023-01-01T00:00:00Z', - } - - act(() => { - result.current.addSource(testModel) - }) - - expect(result.current.sources).toHaveLength(1) - expect(result.current.sources[0]).toEqual(testModel) - }) - - it('should replace existing source with same model_name', () => { - const { result } = renderHook(() => useModelSources()) - - const originalModel: CatalogModel = { - model_name: 'duplicate-model', - description: 'Original description', - developer: 'original-developer', - downloads: 50, - num_quants: 1, - quants: [], - created_at: '2023-01-01T00:00:00Z', - } - - const updatedModel: CatalogModel = { - model_name: 'duplicate-model', - description: 'Updated description', - developer: 'updated-developer', - downloads: 150, - num_quants: 2, - quants: [ - { - model_id: 'duplicate-model-q4', - path: 'https://example.com/duplicate-model-q4.gguf', - file_size: '3.0 GB', - }, - ], - created_at: '2023-02-01T00:00:00Z', - } - - act(() => { - result.current.addSource(originalModel) - }) - - expect(result.current.sources).toHaveLength(1) - - act(() => { - result.current.addSource(updatedModel) - }) - - expect(result.current.sources).toHaveLength(1) - expect(result.current.sources[0]).toEqual(updatedModel) - }) - - it('should handle multiple different sources', () => { - const { result } = renderHook(() => useModelSources()) - - const model1: CatalogModel = { - model_name: 'model-1', - description: 'First model', - developer: 'developer-1', - downloads: 100, - num_quants: 1, - quants: [], - created_at: '2023-01-01T00:00:00Z', - } - - const model2: CatalogModel = { - model_name: 'model-2', - description: 'Second model', - developer: 'developer-2', - downloads: 200, - num_quants: 1, - quants: [], - created_at: '2023-01-02T00:00:00Z', - } - - act(() => { - result.current.addSource(model1) - }) - - act(() => { - result.current.addSource(model2) - }) - - expect(result.current.sources).toHaveLength(2) - expect(result.current.sources).toContainEqual(model1) - expect(result.current.sources).toContainEqual(model2) - }) - - it('should handle CatalogModel with complete quants data', () => { - const { result } = renderHook(() => useModelSources()) - - const modelWithQuants: CatalogModel = { - model_name: 'model-with-quants', - description: 'Model with quantizations', - developer: 'quant-developer', - downloads: 500, - num_quants: 3, - quants: [ - { - model_id: 'model-q4_k_m', - path: 'https://example.com/model-q4_k_m.gguf', - file_size: '2.0 GB', - }, - { - model_id: 'model-q8_0', - path: 'https://example.com/model-q8_0.gguf', - file_size: '4.0 GB', - }, - { - model_id: 'model-f16', - path: 'https://example.com/model-f16.gguf', - file_size: '8.0 GB', - }, - ], - created_at: '2023-01-01T00:00:00Z', - readme: 'https://example.com/readme.md', - } - - act(() => { - result.current.addSource(modelWithQuants) - }) - - expect(result.current.sources).toHaveLength(1) - expect(result.current.sources[0]).toEqual(modelWithQuants) - expect(result.current.sources[0].quants).toHaveLength(3) - }) - }) - describe('state management', () => { it('should maintain state across multiple hook instances', () => { const { result: result1 } = renderHook(() => useModelSources())