| Index: content/browser/browser_plugin/test_guest_manager_delegate.cc
|
| diff --git a/content/browser/browser_plugin/test_guest_manager_delegate.cc b/content/browser/browser_plugin/test_guest_manager_delegate.cc
|
| index 2eec2aed6d05f5b70123440101cdfbd965244038..5b29914f38c67a3720bf1f9fa3260b3942a95f4d 100644
|
| --- a/content/browser/browser_plugin/test_guest_manager_delegate.cc
|
| +++ b/content/browser/browser_plugin/test_guest_manager_delegate.cc
|
| @@ -8,10 +8,12 @@
|
| #include "base/memory/singleton.h"
|
| #include "base/strings/stringprintf.h"
|
| #include "base/values.h"
|
| +#include "content/browser/web_contents/web_contents_impl.h"
|
| #include "content/public/browser/site_instance.h"
|
| #include "content/public/browser/web_contents.h"
|
| #include "content/public/browser/web_contents_observer.h"
|
| #include "content/public/common/url_constants.h"
|
| +#include "content/public/test/test_utils.h"
|
| #include "net/base/escape.h"
|
|
|
| namespace content {
|
| @@ -38,7 +40,8 @@ class GuestWebContentsObserver
|
| };
|
|
|
| TestGuestManagerDelegate::TestGuestManagerDelegate()
|
| - : next_instance_id_(0) {
|
| + : last_guest_added_(NULL),
|
| + next_instance_id_(0) {
|
| }
|
|
|
| TestGuestManagerDelegate::~TestGuestManagerDelegate() {
|
| @@ -49,6 +52,21 @@ TestGuestManagerDelegate* TestGuestManagerDelegate::GetInstance() {
|
| return Singleton<TestGuestManagerDelegate>::get();
|
| }
|
|
|
| +WebContentsImpl* TestGuestManagerDelegate::WaitForGuestAdded() {
|
| + // Check if guests were already created.
|
| + if (last_guest_added_) {
|
| + WebContentsImpl* last_guest_added = last_guest_added_;
|
| + last_guest_added_ = NULL;
|
| + return last_guest_added;
|
| + }
|
| + // Wait otherwise.
|
| + message_loop_runner_ = new MessageLoopRunner();
|
| + message_loop_runner_->Run();
|
| + WebContentsImpl* last_guest_added = last_guest_added_;
|
| + last_guest_added_ = NULL;
|
| + return last_guest_added;
|
| +}
|
| +
|
| content::WebContents* TestGuestManagerDelegate::CreateGuest(
|
| SiteInstance* embedder_site_instance,
|
| int instance_id,
|
| @@ -98,6 +116,9 @@ void TestGuestManagerDelegate::AddGuest(
|
| guest_web_contents_by_instance_id_.end());
|
| guest_web_contents_by_instance_id_[guest_instance_id] = guest_web_contents;
|
| new GuestWebContentsObserver(guest_web_contents);
|
| + last_guest_added_ = static_cast<WebContentsImpl*>(guest_web_contents);
|
| + if (message_loop_runner_)
|
| + message_loop_runner_->Quit();
|
| }
|
|
|
| void TestGuestManagerDelegate::RemoveGuest(
|
|
|