Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
Test for stop of goflow component with multiple inputs
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()
}
}
@lovromazgon

This comment has been minimized.

Copy link
Owner Author

commented May 25, 2017

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.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
You can’t perform that action at this time.