-
Notifications
You must be signed in to change notification settings - Fork 174
Expand file tree
/
Copy pathsyncmap_test.go
More file actions
65 lines (59 loc) · 1.18 KB
/
syncmap_test.go
File metadata and controls
65 lines (59 loc) · 1.18 KB
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
package maputil
import (
"testing"
"github.com/stackrox/rox/pkg/sync"
"github.com/stretchr/testify/assert"
)
func TestSyncMapStoreLoadContains(t *testing.T) {
m := NewSyncMap[string, int]()
m.Store("a", 1)
m.Store("a", 2)
v, ok := m.Load("a")
assert.True(t, ok)
assert.Equal(t, 2, v)
assert.True(t, m.Contains("a"))
assert.False(t, m.Contains("b"))
}
func TestSyncMap(t *testing.T) {
wg := sync.WaitGroup{}
wg.Add(3)
m := NewSyncMap[string, int]()
go func() {
m.Store("a", 1)
wg.Done()
}()
go func() {
_, _ = m.Load("a")
wg.Done()
}()
go func() {
m.Access(func(m *map[string]int) { (*m)["b"] = 2 })
wg.Done()
}()
wg.Wait()
v, ok := m.Load("a")
assert.True(t, ok)
assert.Equal(t, 1, v)
v, ok = m.Load("b")
assert.True(t, ok)
assert.Equal(t, 2, v)
}
func TestAccess(t *testing.T) {
m := NewCmpMap[string, int](nil)
m.Store("a", 1)
m.Store("a", 2)
m.RAccess(func(m map[string]int) {
assert.Equal(t, 2, m["a"])
})
v, ok := m.Load("a")
assert.True(t, ok)
assert.Equal(t, 2, v)
m.Access(func(m *map[string]int) {
*m = map[string]int{"c": 3}
})
_, ok = m.Load("a")
assert.False(t, ok)
v, ok = m.Load("c")
assert.True(t, ok)
assert.Equal(t, 3, v)
}