File size: 2,114 Bytes
6932c8c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 | package csync
import (
"testing"
"github.com/stretchr/testify/require"
)
func TestVersionedMap_Set(t *testing.T) {
t.Parallel()
vm := NewVersionedMap[string, int]()
require.Equal(t, uint64(0), vm.Version())
vm.Set("key1", 42)
require.Equal(t, uint64(1), vm.Version())
value, ok := vm.Get("key1")
require.True(t, ok)
require.Equal(t, 42, value)
}
func TestVersionedMap_Del(t *testing.T) {
t.Parallel()
vm := NewVersionedMap[string, int]()
vm.Set("key1", 42)
initialVersion := vm.Version()
vm.Del("key1")
require.Equal(t, initialVersion+1, vm.Version())
_, ok := vm.Get("key1")
require.False(t, ok)
}
func TestVersionedMap_VersionIncrement(t *testing.T) {
t.Parallel()
vm := NewVersionedMap[string, int]()
initialVersion := vm.Version()
// Setting a value should increment the version
vm.Set("key1", 42)
require.Equal(t, initialVersion+1, vm.Version())
// Deleting a value should increment the version
vm.Del("key1")
require.Equal(t, initialVersion+2, vm.Version())
// Deleting a non-existent key should still increment the version
vm.Del("nonexistent")
require.Equal(t, initialVersion+3, vm.Version())
}
func TestVersionedMap_ConcurrentAccess(t *testing.T) {
t.Parallel()
vm := NewVersionedMap[int, int]()
const numGoroutines = 100
const numOperations = 100
// Initial version
initialVersion := vm.Version()
// Perform concurrent Set and Del operations
for i := range numGoroutines {
go func(id int) {
for j := range numOperations {
key := id*numOperations + j
vm.Set(key, key*2)
vm.Del(key)
}
}(i)
}
// Wait for operations to complete by checking the version
// This is a simplified check - in a real test you might want to use sync.WaitGroup
expectedMinVersion := initialVersion + uint64(numGoroutines*numOperations*2)
// Allow some time for operations to complete
for vm.Version() < expectedMinVersion {
// Busy wait - in a real test you'd use proper synchronization
}
// Final version should be at least the expected minimum
require.GreaterOrEqual(t, vm.Version(), expectedMinVersion)
require.Equal(t, 0, vm.Len())
}
|