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 IsPrerendering() and
+// allows consumers to set an arbitrary value. Note that by default,
+// ContentBrowserClient returns false for IsPrerendering() so unless
+// set_prerender_override() is called, this will have no effect.
+class PrerenderTestContentBrowserClient : public TestContentBrowserClient {
+ public:
+  PrerenderTestContentBrowserClient()
+    : prerender_override_(false)
+  {}
+  ~PrerenderTestContentBrowserClient() override {}
+
+  void set_prerender_override(bool prerender_override) {
+    prerender_override_ = prerender_override;
+  }
+
+  bool IsPrerendering(RenderFrameHost* render_frame_host) override {
+    return prerender_override_;
+  }
+
+ private:
+  bool prerender_override_;
+};
+
 }  // 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 the focused frame according to WebContents.
@@ skipping 37 matching lines @@
       NavigateToURL(new_shell, GetTestUrl("render_frame_host", "focus.html")));
   EXPECT_TRUE(ToRFHI(new_shell->web_contents()->GetMainFrame())->IsFocused());
 
   // The first opened window is no longer focused. The main frame is still the
   // focused frame in the frame tree but as far as RFH is concerned, the frame
   // is not focused.
   EXPECT_EQ(web_contents->GetMainFrame(), web_contents->GetFocusedFrame());
   EXPECT_FALSE(ToRFHI(web_contents->GetMainFrame())->IsFocused());
 }
 
+// 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,
+            ToRFHI(web_contents->GetMainFrame())->GetVisibilityState());
+
+  web_contents->WasHidden();
+  EXPECT_EQ(blink::WebPageVisibilityStateHidden,
+            ToRFHI(web_contents->GetMainFrame())->GetVisibilityState());
+}
+
+// Test that a frame is marked as prerendered if
+// ContentBrowserClient::IsPrerendering() returns true.
+IN_PROC_BROWSER_TEST_F(RenderFrameHostImplBrowserTest,
+                       GetVisibilityState_Prerender) {
+  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,
+            ToRFHI(web_contents->GetMainFrame())->GetVisibilityState());
+
+  new_client.set_prerender_override(true);
+  EXPECT_EQ(blink::WebPageVisibilityStatePrerender,
+            ToRFHI(web_contents->GetMainFrame())->GetVisibilityState());
+
+  SetBrowserClientForTesting(old_client);
+}
+
 }  // namespace content