Function TestAccessLogIncludesChatID

agent/agentproc/api_test.go:146–172

(t *testing.T)

}

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

	sink := testutil.NewFakeSink(t)
	logger := sink.Logger()
	api := agentproc.NewAPI(logger, agentexec.DefaultExecer, nil, nil, nil)
	t.Cleanup(func() {
		_ = api.Close()
	})
	handler := tracing.StatusWriterMiddleware(loggermw.Logger(logger, nil)(
		agentchat.Middleware(api.Routes()),
	))

	chatID := uuid.New().String()
	w := getListWithChatHeader(t, handler, chatID)
	require.Equal(t, http.StatusOK, w.Code)

	entries := sink.Entries(func(entry slog.SinkEntry) bool {
		return entry.Message == http.MethodGet
	})
	require.Len(t, entries, 1)
	fields := make(map[string]any, len(entries[0].Fields))
	for _, field := range entries[0].Fields {
		fields[field.Name] = field.Value
	}
	require.Equal(t, chatID, fields["chat_id"])
}

// waitForExit polls the output endpoint until the process is
// no longer running or the context expires.