package middleware

import "testing"

func TestForEachDriver(t *testing.T) {
	middleware := forEachDriver(nil)

	testCase := make([]int, 100)
	for i := range testCase {
		testCase[i] = i
	}

	middleware.Input <- testCase
	for i := 0; i < 100; i++ {
		result := <-middleware.Output
		if result != i {
			t.Errorf("%d != %d", result, i)
		}
	}
}