Function TestUnixSocketClientStreamsSessionEvents

internal/cli/client_test.go:969–1025

(t *testing.T)

}

func TestUnixSocketClientStreamsSessionEvents(t *testing.T) {
	t.Parallel()

	t.Run("Should stream session events", func(t *testing.T) {
		t.Parallel()

		client := &unixSocketClient{
			socketPath: "/tmp/agh.sock",
			httpClient: &http.Client{
				Transport: roundTripperFunc(func(req *http.Request) (*http.Response, error) {
					if req.Method == http.MethodGet && req.URL.Path == "/api/sessions" {
						return newHTTPResponse(
							http.StatusOK,
							`{"sessions":[{"id":"sess-1","agent_name":"coder","workspace_id":"ws-1","workspace_path":"/tmp","state":"active","created_at":"2026-04-03T12:00:00Z","updated_at":"2026-04-03T12:00:00Z"}]}`,
						), nil
					}
					if req.Method != http.MethodGet || req.URL.Path != "/api/workspaces/ws-1/sessions/sess-1/stream" {
						t.Fatalf("request = %s %s, want GET session stream", req.Method, req.URL.Path)
					}
					if req.Header.Get("Last-Event-ID") != "evt-0" {
						t.Fatalf("Last-Event-ID = %q, want evt-0", req.Header.Get("Last-Event-ID"))
					}
					if req.URL.Query().Get("type") != "tool_call" ||
						req.URL.Query().Get("agent_name") != "coder" ||
						req.URL.Query().Get("limit") != "2" {
						t.Fatalf("stream query = %s, want filters", req.URL.RawQuery)
					}
					return newHTTPResponse(
						http.StatusOK,
						"id: evt-1\n"+
							"event: tool_call\n"+
							`data: {"id":"evt-1","session_id":"sess-1","sequence":1,"type":"tool_call","agent_name":"coder","timestamp":"2026-04-03T12:00:00Z"}`+
							"\n\n",
					), nil
				}),
			},
		}

		var events []SSEEvent
		err := client.StreamSessionEvents(
			context.Background(),
			" sess-1 ",
			SessionEventQuery{Type: "tool_call", AgentName: "coder", Last: 2},
			"evt-0",
			func(event SSEEvent) error {
				events = append(events, event)
				return nil
			},
		)
		if err != nil {
			t.Fatalf("StreamSessionEvents() error = %v", err)
		}
		if len(events) != 1 || events[0].ID != "evt-1" || events[0].Event != "tool_call" {
			t.Fatalf("events = %#v, want one tool_call event", events)
		}
	})
}