Function TestFetchDisallowedServesFromCache

pkg/modelsdev/store_test.go:100–134

TestFetchDisallowedServesFromCache checks the cache-only path: a provider the predicate rejects but that IS present in the on-disk cache resolves from the cache without any network call, and an absent one is a clean "not found".

(t *testing.T)

// predicate rejects but that IS present in the on-disk cache resolves from the
// cache without any network call, and an absent one is a clean "not found".
func TestFetchDisallowedServesFromCache(t *testing.T) {
	t.Parallel()

	cacheFile := filepath.Join(t.TempDir(), "models_dev.json")
	writeCache(t, cacheFile, Database{Providers: map[string]Provider{
		// Present in the catalog but not in the known-provider set below.
		"deepseek": {Models: map[string]Model{
			"deepseek-chat": {Name: "DeepSeek Chat", Limit: Limit{Context: 64000}},
		}},
	}})

	store, err := NewStore(
		WithCache(cacheFile),
		WithKnownProvider(func(p string) bool { return p == "openai" }),
	)
	require.NoError(t, err)

	// Expired context: proves resolution is cache-only (no network) for a
	// fetch-disallowed provider.
	ctx := expiredContext(t)

	m, err := store.GetModel(ctx, NewID("deepseek", "deepseek-chat"))
	require.NoError(t, err)
	assert.Equal(t, 64000, m.Limit.Context)

	// Repeated lookups stay consistent (served from the memoized snapshot).
	again, err := store.GetModel(ctx, NewID("deepseek", "deepseek-chat"))
	require.NoError(t, err)
	assert.Equal(t, m.Limit.Context, again.Limit.Context)

	// A provider absent from the cache is still a clean not-found, no fetch.
	_, err = store.GetModel(ctx, NewID("mistral_gateway", "x"))
	require.Error(t, err)
	assert.NotContains(t, err.Error(), "fetch from API")
}

// BenchmarkGetModelFetchDisallowed guards the hot path: repeatedly resolving a
// fetch-disallowed provider against a warm, sizable cache must not re-read and