goutil/structs/data_test.go at master · pigwantacat/goutil · 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
package structs_test

import (
	"testing"

	"github.com/gookit/goutil/structs"
	"github.com/gookit/goutil/testutil/assert"
)

func TestLiteData_Data(t *testing.T) {
	d := structs.NewLiteData(nil)
	d.SetData(map[string]any{
		"key0": 234,
		"key1": "abc",
		"key2": true,
	})

	assert.NotEmpty(t, d.Data())
	v, ok := d.Value("key0")
	assert.True(t, ok)
	assert.Eq(t, 234, v)

	d.Merge(map[string]any{
		"key1": "def",
		"key4": "value4",
	})
	assert.Eq(t, "def", d.StrVal("key1"))
	assert.Eq(t, "value4", d.GetVal("key4"))

	d.ResetData()
	assert.Empty(t, d.Data())
}

func TestNewData(t *testing.T) {
	md := structs.NewData()
	assert.Eq(t, 0, md.DataLen())

	md.SetData(map[string]any{
		"key0": 234,
		"sub": map[string]any{
			"skey1": "abc",
			"skey2": true,
		},
	})
	assert.NotEmpty(t, md.Data())
	assert.Eq(t, 234, md.IntVal("key0"))

	v, ok := md.Value("key0")
	assert.True(t, ok)
	assert.Eq(t, 234, v)

	md.SetValue("key1", "val1")
	assert.Eq(t, "val1", md.GetVal("key1"))
	assert.Eq(t, "val1", md.StrVal("key1"))
	assert.False(t, md.BoolVal("key1"))
	assert.False(t, md.BoolVal("not-exist"))

	md.SetValue("bol", true)
	assert.True(t, md.BoolVal("bol"))

	md.ResetData()
	assert.Eq(t, 0, md.DataLen())
}

func TestDataStore_EnableLock(t *testing.T) {
	md := structs.NewData()
	md.EnableLock()

	md.SetData(map[string]any{
		"key0": 234,
		"key1": "abc",
		"key2": true,
	})

	md.Set("key1", "def")
	assert.Eq(t, "def", md.Get("key1"))
	assert.NotEmpty(t, md.String())
}

func TestNewOrderedData(t *testing.T) {
	od := structs.NewOrderedData(10)
	od.Set("key0", 234)
	od.Load(map[string]any{
		"key1": "abc",
		"key2": true,
	})

	assert.NotEmpty(t, od.Keys())
}