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. 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. 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 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 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. 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: 166 BrowserTestWithProfileShortcutManager() {} 167 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. 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 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
