1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
|
// Copyright 2014 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/browser/extensions/extension_install_prompt.h"
#include "base/run_loop.h"
#include "chrome/browser/extensions/extension_install_prompt_show_params.h"
#include "chrome/browser/ui/browser.h"
#include "chrome/browser/ui/browser_window.h"
#include "chrome/browser/ui/tabs/tab_strip_model.h"
#include "chrome/test/base/in_process_browser_test.h"
#include "content/public/test/test_utils.h"
#include "extensions/common/extension.h"
#include "extensions/common/extension_builder.h"
#include "extensions/common/value_builder.h"
namespace {
scoped_refptr<extensions::Extension> BuildTestExtension() {
return extensions::ExtensionBuilder()
.SetManifest(std::move(extensions::DictionaryBuilder()
.Set("name", "foo")
.Set("version", "1.0")))
.Build();
}
// ExtensionInstallPrompt::ShowDialogCallback which proceeds without showing the
// prompt.
void TestShowDialogCallback(ExtensionInstallPromptShowParams* params,
ExtensionInstallPrompt::Delegate* delegate,
scoped_ptr<ExtensionInstallPrompt::Prompt> prompt) {
delegate->InstallUIProceed();
}
class TestExtensionInstallPromptDelegate
: public ExtensionInstallPrompt::Delegate {
public:
explicit TestExtensionInstallPromptDelegate(const base::Closure& quit_closure)
: quit_closure_(quit_closure), proceeded_(false), aborted_(false) {}
~TestExtensionInstallPromptDelegate() override {
}
bool DidProceed() {
return proceeded_;
}
bool DidAbort() {
return aborted_;
}
private:
void InstallUIProceed() override {
proceeded_ = true;
quit_closure_.Run();
}
void InstallUIAbort(bool user_initiated) override {
aborted_ = true;
quit_closure_.Run();
}
base::Closure quit_closure_;
bool proceeded_;
bool aborted_;
DISALLOW_COPY_AND_ASSIGN(TestExtensionInstallPromptDelegate);
};
} // namespace
typedef InProcessBrowserTest ExtensionInstallPromptBrowserTest;
// Test that ExtensionInstallPrompt aborts the install if the web contents which
// were passed to the ExtensionInstallPrompt constructor get destroyed.
// CrxInstaller takes in ExtensionInstallPrompt in the constructor and does a
// bunch of asynchronous processing prior to confirming the install. A user may
// close the current tab while this processing is taking place.
IN_PROC_BROWSER_TEST_F(ExtensionInstallPromptBrowserTest,
TrackParentWebContentsDestruction) {
AddBlankTabAndShow(browser());
TabStripModel* tab_strip_model = browser()->tab_strip_model();
content::WebContents* web_contents = tab_strip_model->GetActiveWebContents();
int web_contents_index = tab_strip_model->GetIndexOfWebContents(web_contents);
scoped_refptr<extensions::Extension> extension(BuildTestExtension());
ExtensionInstallPrompt prompt(web_contents);
tab_strip_model->CloseWebContentsAt(web_contents_index,
TabStripModel::CLOSE_NONE);
content::RunAllPendingInMessageLoop();
base::RunLoop run_loop;
TestExtensionInstallPromptDelegate delegate(run_loop.QuitClosure());
prompt.ShowDialog(&delegate, extension.get(), nullptr,
base::Bind(&TestShowDialogCallback));
run_loop.Run();
EXPECT_TRUE(delegate.DidAbort());
}
// Test that ExtensionInstallPrompt aborts the install if the gfx::NativeWindow
// which is passed to the ExtensionInstallPrompt constructor is destroyed.
IN_PROC_BROWSER_TEST_F(ExtensionInstallPromptBrowserTest,
TrackParentWindowDestruction) {
// Create a second browser to prevent the app from exiting when the browser is
// closed.
CreateBrowser(browser()->profile());
scoped_refptr<extensions::Extension> extension(BuildTestExtension());
ExtensionInstallPrompt prompt(browser()->profile(),
browser()->window()->GetNativeWindow());
browser()->window()->Close();
content::RunAllPendingInMessageLoop();
base::RunLoop run_loop;
TestExtensionInstallPromptDelegate delegate(run_loop.QuitClosure());
prompt.ShowDialog(&delegate, extension.get(), nullptr,
base::Bind(&TestShowDialogCallback));
run_loop.Run();
EXPECT_TRUE(delegate.DidAbort());
}
// Test that ExtensionInstallPrompt shows the dialog normally if no parent
// web contents or parent gfx::NativeWindow is passed to the
// ExtensionInstallPrompt constructor.
IN_PROC_BROWSER_TEST_F(ExtensionInstallPromptBrowserTest, NoParent) {
scoped_refptr<extensions::Extension> extension(BuildTestExtension());
ExtensionInstallPrompt prompt(browser()->profile(), NULL);
base::RunLoop run_loop;
TestExtensionInstallPromptDelegate delegate(run_loop.QuitClosure());
prompt.ShowDialog(&delegate, extension.get(), nullptr,
base::Bind(&TestShowDialogCallback));
run_loop.Run();
// TestShowDialogCallback() should have signaled the install to proceed.
EXPECT_TRUE(delegate.DidProceed());
}
|
