1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
|
package stats
import (
"fmt"
"net/http"
"path/filepath"
"testing"
"time"
"github.com/stretchr/testify/require"
"gitlab.com/gitlab-org/gitaly/internal/git/gittest"
"gitlab.com/gitlab-org/gitaly/internal/testhelper"
"gitlab.com/gitlab-org/gitaly/internal/testhelper/testcfg"
)
func TestClone(t *testing.T) {
cfg, _, repoPath := testcfg.BuildWithRepo(t)
gitCmdFactory := gittest.NewCommandFactory(t, cfg)
ctx := testhelper.Context(t)
serverPort, stopGitServer := gittest.HTTPServer(ctx, t, gitCmdFactory, repoPath, nil)
defer func() {
require.NoError(t, stopGitServer())
}()
clone, err := PerformHTTPClone(ctx, fmt.Sprintf("http://localhost:%d/%s", serverPort, filepath.Base(repoPath)), "", "", false)
require.NoError(t, err, "perform analysis clone")
const expectedRequests = 90 // based on contents of _support/gitlab-test.git-packed-refs
require.Greater(t, clone.FetchPack.RefsWanted(), expectedRequests, "number of wanted refs")
require.Equal(t, 200, clone.ReferenceDiscovery.HTTPStatus(), "get status")
require.Greater(t, clone.ReferenceDiscovery.Packets(), 0, "number of get packets")
require.Greater(t, clone.ReferenceDiscovery.PayloadSize(), int64(0), "get payload size")
require.Greater(t, len(clone.ReferenceDiscovery.Caps()), 10, "get capabilities")
previousValue := time.Duration(0)
for _, m := range []struct {
desc string
value time.Duration
}{
{"time to receive response header", clone.ReferenceDiscovery.ResponseHeader()},
{"time to first packet", clone.ReferenceDiscovery.FirstGitPacket()},
{"time to receive response body", clone.ReferenceDiscovery.ResponseBody()},
} {
require.True(t, m.value > previousValue, "get: expect %s (%v) to be greater than previous value %v", m.desc, m.value, previousValue)
previousValue = m.value
}
require.Equal(t, 200, clone.FetchPack.HTTPStatus(), "post status")
require.Greater(t, clone.FetchPack.Packets(), 0, "number of post packets")
require.Greater(t, clone.FetchPack.BandPackets("progress"), 0, "number of progress packets")
require.Greater(t, clone.FetchPack.BandPackets("pack"), 0, "number of pack packets")
require.Greater(t, clone.FetchPack.BandPayloadSize("progress"), int64(0), "progress payload bytes")
require.Greater(t, clone.FetchPack.BandPayloadSize("pack"), int64(0), "pack payload bytes")
previousValue = time.Duration(0)
for _, m := range []struct {
desc string
value time.Duration
}{
{"time to receive response header", clone.FetchPack.ResponseHeader()},
{"time to receive NAK", clone.FetchPack.NAK()},
{"time to receive first progress message", clone.FetchPack.BandFirstPacket("progress")},
{"time to receive first pack message", clone.FetchPack.BandFirstPacket("pack")},
{"time to receive response body", clone.FetchPack.ResponseBody()},
} {
require.True(t, m.value > previousValue, "post: expect %s (%v) to be greater than previous value %v", m.desc, m.value, previousValue)
previousValue = m.value
}
}
func TestCloneWithAuth(t *testing.T) {
cfg, _, repoPath := testcfg.BuildWithRepo(t)
gitCmdFactory := gittest.NewCommandFactory(t, cfg)
ctx := testhelper.Context(t)
const (
user = "test-user"
password = "test-password"
)
authWasChecked := false
serverPort, stopGitServer := gittest.HTTPServer(ctx, t, gitCmdFactory, repoPath, func(w http.ResponseWriter, r *http.Request, next http.Handler) {
authWasChecked = true
actualUser, actualPassword, ok := r.BasicAuth()
require.True(t, ok, "request should have basic auth")
require.Equal(t, user, actualUser)
require.Equal(t, password, actualPassword)
next.ServeHTTP(w, r)
})
defer func() {
require.NoError(t, stopGitServer())
}()
_, err := PerformHTTPClone(
ctx,
fmt.Sprintf("http://localhost:%d/%s", serverPort, filepath.Base(repoPath)),
user,
password,
false,
)
require.NoError(t, err, "perform analysis clone")
require.True(t, authWasChecked, "authentication middleware should have gotten triggered")
}
func TestBandToHuman(t *testing.T) {
testCases := []struct {
in byte
out string
fail bool
}{
{in: 0, fail: true},
{in: 1, out: "pack"},
{in: 2, out: "progress"},
{in: 3, out: "error"},
{in: 4, fail: true},
}
for _, tc := range testCases {
t.Run(fmt.Sprintf("band index %d", tc.in), func(t *testing.T) {
out, err := bandToHuman(tc.in)
if tc.fail {
require.Error(t, err)
return
}
require.NoError(t, err)
require.Equal(t, tc.out, out, "band name")
})
}
}
|
