Function TestWithPolicyCache

internal/policy/cache/cache_test.go:269–328

(t *testing.T)

}

func TestWithPolicyCache(t *testing.T) {
	tests := []struct {
		name           string
		setupContext   func() context.Context
		setupCache     func() *PolicyCache
		expectOriginal bool
		expectNew      bool
	}{
		{
			name: "add cache to empty context",
			setupContext: func() context.Context {
				return context.Background()
			},
			setupCache: func() *PolicyCache {
				cache := NewPolicyCache(context.Background())
				return cache
			},
			expectOriginal: false,
			expectNew:      true,
		},
		{
			name: "add cache to context with existing cache",
			setupContext: func() context.Context {
				ctx := context.Background()
				existingCache := NewPolicyCache(ctx)
				return WithPolicyCache(ctx, existingCache)
			},
			setupCache: func() *PolicyCache {
				newCache := NewPolicyCache(context.Background())
				return newCache
			},
			expectOriginal: true,
			expectNew:      true,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			originalCtx := tt.setupContext()
			testCache := tt.setupCache()

			// Execute test
			newCtx := WithPolicyCache(originalCtx, testCache)

			// Verify cache is in the new context
			retrievedCache, found := PolicyCacheFromContext(newCtx)
			assert.Equal(t, tt.expectNew, found)
			if tt.expectNew {
				assert.Equal(t, testCache, retrievedCache)
			}

			// Verify original context is unchanged
			originalCache, found := PolicyCacheFromContext(originalCtx)
			assert.Equal(t, tt.expectOriginal, found)
			if !tt.expectOriginal {
				assert.Nil(t, originalCache)
			}
		})