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 "base/run_loop.h"
6#include "chrome/browser/extensions/extension_uninstall_dialog.h"
7#include "chrome/browser/ui/browser.h"
8#include "chrome/browser/ui/browser_window.h"
9#include "chrome/test/base/in_process_browser_test.h"
10#include "content/public/test/test_utils.h"
11#include "extensions/common/extension.h"
12#include "extensions/common/extension_builder.h"
13#include "extensions/common/value_builder.h"
14
15namespace {
16
17scoped_refptr<extensions::Extension> BuildTestExtension() {
18  return extensions::ExtensionBuilder()
19      .SetManifest(extensions::DictionaryBuilder()
20                       .Set("name", "foo")
21                       .Set("version", "1.0"))
22      .Build();
23}
24
25class TestExtensionUninstallDialogDelegate
26    : public extensions::ExtensionUninstallDialog::Delegate {
27 public:
28  explicit TestExtensionUninstallDialogDelegate(
29      const base::Closure& quit_closure)
30      : quit_closure_(quit_closure), canceled_(false) {}
31
32  virtual ~TestExtensionUninstallDialogDelegate() {}
33
34  bool canceled() { return canceled_; }
35
36 private:
37  virtual void ExtensionUninstallAccepted() OVERRIDE { quit_closure_.Run(); }
38
39  virtual void ExtensionUninstallCanceled() OVERRIDE {
40    canceled_ = true;
41    quit_closure_.Run();
42  }
43
44  base::Closure quit_closure_;
45  bool canceled_;
46
47  DISALLOW_COPY_AND_ASSIGN(TestExtensionUninstallDialogDelegate);
48};
49
50}  // namespace
51
52typedef InProcessBrowserTest ExtensionUninstallDialogViewBrowserTest;
53
54// Test that ExtensionUninstallDialog cancels the uninstall if the aura::Window
55// which is passed to ExtensionUninstallDialog::Create() is destroyed.
56IN_PROC_BROWSER_TEST_F(ExtensionUninstallDialogViewBrowserTest,
57                       TrackParentWindowDestruction) {
58  // Create a second browser to prevent the app from exiting when the browser is
59  // closed.
60  CreateBrowser(browser()->profile());
61
62  scoped_refptr<extensions::Extension> extension(BuildTestExtension());
63
64  base::RunLoop run_loop;
65  TestExtensionUninstallDialogDelegate delegate(run_loop.QuitClosure());
66  scoped_ptr<extensions::ExtensionUninstallDialog> dialog(
67      extensions::ExtensionUninstallDialog::Create(
68          browser()->profile(), browser()->window()->GetNativeWindow(),
69          &delegate));
70  browser()->window()->Close();
71  content::RunAllPendingInMessageLoop();
72
73  dialog->ConfirmUninstall(extension.get());
74  run_loop.Run();
75  EXPECT_TRUE(delegate.canceled());
76}
77