Test for stop of goflow component with multiple inputs

multiinput_test.go

package test

import (
	"github.com/trustmaster/goflow"
	"fmt"
	"testing"
	"time"
)

// component 1
type C1 struct {
	flow.Component
	In <-chan int
	Out chan<- int
}

func (c *C1) OnIn(i int) {
	c.Out<- i
}

func (c *C1) Finish() {
	fmt.Println("Finish c1")
}

// component 2
type C2 struct {
	flow.Component
	In1 <-chan int
	In2 <- chan int
}

func (c *C2) OnIn1(i int) {
	fmt.Printf("in1: %d\n", i)
}

func (c *C2) OnIn2(i int) {
	fmt.Printf("in2: %d\n", i)
}

func (c *C2) Finish() {
	fmt.Println("Finish c2")
}

// test
func TestMultiInput(t *testing.T) {
	flow.DefaultComponentMode = flow.ComponentModeSync
	n := new(flow.Graph) // creates the object in heap
	n.InitGraphState()   // allocates memory for the graph

	// Add processes to the network
	n.Add(new(C1), "c1-1")
	n.Add(new(C1), "c1-2")
	n.Add(new(C2), "c2")

	n.Connect("c1-1", "Out", "c2", "In1")
	n.Connect("c1-2", "Out", "c2", "In2")

	n.MapInPort("In1", "c1-1", "In")
	n.MapInPort("In2", "c1-2", "In")

	in1 := make(chan int)
	in2 := make(chan int)
	n.SetInPort("In1", in1)
	n.SetInPort("In2", in2)

	flow.RunNet(n)

	in1 <- 1
	in2 <- 2
	in1 <- 3
	in2 <- 4
	in1 <- 5

	time.Sleep(time.Second)
	close(in1)
	time.Sleep(time.Second)
	close(in2)

	select {
	case <-n.Wait():
		t.Log("Success")
	case <-time.NewTimer(time.Second*3).C:
		t.Log("Waited 3 seconds for close!")
		t.Fail()
	}
}

Test output:

in1: 1
in2: 2
in2: 4
in1: 3
in1: 5
Finish c1
Finish c1
	multiinput_test.go:83: Waited 3 seconds for close!

This test shouldn't fail.