Function TestWorkingMemoryManager_Basic

pkg/memory/working_test.go:12–68

(t *testing.T)

)

func TestWorkingMemoryManager_Basic(t *testing.T) {
	ctx := context.Background()
	backend := backends.NewStateBackend()

	// 创建 Working Memory 管理器（thread scope）
	manager, err := NewWorkingMemoryManager(&WorkingMemoryConfig{
		Backend:  backend,
		BasePath: "/working_memory/",
		Scope:    ScopeThread,
	})
	if err != nil {
		t.Fatalf("create manager: %v", err)
	}

	threadID := "test-thread-1"
	resourceID := "test-resource-1"

	// 初始状态应该为空
	content, err := manager.Get(ctx, threadID, resourceID)
	if err != nil {
		t.Fatalf("get initial: %v", err)
	}
	if content != "" {
		t.Errorf("expected empty content, got: %s", content)
	}

	// 更新 Working Memory
	testContent := "# User Profile\n\nName: Alice\nPreferences: concise code reviews"
	err = manager.Update(ctx, threadID, resourceID, testContent)
	if err != nil {
		t.Fatalf("update: %v", err)
	}

	// 读取更新后的内容
	content, err = manager.Get(ctx, threadID, resourceID)
	if err != nil {
		t.Fatalf("get after update: %v", err)
	}
	if content != testContent {
		t.Errorf("expected %q, got %q", testContent, content)
	}

	// 再次更新（覆盖）
	newContent := "# Updated Profile\n\nName: Alice Smith\nRole: Engineer"
	err = manager.Update(ctx, threadID, resourceID, newContent)
	if err != nil {
		t.Fatalf("update again: %v", err)
	}

	content, err = manager.Get(ctx, threadID, resourceID)
	if err != nil {
		t.Fatalf("get after second update: %v", err)
	}
	if content != newContent {
		t.Errorf("expected %q, got %q", newContent, content)
	}
}