AndroidOverlay implementation using Dialog.

content/renderer/render_frame_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 <stdint.h>
 
 #include "base/command_line.h"
 #include "base/debug/leak_annotations.h"
 #include "base/strings/utf_string_conversions.h"
 #include "build/build_config.h"
@@ skipping 86 matching lines @@
   RenderFrameImpl* GetMainRenderFrame() {
     return static_cast<RenderFrameImpl*>(view_->GetMainRenderFrame());
   }
 
   RenderFrameImpl* frame() { return frame_; }
 
   content::RenderWidget* frame_widget() const {
     return frame_->render_widget_.get();
   }
 
+#if defined(OS_ANDROID)
+  void ReceiveOverlayRoutingToken(const base::UnguessableToken& token) {
+    overlay_routing_token_ = token;
+  }
+
+  base::Optional<base::UnguessableToken> overlay_routing_token_;
+#endif
+
  private:
   RenderFrameImpl* frame_;
   FakeCompositorDependencies compositor_deps_;
 };
 
 class RenderFrameTestObserver : public RenderFrameObserver {
  public:
   explicit RenderFrameTestObserver(RenderFrame* render_frame)
       : RenderFrameObserver(render_frame), visible_(false) {}
 
@@ skipping 252 matching lines @@
 TEST_F(RenderFrameImplTest, NoCrashWhenDeletingFrameDuringFind) {
   blink::WebFindOptions options;
   options.force = true;
   FrameMsg_Find find_message(0, 1, base::ASCIIToUTF16("foo"), options);
   frame()->OnMessageReceived(find_message);
 
   FrameMsg_Delete delete_message(0);
   frame()->OnMessageReceived(delete_message);
 }
 
+#if defined(OS_ANDROID)
+// Verify that RFI defers token requests if the token hasn't arrived yet.
+TEST_F(RenderFrameImplTest, TestOverlayRoutingTokenSendsLater) {
+  ASSERT_FALSE(overlay_routing_token_.has_value());
+
+  frame()->GetOverlayRoutingToken(
+      base::Bind(&RenderFrameImplTest::ReceiveOverlayRoutingToken,
+                 base::Unretained(this)));
+  ASSERT_FALSE(overlay_routing_token_.has_value());
+
+  // The host should receive a request for it sent to the frame.
+  const IPC::Message* msg = render_thread_->sink().GetFirstMessageMatching(
+      FrameHostMsg_RequestOverlayRoutingToken::ID);
+  EXPECT_TRUE(msg);
+
+  // Send a token.
+  base::UnguessableToken token = base::UnguessableToken::Create();
+  FrameMsg_SetOverlayRoutingToken token_message(
+      0, token.GetHighForSerialization(), token.GetLowForSerialization());
+  frame()->OnMessageReceived(token_message);
+
+  ProcessPendingMessages();
+  ASSERT_TRUE(overlay_routing_token_.has_value());
+  ASSERT_EQ(overlay_routing_token_.value(), token);
+}
+
+// Verify that RFI sends tokens if they're already available.
+TEST_F(RenderFrameImplTest, TestOverlayRoutingTokenSendsNow) {
+  ASSERT_FALSE(overlay_routing_token_.has_value());
+  base::UnguessableToken token = base::UnguessableToken::Create();
+  FrameMsg_SetOverlayRoutingToken token_message(
+      0, token.GetHighForSerialization(), token.GetLowForSerialization());
+  frame()->OnMessageReceived(token_message);
+
+  // The frame now has a token.  We don't care if it sends the token before
+  // returning or posts a message.
+  ProcessPendingMessages();
+  frame()->GetOverlayRoutingToken(
+      base::Bind(&RenderFrameImplTest::ReceiveOverlayRoutingToken,
+                 base::Unretained(this)));
+  ASSERT_TRUE(overlay_routing_token_.has_value());
+  ASSERT_EQ(overlay_routing_token_.value(), token);
+
+  // Since the token already arrived, a request for it shouldn't be sent.
+  const IPC::Message* msg = render_thread_->sink().GetFirstMessageMatching(
+      FrameHostMsg_RequestOverlayRoutingToken::ID);
+  EXPECT_FALSE(msg);
+}
+#endif
+
 }  // namespace