container_exec_test.go 4.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157
  1. package client
  2. import (
  3. "bytes"
  4. "encoding/json"
  5. "fmt"
  6. "io/ioutil"
  7. "net/http"
  8. "strings"
  9. "testing"
  10. "golang.org/x/net/context"
  11. "github.com/docker/docker/api/types"
  12. )
  13. func TestContainerExecCreateError(t *testing.T) {
  14. client := &Client{
  15. client: newMockClient(errorMock(http.StatusInternalServerError, "Server error")),
  16. }
  17. _, err := client.ContainerExecCreate(context.Background(), "container_id", types.ExecConfig{})
  18. if err == nil || err.Error() != "Error response from daemon: Server error" {
  19. t.Fatalf("expected a Server Error, got %v", err)
  20. }
  21. }
  22. func TestContainerExecCreate(t *testing.T) {
  23. expectedURL := "/containers/container_id/exec"
  24. client := &Client{
  25. client: newMockClient(func(req *http.Request) (*http.Response, error) {
  26. if !strings.HasPrefix(req.URL.Path, expectedURL) {
  27. return nil, fmt.Errorf("expected URL '%s', got '%s'", expectedURL, req.URL)
  28. }
  29. if req.Method != "POST" {
  30. return nil, fmt.Errorf("expected POST method, got %s", req.Method)
  31. }
  32. // FIXME validate the content is the given ExecConfig ?
  33. if err := req.ParseForm(); err != nil {
  34. return nil, err
  35. }
  36. execConfig := &types.ExecConfig{}
  37. if err := json.NewDecoder(req.Body).Decode(execConfig); err != nil {
  38. return nil, err
  39. }
  40. if execConfig.User != "user" {
  41. return nil, fmt.Errorf("expected an execConfig with User == 'user', got %v", execConfig)
  42. }
  43. b, err := json.Marshal(types.IDResponse{
  44. ID: "exec_id",
  45. })
  46. if err != nil {
  47. return nil, err
  48. }
  49. return &http.Response{
  50. StatusCode: http.StatusOK,
  51. Body: ioutil.NopCloser(bytes.NewReader(b)),
  52. }, nil
  53. }),
  54. }
  55. r, err := client.ContainerExecCreate(context.Background(), "container_id", types.ExecConfig{
  56. User: "user",
  57. })
  58. if err != nil {
  59. t.Fatal(err)
  60. }
  61. if r.ID != "exec_id" {
  62. t.Fatalf("expected `exec_id`, got %s", r.ID)
  63. }
  64. }
  65. func TestContainerExecStartError(t *testing.T) {
  66. client := &Client{
  67. client: newMockClient(errorMock(http.StatusInternalServerError, "Server error")),
  68. }
  69. err := client.ContainerExecStart(context.Background(), "nothing", types.ExecStartCheck{})
  70. if err == nil || err.Error() != "Error response from daemon: Server error" {
  71. t.Fatalf("expected a Server Error, got %v", err)
  72. }
  73. }
  74. func TestContainerExecStart(t *testing.T) {
  75. expectedURL := "/exec/exec_id/start"
  76. client := &Client{
  77. client: newMockClient(func(req *http.Request) (*http.Response, error) {
  78. if !strings.HasPrefix(req.URL.Path, expectedURL) {
  79. return nil, fmt.Errorf("Expected URL '%s', got '%s'", expectedURL, req.URL)
  80. }
  81. if err := req.ParseForm(); err != nil {
  82. return nil, err
  83. }
  84. execStartCheck := &types.ExecStartCheck{}
  85. if err := json.NewDecoder(req.Body).Decode(execStartCheck); err != nil {
  86. return nil, err
  87. }
  88. if execStartCheck.Tty || !execStartCheck.Detach {
  89. return nil, fmt.Errorf("expected execStartCheck{Detach:true,Tty:false}, got %v", execStartCheck)
  90. }
  91. return &http.Response{
  92. StatusCode: http.StatusOK,
  93. Body: ioutil.NopCloser(bytes.NewReader([]byte(""))),
  94. }, nil
  95. }),
  96. }
  97. err := client.ContainerExecStart(context.Background(), "exec_id", types.ExecStartCheck{
  98. Detach: true,
  99. Tty: false,
  100. })
  101. if err != nil {
  102. t.Fatal(err)
  103. }
  104. }
  105. func TestContainerExecInspectError(t *testing.T) {
  106. client := &Client{
  107. client: newMockClient(errorMock(http.StatusInternalServerError, "Server error")),
  108. }
  109. _, err := client.ContainerExecInspect(context.Background(), "nothing")
  110. if err == nil || err.Error() != "Error response from daemon: Server error" {
  111. t.Fatalf("expected a Server Error, got %v", err)
  112. }
  113. }
  114. func TestContainerExecInspect(t *testing.T) {
  115. expectedURL := "/exec/exec_id/json"
  116. client := &Client{
  117. client: newMockClient(func(req *http.Request) (*http.Response, error) {
  118. if !strings.HasPrefix(req.URL.Path, expectedURL) {
  119. return nil, fmt.Errorf("Expected URL '%s', got '%s'", expectedURL, req.URL)
  120. }
  121. b, err := json.Marshal(types.ContainerExecInspect{
  122. ExecID: "exec_id",
  123. ContainerID: "container_id",
  124. })
  125. if err != nil {
  126. return nil, err
  127. }
  128. return &http.Response{
  129. StatusCode: http.StatusOK,
  130. Body: ioutil.NopCloser(bytes.NewReader(b)),
  131. }, nil
  132. }),
  133. }
  134. inspect, err := client.ContainerExecInspect(context.Background(), "exec_id")
  135. if err != nil {
  136. t.Fatal(err)
  137. }
  138. if inspect.ExecID != "exec_id" {
  139. t.Fatalf("expected ExecID to be `exec_id`, got %s", inspect.ExecID)
  140. }
  141. if inspect.ContainerID != "container_id" {
  142. t.Fatalf("expected ContainerID `container_id`, got %s", inspect.ContainerID)
  143. }
  144. }