mirror of
https://github.com/moby/moby.git
synced 2022-11-09 12:21:53 -05:00
Move TestAttachDetach to integration-cli
Signed-off-by: Brian Goff <cpuguy83@gmail.com>
This commit is contained in:
parent
f7538c77ef
commit
ae0883ce00
2 changed files with 76 additions and 88 deletions
|
@ -3,6 +3,7 @@
|
||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"bufio"
|
||||||
"os/exec"
|
"os/exec"
|
||||||
"strings"
|
"strings"
|
||||||
"testing"
|
"testing"
|
||||||
|
@ -137,3 +138,78 @@ func TestAttachAfterDetach(t *testing.T) {
|
||||||
|
|
||||||
logDone("attach - reconnect after detaching")
|
logDone("attach - reconnect after detaching")
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// TestAttachDetach checks that attach in tty mode can be detached using the long container ID
|
||||||
|
func TestAttachDetach(t *testing.T) {
|
||||||
|
out, _, _ := dockerCmd(t, "run", "-itd", "busybox", "cat")
|
||||||
|
id := strings.TrimSpace(out)
|
||||||
|
if err := waitRun(id); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
cpty, tty, err := pty.Open()
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
defer cpty.Close()
|
||||||
|
|
||||||
|
cmd := exec.Command(dockerBinary, "attach", id)
|
||||||
|
cmd.Stdin = tty
|
||||||
|
stdout, err := cmd.StdoutPipe()
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
defer stdout.Close()
|
||||||
|
if err := cmd.Start(); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
if err := waitRun(id); err != nil {
|
||||||
|
t.Fatalf("error waiting for container to start: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if _, err := cpty.Write([]byte("hello\n")); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
out, err = bufio.NewReader(stdout).ReadString('\n')
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
if strings.TrimSpace(out) != "hello" {
|
||||||
|
t.Fatalf("exepected 'hello', got %q", out)
|
||||||
|
}
|
||||||
|
|
||||||
|
// escape sequence
|
||||||
|
if _, err := cpty.Write([]byte{16}); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
time.Sleep(100 * time.Millisecond)
|
||||||
|
if _, err := cpty.Write([]byte{17}); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
ch := make(chan struct{})
|
||||||
|
go func() {
|
||||||
|
cmd.Wait()
|
||||||
|
ch <- struct{}{}
|
||||||
|
}()
|
||||||
|
|
||||||
|
running, err := inspectField(id, "State.Running")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
if running != "true" {
|
||||||
|
t.Fatal("exepected container to still be running")
|
||||||
|
}
|
||||||
|
|
||||||
|
go func() {
|
||||||
|
dockerCmd(t, "kill", id)
|
||||||
|
}()
|
||||||
|
|
||||||
|
select {
|
||||||
|
case <-ch:
|
||||||
|
case <-time.After(10 * time.Millisecond):
|
||||||
|
t.Fatal("timed out waiting for container to exit")
|
||||||
|
}
|
||||||
|
|
||||||
|
logDone("attach - detach")
|
||||||
|
}
|
||||||
|
|
|
@ -113,94 +113,6 @@ func assertPipe(input, output string, r io.Reader, w io.Writer, count int) error
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// TestAttachDetach checks that attach in tty mode can be detached using the long container ID
|
|
||||||
func TestAttachDetach(t *testing.T) {
|
|
||||||
stdout, stdoutPipe := io.Pipe()
|
|
||||||
cpty, tty, err := pty.Open()
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
cli := client.NewDockerCli(tty, stdoutPipe, ioutil.Discard, "", testDaemonProto, testDaemonAddr, nil)
|
|
||||||
defer cleanup(globalEngine, t)
|
|
||||||
|
|
||||||
ch := make(chan struct{})
|
|
||||||
go func() {
|
|
||||||
defer close(ch)
|
|
||||||
if err := cli.CmdRun("-i", "-t", "-d", unitTestImageID, "cat"); err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
|
|
||||||
container := waitContainerStart(t, 10*time.Second)
|
|
||||||
|
|
||||||
setTimeout(t, "Reading container's id timed out", 10*time.Second, func() {
|
|
||||||
buf := make([]byte, 1024)
|
|
||||||
n, err := stdout.Read(buf)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
if strings.Trim(string(buf[:n]), " \r\n") != container.ID {
|
|
||||||
t.Fatalf("Wrong ID received. Expect %s, received %s", container.ID, buf[:n])
|
|
||||||
}
|
|
||||||
})
|
|
||||||
setTimeout(t, "Starting container timed out", 10*time.Second, func() {
|
|
||||||
<-ch
|
|
||||||
})
|
|
||||||
|
|
||||||
state := setRaw(t, container)
|
|
||||||
defer unsetRaw(t, container, state)
|
|
||||||
|
|
||||||
stdout, stdoutPipe = io.Pipe()
|
|
||||||
cpty, tty, err = pty.Open()
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
cli = client.NewDockerCli(tty, stdoutPipe, ioutil.Discard, "", testDaemonProto, testDaemonAddr, nil)
|
|
||||||
|
|
||||||
ch = make(chan struct{})
|
|
||||||
go func() {
|
|
||||||
defer close(ch)
|
|
||||||
if err := cli.CmdAttach(container.ID); err != nil {
|
|
||||||
if err != io.ErrClosedPipe {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
|
|
||||||
setTimeout(t, "First read/write assertion timed out", 2*time.Second, func() {
|
|
||||||
if err := assertPipe("hello\n", "hello", stdout, cpty, 150); err != nil {
|
|
||||||
if err != io.ErrClosedPipe {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
})
|
|
||||||
|
|
||||||
setTimeout(t, "Escape sequence timeout", 5*time.Second, func() {
|
|
||||||
cpty.Write([]byte{16})
|
|
||||||
time.Sleep(100 * time.Millisecond)
|
|
||||||
cpty.Write([]byte{17})
|
|
||||||
})
|
|
||||||
|
|
||||||
// wait for CmdRun to return
|
|
||||||
setTimeout(t, "Waiting for CmdAttach timed out", 15*time.Second, func() {
|
|
||||||
<-ch
|
|
||||||
})
|
|
||||||
|
|
||||||
closeWrap(cpty, stdout, stdoutPipe)
|
|
||||||
|
|
||||||
time.Sleep(500 * time.Millisecond)
|
|
||||||
if !container.IsRunning() {
|
|
||||||
t.Fatal("The detached container should be still running")
|
|
||||||
}
|
|
||||||
|
|
||||||
setTimeout(t, "Waiting for container to die timedout", 5*time.Second, func() {
|
|
||||||
container.Kill()
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
// TestAttachDetachTruncatedID checks that attach in tty mode can be detached
|
// TestAttachDetachTruncatedID checks that attach in tty mode can be detached
|
||||||
func TestAttachDetachTruncatedID(t *testing.T) {
|
func TestAttachDetachTruncatedID(t *testing.T) {
|
||||||
stdout, stdoutPipe := io.Pipe()
|
stdout, stdoutPipe := io.Pipe()
|
||||||
|
|
Loading…
Reference in a new issue