diff options
Diffstat (limited to 'internal/gitlabnet/lfsauthenticate')
| -rw-r--r-- | internal/gitlabnet/lfsauthenticate/client.go | 5 | ||||
| -rw-r--r-- | internal/gitlabnet/lfsauthenticate/client_test.go | 5 |
2 files changed, 6 insertions, 4 deletions
diff --git a/internal/gitlabnet/lfsauthenticate/client.go b/internal/gitlabnet/lfsauthenticate/client.go index fffc225..834cbe1 100644 --- a/internal/gitlabnet/lfsauthenticate/client.go +++ b/internal/gitlabnet/lfsauthenticate/client.go @@ -1,6 +1,7 @@ package lfsauthenticate import ( + "context" "fmt" "net/http" "strings" @@ -40,7 +41,7 @@ func NewClient(config *config.Config, args *commandargs.Shell) (*Client, error) return &Client{config: config, client: client, args: args}, nil } -func (c *Client) Authenticate(operation, repo, userId string) (*Response, error) { +func (c *Client) Authenticate(ctx context.Context, operation, repo, userId string) (*Response, error) { request := &Request{Operation: operation, Repo: repo} if c.args.GitlabKeyId != "" { request.KeyId = c.args.GitlabKeyId @@ -48,7 +49,7 @@ func (c *Client) Authenticate(operation, repo, userId string) (*Response, error) request.UserId = strings.TrimPrefix(userId, "user-") } - response, err := c.client.Post("/lfs_authenticate", request) + response, err := c.client.Post(ctx, "/lfs_authenticate", request) if err != nil { return nil, err } diff --git a/internal/gitlabnet/lfsauthenticate/client_test.go b/internal/gitlabnet/lfsauthenticate/client_test.go index 82e364b..2bd0451 100644 --- a/internal/gitlabnet/lfsauthenticate/client_test.go +++ b/internal/gitlabnet/lfsauthenticate/client_test.go @@ -1,6 +1,7 @@ package lfsauthenticate import ( + "context" "encoding/json" "io/ioutil" "net/http" @@ -85,7 +86,7 @@ func TestFailedRequests(t *testing.T) { operation := tc.args.SshArgs[2] - _, err = client.Authenticate(operation, repo, "") + _, err = client.Authenticate(context.Background(), operation, repo, "") require.Error(t, err) require.Equal(t, tc.expectedOutput, err.Error()) @@ -119,7 +120,7 @@ func TestSuccessfulRequests(t *testing.T) { client, err := NewClient(&config.Config{GitlabUrl: url}, args) require.NoError(t, err) - response, err := client.Authenticate(operation, repo, "") + response, err := client.Authenticate(context.Background(), operation, repo, "") require.NoError(t, err) expectedResponse := &Response{ |