FilesExpand file tree

 master

/

docker_api_attach_test.go

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

79 lines (66 loc) · 1.43 KB

 master

/

docker_api_attach_test.go

Copy path

Top

File metadata and controls

• Code
• Blame

79 lines (66 loc) · 1.43 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

package main

import (

"bytes"

"os/exec"

"strings"

"time"

"github.com/go-check/check"

"code.google.com/p/go.net/websocket"

)

func (s *DockerSuite) TestGetContainersAttachWebsocket(c *check.C) {

runCmd := exec.Command(dockerBinary, "run", "-dit", "busybox", "cat")

out, _, err := runCommandWithOutput(runCmd)

if err != nil {

c.Fatalf(out, err)

}

rwc, err := sockConn(time.Duration(10 * time.Second))

if err != nil {

c.Fatal(err)

}

cleanedContainerID := strings.TrimSpace(out)

config, err := websocket.NewConfig(

"/containers/"+cleanedContainerID+"/attach/ws?stream=1&stdin=1&stdout=1&stderr=1",

"http://localhost",

)

if err != nil {

c.Fatal(err)

}

ws, err := websocket.NewClient(config, rwc)

if err != nil {

c.Fatal(err)

}

defer ws.Close()

expected := []byte("hello")

actual := make([]byte, len(expected))

outChan := make(chan error)

go func() {

_, err := ws.Read(actual)

outChan <- err

close(outChan)

}()

inChan := make(chan error)

go func() {

_, err := ws.Write(expected)

inChan <- err

close(inChan)

}()

select {

case err := <-inChan:

if err != nil {

c.Fatal(err)

}

case <-time.After(5 * time.Second):

c.Fatal("Timeout writing to ws")

}

select {

case err := <-outChan:

if err != nil {

c.Fatal(err)

}

case <-time.After(5 * time.Second):

c.Fatal("Timeout reading from ws")

}

if !bytes.Equal(expected, actual) {

c.Fatal("Expected output on websocket to match input")

}

}