Chromium Code Reviews| 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 04358ff7f7ad2396217fe209aca23bbb99879a66..fd4f33a1dbf09fb351d742ca55e73def59b1ab7a 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,22 @@ TestGuestManagerDelegate* TestGuestManagerDelegate::GetInstance() { |
| return Singleton<TestGuestManagerDelegate>::get(); |
| } |
| +WebContentsImpl* TestGuestManagerDelegate::WaitForGuestAdded() { |
| + fprintf(stderr, ">>>%s\n", __PRETTY_FUNCTION__); |
|
lazyboy
2014/05/05 21:40:54
Remove debug code.
Fady Samuel
2014/05/07 17:32:59
Done.
|
| + // 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, |
| @@ -97,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( |