client/swarm_get_unlock_key_test.go
4f0d95fa
 package client // import "github.com/docker/docker/client"
a86023ea
 
 import (
 	"bytes"
7d62e40f
 	"context"
a86023ea
 	"encoding/json"
 	"fmt"
 	"io/ioutil"
 	"net/http"
 	"strings"
 	"testing"
 
 	"github.com/docker/docker/api/types"
38457285
 	"gotest.tools/assert"
 	is "gotest.tools/assert/cmp"
a86023ea
 )
 
 func TestSwarmGetUnlockKeyError(t *testing.T) {
 	client := &Client{
 		client: newMockClient(errorMock(http.StatusInternalServerError, "Server error")),
 	}
 
 	_, err := client.SwarmGetUnlockKey(context.Background())
55bebbae
 	assert.Check(t, is.ErrorContains(err, "Error response from daemon: Server error"))
a86023ea
 }
 
 func TestSwarmGetUnlockKey(t *testing.T) {
 	expectedURL := "/swarm/unlockkey"
 	unlockKey := "SWMKEY-1-y6guTZNTwpQeTL5RhUfOsdBdXoQjiB2GADHSRJvbXeE"
 
 	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 != "GET" {
 				return nil, fmt.Errorf("expected GET method, got %s", req.Method)
 			}
 
 			key := types.SwarmUnlockKeyResponse{
 				UnlockKey: unlockKey,
 			}
 
 			b, err := json.Marshal(key)
 			if err != nil {
 				return nil, err
 			}
 
 			return &http.Response{
 				StatusCode: http.StatusOK,
 				Body:       ioutil.NopCloser(bytes.NewReader(b)),
 			}, nil
 		}),
 	}
 
 	resp, err := client.SwarmGetUnlockKey(context.Background())
6be0f709
 	assert.NilError(t, err)
 	assert.Check(t, is.Equal(unlockKey, resp.UnlockKey))
a86023ea
 }