1 // Copyright 2013 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 <string>
6
7 #include <shlobj.h> // Must be before propkey.
8 #include <propkey.h>
9 #include <shellapi.h>
10
11 #include "base/command_line.h"
12 #include "base/strings/string16.h"
13 #include "base/strings/utf_string_conversions.h"
14 #include "base/win/scoped_comptr.h"
15 #include "base/win/scoped_propvariant.h"
16 #include "base/win/windows_version.h"
17 #include "chrome/browser/browser_process.h"
18 #include "chrome/browser/extensions/extension_browsertest.h"
19 #include "chrome/browser/profiles/profile.h"
20 #include "chrome/browser/profiles/profile_info_cache.h"
21 #include "chrome/browser/profiles/profile_manager.h"
22 #include "chrome/browser/profiles/profile_shortcut_manager_win.h"
23 #include "chrome/browser/profiles/profiles_state.h"
24 #include "chrome/browser/ui/browser.h"
25 #include "chrome/browser/ui/browser_finder.h"
26 #include "chrome/browser/ui/browser_iterator.h"
27 #include "chrome/browser/ui/browser_window.h"
28 #include "chrome/browser/ui/extensions/application_launch.h"
29 #include "chrome/browser/web_applications/web_app.h"
30 #include "chrome/browser/web_applications/web_app_win.h"
31 #include "chrome/common/chrome_switches.h"
32 #include "chrome/common/extensions/extension_constants.h"
33 #include "chrome/installer/util/browser_distribution.h"
34 #include "chrome/test/base/in_process_browser_test.h"
35 #include "chrome/test/base/test_switches.h"
36 #include "content/public/test/test_utils.h"
37 #include "extensions/common/extension.h"
38 #include "ui/views/win/hwnd_util.h"
39
40 typedef ExtensionBrowserTest BrowserWindowPropertyManagerTest;
41
42 namespace {
43
44 // An observer that resumes test code after a new profile is initialized by
45 // quitting the message loop it's blocked on.
UnblockOnProfileCreation(Profile * profile,Profile::CreateStatus status)46 void UnblockOnProfileCreation(Profile* profile,
47 Profile::CreateStatus status) {
48 if (status == Profile::CREATE_STATUS_INITIALIZED)
49 base::MessageLoop::current()->Quit();
50 }
51
52 // Checks that the relaunch name, relaunch command and app icon for the given
53 // |browser| are correct.
ValidateBrowserWindowProperties(const Browser * browser,const base::string16 & expected_profile_name)54 void ValidateBrowserWindowProperties(
55 const Browser* browser,
56 const base::string16& expected_profile_name) {
57 HWND hwnd = views::HWNDForNativeWindow(browser->window()->GetNativeWindow());
58
59 base::win::ScopedComPtr<IPropertyStore> pps;
60 HRESULT result = SHGetPropertyStoreForWindow(hwnd, IID_IPropertyStore,
61 pps.ReceiveVoid());
62 EXPECT_TRUE(SUCCEEDED(result));
63
64 base::win::ScopedPropVariant prop_var;
65 // The relaunch name should be of the form "Chromium" if there is only 1
66 // profile and "First User - Chromium" if there are more. The expected value
67 // is given by |expected_profile_name|.
68 EXPECT_EQ(S_OK, pps->GetValue(PKEY_AppUserModel_RelaunchDisplayNameResource,
69 prop_var.Receive()));
70 EXPECT_EQ(VT_LPWSTR, prop_var.get().vt);
71 EXPECT_EQ(
72 base::FilePath(profiles::internal::GetShortcutFilenameForProfile(
73 expected_profile_name,
74 BrowserDistribution::GetDistribution())).RemoveExtension().value(),
75 prop_var.get().pwszVal);
76 prop_var.Reset();
77
78 // The relaunch command should specify the profile.
79 EXPECT_EQ(S_OK, pps->GetValue(PKEY_AppUserModel_RelaunchCommand,
80 prop_var.Receive()));
81 EXPECT_EQ(VT_LPWSTR, prop_var.get().vt);
82 CommandLine cmd_line(CommandLine::FromString(prop_var.get().pwszVal));
83 EXPECT_EQ(browser->profile()->GetPath().BaseName().value(),
84 cmd_line.GetSwitchValueNative(switches::kProfileDirectory));
85 prop_var.Reset();
86
87 // The app icon should be set to the profile icon.
88 EXPECT_EQ(S_OK, pps->GetValue(PKEY_AppUserModel_RelaunchIconResource,
89 prop_var.Receive()));
90 EXPECT_EQ(VT_LPWSTR, prop_var.get().vt);
91 EXPECT_EQ(profiles::internal::GetProfileIconPath(
92 browser->profile()->GetPath()).value(),
93 prop_var.get().pwszVal);
94 prop_var.Reset();
95 base::MessageLoop::current()->Quit();
96 }
97
ValidateHostedAppWindowProperties(const Browser * browser,const extensions::Extension * extension)98 void ValidateHostedAppWindowProperties(const Browser* browser,
99 const extensions::Extension* extension) {
100 HWND hwnd = views::HWNDForNativeWindow(browser->window()->GetNativeWindow());
101
102 base::win::ScopedComPtr<IPropertyStore> pps;
103 HRESULT result =
104 SHGetPropertyStoreForWindow(hwnd, IID_IPropertyStore, pps.ReceiveVoid());
105 EXPECT_TRUE(SUCCEEDED(result));
106
107 base::win::ScopedPropVariant prop_var;
108 // The relaunch name should be the extension name.
109 EXPECT_EQ(S_OK,
110 pps->GetValue(PKEY_AppUserModel_RelaunchDisplayNameResource,
111 prop_var.Receive()));
112 EXPECT_EQ(VT_LPWSTR, prop_var.get().vt);
113 EXPECT_EQ(base::UTF8ToWide(extension->name()), prop_var.get().pwszVal);
114 prop_var.Reset();
115
116 // The relaunch command should specify the profile and the app id.
117 EXPECT_EQ(
118 S_OK,
119 pps->GetValue(PKEY_AppUserModel_RelaunchCommand, prop_var.Receive()));
120 EXPECT_EQ(VT_LPWSTR, prop_var.get().vt);
121 CommandLine cmd_line(CommandLine::FromString(prop_var.get().pwszVal));
122 EXPECT_EQ(browser->profile()->GetPath().BaseName().value(),
123 cmd_line.GetSwitchValueNative(switches::kProfileDirectory));
124 EXPECT_EQ(base::UTF8ToWide(extension->id()),
125 cmd_line.GetSwitchValueNative(switches::kAppId));
126 prop_var.Reset();
127
128 // The app icon should be set to the extension app icon.
129 base::FilePath web_app_dir = web_app::GetWebAppDataDirectory(
130 browser->profile()->GetPath(), extension->id(), GURL());
131 EXPECT_EQ(S_OK,
132 pps->GetValue(PKEY_AppUserModel_RelaunchIconResource,
133 prop_var.Receive()));
134 EXPECT_EQ(VT_LPWSTR, prop_var.get().vt);
135 EXPECT_EQ(web_app::internals::GetIconFilePath(
136 web_app_dir, base::UTF8ToUTF16(extension->name())).value(),
137 prop_var.get().pwszVal);
138 prop_var.Reset();
139 base::MessageLoop::current()->Quit();
140 }
141
PostValidationTaskToUIThread(const base::Closure & validation_task)142 void PostValidationTaskToUIThread(const base::Closure& validation_task) {
143 content::BrowserThread::PostTask(
144 content::BrowserThread::UI, FROM_HERE, validation_task);
145 }
146
147 // Posts a validation task to the FILE thread which bounces back to the UI
148 // thread and then does validation. This is necessary because the icon profile
149 // pref only gets set at the end of icon creation (which happens on the FILE
150 // thread) and is set on the UI thread.
WaitAndValidateBrowserWindowProperties(const base::Closure & validation_task)151 void WaitAndValidateBrowserWindowProperties(
152 const base::Closure& validation_task) {
153 content::BrowserThread::PostTask(
154 content::BrowserThread::FILE,
155 FROM_HERE,
156 base::Bind(&PostValidationTaskToUIThread, validation_task));
157 content::RunMessageLoop();
158 }
159
160 } // namespace
161
162 // Tests that require the profile shortcut manager to be instantiated despite
163 // having --user-data-dir specified.
164 class BrowserTestWithProfileShortcutManager : public InProcessBrowserTest {
165 public:
BrowserTestWithProfileShortcutManager()166 BrowserTestWithProfileShortcutManager() {}
167
SetUpCommandLine(CommandLine * command_line)168 virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE {
169 command_line->AppendSwitch(switches::kEnableProfileShortcutManager);
170 }
171
172 private:
173 DISALLOW_COPY_AND_ASSIGN(BrowserTestWithProfileShortcutManager);
174 };
175
176 // Check that the window properties on Windows are properly set.
IN_PROC_BROWSER_TEST_F(BrowserTestWithProfileShortcutManager,WindowProperties)177 IN_PROC_BROWSER_TEST_F(BrowserTestWithProfileShortcutManager,
178 WindowProperties) {
179 // Disable this test in Metro+Ash where Windows window properties aren't used.
180 if (CommandLine::ForCurrentProcess()->HasSwitch(switches::kAshBrowserTests))
181 return;
182
183 // This test checks HWND properties that are only available on Win7+.
184 if (base::win::GetVersion() < base::win::VERSION_WIN7)
185 return;
186
187 // Single profile case. The profile name should not be shown.
188 WaitAndValidateBrowserWindowProperties(base::Bind(
189 &ValidateBrowserWindowProperties, browser(), base::string16()));
190
191 // If multiprofile mode is not enabled, we can't test the behavior when there
192 // are multiple profiles.
193 if (!profiles::IsMultipleProfilesEnabled())
194 return;
195
196 // Two profile case. Both profile names should be shown.
197 ProfileManager* profile_manager = g_browser_process->profile_manager();
198 ProfileInfoCache& cache = profile_manager->GetProfileInfoCache();
199
200 base::FilePath path_profile2 =
201 profile_manager->GenerateNextProfileDirectoryPath();
202 profile_manager->CreateProfileAsync(path_profile2,
203 base::Bind(&UnblockOnProfileCreation),
204 base::string16(), base::string16(),
205 std::string());
206
207 // Spin to allow profile creation to take place, loop is terminated
208 // by UnblockOnProfileCreation when the profile is created.
209 content::RunMessageLoop();
210
211 // The default profile's name should be part of the relaunch name.
212 WaitAndValidateBrowserWindowProperties(
213 base::Bind(&ValidateBrowserWindowProperties,
214 browser(),
215 base::UTF8ToUTF16(browser()->profile()->GetProfileName())));
216
217 // The second profile's name should be part of the relaunch name.
218 Browser* profile2_browser =
219 CreateBrowser(profile_manager->GetProfileByPath(path_profile2));
220 size_t profile2_index = cache.GetIndexOfProfileWithPath(path_profile2);
221 WaitAndValidateBrowserWindowProperties(
222 base::Bind(&ValidateBrowserWindowProperties,
223 profile2_browser,
224 cache.GetNameOfProfileAtIndex(profile2_index)));
225 }
226
IN_PROC_BROWSER_TEST_F(BrowserWindowPropertyManagerTest,HostedApp)227 IN_PROC_BROWSER_TEST_F(BrowserWindowPropertyManagerTest, HostedApp) {
228 #if defined(USE_ASH)
229 // Disable this test in Metro+Ash where Windows window properties aren't used.
230 if (CommandLine::ForCurrentProcess()->HasSwitch(switches::kAshBrowserTests))
231 return;
232 #endif
233
234 // This test checks HWND properties that are only available on Win7+.
235 if (base::win::GetVersion() < base::win::VERSION_WIN7)
236 return;
237
238 // Load an app.
239 const extensions::Extension* extension =
240 LoadExtension(test_data_dir_.AppendASCII("app/"));
241 EXPECT_TRUE(extension);
242
243 OpenApplication(AppLaunchParams(browser()->profile(),
244 extension,
245 extensions::LAUNCH_CONTAINER_WINDOW,
246 NEW_FOREGROUND_TAB));
247
248 // Check that the new browser has an app name.
249 // The launch should have created a new browser.
250 ASSERT_EQ(2u,
251 chrome::GetBrowserCount(browser()->profile(),
252 browser()->host_desktop_type()));
253
254 // Find the new browser.
255 Browser* app_browser = NULL;
256 for (chrome::BrowserIterator it; !it.done() && !app_browser; it.Next()) {
257 if (*it != browser())
258 app_browser = *it;
259 }
260 ASSERT_TRUE(app_browser);
261 ASSERT_TRUE(app_browser != browser());
262
263 WaitAndValidateBrowserWindowProperties(
264 base::Bind(&ValidateHostedAppWindowProperties, app_browser, extension));
265 }
266