summaryrefslogtreecommitdiff
path: root/internal/gitlabnet/testserver/gitalyserver.go
diff options
context:
space:
mode:
Diffstat (limited to 'internal/gitlabnet/testserver/gitalyserver.go')
-rw-r--r--internal/gitlabnet/testserver/gitalyserver.go86
1 files changed, 0 insertions, 86 deletions
diff --git a/internal/gitlabnet/testserver/gitalyserver.go b/internal/gitlabnet/testserver/gitalyserver.go
deleted file mode 100644
index 6d0c130..0000000
--- a/internal/gitlabnet/testserver/gitalyserver.go
+++ /dev/null
@@ -1,86 +0,0 @@
-package testserver
-
-import (
- "io/ioutil"
- "net"
- "os"
- "path"
- "path/filepath"
- "testing"
-
- "github.com/stretchr/testify/require"
- "google.golang.org/grpc"
- "google.golang.org/grpc/metadata"
-
- pb "gitlab.com/gitlab-org/gitaly/proto/go/gitalypb"
-)
-
-type TestGitalyServer struct{ ReceivedMD metadata.MD }
-
-func (s *TestGitalyServer) SSHReceivePack(stream pb.SSHService_SSHReceivePackServer) error {
- req, err := stream.Recv()
- if err != nil {
- return err
- }
-
- s.ReceivedMD, _ = metadata.FromIncomingContext(stream.Context())
-
- response := []byte("ReceivePack: " + req.GlId + " " + req.Repository.GlRepository)
- stream.Send(&pb.SSHReceivePackResponse{Stdout: response})
-
- return nil
-}
-
-func (s *TestGitalyServer) SSHUploadPack(stream pb.SSHService_SSHUploadPackServer) error {
- req, err := stream.Recv()
- if err != nil {
- return err
- }
-
- s.ReceivedMD, _ = metadata.FromIncomingContext(stream.Context())
-
- response := []byte("UploadPack: " + req.Repository.GlRepository)
- stream.Send(&pb.SSHUploadPackResponse{Stdout: response})
-
- return nil
-}
-
-func (s *TestGitalyServer) SSHUploadArchive(stream pb.SSHService_SSHUploadArchiveServer) error {
- req, err := stream.Recv()
- if err != nil {
- return err
- }
-
- s.ReceivedMD, _ = metadata.FromIncomingContext(stream.Context())
-
- response := []byte("UploadArchive: " + req.Repository.GlRepository)
- stream.Send(&pb.SSHUploadArchiveResponse{Stdout: response})
-
- return nil
-}
-
-func StartGitalyServer(t *testing.T) (string, *TestGitalyServer, func()) {
- tempDir, _ := ioutil.TempDir("", "gitlab-shell-test-api")
- gitalySocketPath := path.Join(tempDir, "gitaly.sock")
-
- err := os.MkdirAll(filepath.Dir(gitalySocketPath), 0700)
- require.NoError(t, err)
-
- server := grpc.NewServer()
-
- listener, err := net.Listen("unix", gitalySocketPath)
- require.NoError(t, err)
-
- testServer := TestGitalyServer{}
- pb.RegisterSSHServiceServer(server, &testServer)
-
- go server.Serve(listener)
-
- gitalySocketUrl := "unix:" + gitalySocketPath
- cleanup := func() {
- server.Stop()
- os.RemoveAll(tempDir)
- }
-
- return gitalySocketUrl, &testServer, cleanup
-}