Function TestMiddlewareAccessLog

agent/agentchat/log_test.go:19–43

(t *testing.T)

)

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

	chatID := uuid.New()
	ancestorID := uuid.New()
	sink := testutil.NewFakeSink(t)
	handler := tracing.StatusWriterMiddleware(loggermw.Logger(sink.Logger(), nil)(
		agentchat.Middleware(http.HandlerFunc(func(rw http.ResponseWriter, _ *http.Request) {
			rw.WriteHeader(http.StatusNoContent)
		})),
	))

	req := httptest.NewRequest(http.MethodGet, "/test", nil)
	req.Header.Set(workspacesdk.CoderChatIDHeader, chatID.String())
	req.Header.Set(workspacesdk.CoderAncestorChatIDsHeader, mustMarshalJSON(t, []string{ancestorID.String()}))
	rw := httptest.NewRecorder()
	handler.ServeHTTP(rw, req)
	require.Equal(t, http.StatusNoContent, rw.Code)

	entries := sink.Entries()
	require.Len(t, entries, 1)
	fields := fieldsByName(entries[0].Fields)
	require.Equal(t, chatID.String(), fields["chat_id"])
	require.Equal(t, []string{ancestorID.String()}, fields["ancestor_chat_ids"])
}

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