Fix failed UT on CI

vesoft-inc · Jun 12, 2019 · acfa2b1 · acfa2b1
1 parent fe82c25
commit acfa2b1
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/meta/test/BalancerTest.cpp b/src/meta/test/BalancerTest.cpp
@@ -5,6 +5,8 @@
  */
 #include "base/Base.h"
 #include <gtest/gtest.h>
+#include <sys/time.h>
+#include <sys/resource.h>
 #include "meta/processors/admin/Balancer.h"
 
 namespace nebula {
@@ -86,6 +88,7 @@ TEST(BalanceTaskTest, SimpleTest) {
         ASSERT_EQ(BalanceTask::Result::FAILED, task.ret_);
         ASSERT_EQ(BalanceTask::Status::CHANGE_LEADER, task.status_);
     }
+    LOG(INFO) << "Test finished!";
 }
 
 }  // namespace meta
@@ -96,6 +99,9 @@ int main(int argc, char** argv) {
     testing::InitGoogleTest(&argc, argv);
     folly::init(&argc, &argv, true);
     google::SetStderrLogging(google::INFO);
+    struct ::rlimit rl;
+    rl.rlim_max = 32L * 1024L * 1024L;
+    ::setrlimit(::RLIMIT_STACK, &rl);
     return RUN_ALL_TESTS();
 }