update sources to v12.1.2

[ceph.git] / ceph / src / test / librados_test_stub / MockTestMemRadosClient.h

diff --git a/ceph/src/test/librados_test_stub/MockTestMemRadosClient.h b/ceph/src/test/librados_test_stub/MockTestMemRadosClient.h
index c1f1756e2d1e09c833bb943ae7561a4e30a7258b..5fd0aa9fed988e342abb4839b70e7e1f628f6831 100644 (file)
--- a/ceph/src/test/librados_test_stub/MockTestMemRadosClient.h
+++ b/ceph/src/test/librados_test_stub/MockTestMemRadosClient.h
@@ -35,11 +35,29 @@ public:
     return TestMemRadosClient::blacklist_add(client_address, expire_seconds);
   }
 
+  MOCK_METHOD3(service_daemon_register,
+               int(const std::string&,
+                   const std::string&,
+                   const std::map<std::string,std::string>&));
+  int do_service_daemon_register(const std::string& service,
+                                 const std::string& name,
+                                 const std::map<std::string,std::string>& metadata) {
+    return TestMemRadosClient::service_daemon_register(service, name, metadata);
+  }
+
+  MOCK_METHOD1(service_daemon_update_status,
+               int(const std::map<std::string,std::string>&));
+  int do_service_daemon_update_status(const std::map<std::string,std::string>& status) {
+    return TestMemRadosClient::service_daemon_update_status(status);
+  }
+
   void default_to_dispatch() {
     using namespace ::testing;
 
     ON_CALL(*this, create_ioctx(_, _)).WillByDefault(Invoke(this, &MockTestMemRadosClient::do_create_ioctx));
     ON_CALL(*this, blacklist_add(_, _)).WillByDefault(Invoke(this, &MockTestMemRadosClient::do_blacklist_add));
+    ON_CALL(*this, service_daemon_register(_, _, _)).WillByDefault(Invoke(this, &MockTestMemRadosClient::do_service_daemon_register));
+    ON_CALL(*this, service_daemon_update_status(_)).WillByDefault(Invoke(this, &MockTestMemRadosClient::do_service_daemon_update_status));
   }
 };