aboutsummaryrefslogtreecommitdiffstatshomepage
path: root/cli/internal/server/server_test.go
diff options
context:
space:
mode:
author简律纯 <hsiangnianian@outlook.com>2023-04-28 01:36:55 +0800
committer简律纯 <hsiangnianian@outlook.com>2023-04-28 01:36:55 +0800
commitfc8c5fdce62fb229202659408798a7b6c98f6e8b (patch)
tree7554f80e50de4af6fd255afa7c21bcdd58a7af34 /cli/internal/server/server_test.go
parentdd84b9d64fb98746a230cd24233ff50a562c39c9 (diff)
downloadHydroRoll-fc8c5fdce62fb229202659408798a7b6c98f6e8b.tar.gz
HydroRoll-fc8c5fdce62fb229202659408798a7b6c98f6e8b.zip
Diffstat (limited to 'cli/internal/server/server_test.go')
-rw-r--r--cli/internal/server/server_test.go73
1 files changed, 0 insertions, 73 deletions
diff --git a/cli/internal/server/server_test.go b/cli/internal/server/server_test.go
deleted file mode 100644
index b7dcf3a..0000000
--- a/cli/internal/server/server_test.go
+++ /dev/null
@@ -1,73 +0,0 @@
-package server
-
-import (
- "context"
- "testing"
- "time"
-
- "github.com/hashicorp/go-hclog"
- "google.golang.org/grpc"
- "gotest.tools/v3/assert"
-
- turbofs "github.com/vercel/turbo/cli/internal/fs"
- "github.com/vercel/turbo/cli/internal/turbodprotocol"
-)
-
-type mockGrpc struct {
- stopped chan struct{}
-}
-
-func (m *mockGrpc) GracefulStop() {
- close(m.stopped)
-}
-
-func (m *mockGrpc) RegisterService(desc *grpc.ServiceDesc, impl interface{}) {}
-
-func TestDeleteRepoRoot(t *testing.T) {
- logger := hclog.Default()
- logger.SetLevel(hclog.Debug)
- repoRootRaw := t.TempDir()
- repoRoot := turbofs.AbsoluteSystemPathFromUpstream(repoRootRaw)
-
- grpcServer := &mockGrpc{
- stopped: make(chan struct{}),
- }
-
- s, err := New("testServer", logger, repoRoot, "some-version", "/log/file/path")
- assert.NilError(t, err, "New")
- s.Register(grpcServer)
-
- // Delete the repo root, ensure that GracefulStop got called
- err = repoRoot.Remove()
- assert.NilError(t, err, "Remove")
-
- select {
- case <-grpcServer.stopped:
- case <-time.After(2 * time.Second):
- t.Error("timed out waiting for graceful stop to be called")
- }
-}
-
-func TestShutdown(t *testing.T) {
- logger := hclog.Default()
- repoRootRaw := t.TempDir()
- repoRoot := turbofs.AbsoluteSystemPathFromUpstream(repoRootRaw)
-
- grpcServer := &mockGrpc{
- stopped: make(chan struct{}),
- }
-
- s, err := New("testServer", logger, repoRoot, "some-version", "/log/file/path")
- assert.NilError(t, err, "New")
- s.Register(grpcServer)
-
- ctx := context.Background()
- _, err = s.Shutdown(ctx, &turbodprotocol.ShutdownRequest{})
- assert.NilError(t, err, "Shutdown")
- // Ensure that graceful stop gets called
- select {
- case <-grpcServer.stopped:
- case <-time.After(2 * time.Second):
- t.Error("timed out waiting for graceful stop to be called")
- }
-}