Function TestInMemoryStore_BasicCRUD

pkg/memory/logic/store_test.go:13–66

(t *testing.T)

)

func TestInMemoryStore_BasicCRUD(t *testing.T) {
	store := NewInMemoryStore()
	defer func() { _ = store.Close() }()

	ctx := context.Background()
	namespace := "user:123"

	// 创建 Memory
	mem := &LogicMemory{
		ID:          "mem-1",
		Namespace:   namespace,
		Scope:       ScopeUser,
		Type:        "preference",
		Key:         "writing_tone",
		Value:       map[string]any{"tone": "casual"},
		Description: "用户偏好口语化表达",
		Provenance: &memory.MemoryProvenance{
			SourceType: memory.SourceUserInput,
			Confidence: 0.8,
		},
	}

	// Save
	err := store.Save(ctx, mem)
	require.NoError(t, err)

	// Get
	retrieved, err := store.Get(ctx, namespace, "writing_tone")
	require.NoError(t, err)
	assert.Equal(t, "mem-1", retrieved.ID)
	assert.Equal(t, namespace, retrieved.Namespace)
	assert.Equal(t, "preference", retrieved.Type)
	assert.Equal(t, "用户偏好口语化表达", retrieved.Description)
	assert.NotZero(t, retrieved.CreatedAt)
	assert.NotZero(t, retrieved.UpdatedAt)

	// Update
	mem.Description = "用户偏好简洁口语"
	mem.Provenance.Confidence = 0.9
	err = store.Save(ctx, mem)
	require.NoError(t, err)

	updated, err := store.Get(ctx, namespace, "writing_tone")
	require.NoError(t, err)
	assert.Equal(t, "用户偏好简洁口语", updated.Description)
	assert.Equal(t, 0.9, updated.Provenance.Confidence)

	// Delete
	err = store.Delete(ctx, namespace, "writing_tone")
	require.NoError(t, err)

	_, err = store.Get(ctx, namespace, "writing_tone")
	assert.ErrorIs(t, err, ErrMemoryNotFound)
}

func TestInMemoryStore_List(t *testing.T) {
	store := NewInMemoryStore()