where_test.go

package underscore

import (
	"testing"

	"github.com/stretchr/testify/assert"
)

func Benchmark_Where(b *testing.B) {
	for n := 0; n < b.N; n++ {
		Range(0, benchmarkSize, 1).Where(func(r, _ int) bool {
			return r > 100
		}).Any(func(r, _ int) bool {
			return true
		})
	}
}

func Test_Where(t *testing.T) {
	src := []testModel{
		{ID: 1, Name: "one"},
		{ID: 2, Name: "one"},
		{ID: 3, Name: "three"},
		{ID: 4, Name: "three"},
	}
	var res []testModel
	Chain(src).Where(func(r testModel, _ int) bool {
		return r.ID%2 == 0
	}).Value(&res)
	assert.EqualValues(
		t,
		res,
		[]testModel{src[1], src[3]},
	)
}

func Test_WhereBy(t *testing.T) {
	src := []testModel{
		{ID: 1, Name: "one"},
		{ID: 2, Name: "one"},
		{ID: 3, Name: "three"},
		{ID: 4, Name: "three"},
	}
	var res []testModel
	Chain(src).WhereBy(map[string]interface{}{
		"Name": "one",
	}).Value(&res)
	assert.EqualValues(
		t,
		res,
		[]testModel{src[0], src[1]},
	)
}