From acfa2b10c8539a6a9624aa61180f56c47465068d Mon Sep 17 00:00:00 2001
From: heng <heng.chen@vesoft.com>
Date: Wed, 12 Jun 2019 19:40:08 +0800
Subject: [PATCH] Fix failed UT on CI

---
 src/meta/test/BalancerTest.cpp | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/meta/test/BalancerTest.cpp b/src/meta/test/BalancerTest.cpp
index f4c9e30315f..ea4f0a6fe40 100644
--- 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();
 }