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

package git

import (
	"encoding/json"
	"testing"

	"github.com/alecthomas/assert/v2"
)

func TestTagSet(t *testing.T) {
	set := make(TagSet)
	assert.Equal(t, 0, len(set))
	assert.Equal(t, 0, len(set.Slice()))

	set.Add("foo")
	assert.Equal(t, 1, len(set))
	assert.Equal(t, 1, len(set.Slice()))
	assert.True(t, set.Contains("foo"))

	set.Add("bar")
	assert.Equal(t, 2, len(set))
	assert.Equal(t, 2, len(set.Slice()))
	assert.True(t, set.Contains("foo"))
	assert.True(t, set.Contains("bar"))

	set.Add("bar")
	assert.Equal(t, 2, len(set))
	assert.Equal(t, 2, len(set.Slice()))
	assert.True(t, set.Contains("foo"))
	assert.True(t, set.Contains("bar"))

	set.Remove("foo")
	assert.Equal(t, 1, len(set))
	assert.Equal(t, 1, len(set.Slice()))
	assert.False(t, set.Contains("foo"))
	assert.True(t, set.Contains("bar"))

	set.Add("foo")
	set.Add("baz")
	j, err := json.Marshal(set)
	assert.NoError(t, err)
	assert.Equal(t, `["bar","baz","foo"]`, string(j))

	set = make(TagSet)
	b := []byte(`["foo","bar","baz"]`)
	err = json.Unmarshal(b, &set)
	assert.NoError(t, err)
	assert.Equal(t, 3, len(set))
	assert.Equal(t, 3, len(set.Slice()))
	assert.True(t, set.Contains("foo"))
	assert.True(t, set.Contains("bar"))
	assert.True(t, set.Contains("baz"))
}