find_test.go

package goterators

import (
	"fmt"
	"testing"
)

func TestFindInt(t *testing.T) {
	testSource := []int{1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20}

	expectedItem1 := 18
	expectedIndex1 := 17
	matchedItem, index, err := Find(testSource, func(item int) bool {
		return item == expectedItem1
	})
	if err != nil {
		t.Errorf("Find item1 = %v, expected no error, got = %v", expectedItem1, err)
	}
	if matchedItem != expectedItem1 {
		t.Errorf("Find item1 = %v, Expected = %v , got = %v", expectedItem1, expectedItem1, matchedItem)
	}
	if index != expectedIndex1 {
		t.Errorf("Find item1 = %v, Expected = %v , got = %v", expectedItem1, expectedIndex1, index)
	}

	expectedItem2 := 50
	matchedItem, index, err = Find(testSource, func(item int) bool {
		return item == expectedItem2
	})
	if err == nil {
		t.Errorf("Find item2 = %v, expected <not_found>, got = no-error", expectedItem1)
	}
}

func TestFindString(t *testing.T) {
	testSource := []string{"basis", "pray", "platform", "top", "border", "pole", "kidnap", "eaux", "stream", "lemon", "helpless", "indulge", "highlight", "city", "miner", "photography", "squeeze", "king", "offender", "rugby"}

	expectedItem1 := "king"
	expectedIndex1 := 17
	matchedItem, index, err := Find(testSource, func(item string) bool {
		return item == expectedItem1
	})
	if err != nil {
		t.Errorf("Find item1 = %v, expected no error, got = %v", expectedItem1, err)
	}
	if matchedItem != expectedItem1 {
		t.Errorf("Find item1 = %v, Expected = %v , got = %v", expectedItem1, expectedItem1, matchedItem)
	}
	if index != expectedIndex1 {
		t.Errorf("Find item1 = %v, Expected = %v , got = %v", expectedItem1, expectedIndex1, index)
	}

	expectedItem2 := "webuild"
	matchedItem, index, err = Find(testSource, func(item string) bool {
		return item == expectedItem2
	})
	if err == nil {
		t.Errorf("Find item2 = %v, expected <not_found>, got = no-error", expectedItem1)
	}
}

func ExampleFind() {
	testSource := []int{1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20}
	firstFoundItem, index, _ := Find(testSource, func(item int) bool {
		return item%10 == 0
	})
	fmt.Println("Find: ", firstFoundItem, index)
}