Function TestAddError_SQLiteRoundTrip

pkg/session/store_test.go:899–934

(t *testing.T)

}

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

	tempDB := filepath.Join(t.TempDir(), "test_add_error.db")

	store, err := NewSQLiteSessionStore(t.Context(), tempDB)
	require.NoError(t, err)
	defer store.(*SQLiteSessionStore).Close()

	session := &Session{ID: "error-session", CreatedAt: time.Now()}
	require.NoError(t, store.AddSession(t.Context(), session))

	_, err = store.AddMessage(t.Context(), "error-session", UserMessage("do something"))
	require.NoError(t, err)

	err = store.AddError(t.Context(), "error-session", &Error{
		Message:   "model stream failed",
		Code:      "model_error",
		AgentName: "root",
		CreatedAt: time.Now().UTC().Format(time.RFC3339),
	})
	require.NoError(t, err)

	retrieved, err := store.GetSession(t.Context(), "error-session")
	require.NoError(t, err)
	require.Len(t, retrieved.Messages, 2)

	errItem := retrieved.Messages[1]
	require.True(t, errItem.IsError())
	assert.Equal(t, "model stream failed", errItem.Error.Message)
	assert.Equal(t, "model_error", errItem.Error.Code)
	assert.Equal(t, "root", errItem.Error.AgentName)

	// Errors are not conversation messages, so GetAllMessages skips them.
	assert.Len(t, retrieved.GetAllMessages(), 1)
}

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