mirror of
https://github.com/wassname/ray.git
synced 2026-06-27 19:16:19 +08:00
Fix GCS related tests (#9783)
This commit is contained in:
@@ -41,6 +41,10 @@ void TestSetupUtil::StartUpRedisServers(const std::vector<int> &redis_server_por
|
||||
int TestSetupUtil::StartUpRedisServer(const int &port) {
|
||||
int actual_port = port;
|
||||
if (port == 0) {
|
||||
static std::atomic<bool> srand_called(false);
|
||||
if (!srand_called.exchange(true)) {
|
||||
srand(current_time_ms() % RAND_MAX);
|
||||
}
|
||||
// Use random port (in range [2000, 7000) to avoid port conflicts between UTs.
|
||||
actual_port = rand() % 5000 + 2000;
|
||||
}
|
||||
|
||||
@@ -68,8 +68,8 @@ class GlobalStateAccessorTest : public ::testing::Test {
|
||||
void TearDown() override {
|
||||
gcs_server_->Stop();
|
||||
io_service_->stop();
|
||||
gcs_server_.reset();
|
||||
thread_io_service_->join();
|
||||
gcs_server_.reset();
|
||||
|
||||
gcs_client_->Disconnect();
|
||||
global_state_->Disconnect();
|
||||
@@ -272,6 +272,10 @@ TEST_F(GlobalStateAccessorTest, TestWorkerTable) {
|
||||
} // namespace ray
|
||||
|
||||
int main(int argc, char **argv) {
|
||||
InitShutdownRAII ray_log_shutdown_raii(ray::RayLog::StartRayLog,
|
||||
ray::RayLog::ShutDownRayLog, argv[0],
|
||||
ray::RayLogLevel::INFO,
|
||||
/*log_dir=*/"");
|
||||
::testing::InitGoogleTest(&argc, argv);
|
||||
RAY_CHECK(argc == 4);
|
||||
ray::TEST_REDIS_SERVER_EXEC_PATH = argv[1];
|
||||
|
||||
@@ -71,8 +71,8 @@ class ServiceBasedGcsClientTest : public ::testing::Test {
|
||||
|
||||
gcs_server_->Stop();
|
||||
server_io_service_->stop();
|
||||
gcs_server_.reset();
|
||||
server_io_service_thread_->join();
|
||||
gcs_server_.reset();
|
||||
TestSetupUtil::FlushAllRedisServers();
|
||||
client_io_service_thread_->join();
|
||||
}
|
||||
@@ -81,8 +81,8 @@ class ServiceBasedGcsClientTest : public ::testing::Test {
|
||||
RAY_LOG(INFO) << "Stopping GCS service, port = " << gcs_server_->GetPort();
|
||||
gcs_server_->Stop();
|
||||
server_io_service_->stop();
|
||||
gcs_server_.reset();
|
||||
server_io_service_thread_->join();
|
||||
gcs_server_.reset();
|
||||
RAY_LOG(INFO) << "Finished stopping GCS service.";
|
||||
|
||||
server_io_service_.reset(new boost::asio::io_service());
|
||||
@@ -1182,6 +1182,10 @@ TEST_F(ServiceBasedGcsClientTest, TestMultiThreadSubAndUnsub) {
|
||||
} // namespace ray
|
||||
|
||||
int main(int argc, char **argv) {
|
||||
InitShutdownRAII ray_log_shutdown_raii(ray::RayLog::StartRayLog,
|
||||
ray::RayLog::ShutDownRayLog, argv[0],
|
||||
ray::RayLogLevel::INFO,
|
||||
/*log_dir=*/"");
|
||||
::testing::InitGoogleTest(&argc, argv);
|
||||
RAY_CHECK(argc == 4);
|
||||
ray::TEST_REDIS_SERVER_EXEC_PATH = argv[1];
|
||||
|
||||
@@ -57,8 +57,8 @@ class GcsServerTest : public ::testing::Test {
|
||||
void TearDown() override {
|
||||
gcs_server_->Stop();
|
||||
io_service_.stop();
|
||||
gcs_server_.reset();
|
||||
thread_io_service_->join();
|
||||
gcs_server_.reset();
|
||||
}
|
||||
|
||||
bool AddJob(const rpc::AddJobRequest &request) {
|
||||
|
||||
@@ -60,6 +60,10 @@ TEST_F(RedisStoreClientTest, AsyncGetAllAndBatchDeleteTest) {
|
||||
} // namespace ray
|
||||
|
||||
int main(int argc, char **argv) {
|
||||
InitShutdownRAII ray_log_shutdown_raii(ray::RayLog::StartRayLog,
|
||||
ray::RayLog::ShutDownRayLog, argv[0],
|
||||
ray::RayLogLevel::INFO,
|
||||
/*log_dir=*/"");
|
||||
::testing::InitGoogleTest(&argc, argv);
|
||||
RAY_CHECK(argc == 4);
|
||||
ray::TEST_REDIS_SERVER_EXEC_PATH = argv[1];
|
||||
|
||||
@@ -74,6 +74,10 @@ TEST_F(RedisAsioTest, TestRedisCommands) {
|
||||
} // namespace ray
|
||||
|
||||
int main(int argc, char **argv) {
|
||||
InitShutdownRAII ray_log_shutdown_raii(ray::RayLog::StartRayLog,
|
||||
ray::RayLog::ShutDownRayLog, argv[0],
|
||||
ray::RayLogLevel::INFO,
|
||||
/*log_dir=*/"");
|
||||
::testing::InitGoogleTest(&argc, argv);
|
||||
RAY_CHECK(argc == 4);
|
||||
ray::TEST_REDIS_SERVER_EXEC_PATH = argv[1];
|
||||
|
||||
@@ -1492,6 +1492,10 @@ TEST_F(TestGcsWithAsio, TestHashTable) {
|
||||
} // namespace ray
|
||||
|
||||
int main(int argc, char **argv) {
|
||||
InitShutdownRAII ray_log_shutdown_raii(ray::RayLog::StartRayLog,
|
||||
ray::RayLog::ShutDownRayLog, argv[0],
|
||||
ray::RayLogLevel::INFO,
|
||||
/*log_dir=*/"");
|
||||
::testing::InitGoogleTest(&argc, argv);
|
||||
RAY_CHECK(argc == 4);
|
||||
ray::TEST_REDIS_SERVER_EXEC_PATH = argv[1];
|
||||
|
||||
@@ -144,6 +144,10 @@ TEST_F(RedisObjectInfoAccessorTest, TestGetAddRemove) {
|
||||
} // namespace ray
|
||||
|
||||
int main(int argc, char **argv) {
|
||||
InitShutdownRAII ray_log_shutdown_raii(ray::RayLog::StartRayLog,
|
||||
ray::RayLog::ShutDownRayLog, argv[0],
|
||||
ray::RayLogLevel::INFO,
|
||||
/*log_dir=*/"");
|
||||
::testing::InitGoogleTest(&argc, argv);
|
||||
RAY_CHECK(argc == 4);
|
||||
ray::TEST_REDIS_SERVER_EXEC_PATH = argv[1];
|
||||
|
||||
Reference in New Issue
Block a user