Migrate CommitService.CommitStats

1 files changed, 41 insertions, 0 deletions

diff --git a/internal/rubyserver/rubyserver_test.go b/internal/rubyserver/rubyserver_test.go
index a638c2ee4..fbc9b033a 100644
--- a/internal/rubyserver/rubyserver_test.go
+++ b/internal/rubyserver/rubyserver_test.go
@@ -1,7 +1,15 @@
 package rubyserver
 
 import (
+	"context"
 	"testing"
+
+	"github.com/stretchr/testify/assert"
+
+	pb "gitlab.com/gitlab-org/gitaly-proto/go"
+	"gitlab.com/gitlab-org/gitaly/internal/testhelper"
+
+	"google.golang.org/grpc/codes"
 )
 
 func TestStopSafe(t *testing.T) {
@@ -14,3 +22,36 @@ func TestStopSafe(t *testing.T) {
 		bs.Stop()
 	}
 }
+
+func TestSetHeaders(t *testing.T) {
+	testRepo := testhelper.TestRepository()
+
+	testCases := []struct {
+		repo    *pb.Repository
+		errType codes.Code
+	}{
+		{
+			repo:    &pb.Repository{StorageName: "foo", RelativePath: "bar.git"},
+			errType: codes.InvalidArgument,
+		},
+		{
+			repo:    &pb.Repository{StorageName: testRepo.GetStorageName(), RelativePath: "bar.git"},
+			errType: codes.NotFound,
+		},
+		{
+			repo:    testRepo,
+			errType: codes.OK,
+		},
+	}
+
+	for _, tc := range testCases {
+		ctx, err := SetHeaders(context.Background(), tc.repo)
+		if tc.errType != codes.OK {
+			testhelper.AssertGrpcError(t, err, tc.errType, "")
+			assert.Nil(t, ctx)
+		} else {
+			assert.NoError(t, err)
+			assert.NotNil(t, ctx)
+		}
+	}
+}