[2205] supported manager's shutdown command

diff --git a/src/bin/auth/datasrc_clients_mgr.h b/src/bin/auth/datasrc_clients_mgr.h
index 2afd3cc57cf20cc838c9b1b988b5d3d090fff060..6b5d0172bee01a402b76c53ec599dba621623508 100644 (file)
--- a/src/bin/auth/datasrc_clients_mgr.h
+++ b/src/bin/auth/datasrc_clients_mgr.h
@@ -78,11 +78,16 @@ public:
         builder_thread_(boost::bind(&BuilderType::run, &builder_))
     {}
     ~DataSrcClientsMgrBase() {}
-#ifdef notyet
     void shutdown() {
+        {
+            using namespace internal;
+            typename MutexType::Locker locker(queue_mutex_);
+            command_queue_.push_back(Command(SHUTDOWN,
+                                             data::ConstElementPtr()));
+        }
+        cond_.signal();
         builder_thread_.wait();
     }
-#endif
 
 private:
     std::list<internal::Command> command_queue_;

diff --git a/src/bin/auth/tests/datasrc_clients_mgr_unittest.cc b/src/bin/auth/tests/datasrc_clients_mgr_unittest.cc
index aa5dc93368f1822d97ed6dd215be13ccd088fdf1..00e6926299a49d8b525fe0d26222d85d803c0bb4 100644 (file)
--- a/src/bin/auth/tests/datasrc_clients_mgr_unittest.cc
+++ b/src/bin/auth/tests/datasrc_clients_mgr_unittest.cc
@@ -36,4 +36,26 @@ TEST_F(DataSrcClientsMgrTest, start) {
     EXPECT_TRUE(FakeDataSrcClientsBuilder::command_queue->empty());
 }
 
+TEST_F(DataSrcClientsMgrTest, shutdown) {
+    // Invoke shutdown on the manager.
+    TestDataSrcClientsMgr mgr;
+    EXPECT_TRUE(FakeDataSrcClientsBuilder::started);
+
+    // Check pre-command conditions
+    EXPECT_EQ(0, FakeDataSrcClientsBuilder::cond->signal_count);
+    EXPECT_FALSE(FakeDataSrcClientsBuilder::thread_waited);
+
+    mgr.shutdown();
+
+    // The manager should have acquired the lock, put a SHUTDOWN command
+    // to the queue, and should have signaled the builder.
+    EXPECT_EQ(1, FakeDataSrcClientsBuilder::queue_mutex->lock_count);
+    EXPECT_EQ(1, FakeDataSrcClientsBuilder::cond->signal_count);
+    EXPECT_EQ(1, FakeDataSrcClientsBuilder::command_queue->size());
+    const Command& cmd = FakeDataSrcClientsBuilder::command_queue->front();
+    EXPECT_EQ(SHUTDOWN, cmd.first);
+    EXPECT_FALSE(cmd.second);
+    EXPECT_TRUE(FakeDataSrcClientsBuilder::thread_waited);
+}
+
 } // unnamed namespace

diff --git a/src/bin/auth/tests/test_datasrc_clients_mgr.cc b/src/bin/auth/tests/test_datasrc_clients_mgr.cc
index 9d3565e80efa20bf0b8f7bc9d6df50356b05779d..494fb6cad8532157d11fca718f1a06c9c9a38f34 100644 (file)
--- a/src/bin/auth/tests/test_datasrc_clients_mgr.cc
+++ b/src/bin/auth/tests/test_datasrc_clients_mgr.cc
@@ -21,6 +21,7 @@ bool FakeDataSrcClientsBuilder::started = false;
 std::list<internal::Command>* FakeDataSrcClientsBuilder::command_queue = NULL;
 internal::TestCondVar* FakeDataSrcClientsBuilder::cond = NULL;
 internal::TestMutex* FakeDataSrcClientsBuilder::queue_mutex = NULL;
+bool FakeDataSrcClientsBuilder::thread_waited = false;
 
 namespace internal {
 template<>

diff --git a/src/bin/auth/tests/test_datasrc_clients_mgr.h b/src/bin/auth/tests/test_datasrc_clients_mgr.h
index 57ef57205f93d41b8b823d5f58cd7653d3839f5d..618640edb4c76f82cfcfbcd96452ce5d37864b7b 100644 (file)
--- a/src/bin/auth/tests/test_datasrc_clients_mgr.h
+++ b/src/bin/auth/tests/test_datasrc_clients_mgr.h
@@ -47,7 +47,7 @@ public:
 
 class TestCondVar {
 public:
-    TestCondVar() : wait_count(0), command_queue_(NULL),
+    TestCondVar() : wait_count(0), signal_count(0), command_queue_(NULL),
                     delayed_command_queue_(NULL)
     {}
     TestCondVar(std::list<Command>& command_queue,
@@ -66,7 +66,11 @@ public:
         // make the delayed commands available
         command_queue_->splice(command_queue_->end(), *delayed_command_queue_);
     }
+    void signal() {
+        ++signal_count;
+    }
     size_t wait_count;
+    size_t signal_count;
 private:
     std::list<Command>* command_queue_;
     std::list<Command>* delayed_command_queue_;
@@ -98,15 +102,22 @@ public:
         FakeDataSrcClientsBuilder::command_queue = command_queue;
         FakeDataSrcClientsBuilder::cond = cond;
         FakeDataSrcClientsBuilder::queue_mutex = queue_mutex;
+        FakeDataSrcClientsBuilder::thread_waited = false;
     }
     void run() {
         FakeDataSrcClientsBuilder::started = true;
     }
 
+    // true iff a builder has started.
     static bool started;
+
+    // These three correspond to the resource shared with the manager.
     static std::list<internal::Command>* command_queue;
     static internal::TestCondVar* cond;
     static internal::TestMutex* queue_mutex;
+
+    // true iff the manager waited on the thread running the builder.
+    static bool thread_waited;
 };
 
 class TestThread {
@@ -114,6 +125,9 @@ public:
     TestThread(const boost::function<void()>& main) {
         main();
     }
+    void wait() {
+        FakeDataSrcClientsBuilder::thread_waited = true;
+    }
 };
 
 // Convenient shortcut