Mojoify PoweMonitorMessageBroadcaster IPC from browser to child process

content/browser/power_monitor_message_broadcaster_unittest.cc

 // Copyright 2013 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 "base/macros.h"
 #include "base/message_loop/message_loop.h"
+#include "base/run_loop.h"
 #include "base/test/power_monitor_test_base.h"
 #include "content/browser/power_monitor_message_broadcaster.h"
-#include "content/common/power_monitor_messages.h"
-#include "ipc/ipc_sender.h"
+#include "mojo/public/cpp/bindings/binding.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace content {
 
-class PowerMonitorMessageSender : public IPC::Sender {
+class FakePowerMonitorClient : public device::mojom::PowerMonitorClient {
  public:
-  PowerMonitorMessageSender()
-      : power_state_changes_(0),
+  FakePowerMonitorClient(device::mojom::PowerMonitorClientRequest request)
+      : binding_(this, std::move(request)),
+        power_state_changes_(0),
         suspends_(0),
-        resumes_(0) {
+        resumes_(0) {}
+  ~FakePowerMonitorClient() override {}
+
+  // Implement device::mojom::PowerMonitorClient
+  void PowerStateChange(bool on_battery_power) override {
+    power_state_changes_++;
   }
-  ~PowerMonitorMessageSender() override {}
-
-  bool Send(IPC::Message* msg) override {
-    switch (msg->type()) {
-      case PowerMonitorMsg_Suspend::ID:
-        suspends_++;
-        break;
-      case PowerMonitorMsg_Resume::ID:
-        resumes_++;
-        break;
-      case PowerMonitorMsg_PowerStateChange::ID:
-        power_state_changes_++;
-        break;
-    }
-    delete msg;
-    return true;
-  };
+  void Suspend() override { suspends_++; }
+  void Resume() override { resumes_++; }
 
   // Test status counts.
   int power_state_changes() { return power_state_changes_; }
   int suspends() { return suspends_; }
   int resumes() { return resumes_; }
 
  private:
+  mojo::Binding<device::mojom::PowerMonitorClient> binding_;
   int power_state_changes_;  // Count of OnPowerStateChange notifications.
   int suspends_;  // Count of OnSuspend notifications.
   int resumes_;  // Count of OnResume notifications.
 };
 
 class PowerMonitorMessageBroadcasterTest : public testing::Test {
  protected:
   PowerMonitorMessageBroadcasterTest() {
     power_monitor_source_ = new base::PowerMonitorTestSource();
     power_monitor_.reset(new base::PowerMonitor(
         std::unique_ptr<base::PowerMonitorSource>(power_monitor_source_)));
   }
   ~PowerMonitorMessageBroadcasterTest() override {}
 
   base::PowerMonitorTestSource* source() { return power_monitor_source_; }
   base::PowerMonitor* monitor() { return power_monitor_.get(); }
 
  private:
   base::MessageLoop message_loop_;
   base::PowerMonitorTestSource* power_monitor_source_;
   std::unique_ptr<base::PowerMonitor> power_monitor_;
 
   DISALLOW_COPY_AND_ASSIGN(PowerMonitorMessageBroadcasterTest);
 };
 
 TEST_F(PowerMonitorMessageBroadcasterTest, PowerMessageBroadcast) {
-  PowerMonitorMessageSender sender;
-  PowerMonitorMessageBroadcaster broadcaster(&sender);
+  device::mojom::PowerMonitorClientPtr proxy;
+  FakePowerMonitorClient client(mojo::GetProxy(&proxy));
+  PowerMonitorMessageBroadcaster broadcaster;
 
-  // Calling Init should invoke a power state change.
-  broadcaster.Init();
-  EXPECT_EQ(sender.power_state_changes(), 1);
+  // Calling SetClient should invoke a power state change.
+  broadcaster.SetClient(std::move(proxy));
+
+  base::RunLoop().RunUntilIdle();
+  EXPECT_EQ(client.power_state_changes(), 1);
 
   // Sending resume when not suspended should have no effect.
   source()->GenerateResumeEvent();
-  EXPECT_EQ(sender.resumes(), 0);
+  EXPECT_EQ(client.resumes(), 0);
 
   // Pretend we suspended.
   source()->GenerateSuspendEvent();
-  EXPECT_EQ(sender.suspends(), 1);
+  EXPECT_EQ(client.suspends(), 1);
 
   // Send a second suspend notification.  This should be suppressed.
   source()->GenerateSuspendEvent();
-  EXPECT_EQ(sender.suspends(), 1);
+  EXPECT_EQ(client.suspends(), 1);
 
   // Pretend we were awakened.
   source()->GenerateResumeEvent();
-  EXPECT_EQ(sender.resumes(), 1);
+  EXPECT_EQ(client.resumes(), 1);
 
   // Send a duplicate resume notification.  This should be suppressed.
   source()->GenerateResumeEvent();
-  EXPECT_EQ(sender.resumes(), 1);
+  EXPECT_EQ(client.resumes(), 1);
 
   // Pretend the device has gone on battery power
   source()->GeneratePowerStateEvent(true);
-  EXPECT_EQ(sender.power_state_changes(), 2);
+  EXPECT_EQ(client.power_state_changes(), 2);
 
   // Repeated indications the device is on battery power should be suppressed.
   source()->GeneratePowerStateEvent(true);
-  EXPECT_EQ(sender.power_state_changes(), 2);
+  EXPECT_EQ(client.power_state_changes(), 2);
 
   // Pretend the device has gone off battery power
   source()->GeneratePowerStateEvent(false);
-  EXPECT_EQ(sender.power_state_changes(), 3);
+  EXPECT_EQ(client.power_state_changes(), 3);
 
   // Repeated indications the device is off battery power should be suppressed.
   source()->GeneratePowerStateEvent(false);
-  EXPECT_EQ(sender.power_state_changes(), 3);
+  EXPECT_EQ(client.power_state_changes(), 3);
 }
 
 }  // namespace base