Function TestServer_SSEMiddleware_Events

lib/httpapi/server_test.go:614–649

(t *testing.T)

}

func TestServer_SSEMiddleware_Events(t *testing.T) {
	t.Parallel()
	ctx := logctx.WithLogger(context.Background(), slog.New(slog.NewTextHandler(os.Stdout, nil)))
	srv, err := httpapi.NewServer(ctx, httpapi.ServerConfig{
		AgentType:      msgfmt.AgentTypeClaude,
		AgentIO:        nil,
		Port:           0,
		ChatBasePath:   "/chat",
		AllowedHosts:   []string{"*"},
		AllowedOrigins: []string{"*"},
	})
	require.NoError(t, err)
	tsServer := httptest.NewServer(srv.Handler())
	t.Cleanup(tsServer.Close)

	t.Run("events", func(t *testing.T) {
		t.Parallel()
		resp, err := tsServer.Client().Get(tsServer.URL + "/events")
		require.NoError(t, err)
		t.Cleanup(func() {
			_ = resp.Body.Close()
		})
		assertSSEHeaders(t, resp)
	})

	t.Run("internal/screen", func(t *testing.T) {
		t.Parallel()

		resp, err := tsServer.Client().Get(tsServer.URL + "/internal/screen")
		require.NoError(t, err)
		t.Cleanup(func() {
			_ = resp.Body.Close()
		})
		assertSSEHeaders(t, resp)
	})
}

func assertSSEHeaders(t testing.TB, resp *http.Response) {
	t.Helper()