extensions/browser/guest_view/test_guest_view_manager.cc

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "extensions/browser/guest_view/test_guest_view_manager.h"
   6
   7 #include "base/callback.h"
   8 #include "base/strings/stringprintf.h"
   9 #include "content/public/test/browser_test_utils.h"
  10 #include "extensions/browser/app_window/app_window.h"
  11 #include "extensions/browser/app_window/app_window_registry.h"
  12 #include "extensions/browser/extension_host.h"
  13 #include "extensions/browser/process_manager.h"
  14 #include "extensions/common/extension.h"
  15 #include "extensions/common/extension_paths.h"
  16 #include "extensions/shell/browser/shell_extension_system.h"
  17 #include "extensions/shell/test/shell_test.h"
  18 #include "extensions/test/extension_test_message_listener.h"
  19
  20 namespace extensions {
  21
  22 TestGuestViewManager::TestGuestViewManager(content::BrowserContext* context)
  23     : GuestViewManager(context) {
  24 }
  25
  26 TestGuestViewManager::~TestGuestViewManager() {
  27 }
  28
  29 int TestGuestViewManager::GetNumGuests() const {
  30   return guest_web_contents_by_instance_id_.size();
  31 }
  32
  33 content::WebContents* TestGuestViewManager::GetLastGuestCreated() {
  34   content::WebContents* web_contents = nullptr;
  35   for (int i = current_instance_id_; i >= 0; i--) {
  36     web_contents = GetGuestByInstanceID(i);
  37     if (web_contents) {
  38       break;
  39     }
  40   }
  41   return web_contents;
  42 }
  43
  44 void TestGuestViewManager::WaitForAllGuestsDeleted() {
  45   // Make sure that every guest that was created have been removed.
  46   for (auto& watcher : guest_web_contents_watchers_)
  47     watcher->Wait();
  48 }
  49
  50 void TestGuestViewManager::WaitForGuestCreated() {
  51   created_message_loop_runner_ = new content::MessageLoopRunner;
  52   created_message_loop_runner_->Run();
  53 }
  54
  55 content::WebContents* TestGuestViewManager::WaitForSingleGuestCreated() {
  56   if (GetNumGuests() == 0)
  57     WaitForGuestCreated();
  58
  59   return GetLastGuestCreated();
  60 }
  61
  62 void TestGuestViewManager::AddGuest(int guest_instance_id,
  63                                     content::WebContents* guest_web_contents) {
  64   GuestViewManager::AddGuest(guest_instance_id, guest_web_contents);
  65
  66   guest_web_contents_watchers_.push_back(
  67       linked_ptr<content::WebContentsDestroyedWatcher>(
  68           new content::WebContentsDestroyedWatcher(guest_web_contents)));
  69
  70   if (created_message_loop_runner_.get())
  71     created_message_loop_runner_->Quit();
  72 }
  73
  74 void TestGuestViewManager::RemoveGuest(int guest_instance_id) {
  75   GuestViewManager::RemoveGuest(guest_instance_id);
  76 }
  77
  78 // Test factory for creating test instances of GuestViewManager.
  79 TestGuestViewManagerFactory::TestGuestViewManagerFactory()
  80     : test_guest_view_manager_(NULL) {
  81 }
  82
  83 TestGuestViewManagerFactory::~TestGuestViewManagerFactory() {
  84 }
  85
  86 GuestViewManager* TestGuestViewManagerFactory::CreateGuestViewManager(
  87     content::BrowserContext* context) {
  88   return GetManager(context);
  89 }
  90
  91 // This function gets called from GuestViewManager::FromBrowserContext(),
  92 // where test_guest_view_manager_ is assigned to a linked_ptr that takes care
  93 // of deleting it.
  94 TestGuestViewManager* TestGuestViewManagerFactory::GetManager(
  95     content::BrowserContext* context) {
  96   DCHECK(!test_guest_view_manager_);
  97   test_guest_view_manager_ = new TestGuestViewManager(context);
  98   return test_guest_view_manager_;
  99 }
 100
 101 }  // namespace extensions