initial import

core/collection/safemap_test.go

@@ -0,0 +1,110 @@
+package collection
+
+import (
+	"testing"
+
+	"zero/core/stringx"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestSafeMap(t *testing.T) {
+	tests := []struct {
+		size      int
+		exception int
+	}{
+		{
+			100000,
+			2000,
+		},
+		{
+			100000,
+			50,
+		},
+	}
+	for _, test := range tests {
+		t.Run(stringx.Rand(), func(t *testing.T) {
+			testSafeMapWithParameters(t, test.size, test.exception)
+		})
+	}
+}
+
+func TestSafeMap_CopyNew(t *testing.T) {
+	const (
+		size       = 100000
+		exception1 = 5
+		exception2 = 500
+	)
+	m := NewSafeMap()
+
+	for i := 0; i < size; i++ {
+		m.Set(i, i)
+	}
+	for i := 0; i < size; i++ {
+		if i%exception1 == 0 {
+			m.Del(i)
+		}
+	}
+
+	for i := size; i < size<<1; i++ {
+		m.Set(i, i)
+	}
+	for i := size; i < size<<1; i++ {
+		if i%exception2 != 0 {
+			m.Del(i)
+		}
+	}
+
+	for i := 0; i < size; i++ {
+		val, ok := m.Get(i)
+		if i%exception1 != 0 {
+			assert.True(t, ok)
+			assert.Equal(t, i, val.(int))
+		} else {
+			assert.False(t, ok)
+		}
+	}
+	for i := size; i < size<<1; i++ {
+		val, ok := m.Get(i)
+		if i%exception2 == 0 {
+			assert.True(t, ok)
+			assert.Equal(t, i, val.(int))
+		} else {
+			assert.False(t, ok)
+		}
+	}
+}
+
+func testSafeMapWithParameters(t *testing.T, size, exception int) {
+	m := NewSafeMap()
+
+	for i := 0; i < size; i++ {
+		m.Set(i, i)
+	}
+	for i := 0; i < size; i++ {
+		if i%exception != 0 {
+			m.Del(i)
+		}
+	}
+
+	assert.Equal(t, size/exception, m.Size())
+
+	for i := size; i < size<<1; i++ {
+		m.Set(i, i)
+	}
+	for i := size; i < size<<1; i++ {
+		if i%exception != 0 {
+			m.Del(i)
+		}
+	}
+
+	for i := 0; i < size<<1; i++ {
+		val, ok := m.Get(i)
+		if i%exception == 0 {
+			assert.True(t, ok)
+			assert.Equal(t, i, val.(int))
+		} else {
+			assert.False(t, ok)
+		}
+	}
+}