components/guest_view/browser/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 "components/guest_view/browser/test_guest_view_manager.h"
   6
   7 #include "base/memory/scoped_ptr.h"
   8 #include "components/guest_view/browser/guest_view_manager_delegate.h"
   9
  10 namespace guest_view {
  11
  12 TestGuestViewManager::TestGuestViewManager(
  13     content::BrowserContext* context,
  14     scoped_ptr<GuestViewManagerDelegate> delegate)
  15     : GuestViewManager(context, delegate.Pass()),
  16       num_embedder_processes_destroyed_(0),
  17       num_guests_created_(0),
  18       expected_num_guests_created_(0),
  19       num_views_garbage_collected_(0),
  20       waiting_for_guests_created_(false) {
  21 }
  22
  23 TestGuestViewManager::~TestGuestViewManager() {
  24 }
  25
  26 size_t TestGuestViewManager::GetNumGuestsActive() const {
  27   return guest_web_contents_by_instance_id_.size();
  28 }
  29
  30 size_t TestGuestViewManager::GetNumRemovedInstanceIDs() const {
  31   return removed_instance_ids_.size();
  32 }
  33
  34 content::WebContents* TestGuestViewManager::GetLastGuestCreated() {
  35   content::WebContents* web_contents = nullptr;
  36   for (int i = current_instance_id_; i >= 0; i--) {
  37     web_contents = GetGuestByInstanceID(i);
  38     if (web_contents) {
  39       break;
  40     }
  41   }
  42   return web_contents;
  43 }
  44
  45 void TestGuestViewManager::WaitForAllGuestsDeleted() {
  46   // Make sure that every guest that was created has been removed.
  47   for (auto& watcher : guest_web_contents_watchers_)
  48     watcher->Wait();
  49 }
  50
  51 void TestGuestViewManager::WaitForLastGuestDeleted() {
  52   // Wait for the last guest that was created to be deleted.
  53   guest_web_contents_watchers_.back()->Wait();
  54 };
  55
  56 content::WebContents* TestGuestViewManager::WaitForSingleGuestCreated() {
  57   if (!GetNumGuestsActive()) {
  58     // Guests have been created and subsequently destroyed.
  59     if (num_guests_created() > 0)
  60       return nullptr;
  61     WaitForNumGuestsCreated(1u);
  62   }
  63
  64   return GetLastGuestCreated();
  65 }
  66
  67 content::WebContents* TestGuestViewManager::WaitForNextGuestCreated() {
  68   created_message_loop_runner_ = new content::MessageLoopRunner();
  69   created_message_loop_runner_->Run();
  70   return GetLastGuestCreated();
  71 }
  72
  73 void TestGuestViewManager::WaitForNumGuestsCreated(size_t count) {
  74   if (count == num_guests_created_)
  75     return;
  76
  77   waiting_for_guests_created_ = true;
  78   expected_num_guests_created_ = count;
  79
  80   num_created_message_loop_runner_ = new content::MessageLoopRunner;
  81   num_created_message_loop_runner_->Run();
  82 }
  83
  84 void TestGuestViewManager::WaitForViewGarbageCollected() {
  85   gc_message_loop_runner_ = new content::MessageLoopRunner;
  86   gc_message_loop_runner_->Run();
  87 }
  88
  89 void TestGuestViewManager::WaitForSingleViewGarbageCollected() {
  90   if (!num_views_garbage_collected())
  91     WaitForViewGarbageCollected();
  92 }
  93
  94 void TestGuestViewManager::AddGuest(int guest_instance_id,
  95                                     content::WebContents* guest_web_contents) {
  96   GuestViewManager::AddGuest(guest_instance_id, guest_web_contents);
  97
  98   guest_web_contents_watchers_.push_back(
  99       linked_ptr<content::WebContentsDestroyedWatcher>(
 100           new content::WebContentsDestroyedWatcher(guest_web_contents)));
 101
 102   if (created_message_loop_runner_.get())
 103     created_message_loop_runner_->Quit();
 104
 105   ++num_guests_created_;
 106   if (!waiting_for_guests_created_ &&
 107       num_guests_created_ != expected_num_guests_created_) {
 108     return;
 109   }
 110
 111   if (num_created_message_loop_runner_.get())
 112     num_created_message_loop_runner_->Quit();
 113 }
 114
 115 void TestGuestViewManager::GetGuestWebContentsList(
 116     std::vector<content::WebContents*>* guest_web_contents_list) {
 117   for (auto& watcher : guest_web_contents_watchers_)
 118     guest_web_contents_list->push_back(watcher->web_contents());
 119 }
 120
 121 void TestGuestViewManager::RemoveGuest(int guest_instance_id) {
 122   GuestViewManager::RemoveGuest(guest_instance_id);
 123 }
 124
 125 void TestGuestViewManager::EmbedderProcessDestroyed(int embedder_process_id) {
 126   ++num_embedder_processes_destroyed_;
 127   GuestViewManager::EmbedderProcessDestroyed(embedder_process_id);
 128 }
 129
 130 void TestGuestViewManager::ViewGarbageCollected(int embedder_process_id,
 131                                                 int view_instance_id) {
 132   GuestViewManager::ViewGarbageCollected(embedder_process_id, view_instance_id);
 133   ++num_views_garbage_collected_;
 134   if (gc_message_loop_runner_.get())
 135     gc_message_loop_runner_->Quit();
 136 }
 137
 138 // Test factory for creating test instances of GuestViewManager.
 139 TestGuestViewManagerFactory::TestGuestViewManagerFactory()
 140     : test_guest_view_manager_(NULL) {
 141 }
 142
 143 TestGuestViewManagerFactory::~TestGuestViewManagerFactory() {
 144 }
 145
 146 GuestViewManager* TestGuestViewManagerFactory::CreateGuestViewManager(
 147     content::BrowserContext* context,
 148     scoped_ptr<GuestViewManagerDelegate> delegate) {
 149   if (!test_guest_view_manager_) {
 150     test_guest_view_manager_ =
 151         new TestGuestViewManager(context, delegate.Pass());
 152   }
 153   return test_guest_view_manager_;
 154 }
 155
 156 }  // namespace guest_view