Write Contains, IndexOf & LastIndexOf predicate tests for LazyBuilder and ChainBuilder

Author: vellotis

chain_builder_test.go

@@ -440,6 +440,12 @@ func TestChainContains(t *testing.T) {
 			In:       []string{"foo", "bar"},
 			Contains: "bar",
 		},
+		{
+			In:       []string{"foo", "bar"},
+			Contains: func (value string) bool {
+				return value == "bar"
+			},
+		},
 		{
 			In:       results,
 			Contains: f,
@@ -468,6 +474,12 @@ func TestChainContains(t *testing.T) {
 			In:       map[int]*Foo{1: f, 3: c},
 			Contains: 2,
 		},
+		{
+			In:       map[int]*Foo{1: f, 3: c},
+			Contains: func (key int, foo *Foo) bool {
+				return key == 3 && foo.FirstName == "Harald"
+			},
+		},
 	}
 
 	for idx, tc := range testCases {
@@ -759,6 +771,12 @@ func TestChainIndexOf(t *testing.T) {
 			In:   []string{"foo", "bar"},
 			Item: "bar",
 		},
+		{
+			In:   []string{"foo", "bar"},
+			Item: func (value string) bool {
+				return value == "bar"
+			},
+		},
 		{
 			In:   results,
 			Item: f,
@@ -837,6 +855,12 @@ func TestChainLastIndexOf(t *testing.T) {
 			In:   []string{"foo", "bar", "bar"},
 			Item: "bar",
 		},
+		{
+			In:   []string{"foo", "bar", "bar"},
+			Item: func (value string) bool {
+				return value == "bar"
+			},
+		},
 		{
 			In:   []int{1, 2, 2, 3},
 			Item: 2,

lazy_builder_test.go

@@ -440,6 +440,12 @@ func TestLazyContains(t *testing.T) {
 			In:       []string{"foo", "bar"},
 			Contains: "bar",
 		},
+		{
+			In:       []string{"foo", "bar"},
+			Contains: func (value string) bool {
+				return value == "bar"
+			},
+		},
 		{
 			In:       results,
 			Contains: f,
@@ -468,6 +474,12 @@ func TestLazyContains(t *testing.T) {
 			In:       map[int]*Foo{1: f, 3: c},
 			Contains: 2,
 		},
+		{
+			In:       map[int]*Foo{1: f, 3: c},
+			Contains: func (key int, foo *Foo) bool {
+				return key == 3 && foo.FirstName == "Harald"
+			},
+		},
 	}
 
 	for idx, tc := range testCases {
@@ -759,6 +771,12 @@ func TestLazyIndexOf(t *testing.T) {
 			In:   []string{"foo", "bar"},
 			Item: "bar",
 		},
+		{
+			In:   []string{"foo", "bar"},
+			Item: func (value string) bool {
+				return value == "bar"
+			},
+		},
 		{
 			In:   results,
 			Item: f,
@@ -837,6 +855,12 @@ func TestLazyLastIndexOf(t *testing.T) {
 			In:   []string{"foo", "bar", "bar"},
 			Item: "bar",
 		},
+		{
+			In:   []string{"foo", "bar", "bar"},
+			Item: func (value string) bool {
+				return value == "bar"
+			},
+		},
 		{
 			In:   []int{1, 2, 2, 3},
 			Item: 2,