Function TestMemoryHandler

go/core/internal/httpserver/handlers/memory_test.go:28–237

(t *testing.T)

}

func TestMemoryHandler(t *testing.T) {
	setupHandler := func(t *testing.T) (*handlers.MemoryHandler, *mockErrorResponseWriter) {
		base := &handlers.Base{
			DefaultModelConfig: types.NamespacedName{Namespace: "default", Name: "default"},
			DatabaseService:    setupTestDBClient(t),
			Authorizer:         &auth.NoopAuthorizer{},
		}
		handler := handlers.NewMemoryHandler(base)
		responseRecorder := newMockErrorResponseWriter()
		return handler, responseRecorder
	}

	t.Run("AddSession", func(t *testing.T) {
		t.Run("Success", func(t *testing.T) {
			handler, responseRecorder := setupHandler(t)

			reqBody := handlers.AddSessionMemoryRequest{
				AgentName: "test-agent",
				UserID:    "user123",
				Content:   "This is a test conversation",
				Vector:    makeVector(768, 0.1),
				Metadata:  json.RawMessage(`{"session_id": "session-abc"}`),
			}

			jsonBody, _ := json.Marshal(reqBody)
			req := httptest.NewRequest("POST", "/api/memories/sessions", bytes.NewBuffer(jsonBody))
			req = setUser(req, "test-user")
			req.Header.Set("Content-Type", "application/json")

			handler.AddSession(responseRecorder, req)

			assert.Equal(t, http.StatusCreated, responseRecorder.Code)
			var response map[string]string
			require.NoError(t, json.Unmarshal(responseRecorder.Body.Bytes(), &response))
			assert.Contains(t, response, "id")
		})

		t.Run("MissingFields", func(t *testing.T) {
			handler, responseRecorder := setupHandler(t)

			reqBody := handlers.AddSessionMemoryRequest{UserID: "user123", Vector: makeVector(768, 0.1)}
			jsonBody, _ := json.Marshal(reqBody)
			req := httptest.NewRequest("POST", "/api/memories/sessions", bytes.NewBuffer(jsonBody))
			req = setUser(req, "test-user")

			handler.AddSession(responseRecorder, req)

			assert.Equal(t, http.StatusBadRequest, responseRecorder.Code)
		})

		t.Run("WrongVectorDimension", func(t *testing.T) {
			handler, responseRecorder := setupHandler(t)

			reqBody := handlers.AddSessionMemoryRequest{
				AgentName: "test-agent",
				UserID:    "user123",
				Vector:    makeVector(16, 0.1), // not 768
			}