Make ScreenWakeLock ContextLifecycleObserver instead of LocalFrameLifecycleObserver

content/browser/wake_lock/wake_lock_browsertest.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 "base/command_line.h"
 #include "base/test/test_timeouts.h"
 #include "content/browser/wake_lock/wake_lock_service_context.h"
 #include "content/browser/web_contents/web_contents_impl.h"
 #include "content/public/common/content_switches.h"
 #include "content/public/test/browser_test_utils.h"
@@ skipping 248 matching lines @@
 
   NavigateToURL(shell(), test_in_page_url);
   EXPECT_TRUE(HasWakeLock());
 }
 
 IN_PROC_BROWSER_TEST_F(WakeLockTest, UnlockAfterReload) {
   NavigateToURL(shell(), embedded_test_server()->GetURL("/simple_page.html"));
   ScreenWakeLockInMainFrame();
 
   shell()->Reload();
-  WaitForLoadStop(GetWebContents());
+  WaitForPossibleUpdate();
 
   // Screen wake lock should be released after reload.
   EXPECT_FALSE(HasWakeLock());
 }
 
 IN_PROC_BROWSER_TEST_F(WakeLockTest, BrowserInitiatedFrameNavigation) {
   NavigateToURL(shell(),
                 embedded_test_server()->GetURL("/frame_tree/2-4.html"));
 
   EXPECT_FALSE(HasWakeLock());
@@ skipping 87 matching lines @@
       GetNestedFrame()->GetProcess(),
       RenderProcessHostWatcher::WATCH_FOR_PROCESS_EXIT);
   GetNestedFrame()->GetProcess()->Shutdown(0, false);
   watcher.Wait();
 
   // Screen wake lock should be released.
   EXPECT_FALSE(HasWakeLock());
 }
 
 }  // namespace content