From e95f3fdf0a66ba05ba2c8fb7e755e107f9cf7991 Mon Sep 17 00:00:00 2001 From: Paul Buetow Date: Thu, 11 Jun 2026 08:34:41 +0300 Subject: Thread context.Context through blocking I/O entry points Accept ctx as the first parameter on the blocking I/O entry points and propagate it to downstream blocking calls so the work is cancellable from the process entry point: - appconfig.Load / LoadWithOptions: honor ctx before the blocking file reads, returning defaults on a cancelled context. - LSP: lsp.Server.Run(ctx) ties the serve loop to the caller context via a new watchParentContext bridge (cancels the server context, aborting in-flight LLM work). Threaded through hexailsp.Run/RunWithConfig/ RunWithFactory and runtimeconfig.Store.Reload. - MCP: mcp.Server.Run(ctx) stops accepting requests once ctx is cancelled; threaded through hexaimcp.Run/RunWithFactory/RunBackfill. - editor: RunEditor/OpenTempAndEdit/OpenFile take ctx and use exec.CommandContext so a cancelled context kills the editor subprocess; threaded through hexaicli, hexaiaction and askcli call sites. Top-level callers (cmd/hexai-lsp-server, cmd/hexai-mcp-server) now build a signal-cancelled context (SIGINT/SIGTERM) so shutdown tears the run down cleanly. Updated comments to explain the cancellation flow and added cancellation tests for the LSP/MCP loops, editor, and config load. All tests pass with -race; cross-package coverage 86.2%. Co-Authored-By: Claude Opus 4.8 --- internal/hexaimcp/run_test.go | 29 +++++++++++++++-------------- 1 file changed, 15 insertions(+), 14 deletions(-) (limited to 'internal/hexaimcp/run_test.go') diff --git a/internal/hexaimcp/run_test.go b/internal/hexaimcp/run_test.go index 6bedbfc..09f4f87 100644 --- a/internal/hexaimcp/run_test.go +++ b/internal/hexaimcp/run_test.go @@ -3,6 +3,7 @@ package hexaimcp import ( "bytes" + "context" "encoding/json" "fmt" "io" @@ -22,7 +23,7 @@ type mockServerRunner struct { runFunc func() error } -func (m *mockServerRunner) Run() error { +func (m *mockServerRunner) Run(context.Context) error { if m.runFunc != nil { return m.runFunc() } @@ -66,7 +67,7 @@ func TestFullProtocolFlow(t *testing.T) { overrides := MCPOverrides{PromptsDir: tmpDir} // Note: This will hang waiting for more input, which is expected - _ = RunWithFactory("", "", overrides, inBuf, outBuf, errBuf, serverFactory) + _ = RunWithFactory(context.Background(), "", "", overrides, inBuf, outBuf, errBuf, serverFactory) }() // Give server time to process @@ -231,14 +232,14 @@ func TestLoadConfig(t *testing.T) { logger := log.New(io.Discard, "", 0) t.Run("loads default config when path empty", func(t *testing.T) { - cfg := loadConfig(logger, "") + cfg := loadConfig(context.Background(), logger, "") // Should return a valid config (may be defaults) // Just verify it returns without panic _ = cfg }) t.Run("loads config with nonexistent path", func(t *testing.T) { - cfg := loadConfig(logger, "/nonexistent/config.yaml") + cfg := loadConfig(context.Background(), logger, "/nonexistent/config.yaml") // Should return default config without error // Just verify it returns without panic _ = cfg @@ -281,7 +282,7 @@ func TestRun(t *testing.T) { // Pass prompts dir via overrides instead of environment variable overrides := MCPOverrides{PromptsDir: tmpDir} - err := RunWithFactory(logPath, "", overrides, inBuf, outBuf, errBuf, mockFactory) + err := RunWithFactory(context.Background(), logPath, "", overrides, inBuf, outBuf, errBuf, mockFactory) if err != nil { t.Fatalf("RunWithFactory() error = %v", err) } @@ -312,7 +313,7 @@ func TestRunWithFactory_ServerError(t *testing.T) { // Pass prompts dir via overrides instead of environment variable overrides := MCPOverrides{PromptsDir: tmpDir} - err := RunWithFactory(logPath, "", overrides, inBuf, outBuf, errBuf, mockFactory) + err := RunWithFactory(context.Background(), logPath, "", overrides, inBuf, outBuf, errBuf, mockFactory) if err == nil { t.Fatal("RunWithFactory() expected error, got nil") } @@ -331,7 +332,7 @@ func TestRunWithFactory_LoggerError(t *testing.T) { return &mockServerRunner{} } - err := RunWithFactory(badLogPath, "", MCPOverrides{}, &bytes.Buffer{}, &bytes.Buffer{}, &bytes.Buffer{}, mockFactory) + err := RunWithFactory(context.Background(), badLogPath, "", MCPOverrides{}, &bytes.Buffer{}, &bytes.Buffer{}, &bytes.Buffer{}, mockFactory) if err == nil { t.Fatal("expected error for invalid log path, got nil") } @@ -353,7 +354,7 @@ func TestRunWithFactory_StderrLogger(t *testing.T) { overrides := MCPOverrides{PromptsDir: tmpDir} // Empty logPath causes logger to write to stderr (no file to close) - err := RunWithFactory("", "", overrides, &bytes.Buffer{}, &bytes.Buffer{}, &bytes.Buffer{}, mockFactory) + err := RunWithFactory(context.Background(), "", "", overrides, &bytes.Buffer{}, &bytes.Buffer{}, &bytes.Buffer{}, mockFactory) if err != nil { t.Fatalf("RunWithFactory() error = %v", err) } @@ -371,7 +372,7 @@ func TestRun_CallsDefaultFactory(t *testing.T) { // Run with empty stdin — the real server hits EOF and exits cleanly. // This exercises the full Run -> RunWithFactory -> defaultServerFactory path. - err := Run(logPath, "", overrides, &bytes.Buffer{}, &bytes.Buffer{}, &bytes.Buffer{}) + err := Run(context.Background(), logPath, "", overrides, &bytes.Buffer{}, &bytes.Buffer{}, &bytes.Buffer{}) // The server may return nil or an error depending on how it handles EOF; // the important thing is that Run() itself does not panic. _ = err @@ -524,7 +525,7 @@ func TestRunBackfill_FullHappyPath(t *testing.T) { // RunBackfill should succeed: config sets MCPSlashCommandDir, prompts // dir exists, and SyncAll on an empty store is a no-op. - err := RunBackfill(logPath, cfgPath, overrides) + err := RunBackfill(context.Background(), logPath, cfgPath, overrides) if err != nil { t.Fatalf("RunBackfill() error = %v", err) } @@ -550,7 +551,7 @@ func TestRunBackfill_CreateSyncerError(t *testing.T) { t.Fatalf("cannot write config: %v", err) } - err := RunBackfill(logPath, cfgPath, MCPOverrides{}) + err := RunBackfill(context.Background(), logPath, cfgPath, MCPOverrides{}) if err == nil { t.Fatal("expected error for invalid slash command dir, got nil") } @@ -580,7 +581,7 @@ func TestRunBackfill_StderrLogger(t *testing.T) { overrides := MCPOverrides{PromptsDir: promptsDir} // Empty logPath — logger writes to stderr, defer close is a no-op - err := RunBackfill("", cfgPath, overrides) + err := RunBackfill(context.Background(), "", cfgPath, overrides) if err != nil { t.Fatalf("RunBackfill() error = %v", err) } @@ -589,7 +590,7 @@ func TestRunBackfill_StderrLogger(t *testing.T) { // TestRunBackfill_LoggerError verifies RunBackfill returns an error when // the log path is invalid. func TestRunBackfill_LoggerError(t *testing.T) { - err := RunBackfill("/dev/null/impossible/test.log", "", MCPOverrides{}) + err := RunBackfill(context.Background(), "/dev/null/impossible/test.log", "", MCPOverrides{}) if err == nil { t.Fatal("expected error for invalid log path, got nil") } @@ -617,7 +618,7 @@ func TestRunBackfill_NoCmdDir(t *testing.T) { defer os.Setenv("HEXAI_MCP_SLASHCOMMAND_DIR", oldEnv) os.Setenv("HEXAI_MCP_SLASHCOMMAND_DIR", "") - err := RunBackfill(logPath, emptyCfgPath, MCPOverrides{}) + err := RunBackfill(context.Background(), logPath, emptyCfgPath, MCPOverrides{}) if err == nil { t.Fatal("expected error for empty slash command dir, got nil") } -- cgit v1.2.3