Function TestSnapshotFallbackIsNotMemoized

pkg/modelsdev/snapshot_test.go:91–124

TestSnapshotFallbackIsNotMemoized guards against pinning the embedded fallback for the Store's lifetime: a transient fetch failure must degrade to the snapshot, but a later lookup (once the network recovers) must retry the fetch and serve the fresh catalog rather than the stale build-time snapshot.

(t *testing.T)

// the snapshot, but a later lookup (once the network recovers) must retry the
// fetch and serve the fresh catalog rather than the stale build-time snapshot.
func TestSnapshotFallbackIsNotMemoized(t *testing.T) {
	t.Parallel()

	cacheFile := filepath.Join(t.TempDir(), "models_dev.json")

	// First lookup: the network is down, so the fetch fails and we fall back
	// to the embedded snapshot without memoizing it.
	var firstFetched, secondFetched bool
	fetch := func(context.Context, string) (*Database, string, error) {
		if !firstFetched {
			firstFetched = true
			return nil, "", errors.New("fetch from API: network unreachable")
		}
		secondFetched = true
		return &Database{Providers: map[string]Provider{
			"openai": {Models: map[string]Model{
				"fresh-model": {Name: "Freshly Fetched", Limit: Limit{Context: 12345}},
			}},
		}}, "etag-1", nil
	}
	store, err := NewStore(WithCache(cacheFile), WithFetcher(fetch))
	require.NoError(t, err)

	_, err = store.GetModel(t.Context(), NewID("openai", "gpt-4o"))
	require.NoError(t, err)
	assert.True(t, firstFetched)

	// Network recovers: a second lookup must retry the fetch (the fallback was
	// not pinned) and serve the freshly fetched catalog.
	m, err := store.GetModel(t.Context(), NewID("openai", "fresh-model"))
	require.NoError(t, err)
	assert.True(t, secondFetched, "a fetch failure must not be memoized; the next lookup must retry")
	assert.Equal(t, 12345, m.Limit.Context, "the recovered fetch must serve fresh data, not the snapshot")
}