dbus: Mock MockObjectProxy::SetNameOwnerChangedCallback().

chromeos/dbus/power_manager_client_unittest.cc

 // Copyright 2016 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chromeos/dbus/power_manager_client.h"
 
 #include <map>
 #include <string>
 
 #include "base/bind.h"
 #include "base/macros.h"
 #include "base/memory/ref_counted.h"
 #include "base/memory/weak_ptr.h"
 #include "base/message_loop/message_loop.h"
 #include "base/run_loop.h"
 #include "chromeos/dbus/power_manager/suspend.pb.h"
 #include "dbus/mock_bus.h"
 #include "dbus/mock_object_proxy.h"
 #include "dbus/object_path.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "third_party/cros_system_api/dbus/service_constants.h"
 
 using ::testing::_;
 using ::testing::Return;
+using ::testing::SaveArg;
 
 namespace chromeos {
 
 namespace {
 
 // Shorthand for a few commonly-used constants.
 const char* kInterface = power_manager::kPowerManagerInterface;
 const char* kSuspendImminent = power_manager::kSuspendImminentSignal;
 const char* kHandleSuspendReadiness =
     power_manager::kHandleSuspendReadinessMethod;
@@ skipping 133 matching lines @@
         dbus::ObjectPath(power_manager::kPowerManagerServicePath));
 
     // |client_|'s Init() method should request a proxy for communicating with
     // powerd.
     EXPECT_CALL(*bus_.get(),
                 GetObjectProxy(
                     power_manager::kPowerManagerServiceName,
                     dbus::ObjectPath(power_manager::kPowerManagerServicePath)))
         .WillRepeatedly(Return(proxy_.get()));
 
-    // Save |client_|'s signal callbacks.
+    // Save |client_|'s signal and name-owner-changed callbacks.
     EXPECT_CALL(*proxy_.get(), ConnectToSignal(kInterface, _, _, _))
         .WillRepeatedly(Invoke(this, &PowerManagerClientTest::ConnectToSignal));
+    EXPECT_CALL(*proxy_.get(), SetNameOwnerChangedCallback(_))
+        .WillRepeatedly(SaveArg<0>(&name_owner_changed_callback_));
 
     // |client_|'s Init() method should register regular and dark suspend
     // delays.
     EXPECT_CALL(
         *proxy_.get(),
         CallMethod(HasMember(power_manager::kRegisterSuspendDelayMethod), _, _))
         .WillRepeatedly(
             Invoke(this, &PowerManagerClientTest::RegisterSuspendDelay));
     EXPECT_CALL(
         *proxy_.get(),
@@ skipping 61 matching lines @@
   // Mock bus and proxy for simulating calls to powerd.
   scoped_refptr<dbus::MockBus> bus_;
   scoped_refptr<dbus::MockObjectProxy> proxy_;
 
   std::unique_ptr<PowerManagerClient> client_;
 
   // Maps from powerd signal name to the corresponding callback provided by
   // |client_|.
   std::map<std::string, dbus::ObjectProxy::SignalCallback> signal_callbacks_;
 
+  // Callback passed to |proxy_|'s SetNameOwnerChangedCallback() method.
+  // TODO(derat): Test that |client_| handles powerd restarts.
+  dbus::ObjectProxy::NameOwnerChangedCallback name_owner_changed_callback_;
+
  private:
   // Handles calls to |proxy_|'s ConnectToSignal() method.
   void ConnectToSignal(
       const std::string& interface_name,
       const std::string& signal_name,
       dbus::ObjectProxy::SignalCallback signal_callback,
       dbus::ObjectProxy::OnConnectedCallback on_connected_callback) {
     CHECK_EQ(interface_name, power_manager::kPowerManagerInterface);
     signal_callbacks_[signal_name] = signal_callback;
 
@@ skipping 93 matching lines @@
   EmitSuspendDoneSignal(kSuspendId);
   EXPECT_EQ(1, delegate.num_suspend_imminent());
   EXPECT_EQ(1, delegate.num_suspend_done());
 }
 
 // TODO(derat): Add more tests, e.g. for SuspendDone being received while
 // readiness callbacks are still outstanding (http://crbug.com/646912) and for
 // the handling of DarkSuspendImminent signals.
 
 }  // namespace chromeos