Expose the visibility state of a frame to the browser process.

content/browser/frame_host/render_frame_host_impl_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 "content/browser/frame_host/render_frame_host_impl.h"
 #include "content/public/browser/render_frame_host.h"
 #include "content/public/browser/web_contents.h"
+#include "content/public/common/content_client.h"
 #include "content/public/test/content_browser_test.h"
 #include "content/public/test/content_browser_test_utils.h"
 #include "content/public/test/test_utils.h"
 #include "content/shell/browser/shell.h"
+#include "content/test/test_content_browser_client.h"
 
 namespace content {
 
 namespace {
 
 RenderFrameHostImpl* ToRFHI(RenderFrameHost* render_frame_host) {
   return static_cast<RenderFrameHostImpl*>(render_frame_host);
 }
 
+// Implementation of ContentBrowserClient that overrides
+// OverridePageVisibilityState() and allows consumers to set a value.
+class PrerenderTestContentBrowserClient : public TestContentBrowserClient {
+ public:
+  PrerenderTestContentBrowserClient()
+    : override_enabled_(false),
+      visibility_override_(blink::WebPageVisibilityStateVisible)
+  {}
+  ~PrerenderTestContentBrowserClient() override {}
+
+  void EnableVisibilityOverride(
+      blink::WebPageVisibilityState visibility_override) {
+    override_enabled_ = true;
+    visibility_override_ = visibility_override;
+  }
+
+  void OverridePageVisibilityState(
+      RenderFrameHost* render_frame_host,
+      blink::WebPageVisibilityState* visibility_state) override {
+    if (override_enabled_)
+      *visibility_state = visibility_override_;
+  }
+
+ private:
+  bool override_enabled_;
+  blink::WebPageVisibilityState visibility_override_;
+
+  DISALLOW_COPY_AND_ASSIGN(PrerenderTestContentBrowserClient);
+};
+
 }  // anonymous namespace
 
 using RenderFrameHostImplBrowserTest = ContentBrowserTest;
 
 // Test that when creating a new window, the main frame is correctly focused.
 IN_PROC_BROWSER_TEST_F(RenderFrameHostImplBrowserTest, IsFocused_AtLoad) {
   EXPECT_TRUE(
       NavigateToURL(shell(), GetTestUrl("render_frame_host", "focus.html")));
 
   // The main frame should be focused.
@@ skipping 40 matching lines @@
   // is not focused.
   EXPECT_EQ(web_contents->GetMainFrame(), web_contents->GetFocusedFrame());
 
 #if defined(OS_MACOSX)
   EXPECT_TRUE(ToRFHI(web_contents->GetMainFrame())->IsFocused());
 #else
   EXPECT_FALSE(ToRFHI(web_contents->GetMainFrame())->IsFocused());
 #endif
 }
 
+// Test that a frame is visible/hidden depending on its WebContents visibility
+// state.
+IN_PROC_BROWSER_TEST_F(RenderFrameHostImplBrowserTest,
+                       GetVisibilityState_Basic) {
+  EXPECT_TRUE(NavigateToURL(shell(), GURL("data:text/html,foo")));
+  WebContents* web_contents = shell()->web_contents();
+
+  EXPECT_EQ(blink::WebPageVisibilityStateVisible,
+            web_contents->GetMainFrame()->GetVisibilityState());
+
+  web_contents->WasHidden();
+  EXPECT_EQ(blink::WebPageVisibilityStateHidden,
+            web_contents->GetMainFrame()->GetVisibilityState());
+}
+
+// Test that a frame visibility can be overridden by the ContentBrowserClient.
+IN_PROC_BROWSER_TEST_F(RenderFrameHostImplBrowserTest,
+                       GetVisibilityState_Override) {
+  EXPECT_TRUE(NavigateToURL(shell(), GURL("data:text/html,foo")));
+  WebContents* web_contents = shell()->web_contents();
+
+  PrerenderTestContentBrowserClient new_client;
+  ContentBrowserClient* old_client = SetBrowserClientForTesting(&new_client);
+
+  EXPECT_EQ(blink::WebPageVisibilityStateVisible,
+            web_contents->GetMainFrame()->GetVisibilityState());
+
+  new_client.EnableVisibilityOverride(blink::WebPageVisibilityStatePrerender);
+  EXPECT_EQ(blink::WebPageVisibilityStatePrerender,
+            web_contents->GetMainFrame()->GetVisibilityState());
+
+  SetBrowserClientForTesting(old_client);
+}
+
 }  // namespace content