Wake Lock API implementation (Chromium part)

content/browser/wake_lock/wake_lock_dispatcher_host_unittest.cc

+// Copyright 2015 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/browser/wake_lock/wake_lock_dispatcher_host.h"
+
+#include "base/memory/scoped_ptr.h"
+#include "base/process/kill.h"
+#include "content/browser/web_contents/web_contents_impl.h"
+#include "content/public/browser/web_contents.h"
+#include "content/public/test/test_renderer_host.h"
+
+namespace content {
+
+class RenderFrameHost;
+
+class WakeLockDispatcherHostTest : public RenderViewHostTestHarness {
+ protected:
+  void RequestWakeLock(RenderFrameHost* rfh) {
+    ASSERT_TRUE(GetWakeLockDispatcherHost());
+    GetWakeLockDispatcherHost()->RequestWakeLock(rfh);
+  }
+
+  void CancelWakeLock(RenderFrameHost* rfh) {
+    ASSERT_TRUE(GetWakeLockDispatcherHost());
+    GetWakeLockDispatcherHost()->CancelWakeLock(rfh);
+  }
+
+  WakeLockDispatcherHost* GetWakeLockDispatcherHost() {
+    WebContentsImpl* web_contents_impl =
+        static_cast<WebContentsImpl*>(web_contents());
+    return web_contents_impl->wake_lock_dispatcher_host();
+  }
+};
+
+TEST_F(WakeLockDispatcherHostTest, NoLockInitially) {
+  ASSERT_TRUE(GetWakeLockDispatcherHost());
+  EXPECT_FALSE(GetWakeLockDispatcherHost()->has_blocker_for_testing());
+}
+
+TEST_F(WakeLockDispatcherHostTest, LockUnlock) {
+  ASSERT_TRUE(GetWakeLockDispatcherHost());
+  ASSERT_TRUE(web_contents());
+  ASSERT_TRUE(main_rfh());
+  // Request wake lock for main frame
+  RequestWakeLock(main_rfh());
+  // Should set the blocker
+  EXPECT_TRUE(GetWakeLockDispatcherHost()->has_blocker_for_testing());
+  // Remove wake lock request for main frame
+  CancelWakeLock(main_rfh());
+  // Should remove the blocker
+  EXPECT_FALSE(GetWakeLockDispatcherHost()->has_blocker_for_testing());
+}
+
+TEST_F(WakeLockDispatcherHostTest, RenderFrameDeleted) {
+  ASSERT_TRUE(GetWakeLockDispatcherHost());
+  ASSERT_TRUE(web_contents());
+  ASSERT_TRUE(main_rfh());
+  // Request wake lock for main frame
+  RequestWakeLock(main_rfh());
+  // Should set the blocker
+  EXPECT_TRUE(GetWakeLockDispatcherHost()->has_blocker_for_testing());
+  // Simulate render frame deletion
+  GetWakeLockDispatcherHost()->RenderFrameDeleted(main_rfh());
+  // Should remove the blocker
+  EXPECT_FALSE(GetWakeLockDispatcherHost()->has_blocker_for_testing());
+}
+
+TEST_F(WakeLockDispatcherHostTest, RenderFrameHostChanged) {
+  ASSERT_TRUE(GetWakeLockDispatcherHost());
+  ASSERT_TRUE(web_contents());
+  ASSERT_TRUE(main_rfh());
+  // Request wake lock for main frame
+  RequestWakeLock(main_rfh());
+  // Should set the blocker
+  EXPECT_TRUE(GetWakeLockDispatcherHost()->has_blocker_for_testing());
+  // Simulate render frame host change
+  GetWakeLockDispatcherHost()->RenderFrameHostChanged(main_rfh(), nullptr);
+  // Should remove the blocker
+  EXPECT_FALSE(GetWakeLockDispatcherHost()->has_blocker_for_testing());
+}
+
+}  // namespace content