summaryrefslogtreecommitdiffstats
path: root/src/test/test_stress_watch.cc
diff options
context:
space:
mode:
Diffstat (limited to 'src/test/test_stress_watch.cc')
-rw-r--r--src/test/test_stress_watch.cc123
1 files changed, 123 insertions, 0 deletions
diff --git a/src/test/test_stress_watch.cc b/src/test/test_stress_watch.cc
new file mode 100644
index 000000000..2058bbf1a
--- /dev/null
+++ b/src/test/test_stress_watch.cc
@@ -0,0 +1,123 @@
+#include "include/rados/librados.h"
+#include "include/rados/librados.hpp"
+#include "include/utime.h"
+#include "common/Thread.h"
+#include "common/Clock.h"
+#include "test/librados/test_cxx.h"
+
+#include "gtest/gtest.h"
+#include <semaphore.h>
+#include <errno.h>
+#include <map>
+#include <sstream>
+#include <iostream>
+#include <string>
+#include <atomic>
+
+#include "test/librados/testcase_cxx.h"
+
+
+using namespace librados;
+using std::map;
+using std::ostringstream;
+using std::string;
+
+static sem_t sem;
+static std::atomic<bool> stop_flag = { false };
+
+class WatchNotifyTestCtx : public WatchCtx
+{
+public:
+ void notify(uint8_t opcode, uint64_t ver, bufferlist& bl) override
+ {
+ sem_post(&sem);
+ }
+};
+
+#pragma GCC diagnostic ignored "-Wpragmas"
+#pragma GCC diagnostic push
+#pragma GCC diagnostic ignored "-Wdeprecated-declarations"
+
+struct WatcherUnwatcher : public Thread {
+ string pool;
+ explicit WatcherUnwatcher(string& _pool) : pool(_pool) {}
+
+ void *entry() override {
+ Rados cluster;
+ connect_cluster_pp(cluster);
+ while (!stop_flag) {
+ IoCtx ioctx;
+ cluster.ioctx_create(pool.c_str(), ioctx);
+
+ uint64_t handle;
+ WatchNotifyTestCtx watch_ctx;
+ int r = ioctx.watch("foo", 0, &handle, &watch_ctx);
+ if (r == 0)
+ ioctx.unwatch("foo", handle);
+ ioctx.close();
+ }
+ return NULL;
+ }
+};
+
+typedef RadosTestParamPP WatchStress;
+
+INSTANTIATE_TEST_SUITE_P(WatchStressTests, WatchStress,
+ ::testing::Values("", "cache"));
+
+TEST_P(WatchStress, Stress1) {
+ ASSERT_EQ(0, sem_init(&sem, 0, 0));
+ Rados ncluster;
+ std::string pool_name = get_temp_pool_name();
+ ASSERT_EQ("", create_one_pool_pp(pool_name, ncluster));
+ IoCtx nioctx;
+ ncluster.ioctx_create(pool_name.c_str(), nioctx);
+
+ WatcherUnwatcher *thr = new WatcherUnwatcher(pool_name);
+ thr->create("watcher_unwatch");
+ ASSERT_EQ(0, nioctx.create("foo", false));
+
+ for (unsigned i = 0; i < 75; ++i) {
+ std::cerr << "Iteration " << i << std::endl;
+ uint64_t handle;
+ Rados cluster;
+ IoCtx ioctx;
+ WatchNotifyTestCtx ctx;
+
+ connect_cluster_pp(cluster);
+ cluster.ioctx_create(pool_name.c_str(), ioctx);
+ ASSERT_EQ(0, ioctx.watch("foo", 0, &handle, &ctx));
+
+ bool do_blocklist = i % 2;
+ if (do_blocklist) {
+ cluster.test_blocklist_self(true);
+ std::cerr << "blocklisted" << std::endl;
+ sleep(1);
+ }
+
+ bufferlist bl2;
+ ASSERT_EQ(0, nioctx.notify("foo", 0, bl2));
+
+ if (do_blocklist) {
+ sleep(1); // Give a change to see an incorrect notify
+ } else {
+ TestAlarm alarm;
+ sem_wait(&sem);
+ }
+
+ if (do_blocklist) {
+ cluster.test_blocklist_self(false);
+ }
+
+ ioctx.unwatch("foo", handle);
+ ioctx.close();
+ }
+ stop_flag = true;
+ thr->join();
+ nioctx.close();
+ ASSERT_EQ(0, destroy_one_pool_pp(pool_name, ncluster));
+ sem_destroy(&sem);
+}
+
+#pragma GCC diagnostic pop
+#pragma GCC diagnostic warning "-Wpragmas"