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
|
package repository
import (
"os"
"path"
"testing"
"github.com/stretchr/testify/require"
"gitlab.com/gitlab-org/gitaly-proto/go/gitalypb"
"gitlab.com/gitlab-org/gitaly/internal/config"
"gitlab.com/gitlab-org/gitaly/internal/testhelper"
"golang.org/x/net/context"
"google.golang.org/grpc/codes"
)
func TestSuccessfulRepositorySizeRequest(t *testing.T) {
server, serverSocketPath := runRepoServer(t)
defer server.Stop()
client, conn := newRepositoryClient(t, serverSocketPath)
defer conn.Close()
storageName := "default"
storagePath, found := config.StoragePath(storageName)
if !found {
t.Fatalf("No %q storage was found", storageName)
}
repoCopyPath := path.Join(storagePath, "fixed-size-repo.git")
testhelper.MustRunCommand(t, nil, "cp", "-R", "testdata/fixed-size-repo.git", repoCopyPath)
// run `sync` because some filesystems (e.g. ZFS and BTRFS) do lazy-writes
// which leads to `du` returning 0 bytes used until it's finally written to disk.
testhelper.MustRunCommand(t, nil, "sync")
defer os.RemoveAll(repoCopyPath)
request := &gitalypb.RepositorySizeRequest{
Repository: &gitalypb.Repository{
StorageName: storageName,
RelativePath: "fixed-size-repo.git",
},
}
ctx, cancel := context.WithCancel(context.Background())
defer cancel()
response, err := client.RepositorySize(ctx, request)
require.NoError(t, err)
// We can't test for an exact size because it will be different for systems with different sector sizes,
// so we settle for anything greater than zero.
require.True(t, response.Size > 0, "size must be greater than zero")
}
func TestFailedRepositorySizeRequest(t *testing.T) {
server, serverSocketPath := runRepoServer(t)
defer server.Stop()
client, conn := newRepositoryClient(t, serverSocketPath)
defer conn.Close()
invalidRepo := &gitalypb.Repository{StorageName: "fake", RelativePath: "path"}
testCases := []struct {
description string
repo *gitalypb.Repository
}{
{repo: invalidRepo, description: "Invalid repo"},
}
for _, testCase := range testCases {
t.Run(testCase.description, func(t *testing.T) {
request := &gitalypb.RepositorySizeRequest{
Repository: testCase.repo,
}
ctx, cancel := context.WithCancel(context.Background())
defer cancel()
_, err := client.RepositorySize(ctx, request)
testhelper.RequireGrpcError(t, err, codes.InvalidArgument)
})
}
}
|