diff options
Diffstat (limited to 'internal/command/receivepack/gitalycall_test.go')
-rw-r--r-- | internal/command/receivepack/gitalycall_test.go | 134 |
1 files changed, 70 insertions, 64 deletions
diff --git a/internal/command/receivepack/gitalycall_test.go b/internal/command/receivepack/gitalycall_test.go index eadc43b..9f70189 100644 --- a/internal/command/receivepack/gitalycall_test.go +++ b/internal/command/receivepack/gitalycall_test.go @@ -3,6 +3,7 @@ package receivepack import ( "bytes" "context" + "fmt" "testing" "github.com/stretchr/testify/require" @@ -17,80 +18,85 @@ import ( ) func TestReceivePack(t *testing.T) { - gitalyAddress, testServer := testserver.StartGitalyServer(t) + for _, network := range []string{"unix", "tcp", "dns"} { + t.Run(fmt.Sprintf("via %s network", network), func(t *testing.T) { + gitalyAddress, testServer := testserver.StartGitalyServer(t, network) + t.Log(fmt.Sprintf("Server address: %s", gitalyAddress)) - requests := requesthandlers.BuildAllowedWithGitalyHandlers(t, gitalyAddress) - url := testserver.StartHttpServer(t, requests) + requests := requesthandlers.BuildAllowedWithGitalyHandlers(t, gitalyAddress) + url := testserver.StartHttpServer(t, requests) - testCases := []struct { - username string - keyId string - }{ - { - username: "john.doe", - }, - { - keyId: "123", - }, - } + testCases := []struct { + username string + keyId string + }{ + { + username: "john.doe", + }, + { + keyId: "123", + }, + } - for _, tc := range testCases { - output := &bytes.Buffer{} - input := &bytes.Buffer{} - repo := "group/repo" + for _, tc := range testCases { + output := &bytes.Buffer{} + input := &bytes.Buffer{} + repo := "group/repo" - env := sshenv.Env{ - IsSSHConnection: true, - OriginalCommand: "git-receive-pack " + repo, - RemoteAddr: "127.0.0.1", - } + env := sshenv.Env{ + IsSSHConnection: true, + OriginalCommand: "git-receive-pack " + repo, + RemoteAddr: "127.0.0.1", + } - args := &commandargs.Shell{ - CommandType: commandargs.ReceivePack, - SshArgs: []string{"git-receive-pack", repo}, - Env: env, - } + args := &commandargs.Shell{ + CommandType: commandargs.ReceivePack, + SshArgs: []string{"git-receive-pack", repo}, + Env: env, + } - if tc.username != "" { - args.GitlabUsername = tc.username - } else { - args.GitlabKeyId = tc.keyId - } + if tc.username != "" { + args.GitlabUsername = tc.username + } else { + args.GitlabKeyId = tc.keyId + } - cfg := &config.Config{GitlabUrl: url} - cfg.GitalyClient.InitSidechannelRegistry(context.Background()) - cmd := &Command{ - Config: cfg, - Args: args, - ReadWriter: &readwriter.ReadWriter{ErrOut: output, Out: output, In: input}, - } + cfg := &config.Config{GitlabUrl: url} + cfg.GitalyClient.InitSidechannelRegistry(context.Background()) + cmd := &Command{ + Config: cfg, + Args: args, + ReadWriter: &readwriter.ReadWriter{ErrOut: output, Out: output, In: input}, + } - ctx := correlation.ContextWithCorrelation(context.Background(), "a-correlation-id") - ctx = correlation.ContextWithClientName(ctx, "gitlab-shell-tests") + ctx := correlation.ContextWithCorrelation(context.Background(), "a-correlation-id") + ctx = correlation.ContextWithClientName(ctx, "gitlab-shell-tests") - err := cmd.Execute(ctx) - require.NoError(t, err) + err := cmd.Execute(ctx) + require.NoError(t, err) - if tc.username != "" { - require.Equal(t, "ReceivePack: 1 "+repo, output.String()) - } else { - require.Equal(t, "ReceivePack: key-123 "+repo, output.String()) - } + if tc.username != "" { + require.Equal(t, "ReceivePack: 1 "+repo, output.String()) + } else { + require.Equal(t, "ReceivePack: key-123 "+repo, output.String()) + } - for k, v := range map[string]string{ - "gitaly-feature-cache_invalidator": "true", - "gitaly-feature-inforef_uploadpack_cache": "false", - "x-gitlab-client-name": "gitlab-shell-tests-git-receive-pack", - "key_id": "123", - "user_id": "1", - "remote_ip": "127.0.0.1", - "key_type": "key", - } { - actual := testServer.ReceivedMD[k] - require.Len(t, actual, 1) - require.Equal(t, v, actual[0]) - } - require.Empty(t, testServer.ReceivedMD["some-other-ff"]) - require.Equal(t, testServer.ReceivedMD["x-gitlab-correlation-id"][0], "a-correlation-id") + for k, v := range map[string]string{ + "gitaly-feature-cache_invalidator": "true", + "gitaly-feature-inforef_uploadpack_cache": "false", + "x-gitlab-client-name": "gitlab-shell-tests-git-receive-pack", + "key_id": "123", + "user_id": "1", + "remote_ip": "127.0.0.1", + "key_type": "key", + } { + actual := testServer.ReceivedMD[k] + require.Len(t, actual, 1) + require.Equal(t, v, actual[0]) + } + require.Empty(t, testServer.ReceivedMD["some-other-ff"]) + require.Equal(t, testServer.ReceivedMD["x-gitlab-correlation-id"][0], "a-correlation-id") + } + }) } } |