init push

deps/oblib/unittest/lib/task/test_timer.cpp

@@ -0,0 +1,123 @@
+/**
+ * Copyright (c) 2021 OceanBase
+ * OceanBase CE is licensed under Mulan PubL v2.
+ * You can use this software according to the terms and conditions of the Mulan PubL v2.
+ * You may obtain a copy of Mulan PubL v2 at:
+ *          http://license.coscl.org.cn/MulanPubL-2.0
+ * THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+ * EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+ * MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+ * See the Mulan PubL v2 for more details.
+ */
+
+#include <gtest/gtest.h>
+#include "lib/task/ob_timer.h"
+
+namespace oceanbase {
+namespace common {
+
+class TestTimerTask : public ObTimerTask {
+  public:
+  TestTimerTask() : running_(false), task_run_count_(0)
+  {}
+
+  void runTimerTask()
+  {
+    running_ = true;
+    ++task_run_count_;
+    this_routine::usleep(50000);  // 50ms
+    running_ = false;
+  }
+
+  volatile bool running_;
+  int64_t task_run_count_;
+};
+
+TEST(TestTimer, timer_task)
+{
+  ObTimer timer;
+  ASSERT_EQ(OB_SUCCESS, timer.init());
+  ASSERT_EQ(OB_SUCCESS, timer.start());
+  TestTimerTask task[33];
+  const bool is_repeat = true;
+  ASSERT_EQ(OB_SUCCESS, timer.schedule(task[0], 100, is_repeat));
+  for (int i = 1; i < 32; ++i) {
+    ASSERT_EQ(OB_SUCCESS, timer.schedule(task[i], 5000000000, is_repeat));  // 5000s
+  }
+  this_routine::usleep(5000);  // 5ms
+  ASSERT_EQ(OB_ERR_UNEXPECTED, timer.schedule(task[32], 50000000, is_repeat));
+  this_routine::usleep(1000000);  // 1s
+  ASSERT_GT(task[0].task_run_count_, 1);
+}
+
+TEST(TestTimer, task_cancel)
+{
+  ObTimer timer;
+  ASSERT_FALSE(timer.inited());
+  ASSERT_EQ(OB_SUCCESS, timer.init());
+  ASSERT_TRUE(timer.inited());
+  ASSERT_EQ(OB_SUCCESS, timer.start());
+  TestTimerTask task;
+  ASSERT_EQ(OB_SUCCESS, timer.schedule(task, 1000, true));
+  ASSERT_TRUE(timer.task_exist(task));
+  timer.dump();
+  timer.cancel(task);
+  ASSERT_FALSE(timer.task_exist(task));
+
+  ASSERT_EQ(OB_SUCCESS, timer.schedule(task, 1000, false));
+  ASSERT_TRUE(timer.task_exist(task));
+  this_routine::usleep(10000);
+  ASSERT_FALSE(timer.task_exist(task));
+}
+
+TEST(TestTimer, scheduled_immediately_task)
+{
+  ObTimer timer;
+  TestTimerTask task[2];
+  ASSERT_EQ(OB_SUCCESS, timer.init());
+  ASSERT_EQ(OB_SUCCESS, timer.start());
+  ASSERT_EQ(OB_SUCCESS, timer.schedule(task[0], 500000, true));
+  this_routine::usleep(60000);
+  ASSERT_EQ(task[0].task_run_count_, 0);
+
+  ASSERT_EQ(OB_SUCCESS, timer.schedule_repeate_task_immediately(task[1], 50000));
+  this_routine::usleep(100);
+  ASSERT_EQ(task[1].task_run_count_, 1);
+  this_routine::usleep(600000);
+  ASSERT_GT(task[1].task_run_count_, 1);
+}
+
+TEST(TestTimer, start_stop)
+{
+  ObTimer timer;
+  ASSERT_EQ(OB_SUCCESS, timer.init());
+  ASSERT_EQ(OB_SUCCESS, timer.start());
+  TestTimerTask task;
+  ASSERT_EQ(OB_SUCCESS, timer.schedule(task, 0, true));
+  timer.stop();
+  ASSERT_NE(OB_SUCCESS, timer.schedule(task, 0, true));
+  timer.wait();
+
+  timer.cancel_all();
+
+  timer.start();
+  ASSERT_EQ(OB_SUCCESS, timer.schedule(task, 0, true));
+
+  while (!task.running_) {}
+
+  timer.stop();
+  timer.wait();
+
+  timer.start();
+  ASSERT_EQ(OB_SUCCESS, timer.schedule(task, 0, true));
+  timer.destroy();
+}
+
+}  // end namespace common
+}  // end namespace oceanbase
+
+int main(int argc, char** argv)
+{
+  testing::InitGoogleTest(&argc, argv);
+  return RUN_ALL_TESTS();
+}