components/guest_view/browser/test_guest_view_manager.h

   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 #ifndef COMPONENTS_GUEST_VIEW_BROWSER_TEST_GUEST_VIEW_MANAGER_H_
   6 #define COMPONENTS_GUEST_VIEW_BROWSER_TEST_GUEST_VIEW_MANAGER_H_
   7
   8 #include "base/memory/linked_ptr.h"
   9 #include "components/guest_view/browser/guest_view_manager.h"
  10 #include "components/guest_view/browser/guest_view_manager_factory.h"
  11 #include "content/public/test/browser_test_utils.h"
  12 #include "content/public/test/test_utils.h"
  13
  14 namespace guest_view {
  15
  16 class TestGuestViewManager : public GuestViewManager {
  17  public:
  18   TestGuestViewManager(
  19       content::BrowserContext* context,
  20       scoped_ptr<GuestViewManagerDelegate> delegate);
  21   ~TestGuestViewManager() override;
  22
  23   void WaitForAllGuestsDeleted();
  24
  25   void WaitForLastGuestDeleted();
  26
  27   content::WebContents* WaitForSingleGuestCreated();
  28   void WaitForNumGuestsCreated(size_t count);
  29
  30   void WaitForSingleViewGarbageCollected();
  31
  32   content::WebContents* GetLastGuestCreated();
  33
  34   // Returns the number of guests currently still alive at the time of calling
  35   // this method.
  36   size_t GetNumGuestsActive() const;
  37
  38   // Returns the size of the set of removed instance IDs.
  39   size_t GetNumRemovedInstanceIDs() const;
  40
  41   using GuestViewCreateFunction =
  42       base::Callback<GuestViewBase*(content::WebContents*)>;
  43
  44   template <typename T>
  45   void RegisterTestGuestViewType(
  46       const GuestViewCreateFunction& create_function) {
  47     auto registry_entry = std::make_pair(
  48         T::Type,
  49         GuestViewData(create_function, base::Bind(&T::CleanUp)));
  50     guest_view_registry_.insert(registry_entry);
  51   }
  52
  53   // Returns the number of times EmbedderWillBeDestroyed() was called.
  54   int num_embedder_processes_destroyed() const {
  55     return num_embedder_processes_destroyed_;
  56   }
  57
  58   // Returns the number of guests that have been created since the creation of
  59   // this GuestViewManager.
  60   size_t num_guests_created() const { return num_guests_created_; }
  61
  62   // Returns the number of GuestViews that have been garbage collected in
  63   // JavaScript since the creation of this GuestViewManager.
  64   int num_views_garbage_collected() const {
  65     return num_views_garbage_collected_;
  66   }
  67
  68   // Returns the last guest instance ID removed from the manager.
  69   int last_instance_id_removed() const { return last_instance_id_removed_; }
  70
  71   // Returns the list of guests WebContentses that were created by this
  72   // manager.
  73   void GetGuestWebContentsList(
  74       std::vector<content::WebContents*>* guest_web_contents_list);
  75
  76  private:
  77   FRIEND_TEST_ALL_PREFIXES(GuestViewManagerTest, AddRemove);
  78
  79   // GuestViewManager override:
  80   void AddGuest(int guest_instance_id,
  81                 content::WebContents* guest_web_contents) override;
  82   void RemoveGuest(int guest_instance_id) override;
  83   void EmbedderProcessDestroyed(int embedder_process_id) override;
  84   void ViewGarbageCollected(int embedder_process_id,
  85                             int view_instance_id) override;
  86
  87   void WaitForViewGarbageCollected();
  88
  89   using GuestViewManager::last_instance_id_removed_;
  90   using GuestViewManager::removed_instance_ids_;
  91
  92   int num_embedder_processes_destroyed_;
  93   size_t num_guests_created_;
  94   size_t expected_num_guests_created_;
  95   int num_views_garbage_collected_;
  96   bool waiting_for_guests_created_;
  97
  98   std::vector<linked_ptr<content::WebContentsDestroyedWatcher>>
  99       guest_web_contents_watchers_;
 100   scoped_refptr<content::MessageLoopRunner> created_message_loop_runner_;
 101   scoped_refptr<content::MessageLoopRunner> gc_message_loop_runner_;
 102
 103   DISALLOW_COPY_AND_ASSIGN(TestGuestViewManager);
 104 };
 105
 106 // Test factory for creating test instances of GuestViewManager.
 107 class TestGuestViewManagerFactory : public GuestViewManagerFactory {
 108  public:
 109   TestGuestViewManagerFactory();
 110   ~TestGuestViewManagerFactory() override;
 111
 112   GuestViewManager* CreateGuestViewManager(
 113       content::BrowserContext* context,
 114       scoped_ptr<GuestViewManagerDelegate> delegate) override;
 115
 116  private:
 117   TestGuestViewManager* test_guest_view_manager_;
 118
 119   DISALLOW_COPY_AND_ASSIGN(TestGuestViewManagerFactory);
 120 };
 121
 122 }  // namespace guest_view
 123
 124 #endif  // COMPONENTS_GUEST_VIEW_BROWSER_TEST_GUEST_VIEW_MANAGER_H_