image_list_test.go 3.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163
  1. package client // import "github.com/docker/docker/client"
  2. import (
  3. "bytes"
  4. "context"
  5. "encoding/json"
  6. "fmt"
  7. "io/ioutil"
  8. "net/http"
  9. "strings"
  10. "testing"
  11. "github.com/docker/docker/api/types"
  12. "github.com/docker/docker/api/types/filters"
  13. "github.com/docker/docker/errdefs"
  14. )
  15. func TestImageListError(t *testing.T) {
  16. client := &Client{
  17. client: newMockClient(errorMock(http.StatusInternalServerError, "Server error")),
  18. }
  19. _, err := client.ImageList(context.Background(), types.ImageListOptions{})
  20. if err == nil || err.Error() != "Error response from daemon: Server error" {
  21. t.Fatalf("expected a Server Error, got %v", err)
  22. }
  23. if !errdefs.IsSystem(err) {
  24. t.Fatalf("expected a Server Error, got %T", err)
  25. }
  26. }
  27. func TestImageList(t *testing.T) {
  28. expectedURL := "/images/json"
  29. noDanglingfilters := filters.NewArgs()
  30. noDanglingfilters.Add("dangling", "false")
  31. filters := filters.NewArgs()
  32. filters.Add("label", "label1")
  33. filters.Add("label", "label2")
  34. filters.Add("dangling", "true")
  35. listCases := []struct {
  36. options types.ImageListOptions
  37. expectedQueryParams map[string]string
  38. }{
  39. {
  40. options: types.ImageListOptions{},
  41. expectedQueryParams: map[string]string{
  42. "all": "",
  43. "filter": "",
  44. "filters": "",
  45. },
  46. },
  47. {
  48. options: types.ImageListOptions{
  49. Filters: filters,
  50. },
  51. expectedQueryParams: map[string]string{
  52. "all": "",
  53. "filter": "",
  54. "filters": `{"dangling":{"true":true},"label":{"label1":true,"label2":true}}`,
  55. },
  56. },
  57. {
  58. options: types.ImageListOptions{
  59. Filters: noDanglingfilters,
  60. },
  61. expectedQueryParams: map[string]string{
  62. "all": "",
  63. "filter": "",
  64. "filters": `{"dangling":{"false":true}}`,
  65. },
  66. },
  67. }
  68. for _, listCase := range listCases {
  69. client := &Client{
  70. client: newMockClient(func(req *http.Request) (*http.Response, error) {
  71. if !strings.HasPrefix(req.URL.Path, expectedURL) {
  72. return nil, fmt.Errorf("Expected URL '%s', got '%s'", expectedURL, req.URL)
  73. }
  74. query := req.URL.Query()
  75. for key, expected := range listCase.expectedQueryParams {
  76. actual := query.Get(key)
  77. if actual != expected {
  78. return nil, fmt.Errorf("%s not set in URL query properly. Expected '%s', got %s", key, expected, actual)
  79. }
  80. }
  81. content, err := json.Marshal([]types.ImageSummary{
  82. {
  83. ID: "image_id2",
  84. },
  85. {
  86. ID: "image_id2",
  87. },
  88. })
  89. if err != nil {
  90. return nil, err
  91. }
  92. return &http.Response{
  93. StatusCode: http.StatusOK,
  94. Body: ioutil.NopCloser(bytes.NewReader(content)),
  95. }, nil
  96. }),
  97. }
  98. images, err := client.ImageList(context.Background(), listCase.options)
  99. if err != nil {
  100. t.Fatal(err)
  101. }
  102. if len(images) != 2 {
  103. t.Fatalf("expected 2 images, got %v", images)
  104. }
  105. }
  106. }
  107. func TestImageListApiBefore125(t *testing.T) {
  108. expectedFilter := "image:tag"
  109. client := &Client{
  110. client: newMockClient(func(req *http.Request) (*http.Response, error) {
  111. query := req.URL.Query()
  112. actualFilter := query.Get("filter")
  113. if actualFilter != expectedFilter {
  114. return nil, fmt.Errorf("filter not set in URL query properly. Expected '%s', got %s", expectedFilter, actualFilter)
  115. }
  116. actualFilters := query.Get("filters")
  117. if actualFilters != "" {
  118. return nil, fmt.Errorf("filters should have not been present, were with value: %s", actualFilters)
  119. }
  120. content, err := json.Marshal([]types.ImageSummary{
  121. {
  122. ID: "image_id2",
  123. },
  124. {
  125. ID: "image_id2",
  126. },
  127. })
  128. if err != nil {
  129. return nil, err
  130. }
  131. return &http.Response{
  132. StatusCode: http.StatusOK,
  133. Body: ioutil.NopCloser(bytes.NewReader(content)),
  134. }, nil
  135. }),
  136. version: "1.24",
  137. }
  138. filters := filters.NewArgs()
  139. filters.Add("reference", "image:tag")
  140. options := types.ImageListOptions{
  141. Filters: filters,
  142. }
  143. images, err := client.ImageList(context.Background(), options)
  144. if err != nil {
  145. t.Fatal(err)
  146. }
  147. if len(images) != 2 {
  148. t.Fatalf("expected 2 images, got %v", images)
  149. }
  150. }