Browser Plugin: Reload and Stop operations

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 "base/test/test_timeouts.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"
@@ skipping 10 matching lines @@
 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),
       focus_observed_(false),
       advance_focus_observed_(false),
       was_hidden_observed_(false),
+      stop_observed_(false),
+      reload_observed_(false),
       waiting_for_update_rect_msg_with_size_(false),
       last_update_rect_width_(-1),
       last_update_rect_height_(-1) {
   // Listen to visibility changes so that a test can wait for these changes.
   registrar_.Add(this,
                  NOTIFICATION_WEB_CONTENTS_VISIBILITY_CHANGED,
                  Source<WebContents>(web_contents));
 }
 
 TestBrowserPluginGuest::~TestBrowserPluginGuest() {
@@ skipping 49 matching lines @@
 }
 
 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::ResetUpdateRectCount() {
+  update_rect_count_ = 0;
+}
+
 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;
@@ skipping 37 matching lines @@
 void TestBrowserPluginGuest::WaitUntilHidden() {
   if (was_hidden_observed_) {
     was_hidden_observed_ = false;
     return;
   }
   was_hidden_message_loop_runner_ = new MessageLoopRunner();
   was_hidden_message_loop_runner_->Run();
   was_hidden_observed_ = false;
 }
 
+void TestBrowserPluginGuest::WaitForReload() {
+  if (reload_observed_) {
+    reload_observed_ = false;
+    return;
+  }
+
+  reload_message_loop_runner_ = new MessageLoopRunner();
+  reload_message_loop_runner_->Run();
+  reload_observed_ = false;
+}
+
+void TestBrowserPluginGuest::WaitForStop() {
+  if (stop_observed_) {
+    stop_observed_ = false;
+    return;
+  }
+
+  stop_message_loop_runner_ = new MessageLoopRunner();
+  stop_message_loop_runner_->Run();
+  stop_observed_ = false;
+}
+
+
 void TestBrowserPluginGuest::SetFocus(bool focused) {
   focus_observed_ = true;
   if (focus_message_loop_runner_)
     focus_message_loop_runner_->Quit();
   BrowserPluginGuest::SetFocus(focused);
 }
 
 bool TestBrowserPluginGuest::ViewTakeFocus(bool reverse) {
   advance_focus_observed_ = true;
   if (advance_focus_message_loop_runner_)
     advance_focus_message_loop_runner_->Quit();
   return BrowserPluginGuest::ViewTakeFocus(reverse);
 }
 
+void TestBrowserPluginGuest::Reload() {
+  reload_observed_ = true;
+  if (reload_message_loop_runner_)
+    reload_message_loop_runner_->Quit();
+  BrowserPluginGuest::Reload();
+}
+
+void TestBrowserPluginGuest::Stop() {
+  stop_observed_ = true;
+  if (stop_message_loop_runner_)
+    stop_message_loop_runner_->Quit();
+  BrowserPluginGuest::Stop();
+}
+
 }  // namespace content