Also check for "refs" folder for repo existence

1 files changed, 25 insertions, 0 deletions

diff --git a/internal/helper/repo_test.go b/internal/helper/repo_test.go
index e826f0bb2..d92886eb4 100644
--- a/internal/helper/repo_test.go
+++ b/internal/helper/repo_test.go
@@ -1,6 +1,8 @@
 package helper
 
 import (
+	"os"
+	"path"
 	"testing"
 
 	"gitlab.com/gitlab-org/gitaly/internal/config"
@@ -129,3 +131,26 @@ func TestGetRepoPath(t *testing.T) {
 		assert.Equal(t, tc.path, path, tc.desc)
 	}
 }
+
+func assertInvalidRepoWithoutFile(t *testing.T, repo *pb.Repository, repoPath, file string) {
+	oldRoute := path.Join(repoPath, file)
+	renamedRoute := path.Join(repoPath, file+"moved")
+	os.Rename(oldRoute, renamedRoute)
+	defer func() {
+		os.Rename(renamedRoute, oldRoute)
+	}()
+
+	_, err := GetRepoPath(repo)
+
+	testhelper.AssertGrpcError(t, err, codes.NotFound, "")
+}
+
+func TestGetRepoPathWithCorruptedRepo(t *testing.T) {
+	testRepo := testhelper.TestRepository()
+	testRepoStoragePath := testhelper.GitlabTestStoragePath()
+	testRepoPath := path.Join(testRepoStoragePath, testRepo.RelativePath)
+
+	for _, file := range []string{"objects", "refs", "HEAD"} {
+		assertInvalidRepoWithoutFile(t, testRepo, testRepoPath, file)
+	}
+}