volume_list_test.go 2.5 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798
  1. package client
  2. import (
  3. "bytes"
  4. "encoding/json"
  5. "fmt"
  6. "io/ioutil"
  7. "net/http"
  8. "strings"
  9. "testing"
  10. "github.com/docker/docker/api/types"
  11. "github.com/docker/docker/api/types/filters"
  12. volumetypes "github.com/docker/docker/api/types/volume"
  13. "golang.org/x/net/context"
  14. )
  15. func TestVolumeListError(t *testing.T) {
  16. client := &Client{
  17. client: newMockClient(errorMock(http.StatusInternalServerError, "Server error")),
  18. }
  19. _, err := client.VolumeList(context.Background(), filters.NewArgs())
  20. if err == nil || err.Error() != "Error response from daemon: Server error" {
  21. t.Fatalf("expected a Server Error, got %v", err)
  22. }
  23. }
  24. func TestVolumeList(t *testing.T) {
  25. expectedURL := "/volumes"
  26. noDanglingFilters := filters.NewArgs()
  27. noDanglingFilters.Add("dangling", "false")
  28. danglingFilters := filters.NewArgs()
  29. danglingFilters.Add("dangling", "true")
  30. labelFilters := filters.NewArgs()
  31. labelFilters.Add("label", "label1")
  32. labelFilters.Add("label", "label2")
  33. listCases := []struct {
  34. filters filters.Args
  35. expectedFilters string
  36. }{
  37. {
  38. filters: filters.NewArgs(),
  39. expectedFilters: "",
  40. }, {
  41. filters: noDanglingFilters,
  42. expectedFilters: `{"dangling":{"false":true}}`,
  43. }, {
  44. filters: danglingFilters,
  45. expectedFilters: `{"dangling":{"true":true}}`,
  46. }, {
  47. filters: labelFilters,
  48. expectedFilters: `{"label":{"label1":true,"label2":true}}`,
  49. },
  50. }
  51. for _, listCase := range listCases {
  52. client := &Client{
  53. client: newMockClient(func(req *http.Request) (*http.Response, error) {
  54. if !strings.HasPrefix(req.URL.Path, expectedURL) {
  55. return nil, fmt.Errorf("Expected URL '%s', got '%s'", expectedURL, req.URL)
  56. }
  57. query := req.URL.Query()
  58. actualFilters := query.Get("filters")
  59. if actualFilters != listCase.expectedFilters {
  60. return nil, fmt.Errorf("filters not set in URL query properly. Expected '%s', got %s", listCase.expectedFilters, actualFilters)
  61. }
  62. content, err := json.Marshal(volumetypes.VolumesListOKBody{
  63. Volumes: []*types.Volume{
  64. {
  65. Name: "volume",
  66. Driver: "local",
  67. },
  68. },
  69. })
  70. if err != nil {
  71. return nil, err
  72. }
  73. return &http.Response{
  74. StatusCode: http.StatusOK,
  75. Body: ioutil.NopCloser(bytes.NewReader(content)),
  76. }, nil
  77. }),
  78. }
  79. volumeResponse, err := client.VolumeList(context.Background(), listCase.filters)
  80. if err != nil {
  81. t.Fatal(err)
  82. }
  83. if len(volumeResponse.Volumes) != 1 {
  84. t.Fatalf("expected 1 volume, got %v", volumeResponse.Volumes)
  85. }
  86. }
  87. }