From 8fa31daeba7a6617f08027a5f9f68bb612587772 Mon Sep 17 00:00:00 2001 From: Paul Buetow Date: Mon, 2 Mar 2026 14:05:16 +0200 Subject: lsp: cancel handler contexts on shutdown via server context (task 423) --- internal/lsp/init_shutdown_test.go | 41 ++++++++++++++++++++++++++++++++++++++ 1 file changed, 41 insertions(+) (limited to 'internal/lsp/init_shutdown_test.go') diff --git a/internal/lsp/init_shutdown_test.go b/internal/lsp/init_shutdown_test.go index 2847170..4e1bd2f 100644 --- a/internal/lsp/init_shutdown_test.go +++ b/internal/lsp/init_shutdown_test.go @@ -2,10 +2,13 @@ package lsp import ( "bytes" + "context" "encoding/json" + "errors" "io" "log" "testing" + "time" ) func TestHandleShutdown_Replies(t *testing.T) { @@ -20,3 +23,41 @@ func TestHandleShutdown_Replies(t *testing.T) { t.Fatalf("unexpected shutdown response: %+v", resp) } } + +func TestHandleShutdown_CancelsServerContext(t *testing.T) { + var out bytes.Buffer + s := NewServer(bytes.NewReader(nil), &out, log.New(io.Discard, "", 0), ServerOptions{}) + req := Request{JSONRPC: "2.0", ID: json.RawMessage("12"), Method: "shutdown"} + s.handleShutdown(req) + + ctx, cancel := s.requestTimeoutContext(2 * time.Second) + defer cancel() + select { + case <-ctx.Done(): + if !errors.Is(ctx.Err(), context.Canceled) { + t.Fatalf("expected canceled context, got %v", ctx.Err()) + } + default: + t.Fatalf("expected canceled context after shutdown") + } +} + +func TestHandleExit_CancelsServerContext(t *testing.T) { + var out bytes.Buffer + s := NewServer(bytes.NewReader(nil), &out, log.New(io.Discard, "", 0), ServerOptions{}) + s.handleExit() + if !s.exited.Load() { + t.Fatalf("expected exited flag to be set") + } + + ctx, cancel := s.requestTimeoutContext(2 * time.Second) + defer cancel() + select { + case <-ctx.Done(): + if !errors.Is(ctx.Err(), context.Canceled) { + t.Fatalf("expected canceled context, got %v", ctx.Err()) + } + default: + t.Fatalf("expected canceled context after exit") + } +} -- cgit v1.2.3