diff --git a/tests-ui/tests/services/assetService.test.ts b/tests-ui/tests/services/assetService.test.ts index b6386bc2b..8cb91fae0 100644 --- a/tests-ui/tests/services/assetService.test.ts +++ b/tests-ui/tests/services/assetService.test.ts @@ -297,7 +297,7 @@ describe('assetService', () => { const result = await assetService.getAssetsByTag('models') expect(api.fetchApi).toHaveBeenCalledWith( - '/assets?include_tags=models&limit=500' + '/assets?include_tags=models&limit=500&include_public=true' ) expect(result).toEqual(testAssets) }) @@ -355,5 +355,29 @@ describe('assetService', () => { expect(result[0]).toHaveProperty('asset_hash', 'blake3:full123') expect(result[0]).toHaveProperty('user_metadata') }) + + it('should exclude public assets when includePublic is false', async () => { + const testAssets = [MOCK_ASSETS.checkpoints] + mockApiResponse(testAssets) + + const result = await assetService.getAssetsByTag('input', false) + + expect(api.fetchApi).toHaveBeenCalledWith( + '/assets?include_tags=input&limit=500&include_public=false' + ) + expect(result).toEqual(testAssets) + }) + + it('should include public assets when includePublic is true', async () => { + const testAssets = [MOCK_ASSETS.checkpoints, MOCK_ASSETS.loras] + mockApiResponse(testAssets) + + const result = await assetService.getAssetsByTag('models', true) + + expect(api.fetchApi).toHaveBeenCalledWith( + '/assets?include_tags=models&limit=500&include_public=true' + ) + expect(result).toEqual(testAssets) + }) }) })