mightymap_concurency_test.go

package mightymap_test

import (
	"context"
	"sync"
	"testing"

	"github.com/thisisdevelopment/mightymap"
)

func TestMightyMap_Concurrency(t *testing.T) {
	// Testing concurrent access to MightyMap
	ctx := context.Background()
	cm := mightymap.New[int, int](true)
	var wg sync.WaitGroup

	t.Run("Concurrent Store", func(t *testing.T) {
		for i := 0; i < 1000; i++ {
			wg.Add(1)
			go func(i int) {
				defer wg.Done()
				cm.Store(ctx, i, i*i)
			}(i)
		}
		wg.Wait()

		if cm.Len(ctx) != 1000 {
			t.Errorf("Expected 1000 items, got %d", cm.Len(ctx))
		}

		err := cm.Close(ctx)
		if err != nil {
			t.Errorf("Error closing map: %v", err)
		}
	})

	t.Run("Concurrent Load and Delete", func(t *testing.T) {
		for i := 0; i < 1000; i++ {
			wg.Add(1)
			go func(i int) {
				defer wg.Done()
				value, ok := cm.Load(ctx, i)
				if !ok || value != i*i {
					t.Errorf("Expected to load %d, got %d", i*i, value)
				}
				cm.Delete(ctx, i)
			}(i)
		}
		wg.Wait()

		if cm.Len(ctx) != 0 {
			t.Errorf("Expected map to be empty after deletes, got %d", cm.Len(ctx))
		}
	})
}