From a4dbfa313cbe2affdffa7947d9eff97fe2e187c2 Mon Sep 17 00:00:00 2001
From: David Zafman <david.zafman@inktank.com>
Date: Wed, 5 Jun 2013 19:14:33 -0700
Subject: [PATCH] test: Add blacklisting to test_stress_watch

Signed-off-by: David Zafman <david.zafman@inktank.com>
---
 src/test/test_stress_watch.cc | 43 ++++++++++++++++++++++++++---------
 1 file changed, 32 insertions(+), 11 deletions(-)

diff --git a/src/test/test_stress_watch.cc b/src/test/test_stress_watch.cc
index d34e9ffb53fc6..1b8f1904d411c 100644
--- a/src/test/test_stress_watch.cc
+++ b/src/test/test_stress_watch.cc
@@ -56,34 +56,55 @@ struct WatcherUnwatcher : public Thread {
 };
 TEST(WatchStress, Stress1) {
   ASSERT_EQ(0, sem_init(&sem, 0, 0));
-  Rados cluster;
+  Rados ncluster;
   std::string pool_name = get_temp_pool_name();
-  ASSERT_EQ("", create_one_pool_pp(pool_name, cluster));
-  IoCtx ioctx;
-  cluster.ioctx_create(pool_name.c_str(), ioctx);
+  ASSERT_EQ("", create_one_pool_pp(pool_name, ncluster));
+  IoCtx nioctx;
+  ncluster.ioctx_create(pool_name.c_str(), nioctx);
   WatchNotifyTestCtx ctx;
 
   WatcherUnwatcher *thr = new WatcherUnwatcher(pool_name);
   thr->create();
-  ASSERT_EQ(0, ioctx.create("foo", false));
+  ASSERT_EQ(0, nioctx.create("foo", false));
 
-  for (int i = 0; i < 10000; ++i) {
+  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_blacklist = i % 2;
+    if (do_blacklist) {
+      cluster.test_blacklist_self(true);
+      std::cerr << "blacklisted" << std::endl;
+      sleep(1);
+    }
+
     bufferlist bl2;
-    ASSERT_EQ(0, ioctx.notify("foo", 0, bl2));
+    ASSERT_EQ(0, nioctx.notify("foo", 0, bl2));
+
+    if (do_blacklist) {
+      sleep(1); // Give a change to see an incorrect notify
+    } else {
+      TestAlarm alarm;
+      sem_wait(&sem);
+    }
+
+    if (do_blacklist) {
+      cluster.test_blacklist_self(false);
+    }
 
-    TestAlarm alarm;
-    sem_wait(&sem);
     ioctx.unwatch("foo", handle);
+    ioctx.close();
   }
   stop_flag.set(1);
   thr->join();
-  ioctx.close();
-  ASSERT_EQ(0, destroy_one_pool_pp(pool_name, cluster));
+  nioctx.close();
+  ASSERT_EQ(0, destroy_one_pool_pp(pool_name, ncluster));
   sem_destroy(&sem);
 }
-- 
2.39.5