Towards moving guest management to chrome: Introduce GuestViewManager

content/browser/browser_plugin/test_browser_plugin_guest_manager.cc

Index: content/browser/browser_plugin/test_browser_plugin_guest_manager.cc
diff --git a/content/browser/browser_plugin/test_browser_plugin_guest_manager.cc b/content/browser/browser_plugin/test_browser_plugin_guest_manager.cc
index 2d742880364c862bea538316dbcd4df2b683f89e..3885fc7f6592ce15c73128ccca74a1c648d05d7e 100644
--- a/content/browser/browser_plugin/test_browser_plugin_guest_manager.cc
+++ b/content/browser/browser_plugin/test_browser_plugin_guest_manager.cc
@@ -9,7 +9,10 @@
 
 namespace content {
 
-TestBrowserPluginGuestManager::TestBrowserPluginGuestManager() {
+TestBrowserPluginGuestManager::TestBrowserPluginGuestManager(
+    BrowserContext* context)
+    : BrowserPluginGuestManager(context),
+      last_guest_(NULL) {
 }
 
 TestBrowserPluginGuestManager::~TestBrowserPluginGuestManager() {
@@ -17,19 +20,21 @@ TestBrowserPluginGuestManager::~TestBrowserPluginGuestManager() {
 
 void TestBrowserPluginGuestManager::AddGuest(
     int instance_id,
-    WebContentsImpl* guest_web_contents) {
+    WebContents* guest_web_contents) {
   BrowserPluginGuestManager::AddGuest(instance_id, guest_web_contents);
-  if (message_loop_runner_)
+  last_guest_ = guest_web_contents;
+  if (message_loop_runner_.get())
     message_loop_runner_->Quit();
 }
 
-void TestBrowserPluginGuestManager::WaitForGuestAdded() {
+WebContents* TestBrowserPluginGuestManager::WaitForGuestAdded() {
   // Check if guests were already created.
-  if (guest_web_contents_by_instance_id_.size() > 0)
-    return;
+  if (last_guest_)
+    return last_guest_;
   // Wait otherwise.
   message_loop_runner_ = new MessageLoopRunner();
   message_loop_runner_->Run();
+  return last_guest_;
 }
 
 }  // namespace content