mirror of https://github.com/docker/docs.git
129 lines
2.6 KiB
Go
129 lines
2.6 KiB
Go
package utils
|
|
|
|
import (
|
|
"github.com/docker/beam"
|
|
"github.com/docker/beam/inmem"
|
|
"github.com/docker/beam/unix"
|
|
"github.com/dotcloud/docker/pkg/testutils"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestStackWithPipe(t *testing.T) {
|
|
r, w := inmem.Pipe()
|
|
defer r.Close()
|
|
defer w.Close()
|
|
s := NewStackSender()
|
|
s.Add(w)
|
|
testutils.Timeout(t, func() {
|
|
go func() {
|
|
msg, _, _, err := r.Receive(0)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if msg.Name != "hello" {
|
|
t.Fatalf("%#v", msg)
|
|
}
|
|
if strings.Join(msg.Args, " ") != "wonderful world" {
|
|
t.Fatalf("%#v", msg)
|
|
}
|
|
}()
|
|
_, _, err := s.Send(&beam.Message{Name: "hello", Args: []string{"wonderful", "world"}}, 0)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
})
|
|
}
|
|
|
|
func TestStackWithPair(t *testing.T) {
|
|
r, w, err := unix.Pair()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer r.Close()
|
|
defer w.Close()
|
|
s := NewStackSender()
|
|
s.Add(w)
|
|
testutils.Timeout(t, func() {
|
|
go func() {
|
|
msg, _, _, err := r.Receive(0)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if msg.Name != "hello" {
|
|
t.Fatalf("%#v", msg)
|
|
}
|
|
if strings.Join(msg.Args, " ") != "wonderful world" {
|
|
t.Fatalf("%#v", msg)
|
|
}
|
|
}()
|
|
_, _, err := s.Send(&beam.Message{Name: "hello", Args: []string{"wonderful", "world"}}, 0)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
})
|
|
}
|
|
|
|
func TestStackLen(t *testing.T) {
|
|
s := NewStackSender()
|
|
if s.Len() != 0 {
|
|
t.Fatalf("empty StackSender has length %d", s.Len())
|
|
}
|
|
}
|
|
|
|
func TestStackAdd(t *testing.T) {
|
|
s := NewStackSender()
|
|
a := Buffer{}
|
|
beforeA := s.Add(&a)
|
|
// Add on an empty StackSender should return an empty StackSender
|
|
if beforeA.Len() != 0 {
|
|
t.Fatalf("%s has %d elements", beforeA, beforeA.Len())
|
|
}
|
|
if s.Len() != 1 {
|
|
t.Fatalf("%#v", beforeA)
|
|
}
|
|
// Add a 2nd element
|
|
b := Buffer{}
|
|
beforeB := s.Add(&b)
|
|
if beforeB.Len() != 1 {
|
|
t.Fatalf("%#v", beforeA)
|
|
}
|
|
if s.Len() != 2 {
|
|
t.Fatalf("%#v", beforeA)
|
|
}
|
|
s.Send(&beam.Message{Name: "for b", Args: nil}, 0)
|
|
beforeB.Send(&beam.Message{Name: "for a", Args: nil}, 0)
|
|
beforeA.Send(&beam.Message{Name: "for nobody", Args: nil}, 0)
|
|
if len(a) != 1 {
|
|
t.Fatalf("%#v", a)
|
|
}
|
|
if len(b) != 1 {
|
|
t.Fatalf("%#v", b)
|
|
}
|
|
}
|
|
|
|
// Misbehaving backends must be removed
|
|
func TestStackAddBad(t *testing.T) {
|
|
s := NewStackSender()
|
|
buf := Buffer{}
|
|
s.Add(&buf)
|
|
r, w := inmem.Pipe()
|
|
s.Add(w)
|
|
if s.Len() != 2 {
|
|
t.Fatalf("%#v", s)
|
|
}
|
|
r.Close()
|
|
if _, _, err := s.Send(&beam.Message{Name: "for the buffer", Args: nil}, 0); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if s.Len() != 1 {
|
|
t.Fatalf("%#v")
|
|
}
|
|
if len(buf) != 1 {
|
|
t.Fatalf("%#v", buf)
|
|
}
|
|
if buf[0].Name != "for the buffer" {
|
|
t.Fatalf("%#v", buf)
|
|
}
|
|
}
|