mirror of
https://github.com/moby/moby.git
synced 2026-01-11 18:51:37 +00:00
839c2709af
The WithMockClient option was explicitly resetting the client's API version (see [1]), which differs from the regular client, which is initialized with the current API version used by the client (see [2]). This patch: - reduces the `WithMockClient` to only set the custom HTTP client, leaving other fields un-touched. - adds a test utility and updates tests to handle the API-version prefix - removes redundant uses of `WithVersion()` in tests; for most test-cases it was used to make sure a current API version is used that supports the feature being tested, but there was no test to verify the behavior for lower API versions, so we may as well test against "latest". [1]:5a582729d8/client/client_mock_test.go (L22-L36)[2]:5a582729d8/client/client.go (L167-L190)Signed-off-by: Sebastiaan van Stijn <github@gone.nl>
114 lines
3.4 KiB
Go
114 lines
3.4 KiB
Go
package client
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"testing"
|
|
|
|
cerrdefs "github.com/containerd/errdefs"
|
|
"github.com/moby/moby/api/types/image"
|
|
ocispec "github.com/opencontainers/image-spec/specs-go/v1"
|
|
"gotest.tools/v3/assert"
|
|
is "gotest.tools/v3/assert/cmp"
|
|
)
|
|
|
|
func TestImageInspectError(t *testing.T) {
|
|
client, err := NewClientWithOpts(WithMockClient(errorMock(http.StatusInternalServerError, "Server error")))
|
|
assert.NilError(t, err)
|
|
|
|
_, err = client.ImageInspect(context.Background(), "nothing")
|
|
assert.Check(t, is.ErrorType(err, cerrdefs.IsInternal))
|
|
}
|
|
|
|
func TestImageInspectImageNotFound(t *testing.T) {
|
|
client, err := NewClientWithOpts(WithMockClient(errorMock(http.StatusNotFound, "Server error")))
|
|
assert.NilError(t, err)
|
|
|
|
_, err = client.ImageInspect(context.Background(), "unknown")
|
|
assert.Check(t, is.ErrorType(err, cerrdefs.IsNotFound))
|
|
}
|
|
|
|
func TestImageInspectWithEmptyID(t *testing.T) {
|
|
client, err := NewClientWithOpts(WithMockClient(func(req *http.Request) (*http.Response, error) {
|
|
return nil, errors.New("should not make request")
|
|
}))
|
|
assert.NilError(t, err)
|
|
_, err = client.ImageInspect(context.Background(), "")
|
|
assert.Check(t, is.ErrorType(err, cerrdefs.IsNotFound))
|
|
}
|
|
|
|
func TestImageInspect(t *testing.T) {
|
|
const expectedURL = "/images/image_id/json"
|
|
expectedTags := []string{"tag1", "tag2"}
|
|
client, err := NewClientWithOpts(WithMockClient(func(req *http.Request) (*http.Response, error) {
|
|
if err := assertRequest(req, http.MethodGet, expectedURL); err != nil {
|
|
return nil, err
|
|
}
|
|
content, err := json.Marshal(image.InspectResponse{
|
|
ID: "image_id",
|
|
RepoTags: expectedTags,
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &http.Response{
|
|
StatusCode: http.StatusOK,
|
|
Body: io.NopCloser(bytes.NewReader(content)),
|
|
}, nil
|
|
}))
|
|
assert.NilError(t, err)
|
|
|
|
imageInspect, err := client.ImageInspect(context.Background(), "image_id")
|
|
assert.NilError(t, err)
|
|
assert.Check(t, is.Equal(imageInspect.ID, "image_id"))
|
|
assert.Check(t, is.DeepEqual(imageInspect.RepoTags, expectedTags))
|
|
}
|
|
|
|
func TestImageInspectWithPlatform(t *testing.T) {
|
|
const expectedURL = "/images/image_id/json"
|
|
requestedPlatform := &ocispec.Platform{
|
|
OS: "linux",
|
|
Architecture: "arm64",
|
|
}
|
|
|
|
expectedPlatform, err := encodePlatform(requestedPlatform)
|
|
assert.NilError(t, err)
|
|
|
|
client, err := NewClientWithOpts(WithMockClient(func(req *http.Request) (*http.Response, error) {
|
|
if err := assertRequest(req, http.MethodGet, expectedURL); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// Check if platform parameter is passed correctly
|
|
platform := req.URL.Query().Get("platform")
|
|
if platform != expectedPlatform {
|
|
return nil, fmt.Errorf("Expected platform '%s', got '%s'", expectedPlatform, platform)
|
|
}
|
|
|
|
content, err := json.Marshal(image.InspectResponse{
|
|
ID: "image_id",
|
|
Architecture: "arm64",
|
|
Os: "linux",
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &http.Response{
|
|
StatusCode: http.StatusOK,
|
|
Body: io.NopCloser(bytes.NewReader(content)),
|
|
}, nil
|
|
}))
|
|
assert.NilError(t, err)
|
|
|
|
imageInspect, err := client.ImageInspect(context.Background(), "image_id", ImageInspectWithPlatform(requestedPlatform))
|
|
assert.NilError(t, err)
|
|
assert.Check(t, is.Equal(imageInspect.ID, "image_id"))
|
|
assert.Check(t, is.Equal(imageInspect.Architecture, "arm64"))
|
|
assert.Check(t, is.Equal(imageInspect.Os, "linux"))
|
|
}
|