package testsuite

import (
	"fmt"
	"sync"
	"testing"

	"forge.cadoles.com/arcad/edge/pkg/storage/driver/cache/lfu"
	"github.com/pkg/errors"
)

func testConcurrent(t *testing.T, store lfu.Store[string, string]) error {
	const value = "foobar"
	totalKeys := 25
	totalSize := len(value) * totalKeys
	capacity := totalSize / 2

	cache := lfu.New[string, string](store,
		lfu.WithCapacity[string, string](capacity),
		lfu.WithLog[string, string](t.Logf),
	)

	var wg sync.WaitGroup

	wg.Add(totalKeys)

	loops := totalKeys * 10

	for i := 0; i < totalKeys; i++ {
		key := fmt.Sprintf("key%d", i)
		func(key string) {
			go func() {
				defer wg.Done()
				for i := 0; i < loops; i++ {
					if err := cache.Set(key, value); err != nil {
						t.Errorf("%+v", errors.WithStack(err))
					}
				}
			}()
		}(key)
	}

	wg.Wait()

	t.Logf("cache before final evict [capacity: %d, size: %d, len: %d]", cache.Capacity(), cache.Size(), cache.Len())

	if err := cache.Evict(); err != nil {
		t.Errorf("%+v", errors.WithStack(err))
	}

	t.Logf("cache after final evict [capacity: %d, size: %d, len: %d]", cache.Capacity(), cache.Size(), cache.Len())

	expectedLength := capacity / len(value)
	if e, g := expectedLength, cache.Len(); e < g {
		t.Errorf("cache.Len(): expected <= %d, got %d", e, g)
	}

	if cache.Size() > capacity {
		t.Errorf("cache.Size(): expected <= %d, got %d", capacity, cache.Size())
	}

	if e, g := expectedLength*len(value), cache.Size(); e < g {
		t.Errorf("cache.Size(): expected <= %d, got %d", e, g)
	}

	return nil
}