func TestOrder(t *testing.T) {
input := make(chan stream.Object)
passthruFn := func(obj stream.Object, out mapper.Outputer) {
out.Sending(1).Send(obj)
}
FirstOp := mapper.NewOrderedOp(passthruFn, "First PT o")
FirstOp.SetIn(input)
SecondOp := mapper.NewOrderedOp(passthruFn, "2nd PT o")
ch := stream.NewChain()
ch.Add(FirstOp)
ch.Add(SecondOp)
ch.Start()
output := SecondOp.Out()
go func() {
for i := 0; i < 10000; i++ {
input <- i
}
}()
for i := 0; i < 10000; i++ {
val := <-output
if val != i {
t.Error("Stuff out of order. Got ", val, " Expected ", i)
}
}
ch.Stop()
ch.Wait()
}
func TestNoOrder(t *testing.T) {
input := make(chan stream.Object)
passthruFn := func(obj stream.Object, out mapper.Outputer) {
out.Sending(1).Send(obj)
}
FirstOp := mapper.NewOp(passthruFn, "First PT no")
FirstOp.SetIn(input)
SecondOp := mapper.NewOp(passthruFn, "2nd PT no")
ch := stream.NewChain()
ch.Add(FirstOp)
ch.Add(SecondOp)
output := SecondOp.Out()
ch.Start()
go func() {
for i := 0; i < 10000; i++ {
input <- i
}
}()
found := false
for i := 0; i < 10000; i++ {
val := <-output
if val != i {
found = true
}
}
if !found {
t.Error("Weird no out of order stuff found")
}
ch.Stop()
ch.Wait()
}
func TestGob(t *testing.T) {
input := make(chan stream.Object)
enc := NewGobEncodeOp()
enc.SetIn(input)
ch := stream.NewChain()
ch.Add(enc)
decFn := func(in []byte, decoder func([]byte, interface{})) stream.Object {
var i int
decoder(in, &i)
return i
}
decodeOp := NewGobDecodeOp(decFn)
ch.Add(decodeOp)
/*
The old way not supported:
output := make(chan int)
var i int
decodeOp := NewGobDecodeRopUnsafe(encodeCh, output, reflect.TypeOf(i))*/
ch.Start()
go func() {
for i := 0; i < 10; i++ {
input <- i
}
}()
output := decodeOp.Out()
for i := 0; i < 10; i++ {
obj := <-output
val := obj.(int)
if val < 0 || val > 10 { //cant test value here because order can be messed up since encoder and decoders work in parallel
t.Error("Value is", val, " Expected", i)
}
}
go func() {
input <- 4321
}()
val := <-output
if val != 4321 {
t.Error("Value is", val, " Expected", 4321)
}
go func() {
input <- 1234
}()
val = <-output
if val != 1234 {
t.Error("Value is", val, " Expected", 1234)
}
}