container_wait_test.go 1.8 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374
  1. package client
  2. import (
  3. "bytes"
  4. "encoding/json"
  5. "fmt"
  6. "io/ioutil"
  7. "log"
  8. "net/http"
  9. "strings"
  10. "testing"
  11. "time"
  12. "github.com/docker/docker/api/types/container"
  13. "golang.org/x/net/context"
  14. )
  15. func TestContainerWaitError(t *testing.T) {
  16. client := &Client{
  17. client: newMockClient(errorMock(http.StatusInternalServerError, "Server error")),
  18. }
  19. resultC, errC := client.ContainerWait(context.Background(), "nothing", "")
  20. select {
  21. case result := <-resultC:
  22. t.Fatalf("expected to not get a wait result, got %d", result.StatusCode)
  23. case err := <-errC:
  24. if err.Error() != "Error response from daemon: Server error" {
  25. t.Fatalf("expected a Server Error, got %v", err)
  26. }
  27. }
  28. }
  29. func TestContainerWait(t *testing.T) {
  30. expectedURL := "/containers/container_id/wait"
  31. client := &Client{
  32. client: newMockClient(func(req *http.Request) (*http.Response, error) {
  33. if !strings.HasPrefix(req.URL.Path, expectedURL) {
  34. return nil, fmt.Errorf("Expected URL '%s', got '%s'", expectedURL, req.URL)
  35. }
  36. b, err := json.Marshal(container.ContainerWaitOKBody{
  37. StatusCode: 15,
  38. })
  39. if err != nil {
  40. return nil, err
  41. }
  42. return &http.Response{
  43. StatusCode: http.StatusOK,
  44. Body: ioutil.NopCloser(bytes.NewReader(b)),
  45. }, nil
  46. }),
  47. }
  48. resultC, errC := client.ContainerWait(context.Background(), "container_id", "")
  49. select {
  50. case err := <-errC:
  51. t.Fatal(err)
  52. case result := <-resultC:
  53. if result.StatusCode != 15 {
  54. t.Fatalf("expected a status code equal to '15', got %d", result.StatusCode)
  55. }
  56. }
  57. }
  58. func ExampleClient_ContainerWait_withTimeout() {
  59. ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
  60. defer cancel()
  61. client, _ := NewEnvClient()
  62. _, errC := client.ContainerWait(ctx, "container_id", "")
  63. if err := <-errC; err != nil {
  64. log.Fatal(err)
  65. }
  66. }