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
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
|
package commit
import (
"context"
"testing"
"github.com/stretchr/testify/require"
"gitlab.com/gitlab-org/gitaly/internal/metadata/featureflag"
"gitlab.com/gitlab-org/gitaly/internal/testhelper"
"gitlab.com/gitlab-org/gitaly/proto/go/gitalypb"
"google.golang.org/grpc/codes"
"google.golang.org/grpc/metadata"
)
func TestLanguages(t *testing.T) {
server, serverSocketPath := startTestServices(t)
defer server.Stop()
client, conn := newCommitServiceClient(t, serverSocketPath)
defer conn.Close()
testRepo, _, cleanupFn := testhelper.NewTestRepo(t)
defer cleanupFn()
request := &gitalypb.CommitLanguagesRequest{
Repository: testRepo,
Revision: []byte("cb19058ecc02d01f8e4290b7e79cafd16a8839b6"),
}
ctx, cancel := context.WithCancel(context.Background())
defer cancel()
ctx = EnableLinguistFileCountStatsFeatureFlag(ctx)
resp, err := client.CommitLanguages(ctx, request)
require.NoError(t, err)
require.NotZero(t, len(resp.Languages), "number of languages in response")
expectedLanguages := []gitalypb.CommitLanguagesResponse_Language{
{Name: "Ruby", Share: 66, Color: "#701516", FileCount: 4, Bytes: 2943},
{Name: "JavaScript", Share: 22, Color: "#f1e05a", FileCount: 1, Bytes: 1014},
{Name: "HTML", Share: 7, Color: "#e34c26", FileCount: 1, Bytes: 349},
{Name: "CoffeeScript", Share: 2, Color: "#244776", FileCount: 1, Bytes: 107},
// Modula-2 is a special case because Linguist has no color for it. This
// test case asserts that we invent a color for it (SHA256 of the name).
{Name: "Modula-2", Share: 2, Color: "#3fd5e0", FileCount: 1, Bytes: 95},
}
require.Equal(t, len(expectedLanguages), len(resp.Languages))
for i, el := range expectedLanguages {
actualLanguage := resp.Languages[i]
requireLanguageEqual(t, &el, actualLanguage)
}
}
func TestFileCountIsZeroWhenFeatureIsDisabled(t *testing.T) {
server, serverSocketPath := startTestServices(t)
defer server.Stop()
client, conn := newCommitServiceClient(t, serverSocketPath)
defer conn.Close()
testRepo, _, cleanupFn := testhelper.NewTestRepo(t)
defer cleanupFn()
request := &gitalypb.CommitLanguagesRequest{
Repository: testRepo,
Revision: []byte("cb19058ecc02d01f8e4290b7e79cafd16a8839b6"),
}
ctx, cancel := context.WithCancel(context.Background())
defer cancel()
resp, err := client.CommitLanguages(ctx, request)
require.NoError(t, err)
require.NotZero(t, len(resp.Languages), "number of languages in response")
for i := range resp.Languages {
actualLanguage := resp.Languages[i]
require.Equal(t, uint32(0), actualLanguage.FileCount)
}
}
func requireLanguageEqual(t *testing.T, expected, actual *gitalypb.CommitLanguagesResponse_Language) {
require.Equal(t, expected.Name, actual.Name)
require.Equal(t, expected.Color, actual.Color)
require.False(t, (expected.Share-actual.Share)*(expected.Share-actual.Share) >= 1.0, "shares do not match")
require.Equal(t, expected.FileCount, actual.FileCount)
require.Equal(t, expected.Bytes, actual.Bytes)
}
func TestLanguagesEmptyRevision(t *testing.T) {
server, serverSocketPath := startTestServices(t)
defer server.Stop()
client, conn := newCommitServiceClient(t, serverSocketPath)
defer conn.Close()
testRepo, _, cleanupFn := testhelper.NewTestRepo(t)
defer cleanupFn()
request := &gitalypb.CommitLanguagesRequest{
Repository: testRepo,
}
ctx, cancel := context.WithCancel(context.Background())
defer cancel()
resp, err := client.CommitLanguages(ctx, request)
require.NoError(t, err)
require.NotZero(t, len(resp.Languages), "number of languages in response")
foundRuby := false
for _, l := range resp.Languages {
if l.Name == "Ruby" {
foundRuby = true
}
}
require.True(t, foundRuby, "expected to find Ruby as a language on HEAD")
}
func TestInvalidCommitLanguagesRequestRevision(t *testing.T) {
server, serverSocketPath := startTestServices(t)
defer server.Stop()
client, conn := newCommitServiceClient(t, serverSocketPath)
defer conn.Close()
testRepo, _, cleanupFn := testhelper.NewTestRepo(t)
defer cleanupFn()
ctx, cancel := testhelper.Context()
defer cancel()
_, err := client.CommitLanguages(ctx, &gitalypb.CommitLanguagesRequest{
Repository: testRepo,
Revision: []byte("--output=/meow"),
})
testhelper.RequireGrpcError(t, err, codes.InvalidArgument)
}
func TestAmbiguousRefCommitLanguagesRequestRevision(t *testing.T) {
server, serverSocketPath := startTestServices(t)
defer server.Stop()
client, conn := newCommitServiceClient(t, serverSocketPath)
defer conn.Close()
testRepo, _, cleanupFn := testhelper.NewTestRepo(t)
defer cleanupFn()
ctx, cancel := testhelper.Context()
defer cancel()
// gitlab-test repo has both a branch and a tag named 'v1.1.0'
// b83d6e391c22777fca1ed3012fce84f633d7fed0 refs/heads/v1.1.0
// 8a2a6eb295bb170b34c24c76c49ed0e9b2eaf34b refs/tags/v1.1.0
_, err := client.CommitLanguages(ctx, &gitalypb.CommitLanguagesRequest{
Repository: testRepo,
Revision: []byte("v1.1.0"),
})
require.NoError(t, err)
}
func EnableLinguistFileCountStatsFeatureFlag(ctx context.Context) context.Context {
return metadata.NewOutgoingContext(ctx, metadata.New(map[string]string{
featureflag.HeaderKey(featureflag.LinguistFileCountStats): "true",
}))
}
|
