Connects PostMessage() for OOPIFs

components/html_viewer/html_frame_apptest.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/bind.h"
 #include "base/command_line.h"
 #include "base/json/json_reader.h"
 #include "base/run_loop.h"
 #include "base/strings/stringprintf.h"
 #include "base/test/test_timeouts.h"
@@ skipping 68 matching lines @@
 }  // namespace
 
 class HTMLFrameTest : public ViewManagerTestBase {
  public:
   HTMLFrameTest() {}
   ~HTMLFrameTest() override {}
 
  protected:
   // Creates the frame tree showing an empty page at the root and adds (via
   // script) a frame showing the same empty page.
-  Frame* LoadEmptyPageAndCreateFrame() {
+  Frame* LoadEmptyPageAndCreateFrame(mandoline::FrameTreeDelegate* delegate) {
     View* embed_view = window_manager()->CreateView();
     FrameConnection* root_connection = InitFrameTree(
-        embed_view, nullptr, "http://127.0.0.1:%u/files/empty_page2.html");
+        embed_view, delegate, "http://127.0.0.1:%u/files/empty_page2.html");
     const std::string frame_text =
         GetFrameText(root_connection->application_connection());
     if (frame_text != "child2") {
       ADD_FAILURE() << "unexpected text " << frame_text;
       return nullptr;
     }
 
     return CreateEmptyChildFrame(frame_tree_->root());
   }
 
@@ skipping 219 matching lines @@
   ASSERT_TRUE(child_frame->user_data());
   ASSERT_EQ("child2",
             GetFrameText(static_cast<FrameConnection*>(child_frame->user_data())
                              ->application_connection()));
 }
 
 TEST_F(HTMLFrameTest, DynamicallyAddFrameAndVerifyParent) {
   if (!EnableOOPIFs())
     return;
 
-  Frame* child_frame = LoadEmptyPageAndCreateFrame();
+  Frame* child_frame = LoadEmptyPageAndCreateFrame(nullptr);
   ASSERT_TRUE(child_frame);
 
   mojo::ApplicationConnection* child_frame_connection =
       ApplicationConnectionForFrame(child_frame);
 
   ASSERT_EQ("child", GetFrameText(child_frame_connection));
   // The child's parent should not be itself:
   const char kGetWindowParentNameScript[] =
       "window.parent == window ? 'parent is self' : 'parent not self';";
   scoped_ptr<base::Value> parent_value(
       ExecuteScript(child_frame_connection, kGetWindowParentNameScript));
   ASSERT_TRUE(parent_value->IsType(base::Value::TYPE_LIST));
   base::ListValue* parent_list;
   ASSERT_TRUE(parent_value->GetAsList(&parent_list));
   ASSERT_EQ(1u, parent_list->GetSize());
   std::string parent_name;
   ASSERT_TRUE(parent_list->GetString(0u, &parent_name));
   EXPECT_EQ("parent not self", parent_name);
 }
 
 TEST_F(HTMLFrameTest, DynamicallyAddFrameAndSeeNameChange) {
   if (!EnableOOPIFs())
     return;
 
-  Frame* child_frame = LoadEmptyPageAndCreateFrame();
+  Frame* child_frame = LoadEmptyPageAndCreateFrame(nullptr);
   ASSERT_TRUE(child_frame);
 
   mojo::ApplicationConnection* child_frame_connection =
       ApplicationConnectionForFrame(child_frame);
 
   // Change the name of the child's window.
   ExecuteScript(child_frame_connection, "window.name = 'new_child';");
 
   // Eventually the parent should see the change. There is no convenient way
   // to observe this change, so we repeatedly ask for it and timeout if we
@@ skipping 18 matching lines @@
                TestTimeouts::action_timeout());
   EXPECT_EQ("found frame", find_window_result);
 }
 
 // Triggers dynamic addition and removal of a frame.
 TEST_F(HTMLFrameTest, FrameTreeOfThreeLevels) {
   if (!EnableOOPIFs())
     return;
 
   // Create a child frame, and in that child frame create another child frame.
-  Frame* child_frame = LoadEmptyPageAndCreateFrame();
+  Frame* child_frame = LoadEmptyPageAndCreateFrame(nullptr);
   ASSERT_TRUE(child_frame);
 
   ASSERT_TRUE(CreateEmptyChildFrame(child_frame));
 
   // Make sure the parent can see the child and child's child. There is no
   // convenient way to observe this change, so we repeatedly ask for it and
   // timeout if we never get the right value.
   const char kGetChildChildFrameCount[] =
       "if (window.frames.length > 0)"
       "  window.frames[0].frames.length.toString();"
@@ skipping 33 matching lines @@
           script_value_as_list->GetSize() == 1) {
         script_value_as_list->GetString(0u, &child_child_frame_count);
       }
     }
   } while (child_child_frame_count != "0" &&
            base::TimeTicks::Now() - start_time <
                TestTimeouts::action_timeout());
   ASSERT_EQ("0", child_child_frame_count);
 }
 
+// Verifies PostMessage() works across frames.
+TEST_F(HTMLFrameTest, PostMessage) {
+  if (!EnableOOPIFs())
+    return;
+
+  mandoline::TestFrameTreeDelegate frame_tree_delegate;
+  Frame* child_frame = LoadEmptyPageAndCreateFrame(&frame_tree_delegate);
+  ASSERT_TRUE(child_frame);
+
+  mojo::ApplicationConnection* child_frame_connection =
+      ApplicationConnectionForFrame(child_frame);
+  ASSERT_EQ("child", GetFrameText(child_frame_connection));
+
+  // Register an event handler in the child frame.
+  const char kRegisterPostMessageHandler[] =
+      "window.messageData = null;"
+      "function messageFunction(event) {"
+      "  window.messageData = event.data;"
+      "}"
+      "window.addEventListener('message', messageFunction, false);";
+  ExecuteScript(child_frame_connection, kRegisterPostMessageHandler);
+
+  // Post a message from the parent to the child.
+  const char kPostMessageFromParent[] =
+      "window.frames[0].postMessage('hello from parent', '*');";
+  ExecuteScript(ApplicationConnectionForFrame(frame_tree_->root()),
+                kPostMessageFromParent);
+
+  // Wait for the child frame to see the message.
+  const base::TimeTicks start_time(base::TimeTicks::Now());
+  std::string message_in_child;
+  do {
+    const char kGetMessageData[] = "window.messageData;";
+    scoped_ptr<base::Value> script_value(
+        ExecuteScript(child_frame_connection, kGetMessageData));
+    if (script_value->IsType(base::Value::TYPE_LIST)) {
+      base::ListValue* script_value_as_list;
+      if (script_value->GetAsList(&script_value_as_list) &&
+          script_value_as_list->GetSize() == 1) {
+        script_value_as_list->GetString(0u, &message_in_child);
+      }
+    }
+  } while (message_in_child != "hello from parent" &&
+           base::TimeTicks::Now() - start_time <
+               TestTimeouts::action_timeout());
+  EXPECT_EQ("hello from parent", message_in_child);
+}
+
 }  // namespace mojo