72f49e55 |
package main
import ( |
e4cfd9b3 |
"bufio" |
4203230c |
"fmt" |
01094c15 |
"io" |
72f49e55 |
"os/exec"
"strings"
"sync"
"time" |
dc944ea7 |
"github.com/go-check/check" |
72f49e55 |
)
|
01094c15 |
const attachWait = 5 * time.Second
|
dc944ea7 |
func (s *DockerSuite) TestAttachMultipleAndRestart(c *check.C) { |
01094c15 |
endGroup := &sync.WaitGroup{}
startGroup := &sync.WaitGroup{}
endGroup.Add(3)
startGroup.Add(3)
if err := waitForContainer("attacher", "-d", "busybox", "/bin/sh", "-c", "while true; do sleep 1; echo hello; done"); err != nil { |
dc944ea7 |
c.Fatal(err) |
01094c15 |
}
startDone := make(chan struct{})
endDone := make(chan struct{})
go func() {
endGroup.Wait()
close(endDone) |
eb971633 |
}()
|
72f49e55 |
go func() { |
01094c15 |
startGroup.Wait()
close(startDone) |
72f49e55 |
}()
for i := 0; i < 3; i++ {
go func() { |
dc944ea7 |
cmd := exec.Command(dockerBinary, "attach", "attacher") |
72f49e55 |
|
01094c15 |
defer func() { |
dc944ea7 |
cmd.Wait() |
01094c15 |
endGroup.Done()
}()
|
dc944ea7 |
out, err := cmd.StdoutPipe() |
72f49e55 |
if err != nil { |
dc944ea7 |
c.Fatal(err) |
01094c15 |
}
|
dc944ea7 |
if err := cmd.Start(); err != nil {
c.Fatal(err) |
72f49e55 |
} |
01094c15 |
buf := make([]byte, 1024)
if _, err := out.Read(buf); err != nil && err != io.EOF { |
dc944ea7 |
c.Fatal(err) |
01094c15 |
}
startGroup.Done()
if !strings.Contains(string(buf), "hello") { |
dc944ea7 |
c.Fatalf("unexpected output %s expected hello\n", string(buf)) |
72f49e55 |
}
}()
}
|
01094c15 |
select {
case <-startDone:
case <-time.After(attachWait): |
dc944ea7 |
c.Fatalf("Attaches did not initialize properly") |
01094c15 |
}
cmd := exec.Command(dockerBinary, "kill", "attacher")
if _, err := runCommand(cmd); err != nil { |
dc944ea7 |
c.Fatal(err) |
01094c15 |
}
select {
case <-endDone:
case <-time.After(attachWait): |
dc944ea7 |
c.Fatalf("Attaches did not finish properly") |
01094c15 |
} |
72f49e55 |
} |
67e3ddb7 |
|
dc944ea7 |
func (s *DockerSuite) TestAttachTtyWithoutStdin(c *check.C) { |
67e3ddb7 |
cmd := exec.Command(dockerBinary, "run", "-d", "-ti", "busybox")
out, _, err := runCommandWithOutput(cmd)
if err != nil { |
dc944ea7 |
c.Fatalf("failed to start container: %v (%v)", out, err) |
67e3ddb7 |
}
id := strings.TrimSpace(out)
if err := waitRun(id); err != nil { |
dc944ea7 |
c.Fatal(err) |
67e3ddb7 |
}
defer func() {
cmd := exec.Command(dockerBinary, "kill", id)
if out, _, err := runCommandWithOutput(cmd); err != nil { |
dc944ea7 |
c.Fatalf("failed to kill container: %v (%v)", out, err) |
67e3ddb7 |
}
}()
|
4203230c |
done := make(chan error) |
67e3ddb7 |
go func() {
defer close(done)
cmd := exec.Command(dockerBinary, "attach", id)
if _, err := cmd.StdinPipe(); err != nil { |
4203230c |
done <- err
return |
67e3ddb7 |
}
expected := "cannot enable tty mode"
if out, _, err := runCommandWithOutput(cmd); err == nil { |
4203230c |
done <- fmt.Errorf("attach should have failed")
return |
67e3ddb7 |
} else if !strings.Contains(out, expected) { |
4203230c |
done <- fmt.Errorf("attach failed with error %q: expected %q", out, expected)
return |
67e3ddb7 |
}
}()
select { |
4203230c |
case err := <-done:
c.Assert(err, check.IsNil) |
67e3ddb7 |
case <-time.After(attachWait): |
dc944ea7 |
c.Fatal("attach is running but should have failed") |
67e3ddb7 |
}
} |
e4cfd9b3 |
|
dc944ea7 |
func (s *DockerSuite) TestAttachDisconnect(c *check.C) {
out, _ := dockerCmd(c, "run", "-di", "busybox", "/bin/cat") |
e4cfd9b3 |
id := strings.TrimSpace(out)
cmd := exec.Command(dockerBinary, "attach", id)
stdin, err := cmd.StdinPipe()
if err != nil { |
dc944ea7 |
c.Fatal(err) |
e4cfd9b3 |
}
defer stdin.Close()
stdout, err := cmd.StdoutPipe()
if err != nil { |
dc944ea7 |
c.Fatal(err) |
e4cfd9b3 |
}
defer stdout.Close()
if err := cmd.Start(); err != nil { |
dc944ea7 |
c.Fatal(err) |
e4cfd9b3 |
}
defer cmd.Process.Kill()
if _, err := stdin.Write([]byte("hello\n")); err != nil { |
dc944ea7 |
c.Fatal(err) |
e4cfd9b3 |
}
out, err = bufio.NewReader(stdout).ReadString('\n')
if err != nil { |
dc944ea7 |
c.Fatal(err) |
e4cfd9b3 |
}
if strings.TrimSpace(out) != "hello" { |
3941623f |
c.Fatalf("expected 'hello', got %q", out) |
e4cfd9b3 |
}
if err := stdin.Close(); err != nil { |
dc944ea7 |
c.Fatal(err) |
e4cfd9b3 |
}
// Expect container to still be running after stdin is closed
running, err := inspectField(id, "State.Running")
if err != nil { |
dc944ea7 |
c.Fatal(err) |
e4cfd9b3 |
}
if running != "true" { |
3941623f |
c.Fatal("expected container to still be running") |
e4cfd9b3 |
}
} |