InTheForest/moby
1
0
Fork 0
mirror of https://github.com/moby/moby.git synced 2026-01-11 18:51:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

client: simplify test with mock-responses

Browse Source 
Signed-off-by: Sebastiaan van Stijn <github@gone.nl>
This commit is contained in:
Sebastiaan van Stijn
2025-10-23 14:20:03 +02:00
parent 1d8c8e192f
commit 47fd987af2
93 changed files with 228 additions and 1087 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
client/image_pull_test.go
View File

@@ -1,7 +1,6 @@
package client
import (
"bytes"
"context"
"errors"
"fmt"
@@ -72,10 +71,7 @@ func TestImagePullWithPrivilegedFuncNoError(t *testing.T) {
}
auth := req.Header.Get(registry.AuthHeader)
if auth == invalidAuth {
return &http.Response{
StatusCode: http.StatusUnauthorized,
Body: io.NopCloser(bytes.NewReader([]byte("Invalid credentials"))),
}, nil
return mockResponse(http.StatusUnauthorized, nil, "Invalid credentials")(req)
}
if auth != validAuth {
return nil, fmt.Errorf("invalid auth header: expected %s, got %s", "IAmValid", auth)
@@ -89,10 +85,7 @@ func TestImagePullWithPrivilegedFuncNoError(t *testing.T) {
if tag != "latest" {
return nil, fmt.Errorf("tag not set in URL query properly. Expected '%s', got %s", "latest", tag)
}
return &http.Response{
StatusCode: http.StatusOK,
Body: io.NopCloser(bytes.NewReader([]byte("hello world"))),
}, nil
return mockResponse(http.StatusOK, nil, "hello world")(req)
}))
assert.NilError(t, err)
resp, err := client.ImagePull(context.Background(), "myimage", ImagePullOptions{
@@ -181,10 +174,7 @@ func TestImagePullWithoutErrors(t *testing.T) {
if tag != pullCase.expectedTag {
return nil, fmt.Errorf("tag not set in URL query properly. Expected '%s', got %s", pullCase.expectedTag, tag)
}
return &http.Response{
StatusCode: http.StatusOK,
Body: io.NopCloser(bytes.NewReader([]byte(expectedOutput))),
}, nil
return mockResponse(http.StatusOK, nil, expectedOutput)(req)
}))
assert.NilError(t, err)
resp, err := client.ImagePull(context.Background(), pullCase.reference, ImagePullOptions{