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 "content/public/test/test_utils.h"
8#include "extensions/browser/extension_registry.h"
9
10namespace extensions {
11
12class TestExtensionRegistryObserver::Waiter {
13 public:
14  explicit Waiter(const std::string& extension_id)
15      : observed_(false), runner_(NULL) {}
16
17  void Wait() {
18    if (observed_)
19      return;
20
21    runner_ = new content::MessageLoopRunner();
22    runner_->Run();
23  }
24
25  void OnObserved() {
26    observed_ = true;
27
28    if (runner_.get()) {
29      runner_->Quit();
30      runner_ = NULL;
31    }
32  }
33
34 private:
35  bool observed_;
36  scoped_refptr<content::MessageLoopRunner> runner_;
37
38  DISALLOW_COPY_AND_ASSIGN(Waiter);
39};
40
41TestExtensionRegistryObserver::TestExtensionRegistryObserver(
42    ExtensionRegistry* registry,
43    const std::string& extension_id)
44    : will_be_installed_waiter_(new Waiter(extension_id)),
45      uninstalled_waiter_(new Waiter(extension_id)),
46      loaded_waiter_(new Waiter(extension_id)),
47      unloaded_waiter_(new Waiter(extension_id)),
48      extension_registry_observer_(this),
49      extension_id_(extension_id) {
50  extension_registry_observer_.Add(registry);
51}
52
53TestExtensionRegistryObserver::~TestExtensionRegistryObserver() {
54}
55
56void TestExtensionRegistryObserver::WaitForExtensionUninstalled() {
57  uninstalled_waiter_->Wait();
58}
59
60void TestExtensionRegistryObserver::WaitForExtensionWillBeInstalled() {
61  will_be_installed_waiter_->Wait();
62}
63
64void TestExtensionRegistryObserver::WaitForExtensionLoaded() {
65  loaded_waiter_->Wait();
66}
67
68void TestExtensionRegistryObserver::WaitForExtensionUnloaded() {
69  unloaded_waiter_->Wait();
70}
71
72void TestExtensionRegistryObserver::OnExtensionWillBeInstalled(
73    content::BrowserContext* browser_context,
74    const Extension* extension,
75    bool is_update,
76    bool from_ephemeral,
77    const std::string& old_name) {
78  if (extension->id() == extension_id_)
79    will_be_installed_waiter_->OnObserved();
80}
81
82void TestExtensionRegistryObserver::OnExtensionUninstalled(
83    content::BrowserContext* browser_context,
84    const Extension* extension,
85    extensions::UninstallReason reason) {
86  if (extension->id() == extension_id_)
87    uninstalled_waiter_->OnObserved();
88}
89
90void TestExtensionRegistryObserver::OnExtensionLoaded(
91    content::BrowserContext* browser_context,
92    const Extension* extension) {
93  if (extension->id() == extension_id_)
94    loaded_waiter_->OnObserved();
95}
96
97void TestExtensionRegistryObserver::OnExtensionUnloaded(
98    content::BrowserContext* browser_context,
99    const Extension* extension,
100    UnloadedExtensionInfo::Reason reason) {
101  if (extension->id() == extension_id_)
102    unloaded_waiter_->OnObserved();
103}
104
105}  // namespace extensions
106