diff options
author | Igor Drozdov <idrozdov@gitlab.com> | 2020-04-07 07:00:04 +0300 |
---|---|---|
committer | Ash McKenzie <amckenzie@gitlab.com> | 2020-04-14 23:38:52 +1000 |
commit | 40b3ddc142b906a79f22c726a0a61dd113c39485 (patch) | |
tree | e0b6d0a37d049dbd01e164a95dc0f8728d51d7e2 /internal/command/receivepack/customaction_test.go | |
parent | 74bef786d44b270167c7f9dbb029d197fe32ac4f (diff) | |
download | gitlab-shell-40b3ddc142b906a79f22c726a0a61dd113c39485.tar.gz |
Extract customaction into a separate moduleid-extract-custom-action-in-separate-module
We'll reuse this module for uploadpack in the future
Diffstat (limited to 'internal/command/receivepack/customaction_test.go')
-rw-r--r-- | internal/command/receivepack/customaction_test.go | 103 |
1 files changed, 0 insertions, 103 deletions
diff --git a/internal/command/receivepack/customaction_test.go b/internal/command/receivepack/customaction_test.go deleted file mode 100644 index c55a8f3..0000000 --- a/internal/command/receivepack/customaction_test.go +++ /dev/null @@ -1,103 +0,0 @@ -package receivepack - -import ( - "bytes" - "encoding/json" - "io/ioutil" - "net/http" - "testing" - - "github.com/stretchr/testify/require" - - "gitlab.com/gitlab-org/gitlab-shell/internal/command/commandargs" - "gitlab.com/gitlab-org/gitlab-shell/internal/command/readwriter" - "gitlab.com/gitlab-org/gitlab-shell/internal/config" - "gitlab.com/gitlab-org/gitlab-shell/internal/gitlabnet/accessverifier" - "gitlab.com/gitlab-org/gitlab-shell/internal/gitlabnet/testserver" -) - -func TestCustomReceivePack(t *testing.T) { - repo := "group/repo" - keyId := "1" - - requests := []testserver.TestRequestHandler{ - { - Path: "/api/v4/internal/allowed", - Handler: func(w http.ResponseWriter, r *http.Request) { - b, err := ioutil.ReadAll(r.Body) - require.NoError(t, err) - - var request *accessverifier.Request - require.NoError(t, json.Unmarshal(b, &request)) - - require.Equal(t, "1", request.KeyId) - - body := map[string]interface{}{ - "status": true, - "gl_id": "1", - "payload": map[string]interface{}{ - "action": "geo_proxy_to_primary", - "data": map[string]interface{}{ - "api_endpoints": []string{"/geo/proxy_git_push_ssh/info_refs", "/geo/proxy_git_push_ssh/push"}, - "gl_username": "custom", - "primary_repo": "https://repo/path", - }, - }, - } - w.WriteHeader(http.StatusMultipleChoices) - require.NoError(t, json.NewEncoder(w).Encode(body)) - }, - }, - { - Path: "/geo/proxy_git_push_ssh/info_refs", - Handler: func(w http.ResponseWriter, r *http.Request) { - b, err := ioutil.ReadAll(r.Body) - require.NoError(t, err) - - var request *Request - require.NoError(t, json.Unmarshal(b, &request)) - - require.Equal(t, request.Data.UserId, "key-"+keyId) - require.Empty(t, request.Output) - - err = json.NewEncoder(w).Encode(Response{Result: []byte("custom")}) - require.NoError(t, err) - }, - }, - { - Path: "/geo/proxy_git_push_ssh/push", - Handler: func(w http.ResponseWriter, r *http.Request) { - b, err := ioutil.ReadAll(r.Body) - require.NoError(t, err) - - var request *Request - require.NoError(t, json.Unmarshal(b, &request)) - - require.Equal(t, request.Data.UserId, "key-"+keyId) - require.Equal(t, "input", string(request.Output)) - - err = json.NewEncoder(w).Encode(Response{Result: []byte("output")}) - require.NoError(t, err) - }, - }, - } - - url, cleanup := testserver.StartSocketHttpServer(t, requests) - defer cleanup() - - outBuf := &bytes.Buffer{} - errBuf := &bytes.Buffer{} - input := bytes.NewBufferString("input") - - cmd := &Command{ - Config: &config.Config{GitlabUrl: url}, - Args: &commandargs.Shell{GitlabKeyId: keyId, CommandType: commandargs.ReceivePack, SshArgs: []string{"git-receive-pack", repo}}, - ReadWriter: &readwriter.ReadWriter{ErrOut: errBuf, Out: outBuf, In: input}, - } - - require.NoError(t, cmd.Execute()) - - // expect printing of info message, "custom" string from the first request - // and "output" string from the second request - require.Equal(t, "customoutput", outBuf.String()) -} |