Function TestPolicyCacheFromContext

internal/policy/cache/cache_test.go:212–267

(t *testing.T)

}

func TestPolicyCacheFromContext(t *testing.T) {
	tests := []struct {
		name          string
		setupContext  func() context.Context
		expectedFound bool
		expectedCache *PolicyCache
	}{
		{
			name: "PolicyCache not in context",
			setupContext: func() context.Context {
				return context.Background()
			},
			expectedFound: false,
			expectedCache: nil,
		},
		{
			name: "PolicyCache in context",
			setupContext: func() context.Context {
				ctx := context.Background()
				cache := NewPolicyCache(ctx)
				return WithPolicyCache(ctx, cache)
			},
			expectedFound: true,
			expectedCache: func() *PolicyCache {
				cache := NewPolicyCache(context.Background())
				return cache
			}(),
		},
		{
			name: "nil PolicyCache in context",
			setupContext: func() context.Context {
				ctx := context.Background()
				return context.WithValue(ctx, policyCacheKey, (*PolicyCache)(nil))
			},
			expectedFound: true,
			expectedCache: nil,
		},
	}

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

			// Execute test
			retrievedCache, found := PolicyCacheFromContext(ctx)

			// Assert results
			assert.Equal(t, tt.expectedFound, found)
			if tt.expectedCache == nil {
				assert.Nil(t, retrievedCache)
			} else {
				assert.NotNil(t, retrievedCache)
			}
		})
	}
}

func TestWithPolicyCache(t *testing.T) {