inmem: use pkg/testutils

Signed-off-by: Solomon Hykes <solomon@docker.com>
This commit is contained in:
Solomon Hykes 2014-05-09 21:12:27 -07:00
parent a1ed1ec1fa
commit caa6004064
1 changed files with 25 additions and 31 deletions

View File

@ -1,49 +1,43 @@
package inmem
import (
"github.com/dotcloud/docker/pkg/testutils"
"testing"
"time"
)
func TestSimpleSend(t *testing.T) {
r, w := Pipe()
onTimeout := time.After(100 * time.Millisecond)
onRcv := make(chan bool)
go func() {
msg, in, out, err := r.Receive(0)
testutils.Timeout(t, func() {
go func() {
msg, in, out, err := r.Receive(0)
if err != nil {
t.Fatal(err)
}
if msg.Data != "hello world" {
t.Fatalf("%#v", *msg)
}
if msg.Name != "print" {
t.Fatalf("%#v", *msg)
}
if len(msg.Args) != 0 {
t.Fatalf("%#v", *msg)
}
if in != nil {
t.Fatalf("%#v", in)
}
if out != nil {
t.Fatalf("%#v", out)
}
}()
in, out, err := w.Send(&Message{Name: "print", Data: "hello world"}, 0)
if err != nil {
t.Fatal(err)
}
if msg.Data != "hello world" {
t.Fatalf("%#v", *msg)
}
if msg.Name != "print" {
t.Fatalf("%#v", *msg)
}
if len(msg.Args) != 0 {
t.Fatalf("%#v", *msg)
}
if in != nil {
t.Fatalf("%#v", in)
}
if out != nil {
t.Fatalf("%#v", out)
}
close(onRcv)
}()
in, out, err := w.Send(&Message{Name: "print", Data: "hello world"}, 0)
if err != nil {
t.Fatal(err)
}
if in != nil {
t.Fatalf("%#v", in)
}
if out != nil {
t.Fatalf("%#v", out)
}
select {
case <-onTimeout:
t.Fatalf("timeout")
case <-onRcv:
}
})
}