[DeviceService] Add end-to-end browsertest for PowerMonitor

content/shell/renderer/shell_content_renderer_client.cc

 // Copyright (c) 2012 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 "content/shell/renderer/shell_content_renderer_client.h"
 
 #include <string>
 
 #include "base/bind.h"
 #include "base/command_line.h"
 #include "base/logging.h"
 #include "base/macros.h"
+#include "base/power_monitor/power_monitor.h"
 #include "components/cdm/renderer/external_clear_key_key_system_properties.h"
 #include "components/web_cache/renderer/web_cache_impl.h"
 #include "content/public/child/child_thread.h"
 #include "content/public/common/service_manager_connection.h"
 #include "content/public/common/simple_connection_filter.h"
 #include "content/public/test/test_service.mojom.h"
+#include "content/shell/common/power_monitor_test.mojom.h"
 #include "content/shell/common/shell_switches.h"
 #include "content/shell/renderer/shell_render_view_observer.h"
 #include "mojo/public/cpp/bindings/binding.h"
+#include "mojo/public/cpp/bindings/strong_binding.h"
 #include "mojo/public/cpp/system/message_pipe.h"
 #include "ppapi/features/features.h"
 #include "services/service_manager/public/cpp/binder_registry.h"
 #include "third_party/WebKit/public/web/WebTestingSupport.h"
 #include "third_party/WebKit/public/web/WebView.h"
 #include "v8/include/v8.h"
 
 #if BUILDFLAG(ENABLE_PLUGINS)
 #include "ppapi/shared_impl/ppapi_switches.h"  // nogncheck
 #endif
@@ skipping 56 matching lines @@
 
   DISALLOW_COPY_AND_ASSIGN(TestServiceImpl);
 };
 
 void CreateTestService(const service_manager::BindSourceInfo& source_info,
                        mojom::TestServiceRequest request) {
   // Owns itself.
   new TestServiceImpl(std::move(request));
 }
 
+class PowerMonitorTestImpl : public base::PowerObserver,
+                             public mojom::PowerMonitorTest {
+ public:
+  static void MakeStrongBinding(
+      std::unique_ptr<PowerMonitorTestImpl> instance,
+      const service_manager::BindSourceInfo& source_info,
+      mojom::PowerMonitorTestRequest request) {
+    mojo::MakeStrongBinding(std::move(instance), std::move(request));
+  }
+
+  PowerMonitorTestImpl() {
+    base::PowerMonitor* power_monitor = base::PowerMonitor::Get();
+    if (power_monitor)
+      power_monitor->AddObserver(this);
+  }
+  ~PowerMonitorTestImpl() override {
+    base::PowerMonitor* power_monitor = base::PowerMonitor::Get();
+    if (power_monitor)
+      power_monitor->RemoveObserver(this);
+  }
+
+ private:
+  // mojom::PowerMonitorTest:
+  void QueryNextState(QueryNextStateCallback callback) override {
+    // Do not allow overlapping call.
+    DCHECK(callback_.is_null());
+    callback_ = std::move(callback);
+
+    if (need_to_report_)
+      ReportState();
+  }
+
+  // base::PowerObserver:
+  void OnPowerStateChange(bool on_battery_power) override {
+    on_battery_power_ = on_battery_power;
+    need_to_report_ = true;
+
+    if (!callback_.is_null())
+      ReportState();
+  }
+  void OnSuspend() override {}
+  void OnResume() override {}
+
+  void ReportState() {
+    std::move(callback_).Run(on_battery_power_);
+    need_to_report_ = false;
+  }
+
+  QueryNextStateCallback callback_;
+  bool on_battery_power_ = false;
+  bool need_to_report_ = false;
+
+  DISALLOW_COPY_AND_ASSIGN(PowerMonitorTestImpl);
+};
+
 }  // namespace
 
 ShellContentRendererClient::ShellContentRendererClient() {}
 
 ShellContentRendererClient::~ShellContentRendererClient() {
 }
 
 void ShellContentRendererClient::RenderThreadStarted() {
   web_cache_impl_.reset(new web_cache::WebCacheImpl());
 
   auto registry = base::MakeUnique<service_manager::BinderRegistry>();
   registry->AddInterface<mojom::TestService>(
       base::Bind(&CreateTestService), base::ThreadTaskRunnerHandle::Get());
+  registry->AddInterface<mojom::PowerMonitorTest>(
+      base::Bind(&PowerMonitorTestImpl::MakeStrongBinding,
+                 base::Passed(base::MakeUnique<PowerMonitorTestImpl>())),
+      base::ThreadTaskRunnerHandle::Get());
   content::ChildThread::Get()
       ->GetServiceManagerConnection()
       ->AddConnectionFilter(
           base::MakeUnique<SimpleConnectionFilter>(std::move(registry)));
 }
 
 void ShellContentRendererClient::RenderViewCreated(RenderView* render_view) {
   new ShellRenderViewObserver(render_view);
 }
 
@@ skipping 31 matching lines @@
     return;
 
   static const char kExternalClearKeyKeySystem[] =
       "org.chromium.externalclearkey";
   key_systems->emplace_back(
       new cdm::ExternalClearKeyProperties(kExternalClearKeyKeySystem));
 }
 #endif
 
 }  // namespace content