From e6918187568dbd01842d8d1d2c808ce16a894239 Mon Sep 17 00:00:00 2001
From: Daniel Baumann <daniel.baumann@progress-linux.org>
Date: Sun, 21 Apr 2024 13:54:28 +0200
Subject: Adding upstream version 18.2.2.

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
---
 src/test/test_workqueue.cc | 98 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 98 insertions(+)
 create mode 100644 src/test/test_workqueue.cc

(limited to 'src/test/test_workqueue.cc')

diff --git a/src/test/test_workqueue.cc b/src/test/test_workqueue.cc
new file mode 100644
index 000000000..d3354d633
--- /dev/null
+++ b/src/test/test_workqueue.cc
@@ -0,0 +1,98 @@
+#include "gtest/gtest.h"
+
+#include "common/WorkQueue.h"
+#include "common/ceph_argparse.h"
+
+using namespace std;
+
+TEST(WorkQueue, StartStop)
+{
+  ThreadPool tp(g_ceph_context, "foo", "tp_foo", 10, "");
+  
+  tp.start();
+  tp.pause();
+  tp.pause_new();
+  tp.unpause();
+  tp.unpause();
+  tp.drain();
+  tp.stop();
+}
+
+TEST(WorkQueue, Resize)
+{
+  ThreadPool tp(g_ceph_context, "bar", "tp_bar", 2, "filestore_op_threads");
+  
+  tp.start();
+
+  sleep(1);
+  ASSERT_EQ(2, tp.get_num_threads());
+
+  g_conf().set_val("filestore op threads", "5");
+  g_conf().apply_changes(&cout);
+  sleep(1);
+  ASSERT_EQ(5, tp.get_num_threads());
+
+  g_conf().set_val("filestore op threads", "3");
+  g_conf().apply_changes(&cout);
+  sleep(1);
+  ASSERT_EQ(3, tp.get_num_threads());
+
+  g_conf().set_val("filestore op threads", "0");
+  g_conf().apply_changes(&cout);
+  sleep(1);
+  ASSERT_EQ(0, tp.get_num_threads());
+
+  g_conf().set_val("filestore op threads", "15");
+  g_conf().apply_changes(&cout);
+  sleep(1);
+  ASSERT_EQ(15, tp.get_num_threads());
+
+  g_conf().set_val("filestore op threads", "-1");
+  g_conf().apply_changes(&cout);
+  sleep(1);
+  ASSERT_EQ(15, tp.get_num_threads());
+
+  sleep(1);
+  tp.stop();
+}
+
+class twq : public ThreadPool::WorkQueue<int> {
+public:
+    twq(time_t timeout, time_t suicide_timeout, ThreadPool *tp)
+        : ThreadPool::WorkQueue<int>("test_wq", ceph::make_timespan(timeout), ceph::make_timespan(suicide_timeout), tp) {}
+
+    bool _enqueue(int* item) override {
+        return true;
+    }
+    void _dequeue(int* item) override {
+        ceph_abort();
+    }
+    bool _empty() override {
+        return true;
+    }
+    int *_dequeue() override {
+        return nullptr;
+    }
+    void _process(int *osr, ThreadPool::TPHandle &handle) override {
+    }
+    void _process_finish(int *osr) override {
+    }
+    void _clear() override {
+    }
+};
+
+TEST(WorkQueue, change_timeout){
+    ThreadPool tp(g_ceph_context, "bar", "tp_bar", 2, "filestore_op_threads");
+    tp.start();
+    twq wq(2, 20, &tp);
+    // check timeout and suicide
+    ASSERT_EQ(ceph::make_timespan(2), wq.timeout_interval);
+    ASSERT_EQ(ceph::make_timespan(20), wq.suicide_interval);
+
+    // change the timeout and suicide and then check them
+    wq.set_timeout(4);
+    wq.set_suicide_timeout(40);
+    ASSERT_EQ(ceph::make_timespan(4), wq.timeout_interval);
+    ASSERT_EQ(ceph::make_timespan(40), wq.suicide_interval);
+    tp.stop();
+}
-- 
cgit v1.2.3