Function TestSession_PostToolUseFailureHook

go/session_test.go:1030–1094

(t *testing.T)

}

func TestSession_PostToolUseFailureHook(t *testing.T) {
	t.Run("dispatches with parsed input and returns additional context", func(t *testing.T) {
		session, cleanup := newTestSession()
		defer cleanup()

		var captured PostToolUseFailureHookInput
		session.registerHooks(&SessionHooks{
			OnPostToolUseFailure: func(input PostToolUseFailureHookInput, _ HookInvocation) (*PostToolUseFailureHookOutput, error) {
				captured = input
				return &PostToolUseFailureHookOutput{
					AdditionalContext: "extra-context: " + input.Error,
				}, nil
			},
		})

		raw := json.RawMessage(`{
			"sessionId": "sess-1",
			"timestamp": 1700000000,
			"cwd": "/work",
			"toolName": "tool-x",
			"toolArgs": {"foo": "bar"},
			"error": "boom"
		}`)
		output, err := session.handleHooksInvoke("postToolUseFailure", raw)
		if err != nil {
			t.Fatalf("unexpected error: %v", err)
		}
		if captured.SessionID != "sess-1" {
			t.Errorf("expected sessionId 'sess-1', got %q", captured.SessionID)
		}
		if captured.ToolName != "tool-x" {
			t.Errorf("expected toolName 'tool-x', got %q", captured.ToolName)
		}
		if captured.Error != "boom" {
			t.Errorf("expected error 'boom', got %q", captured.Error)
		}
		if !captured.Timestamp.Equal(time.UnixMilli(1700000000)) {
			t.Errorf("expected timestamp %v, got %v", time.UnixMilli(1700000000), captured.Timestamp)
		}
		if captured.WorkingDirectory != "/work" {
			t.Errorf("expected WorkingDirectory '/work', got %q", captured.WorkingDirectory)
		}
		out, ok := output.(*PostToolUseFailureHookOutput)
		if !ok {
			t.Fatalf("expected *PostToolUseFailureHookOutput, got %T", output)
		}
		if out.AdditionalContext != "extra-context: boom" {
			t.Errorf("unexpected AdditionalContext: %q", out.AdditionalContext)
		}
	})

	t.Run("no handler registered returns nil without error", func(t *testing.T) {
		session, cleanup := newTestSession()
		defer cleanup()
		session.registerHooks(&SessionHooks{})

		output, err := session.handleHooksInvoke("postToolUseFailure", json.RawMessage(`{"sessionId":"sess-1","timestamp":0,"cwd":"","toolName":"t","toolArgs":null,"error":"e"}`))
		if err != nil {