summaryrefslogtreecommitdiffstats
path: root/tpl/collections/complement_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'tpl/collections/complement_test.go')
-rw-r--r--tpl/collections/complement_test.go19
1 files changed, 15 insertions, 4 deletions
diff --git a/tpl/collections/complement_test.go b/tpl/collections/complement_test.go
index 4cae7556f..e5edb23d3 100644
--- a/tpl/collections/complement_test.go
+++ b/tpl/collections/complement_test.go
@@ -23,6 +23,13 @@ import (
"github.com/stretchr/testify/require"
)
+type StructWithSlice struct {
+ A string
+ B []string
+}
+
+type StructWithSlicePointers []*StructWithSlice
+
func TestComplement(t *testing.T) {
t.Parallel()
@@ -33,10 +40,13 @@ func TestComplement(t *testing.T) {
s1 := []TstX{TstX{A: "a"}, TstX{A: "b"}, TstX{A: "d"}, TstX{A: "e"}}
s2 := []TstX{TstX{A: "b"}, TstX{A: "e"}}
- xa, xd := &TstX{A: "a"}, &TstX{A: "d"}
+ xa, xb, xd, xe := &StructWithSlice{A: "a"}, &StructWithSlice{A: "b"}, &StructWithSlice{A: "d"}, &StructWithSlice{A: "e"}
+
+ sp1 := []*StructWithSlice{xa, xb, xd, xe}
+ sp2 := []*StructWithSlice{xb, xe}
- sp1 := []*TstX{xa, &TstX{A: "b"}, xd, &TstX{A: "e"}}
- sp2 := []*TstX{&TstX{A: "b"}, &TstX{A: "e"}}
+ sp1_2 := StructWithSlicePointers{xa, xb, xd, xe}
+ sp2_2 := StructWithSlicePointers{xb, xe}
for i, test := range []struct {
s interface{}
@@ -49,7 +59,8 @@ func TestComplement(t *testing.T) {
{[]int{1, 2, 3, 4, 5}, []interface{}{[]int{1, 3}, []string{"a", "b"}, []int{1, 2}}, []int{4, 5}},
{[]int{1, 2, 3, 4, 5}, []interface{}{[]int64{1, 3}}, []int{2, 4, 5}},
{s1, []interface{}{s2}, []TstX{TstX{A: "a"}, TstX{A: "d"}}},
- {sp1, []interface{}{sp2}, []*TstX{xa, xd}},
+ {sp1, []interface{}{sp2}, []*StructWithSlice{xa, xd}},
+ {sp1_2, []interface{}{sp2_2}, StructWithSlicePointers{xa, xd}},
// Errors
{[]string{"a", "b", "c"}, []interface{}{"error"}, false},