filter_test.go

package goneric

import (
	"fmt"
	"github.com/stretchr/testify/assert"
	"sync"
	"testing"
)

func TestFilterMap(t *testing.T) {
	data := map[string]int{"a": 1, "b": 9, "c": 3, "d": 2, "e": 10, "f": 5, "g": 6, "h": 7}
	out := FilterMap(func(k string, v int) bool {
		if v > 8 {
			return false
		}
		return true
	}, data)
	assert.Equal(t, map[string]int{"a": 1, "c": 3, "d": 2, "f": 5, "g": 6, "h": 7}, out)
}

func TestFilterSlice(t *testing.T) {
	data := []int{1, 9, 3, 2, 10, 5, 6, 7}
	out := FilterSlice(func(idx int, v int) bool {
		if v > 8 {
			return false
		}
		return true
	}, data)
	assert.Equal(t, []int{1, 3, 2, 5, 6, 7}, out)
}

func TestFilterChan(t *testing.T) {
	data := []int{1, 9, 3, 2, 10, 5, 6, 7}
	inCh := make(chan int)
	go func() {
		for _, v := range data {
			inCh <- v
		}
		close(inCh)
	}()
	outCh := FilterChan(func(i int) bool {
		return i < 7
	}, inCh)
	outData := []int{}
	for v := range outCh {
		outData = append(outData, v)
	}
	assert.Equal(t, []int{1, 3, 2, 5, 6}, outData)
	assert.Panics(t, func() { close(outCh) }, "make sure out channel is closed")
}

func TestFilterChanErr(t *testing.T) {
	data := []int{1, 9, 3, 2, 10, 5, 6, 7}
	inCh := make(chan int)
	go func() {
		for _, v := range data {
			inCh <- v
		}
		close(inCh)
	}()
	outCh, errCh := FilterChanErr(func(i int) (bool, error) {
		if i > 8 {
			return i > 9, fmt.Errorf("err:%d", i)
		}
		return i < 7, nil
	}, inCh)
	outData := []int{}
	outErr := []error{}
	wg := sync.WaitGroup{}
	wg.Add(1)
	go func() {
		defer wg.Done()
		for v := range errCh {
			outErr = append(outErr, v)
		}
	}()
	for v := range outCh {
		outData = append(outData, v)
	}
	wg.Wait()
	assert.Equal(t, []int{1, 3, 2, 5, 6}, outData)
	assert.Len(t, outErr, 2)
	assert.Panics(t, func() { close(outCh) }, "make sure out channel is closed")
	assert.Panics(t, func() { close(errCh) }, "make sure error channel is closed")
}