image_remove_test.go 2.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104
  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/stretchr/testify/assert"
  12. "golang.org/x/net/context"
  13. )
  14. func TestImageRemoveError(t *testing.T) {
  15. client := &Client{
  16. client: newMockClient(errorMock(http.StatusInternalServerError, "Server error")),
  17. }
  18. _, err := client.ImageRemove(context.Background(), "image_id", types.ImageRemoveOptions{})
  19. assert.EqualError(t, err, "Error response from daemon: Server error")
  20. }
  21. func TestImageRemoveImageNotFound(t *testing.T) {
  22. client := &Client{
  23. client: newMockClient(errorMock(http.StatusNotFound, "missing")),
  24. }
  25. _, err := client.ImageRemove(context.Background(), "unknown", types.ImageRemoveOptions{})
  26. assert.EqualError(t, err, "Error: No such image: unknown")
  27. assert.True(t, IsErrNotFound(err))
  28. }
  29. func TestImageRemove(t *testing.T) {
  30. expectedURL := "/images/image_id"
  31. removeCases := []struct {
  32. force bool
  33. pruneChildren bool
  34. expectedQueryParams map[string]string
  35. }{
  36. {
  37. force: false,
  38. pruneChildren: false,
  39. expectedQueryParams: map[string]string{
  40. "force": "",
  41. "noprune": "1",
  42. },
  43. }, {
  44. force: true,
  45. pruneChildren: true,
  46. expectedQueryParams: map[string]string{
  47. "force": "1",
  48. "noprune": "",
  49. },
  50. },
  51. }
  52. for _, removeCase := range removeCases {
  53. client := &Client{
  54. client: newMockClient(func(req *http.Request) (*http.Response, error) {
  55. if !strings.HasPrefix(req.URL.Path, expectedURL) {
  56. return nil, fmt.Errorf("expected URL '%s', got '%s'", expectedURL, req.URL)
  57. }
  58. if req.Method != "DELETE" {
  59. return nil, fmt.Errorf("expected DELETE method, got %s", req.Method)
  60. }
  61. query := req.URL.Query()
  62. for key, expected := range removeCase.expectedQueryParams {
  63. actual := query.Get(key)
  64. if actual != expected {
  65. return nil, fmt.Errorf("%s not set in URL query properly. Expected '%s', got %s", key, expected, actual)
  66. }
  67. }
  68. b, err := json.Marshal([]types.ImageDeleteResponseItem{
  69. {
  70. Untagged: "image_id1",
  71. },
  72. {
  73. Deleted: "image_id",
  74. },
  75. })
  76. if err != nil {
  77. return nil, err
  78. }
  79. return &http.Response{
  80. StatusCode: http.StatusOK,
  81. Body: ioutil.NopCloser(bytes.NewReader(b)),
  82. }, nil
  83. }),
  84. }
  85. imageDeletes, err := client.ImageRemove(context.Background(), "image_id", types.ImageRemoveOptions{
  86. Force: removeCase.force,
  87. PruneChildren: removeCase.pruneChildren,
  88. })
  89. if err != nil {
  90. t.Fatal(err)
  91. }
  92. if len(imageDeletes) != 2 {
  93. t.Fatalf("expected 2 deleted images, got %v", imageDeletes)
  94. }
  95. }
  96. }