coretracer/safemap_test.go at main · InjectiveLabs/coretracer · GitHub

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
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
package coretracer

import (
	"fmt"
	"sync"
	"testing"

	"github.com/stretchr/testify/require"
)

func TestNewSafeMap(t *testing.T) {
	sm := newSafeMap()
	require.NotNil(t, sm.mux, "Expected mux to be initialized")
	require.NotNil(t, sm.m, "Expected map to be initialized")
}

func TestNewSafeMapWith(t *testing.T) {
	key, value := "key", "value"
	sm := newSafeMapWith(key, value)
	require.NotNil(t, sm.mux, "Expected mux to be initialized")
	require.NotNil(t, sm.m, "Expected map to be initialized")
	require.Equal(t, value, sm.m[key], "Expected map to contain %s: %s", key, value)
}

func TestSafeMap_IsValid(t *testing.T) {
	sm := newSafeMap()
	require.True(t, sm.IsValid(), "Expected SafeMap to be valid")
}

func TestSafeMap_Set(t *testing.T) {
	sm := newSafeMap()
	key, value := "key", "value"
	sm = sm.Set(key, value)
	require.Equal(t, value, sm.m[key], "Expected map to contain %s: %s", key, value)
}

func TestSafeMap_RLock(t *testing.T) {
	sm := newSafeMap()
	sm.RLock()
	defer sm.RUnlock()
	// No panic expected

	_ = sm.m["kek"]
}

func TestSafeMap_RUnlock(t *testing.T) {
	sm := newSafeMap()
	sm.RLock()
	_ = sm.m["kek"]
	sm.RUnlock()
	// No panic expected
}

func TestSafeMap_Map(t *testing.T) {
	sm := newSafeMap()
	require.NotNil(t, sm.Map(), "Expected map to be returned")
}

func BenchmarkSafeMap(b *testing.B) {
	sm := newSafeMap()
	var wg sync.WaitGroup

	b.Run("writeOne", func(b *testing.B) {
		for i := 0; i < b.N; i++ {
			wg.Add(1)
			go func(i int) {
				defer wg.Done()
				key := fmt.Sprintf("key%d", i)
				value := fmt.Sprintf("value%d", i)
				sm.Set(key, value)
			}(i)
		}
		wg.Wait()
	})

	b.Run("readOne", func(b *testing.B) {
		for i := 0; i < b.N; i++ {
			wg.Add(1)
			go func() {
				defer wg.Done()
				sm.RLock()
				for k, v := range sm.Map() {
					_ = k
					_ = v
					break
				}
				sm.RUnlock()
			}()
		}
		wg.Wait()
	})

	b.Run("readAll", func(b *testing.B) {
		for i := 0; i < b.N; i++ {
			wg.Add(1)
			go func() {
				defer wg.Done()
				sm.RLock()
				for k, v := range sm.Map() {
					_ = k
					_ = v
				}
				sm.RUnlock()
			}()
		}
		wg.Wait()
	})
}