extensions/browser/test_extension_registry_observer.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/test_extension_registry_observer.h"
   6
   7 #include "base/run_loop.h"
   8 #include "extensions/browser/extension_registry.h"
   9
  10 namespace extensions {
  11
  12 class TestExtensionRegistryObserver::Waiter {
  13  public:
  14   Waiter() : observed_(false), extension_(nullptr) {}
  15
  16   void Wait() {
  17     if (!observed_)
  18       run_loop_.Run();
  19   }
  20
  21   void OnObserved(const Extension* extension) {
  22     observed_ = true;
  23     run_loop_.Quit();
  24     extension_ = extension;
  25   }
  26
  27   const Extension* extension() const { return extension_; }
  28
  29  private:
  30   bool observed_;
  31   base::RunLoop run_loop_;
  32   const Extension* extension_;
  33
  34   DISALLOW_COPY_AND_ASSIGN(Waiter);
  35 };
  36
  37 TestExtensionRegistryObserver::TestExtensionRegistryObserver(
  38     ExtensionRegistry* registry)
  39     : TestExtensionRegistryObserver(registry, std::string()) {
  40 }
  41
  42 TestExtensionRegistryObserver::TestExtensionRegistryObserver(
  43     ExtensionRegistry* registry,
  44     const std::string& extension_id)
  45     : will_be_installed_waiter_(new Waiter()),
  46       uninstalled_waiter_(new Waiter()),
  47       loaded_waiter_(new Waiter()),
  48       unloaded_waiter_(new Waiter()),
  49       extension_registry_observer_(this),
  50       extension_id_(extension_id) {
  51   extension_registry_observer_.Add(registry);
  52 }
  53
  54 TestExtensionRegistryObserver::~TestExtensionRegistryObserver() {
  55 }
  56
  57 const Extension* TestExtensionRegistryObserver::WaitForExtensionUninstalled() {
  58   return Wait(&uninstalled_waiter_);
  59 }
  60
  61 const Extension*
  62 TestExtensionRegistryObserver::WaitForExtensionWillBeInstalled() {
  63   return Wait(&will_be_installed_waiter_);
  64 }
  65
  66 const Extension* TestExtensionRegistryObserver::WaitForExtensionLoaded() {
  67   return Wait(&loaded_waiter_);
  68 }
  69
  70 const Extension* TestExtensionRegistryObserver::WaitForExtensionUnloaded() {
  71   return Wait(&unloaded_waiter_);
  72 }
  73
  74 void TestExtensionRegistryObserver::OnExtensionWillBeInstalled(
  75     content::BrowserContext* browser_context,
  76     const Extension* extension,
  77     bool is_update,
  78     bool from_ephemeral,
  79     const std::string& old_name) {
  80   if (extension_id_.empty() || extension->id() == extension_id_)
  81     will_be_installed_waiter_->OnObserved(extension);
  82 }
  83
  84 void TestExtensionRegistryObserver::OnExtensionUninstalled(
  85     content::BrowserContext* browser_context,
  86     const Extension* extension,
  87     extensions::UninstallReason reason) {
  88   if (extension_id_.empty() || extension->id() == extension_id_)
  89     uninstalled_waiter_->OnObserved(extension);
  90 }
  91
  92 void TestExtensionRegistryObserver::OnExtensionLoaded(
  93     content::BrowserContext* browser_context,
  94     const Extension* extension) {
  95   if (extension_id_.empty() || extension->id() == extension_id_)
  96     loaded_waiter_->OnObserved(extension);
  97 }
  98
  99 void TestExtensionRegistryObserver::OnExtensionUnloaded(
 100     content::BrowserContext* browser_context,
 101     const Extension* extension,
 102     UnloadedExtensionInfo::Reason reason) {
 103   if (extension_id_.empty() || extension->id() == extension_id_)
 104     unloaded_waiter_->OnObserved(extension);
 105 }
 106
 107 const Extension* TestExtensionRegistryObserver::Wait(
 108     scoped_ptr<Waiter>* waiter) {
 109   waiter->get()->Wait();
 110   const Extension* extension = waiter->get()->extension();
 111   // Reset the waiter for future uses.
 112   // We could have a Waiter::Reset method, but it would reset every field in the
 113   // class, so let's just reset the pointer.
 114   waiter->reset(new Waiter());
 115   return extension;
 116 }
 117
 118 }  // namespace extensions