Browser Plugin: New Implementation (Browser Side)

content/browser/browser_plugin/test_browser_plugin_guest.cc

+// Copyright (c) 2012 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/browser_plugin/test_browser_plugin_guest.h"
+
+#include "content/browser/browser_plugin/browser_plugin_guest.h"
+#include "content/browser/renderer_host/render_view_host_impl.h"
+#include "content/browser/web_contents/web_contents_impl.h"
+#include "content/common/browser_plugin_messages.h"
+#include "content/public/test/test_utils.h"
+#include "ipc/ipc_channel_handle.h"
+#include "ipc/ipc_sync_message.h"
+#include "ui/gfx/size.h"
+
+namespace content {
+
+class BrowserPluginGuest;
+
+TestBrowserPluginGuest::TestBrowserPluginGuest(
+    int instance_id,
+    WebContentsImpl* web_contents,
+    RenderViewHost* render_view_host)
+    : BrowserPluginGuest(instance_id, web_contents, render_view_host),
+      update_rect_count_(0),
+      crash_observed_(false),
+      last_update_rect_width_(-1),
+      last_update_rect_height_(-1) {
+}
+
+TestBrowserPluginGuest::~TestBrowserPluginGuest() {
+}
+
+void TestBrowserPluginGuest::SendMessageToEmbedder(IPC::Message* msg) {
+  if (msg->type() == BrowserPluginMsg_UpdateRect::ID) {
+    PickleIterator iter(*msg);
+
+    int instance_id;
+    int message_id;
+    BrowserPluginMsg_UpdateRect_Params update_rect_params;
+
+    if (!IPC::ReadParam(msg, &iter, &instance_id) ||
+        !IPC::ReadParam(msg, &iter, &message_id) ||
+        !IPC::ReadParam(msg, &iter, &update_rect_params)) {
+      NOTREACHED() <<
+          "Cannot read BrowserPluginMsg_UpdateRect params from ipc message";
+    }
+    last_update_rect_width_ = update_rect_params.view_size.width();
+    last_update_rect_height_ = update_rect_params.view_size.height();
+    update_rect_count_++;
+    if (waiting_for_update_rect_msg_with_size_ &&
+        expected_width_ == last_update_rect_width_ &&
+        expected_height_ == last_update_rect_height_) {
+      waiting_for_update_rect_msg_with_size_ = false;
+      if (send_message_loop_runner_)
+        send_message_loop_runner_->Quit();
+    } else if (!waiting_for_update_rect_msg_with_size_) {
+      if (send_message_loop_runner_)
+        send_message_loop_runner_->Quit();
+    }
+  }
+  BrowserPluginGuest::SendMessageToEmbedder(msg);
+}
+
+void TestBrowserPluginGuest::WaitForUpdateRectMsg() {
+  // Check if we already got any UpdateRect message.
+  if (update_rect_count_ > 0)
+    return;
+  send_message_loop_runner_= new MessageLoopRunner();
+  send_message_loop_runner_->Run();
+}
+
+void TestBrowserPluginGuest::WaitForUpdateRectMsgWithSize(int width,
+                                                          int height) {
+  if (update_rect_count_ > 0 &&
+      last_update_rect_width_ == width &&
+      last_update_rect_height_ == height) {
+    // We already saw this message.
+    return;
+  }
+  waiting_for_update_rect_msg_with_size_ = true;
+  expected_width_ = width;
+  expected_height_ = height;
+  send_message_loop_runner_= new MessageLoopRunner();
+  send_message_loop_runner_->Run();
+}
+
+void TestBrowserPluginGuest::RenderViewGone(base::TerminationStatus status) {
+  crash_observed_ = true;
+  if (crash_message_loop_runner_)
+    crash_message_loop_runner_->Quit();
+  BrowserPluginGuest::RenderViewGone(status);
+}
+
+void TestBrowserPluginGuest::WaitForCrashed() {
+  // Check if we already observed a guest crash, return immediately if so.
+  if (crash_observed_)
+    return;
+  crash_message_loop_runner_ = new MessageLoopRunner();
+  crash_message_loop_runner_->Run();

[awong, 2012/09/10 19:21:07]

Per mail from phadjdan@, sounds like the right thing to do is set the hang time
to TestTimeouts::tiny_timeout().

Then make it so this (and any other WaitForXXX event) doesn't wait indefinitely.
 Instead, make it wait for TestTimeouts::action_timeout().

For this code, I would post a delayed task with crash_message_loop_runner_'s
QuitClosure(), and possibly set a flag that tells if this timedout.  So maybe

// Come up with a better name.
class Foo {
 public:
  Foo() : timed_out_(false) {}

  void OnTimeout(const Closure& done) {
    timed_out_ = true;
    done->Run();
  }

  bool timed_out() { return timed_out_; }

 private:
  bool timed_out_;
}

bool TestBrowserPluginGuest::WaitForCrashed() {
  // Check if we already observed a guest crash, return immediately if so.
  if (crash_observed_)
    return;
  Foo f;
  WeakPtrFactory<Foo> weak_factory(&f);  // So the task cancels on function
exit.

  crash_message_loop_runner_ = new MessageLoopRunner();
  MessageLoop::current()->PostDelayedTask(
     FROM_HERE, 
     Bind(&Foo::OnTimeout, weak_factor.GetWeakPtr(),
          crash_message_loop_runner_.QuitClosure().
     TestTimeouts::action_timeout());
  crash_message_loop_runner_->Run();

  return f.timed_out;
}

[lazyboy, 2012/09/10 23:40:57]

Using tiny_timeout() for guest crash timeout.
I've changed all the WaitForXXX methods to do use action_timeout() timeout; I am
still learning about WeakPtr, pretty much copied the code you showed here for
now =P

+}
+
+}  // namespace content