Fix GCS related tests (#9783)

This commit is contained in:
Kai Yang
2020-07-30 11:46:36 +08:00
committed by GitHub
parent 260bc52254
commit 9be5a2f0fc
8 changed files with 32 additions and 4 deletions
+4
View File
@@ -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];
+4
View File
@@ -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];