FilesExpand file tree

 master

/

docker_api_attach_test.go

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

66 lines (55 loc) · 1.26 KB

 master

/

docker_api_attach_test.go

Copy path

Top

File metadata and controls

• Code
• Blame

66 lines (55 loc) · 1.26 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

package main

import (

"bytes"

"os/exec"

"testing"

"time"

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

)

func TestGetContainersAttachWebsocket(t *testing.T) {

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

out, _, err := runCommandWithOutput(runCmd)

if err != nil {

t.Fatalf(out, err)

}

defer deleteAllContainers()

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

if err != nil {

t.Fatal(err)

}

cleanedContainerID := stripTrailingCharacters(out)

config, err := websocket.NewConfig(

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

"http://localhost",

)

if err != nil {

t.Fatal(err)

}

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

if err != nil {

t.Fatal(err)

}

defer ws.Close()

expected := []byte("hello")

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

outChan := make(chan string)

go func() {

if _, err := ws.Read(actual); err != nil {

t.Fatal(err)

}

outChan <- "done"

}()

inChan := make(chan string)

go func() {

if _, err := ws.Write(expected); err != nil {

t.Fatal(err)

}

inChan <- "done"

}()

<-inChan

<-outChan

if !bytes.Equal(expected, actual) {

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

}

logDone("container attach websocket - can echo input via cat")

}