Function TestSessionPromptJSONLOutput

internal/cli/session_test.go:900–989

(t *testing.T)

}

func TestSessionPromptJSONLOutput(t *testing.T) {
	t.Run("Should stream prompt events as JSONL without buffering the completed turn", func(t *testing.T) {
		t.Parallel()

		var streamCalled bool
		deps := newTestDeps(t, &stubClient{
			promptSessionFn: func(context.Context, string, string) ([]AgentEventRecord, error) {
				t.Fatal("PromptSession should not be called for prompt -o jsonl")
				return nil, nil
			},
			streamPromptSessionFn: func(_ context.Context, id string, message string, handler SSEHandler) error {
				streamCalled = true
				if id != "sess-1" || message != "hello" {
					t.Fatalf("StreamPromptSession() = (%q, %q), want (sess-1, hello)", id, message)
				}
				events := []SSEEvent{
					{
						Data: mustJSON(t, map[string]any{
							"type":      "start",
							"messageId": "turn-1",
						}),
					},
					{
						Data: mustJSON(t, map[string]any{
							"type": "text-start",
							"id":   "turn-1-text",
						}),
					},
					{
						Data: mustJSON(t, map[string]any{
							"type":  "text-delta",
							"id":    "turn-1-text",
							"delta": "hello back",
						}),
					},
					{
						Data: mustJSON(t, map[string]any{
							"type": "text-end",
							"id":   "turn-1-text",
						}),
					},
					{
						Data: mustJSON(t, map[string]any{
							"type":         "finish",
							"finishReason": "stop",
						}),
					},
					{
						Data: []byte("[DONE]"),
					},
				}
				for _, event := range events {
					if err := handler(event); err != nil {
						return err
					}
				}
				return nil
			},