Function TestRoom_SendTo

pkg/core/room_test.go:347–404

TestRoom_SendTo 测试点对点消息

(t *testing.T)

// TestRoom_SendTo 测试点对点消息
func TestRoom_SendTo(t *testing.T) {
	deps := createTestDeps(t)
	pool := NewPool(&PoolOptions{
		Dependencies: deps,
		MaxAgents:    10,
	})
	defer func() {
		if err := pool.Shutdown(); err != nil {
			t.Errorf("Shutdown failed: %v", err)
		}
	}()

	room := NewRoom(pool)
	ctx := context.Background()

	// 创建 Agent
	config1 := createTestConfig("agent-1")
	_, err := pool.Create(ctx, config1)
	if err != nil {
		t.Fatalf("Failed to create agent-1: %v", err)
	}

	config2 := createTestConfig("agent-2")
	_, err = pool.Create(ctx, config2)
	if err != nil {
		t.Fatalf("Failed to create agent-2: %v", err)
	}

	// 加入 Room
	if err := room.Join("alice", "agent-1"); err != nil {
		t.Fatalf("Failed to join room: %v", err)
	}
	if err := room.Join("bob", "agent-2"); err != nil {
		t.Fatalf("Failed to join room: %v", err)
	}

	// 发送点对点消息
	err = room.SendTo(ctx, "alice", "bob", "Private message")
	if err != nil {
		t.Fatalf("Failed to send private message: %v", err)
	}

	// 检查历史
	history := room.GetHistory()
	if len(history) != 1 {
		t.Errorf("Expected 1 message in history, got %d", len(history))
	}

	if len(history) > 0 {
		msg := history[0]
		if msg.From != "alice" {
			t.Errorf("Expected message from alice, got %s", msg.From)
		}
		if len(msg.To) != 1 || msg.To[0] != "bob" {
			t.Errorf("Expected message to bob, got %v", msg.To)
		}
	}
}