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/swarm_test: Use functional option to create mock client

Browse Source 
Signed-off-by: Paweł Gronowski <pawel.gronowski@docker.com>
This commit is contained in:
Paweł Gronowski
2025-08-29 15:17:23 +02:00
parent 124bba478a
commit 407af72993
7 changed files with 132 additions and 146 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
client/swarm_leave_test.go
View File

@@ -15,11 +15,10 @@ import (
)
func TestSwarmLeaveError(t *testing.T) {
client := &Client{
client: newMockClient(errorMock(http.StatusInternalServerError, "Server error")),
}
client, err := NewClientWithOpts(WithMockClient(errorMock(http.StatusInternalServerError, "Server error")))
assert.NilError(t, err)
err := client.SwarmLeave(context.Background(), false)
err = client.SwarmLeave(context.Background(), false)
assert.Check(t, is.ErrorType(err, cerrdefs.IsInternal))
}
@@ -40,26 +39,25 @@ func TestSwarmLeave(t *testing.T) {
}
for _, leaveCase := range leaveCases {
client := &Client{
client: newMockClient(func(req *http.Request) (*http.Response, error) {
if !strings.HasPrefix(req.URL.Path, expectedURL) {
return nil, fmt.Errorf("Expected URL '%s', got '%s'", expectedURL, req.URL)
}
if req.Method != http.MethodPost {
return nil, fmt.Errorf("expected POST method, got %s", req.Method)
}
force := req.URL.Query().Get("force")
if force != leaveCase.expectedForce {
return nil, fmt.Errorf("force not set in URL query properly. expected '%s', got %s", leaveCase.expectedForce, force)
}
return &http.Response{
StatusCode: http.StatusOK,
Body: io.NopCloser(bytes.NewReader([]byte(""))),
}, nil
}),
}
client, err := NewClientWithOpts(WithMockClient(func(req *http.Request) (*http.Response, error) {
if !strings.HasPrefix(req.URL.Path, expectedURL) {
return nil, fmt.Errorf("Expected URL '%s', got '%s'", expectedURL, req.URL)
}
if req.Method != http.MethodPost {
return nil, fmt.Errorf("expected POST method, got %s", req.Method)
}
force := req.URL.Query().Get("force")
if force != leaveCase.expectedForce {
return nil, fmt.Errorf("force not set in URL query properly. expected '%s', got %s", leaveCase.expectedForce, force)
}
return &http.Response{
StatusCode: http.StatusOK,
Body: io.NopCloser(bytes.NewReader([]byte(""))),
}, nil
}))
assert.NilError(t, err)
err := client.SwarmLeave(context.Background(), leaveCase.force)
err = client.SwarmLeave(context.Background(), leaveCase.force)
assert.NilError(t, err)
}
}