base-golang/tool/help/groupby_test.go

package help

import (
	"reflect"
	"testing"
)

func TestGroupBy(t *testing.T) {
	input := []int{1, 2, 3, 4, 5, 6, 7, 8, 9}
	wantEven := []interface{}{2, 4, 6, 8}
	wantOdd := []interface{}{1, 3, 5, 7, 9}

	q := From(input).GroupBy(
		func(i interface{}) interface{} { return i.(int) % 2 },
		func(i interface{}) interface{} { return i.(int) },
	)

	next := q.Iterate()
	eq := true
	for item, ok := next(); ok; item, ok = next() {
		group := item.(Group)
		switch group.Key.(int) {
		case 0:
			if !reflect.DeepEqual(group.Group, wantEven) {
				eq = false
			}
		case 1:
			if !reflect.DeepEqual(group.Group, wantOdd) {
				eq = false
			}
		default:
			eq = false
		}
	}
	t.Logf("From(%v).GroupBy()=%v", input, toSlice(q))
	if !eq {
		t.Errorf("From(%v).GroupBy()=%v", input, toSlice(q))
	}
}

func TestGroupByT_PanicWhenKeySelectorFnIsInvalid(t *testing.T) {
	mustPanicWithError(t, "GroupByT: parameter [keySelectorFn] has a invalid function signature. Expected: 'func(T)T', actual: 'func(int,int)bool'", func() {
		var r []int
		From([]int{1, 1, 1, 2, 1, 2, 3, 4, 2}).GroupByT(
			func(i, j int) bool { return true },
			func(i int) int { return i },
		).ToSlice(&r)
	})
}

func TestGroupByT_PanicWhenElementSelectorFnIsInvalid(t *testing.T) {
	mustPanicWithError(t, "GroupByT: parameter [elementSelectorFn] has a invalid function signature. Expected: 'func(T)T', actual: 'func(int,int)int'", func() {
		var r []int
		From([]int{1, 1, 1, 2, 1, 2, 3, 4, 2}).GroupByT(
			func(i int) bool { return true },
			func(i, j int) int { return i },
		).ToSlice(&r)
	})
}