kubescape/core/pkg/policyhandler/cache_test.go

package policyhandler

import (
	"testing"
	"time"
)

func TestTimedCache(t *testing.T) {
	tests := []struct {
		name string
		// value ttl
		ttl time.Duration
		// value to set
		value int
		// time to wait before checking if value exists
		wait time.Duration
		// number of times to check if value exists (with wait in between)
		checks int
		// should the value exist in cache
		exists bool
		// expected cache value
		wantVal int
	}{
		{
			name:    "value exists before ttl",
			ttl:     time.Second * 5,
			value:   42,
			wait:    time.Second * 1,
			checks:  2,
			exists:  true,
			wantVal: 42,
		},
		{
			name:   "value does not exist after ttl",
			ttl:    time.Second * 3,
			value:  55,
			wait:   time.Second * 4,
			checks: 1,
			exists: false,
		},
		{
			name:   "cache is disabled (ttl = 0) always returns false",
			ttl:    0,
			value:  55,
			wait:   0,
			checks: 1,
			exists: false,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			cache := NewTimedCache[int](tt.ttl)
			cache.Set(tt.value)

			for i := 0; i < tt.checks; i++ {
				// Wait for the specified duration
				time.Sleep(tt.wait)

				// Get the value from the cache
				value, exists := cache.Get()

				// Check if value exists
				if exists != tt.exists {
					t.Errorf("Expected exists to be %v, got %v", tt.exists, exists)
				}

				// Check value
				if exists && value != tt.wantVal {
					t.Errorf("Expected value to be %d, got %d", tt.wantVal, value)
				}
			}
		})
	}
}