Function TEST

tests/unit_test/balance_test.cpp:13–38

@brief Test that stealing from a loaded RR scheduler works correctly. This validates the core primitive that Balance() relies on: Dequeue from source scheduler, Enqueue to destination scheduler.

/// This validates the core primitive that Balance() relies on:
/// Dequeue from source scheduler, Enqueue to destination scheduler.
TEST(BalanceTest, StealFromLoadedScheduler) {
  RoundRobinScheduler source;
  RoundRobinScheduler dest;

  // Source has 4 tasks, dest has 0
  TaskControlBlock task1("Task1", 1, nullptr, nullptr);
  TaskControlBlock task2("Task2", 2, nullptr, nullptr);
  TaskControlBlock task3("Task3", 3, nullptr, nullptr);
  TaskControlBlock task4("Task4", 4, nullptr, nullptr);

  source.Enqueue(&task1);
  source.Enqueue(&task2);
  source.Enqueue(&task3);
  source.Enqueue(&task4);

  EXPECT_EQ(source.GetQueueSize(), 4);
  EXPECT_EQ(dest.GetQueueSize(), 0);

  // Simulate steal: pick from source, enqueue to dest
  auto* stolen = source.PickNext();
  ASSERT_NE(stolen, nullptr);
  dest.Enqueue(stolen);

  EXPECT_EQ(source.GetQueueSize(), 3);
  EXPECT_EQ(dest.GetQueueSize(), 1);
}

/// @brief Test that no stealing occurs when queues are balanced.
TEST(BalanceTest, NoStealWhenBalanced) {