github.com/kubeshop/testkube@v1.17.23/pkg/data/set/set_test.go (about) 1 package set 2 3 import ( 4 "testing" 5 6 "github.com/stretchr/testify/assert" 7 ) 8 9 func TestOf(t *testing.T) { 10 out := Of("aaa", "bbb") 11 12 assert.True(t, out.Has("aaa"), "set should have aaa") 13 assert.True(t, out.Has("bbb"), "set should have aaa") 14 } 15 16 func TestSet_ToArray(t *testing.T) { 17 // given 18 out := Of("aaa", "bbb") 19 20 // when 21 arr := out.ToArray() 22 23 // then 24 assert.Contains(t, arr, "aaa") 25 assert.Contains(t, arr, "bbb") 26 } 27 28 func TestSet_Remove(t *testing.T) { 29 // given 30 out := Of("aaa", "bbb", "ccc") 31 32 // when 33 out.Remove("bbb") 34 35 // then 36 assert.False(t, out.Has("bbb")) 37 }