| // Copyright (c) 2012 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 <objbase.h> |
| |
| #include <string> |
| |
| #include "base/base_paths.h" |
| #include "base/files/file_path.h" |
| #include "base/files/file_util.h" |
| #include "base/files/scoped_temp_dir.h" |
| #include "base/memory/scoped_ptr.h" |
| #include "base/strings/string16.h" |
| #include "base/strings/utf_string_conversions.h" |
| #include "base/test/scoped_path_override.h" |
| #include "base/test/test_shortcut_win.h" |
| #include "base/version.h" |
| #include "base/win/shortcut.h" |
| #include "chrome/installer/setup/install.h" |
| #include "chrome/installer/setup/install_worker.h" |
| #include "chrome/installer/setup/setup_constants.h" |
| #include "chrome/installer/util/browser_distribution.h" |
| #include "chrome/installer/util/installer_state.h" |
| #include "chrome/installer/util/master_preferences.h" |
| #include "chrome/installer/util/master_preferences_constants.h" |
| #include "chrome/installer/util/product.h" |
| #include "chrome/installer/util/shell_util.h" |
| #include "chrome/installer/util/util_constants.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| namespace { |
| |
| class CreateVisualElementsManifestTest : public testing::Test { |
| protected: |
| void SetUp() override { |
| // Create a temp directory for testing. |
| ASSERT_TRUE(test_dir_.CreateUniqueTempDir()); |
| |
| version_ = Version("0.0.0.0"); |
| |
| version_dir_ = test_dir_.path().AppendASCII(version_.GetString()); |
| ASSERT_TRUE(base::CreateDirectory(version_dir_)); |
| |
| manifest_path_ = |
| test_dir_.path().Append(installer::kVisualElementsManifest); |
| } |
| |
| void TearDown() override { |
| // Clean up test directory manually so we can fail if it leaks. |
| ASSERT_TRUE(test_dir_.Delete()); |
| } |
| |
| // The temporary directory used to contain the test operations. |
| base::ScopedTempDir test_dir_; |
| |
| // A dummy version number used to create the version directory. |
| Version version_; |
| |
| // The path to |test_dir_|\|version_|. |
| base::FilePath version_dir_; |
| |
| // The path to VisualElementsManifest.xml. |
| base::FilePath manifest_path_; |
| }; |
| |
| class InstallShortcutTest : public testing::Test { |
| protected: |
| void SetUp() override { |
| EXPECT_EQ(S_OK, CoInitialize(NULL)); |
| |
| dist_ = BrowserDistribution::GetDistribution(); |
| ASSERT_TRUE(dist_ != NULL); |
| product_.reset(new installer::Product(dist_)); |
| |
| ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); |
| chrome_exe_ = temp_dir_.path().Append(installer::kChromeExe); |
| EXPECT_EQ(0, base::WriteFile(chrome_exe_, "", 0)); |
| |
| ShellUtil::ShortcutProperties chrome_properties(ShellUtil::CURRENT_USER); |
| product_->AddDefaultShortcutProperties(chrome_exe_, &chrome_properties); |
| |
| expected_properties_.set_target(chrome_exe_); |
| expected_properties_.set_icon(chrome_properties.icon, |
| chrome_properties.icon_index); |
| expected_properties_.set_app_id(chrome_properties.app_id); |
| expected_properties_.set_description(chrome_properties.description); |
| expected_properties_.set_dual_mode(false); |
| expected_start_menu_properties_ = expected_properties_; |
| expected_start_menu_properties_.set_dual_mode(true); |
| |
| prefs_.reset(GetFakeMasterPrefs(false, false, false)); |
| |
| ASSERT_TRUE(fake_user_desktop_.CreateUniqueTempDir()); |
| ASSERT_TRUE(fake_common_desktop_.CreateUniqueTempDir()); |
| ASSERT_TRUE(fake_user_quick_launch_.CreateUniqueTempDir()); |
| ASSERT_TRUE(fake_start_menu_.CreateUniqueTempDir()); |
| ASSERT_TRUE(fake_common_start_menu_.CreateUniqueTempDir()); |
| user_desktop_override_.reset( |
| new base::ScopedPathOverride(base::DIR_USER_DESKTOP, |
| fake_user_desktop_.path())); |
| common_desktop_override_.reset( |
| new base::ScopedPathOverride(base::DIR_COMMON_DESKTOP, |
| fake_common_desktop_.path())); |
| user_quick_launch_override_.reset( |
| new base::ScopedPathOverride(base::DIR_USER_QUICK_LAUNCH, |
| fake_user_quick_launch_.path())); |
| start_menu_override_.reset( |
| new base::ScopedPathOverride(base::DIR_START_MENU, |
| fake_start_menu_.path())); |
| common_start_menu_override_.reset( |
| new base::ScopedPathOverride(base::DIR_COMMON_START_MENU, |
| fake_common_start_menu_.path())); |
| |
| base::string16 shortcut_name( |
| dist_->GetShortcutName(BrowserDistribution::SHORTCUT_CHROME) + |
| installer::kLnkExt); |
| base::string16 alternate_shortcut_name( |
| dist_->GetShortcutName(BrowserDistribution::SHORTCUT_CHROME_ALTERNATE) + |
| installer::kLnkExt); |
| |
| user_desktop_shortcut_ = |
| fake_user_desktop_.path().Append(shortcut_name); |
| user_quick_launch_shortcut_ = |
| fake_user_quick_launch_.path().Append(shortcut_name); |
| user_start_menu_shortcut_ = |
| fake_start_menu_.path().Append( |
| dist_->GetStartMenuShortcutSubfolder( |
| BrowserDistribution::SUBFOLDER_CHROME)) |
| .Append(shortcut_name); |
| system_desktop_shortcut_ = |
| fake_common_desktop_.path().Append(shortcut_name); |
| system_start_menu_shortcut_ = |
| fake_common_start_menu_.path().Append( |
| dist_->GetStartMenuShortcutSubfolder( |
| BrowserDistribution::SUBFOLDER_CHROME)) |
| .Append(shortcut_name); |
| user_alternate_desktop_shortcut_ = |
| fake_user_desktop_.path().Append(alternate_shortcut_name); |
| } |
| |
| void TearDown() override { |
| // Try to unpin potentially pinned shortcuts (although pinning isn't tested, |
| // the call itself might still have pinned the Start Menu shortcuts). |
| base::win::TaskbarUnpinShortcutLink( |
| user_start_menu_shortcut_.value().c_str()); |
| base::win::TaskbarUnpinShortcutLink( |
| system_start_menu_shortcut_.value().c_str()); |
| CoUninitialize(); |
| } |
| |
| installer::MasterPreferences* GetFakeMasterPrefs( |
| bool do_not_create_desktop_shortcut, |
| bool do_not_create_quick_launch_shortcut, |
| bool alternate_desktop_shortcut) { |
| const struct { |
| const char* pref_name; |
| bool is_desired; |
| } desired_prefs[] = { |
| { installer::master_preferences::kDoNotCreateDesktopShortcut, |
| do_not_create_desktop_shortcut }, |
| { installer::master_preferences::kDoNotCreateQuickLaunchShortcut, |
| do_not_create_quick_launch_shortcut }, |
| { installer::master_preferences::kAltShortcutText, |
| alternate_desktop_shortcut }, |
| }; |
| |
| std::string master_prefs("{\"distribution\":{"); |
| for (size_t i = 0; i < arraysize(desired_prefs); ++i) { |
| master_prefs += (i == 0 ? "\"" : ",\""); |
| master_prefs += desired_prefs[i].pref_name; |
| master_prefs += "\":"; |
| master_prefs += desired_prefs[i].is_desired ? "true" : "false"; |
| } |
| master_prefs += "}}"; |
| |
| return new installer::MasterPreferences(master_prefs); |
| } |
| |
| base::win::ShortcutProperties expected_properties_; |
| base::win::ShortcutProperties expected_start_menu_properties_; |
| |
| BrowserDistribution* dist_; |
| base::FilePath chrome_exe_; |
| scoped_ptr<installer::Product> product_; |
| scoped_ptr<installer::MasterPreferences> prefs_; |
| |
| base::ScopedTempDir temp_dir_; |
| base::ScopedTempDir fake_user_desktop_; |
| base::ScopedTempDir fake_common_desktop_; |
| base::ScopedTempDir fake_user_quick_launch_; |
| base::ScopedTempDir fake_start_menu_; |
| base::ScopedTempDir fake_common_start_menu_; |
| scoped_ptr<base::ScopedPathOverride> user_desktop_override_; |
| scoped_ptr<base::ScopedPathOverride> common_desktop_override_; |
| scoped_ptr<base::ScopedPathOverride> user_quick_launch_override_; |
| scoped_ptr<base::ScopedPathOverride> start_menu_override_; |
| scoped_ptr<base::ScopedPathOverride> common_start_menu_override_; |
| |
| base::FilePath user_desktop_shortcut_; |
| base::FilePath user_quick_launch_shortcut_; |
| base::FilePath user_start_menu_shortcut_; |
| base::FilePath system_desktop_shortcut_; |
| base::FilePath system_start_menu_shortcut_; |
| base::FilePath user_alternate_desktop_shortcut_; |
| }; |
| |
| } // namespace |
| |
| // Test that VisualElementsManifest.xml is not created when VisualElements are |
| // not present. |
| TEST_F(CreateVisualElementsManifestTest, VisualElementsManifestNotCreated) { |
| ASSERT_TRUE( |
| installer::CreateVisualElementsManifest(test_dir_.path(), version_)); |
| ASSERT_FALSE(base::PathExists(manifest_path_)); |
| } |
| |
| // Test that VisualElementsManifest.xml is created with the correct content when |
| // VisualElements are present. |
| TEST_F(CreateVisualElementsManifestTest, VisualElementsManifestCreated) { |
| ASSERT_TRUE(base::CreateDirectory( |
| version_dir_.Append(installer::kVisualElements))); |
| ASSERT_TRUE( |
| installer::CreateVisualElementsManifest(test_dir_.path(), version_)); |
| ASSERT_TRUE(base::PathExists(manifest_path_)); |
| |
| std::string read_manifest; |
| ASSERT_TRUE(base::ReadFileToString(manifest_path_, &read_manifest)); |
| |
| static const char kExpectedManifest[] = |
| "<Application>\r\n" |
| " <VisualElements\r\n" |
| " DisplayName='Google Chrome'\r\n" |
| " Logo='0.0.0.0\\VisualElements\\Logo.png'\r\n" |
| " SmallLogo='0.0.0.0\\VisualElements\\SmallLogo.png'\r\n" |
| " ForegroundText='light'\r\n" |
| " BackgroundColor='#323232'>\r\n" |
| " <DefaultTile ShowName='allLogos'/>\r\n" |
| " <SplashScreen Image='0.0.0.0\\VisualElements\\splash-620x300.png'/>" |
| "\r\n" |
| " </VisualElements>\r\n" |
| "</Application>"; |
| |
| ASSERT_STREQ(kExpectedManifest, read_manifest.c_str()); |
| } |
| |
| TEST_F(InstallShortcutTest, CreateAllShortcuts) { |
| installer::CreateOrUpdateShortcuts( |
| chrome_exe_, *product_, *prefs_, installer::CURRENT_USER, |
| installer::INSTALL_SHORTCUT_CREATE_ALL); |
| base::win::ValidateShortcut(user_desktop_shortcut_, expected_properties_); |
| base::win::ValidateShortcut(user_quick_launch_shortcut_, |
| expected_properties_); |
| base::win::ValidateShortcut(user_start_menu_shortcut_, |
| expected_start_menu_properties_); |
| } |
| |
| TEST_F(InstallShortcutTest, CreateAllShortcutsSystemLevel) { |
| installer::CreateOrUpdateShortcuts( |
| chrome_exe_, *product_, *prefs_, installer::ALL_USERS, |
| installer::INSTALL_SHORTCUT_CREATE_ALL); |
| base::win::ValidateShortcut(system_desktop_shortcut_, expected_properties_); |
| base::win::ValidateShortcut(system_start_menu_shortcut_, |
| expected_start_menu_properties_); |
| // The quick launch shortcut is always created per-user for the admin running |
| // the install (other users will get it via Active Setup). |
| base::win::ValidateShortcut(user_quick_launch_shortcut_, |
| expected_properties_); |
| } |
| |
| TEST_F(InstallShortcutTest, CreateAllShortcutsAlternateDesktopName) { |
| scoped_ptr<installer::MasterPreferences> prefs_alt_desktop( |
| GetFakeMasterPrefs(false, false, true)); |
| installer::CreateOrUpdateShortcuts( |
| chrome_exe_, *product_, *prefs_alt_desktop, installer::CURRENT_USER, |
| installer::INSTALL_SHORTCUT_CREATE_ALL); |
| base::win::ValidateShortcut(user_alternate_desktop_shortcut_, |
| expected_properties_); |
| base::win::ValidateShortcut(user_quick_launch_shortcut_, |
| expected_properties_); |
| base::win::ValidateShortcut(user_start_menu_shortcut_, |
| expected_start_menu_properties_); |
| } |
| |
| TEST_F(InstallShortcutTest, CreateAllShortcutsButDesktopShortcut) { |
| scoped_ptr<installer::MasterPreferences> prefs_no_desktop( |
| GetFakeMasterPrefs(true, false, false)); |
| installer::CreateOrUpdateShortcuts( |
| chrome_exe_, *product_, *prefs_no_desktop, installer::CURRENT_USER, |
| installer::INSTALL_SHORTCUT_CREATE_ALL); |
| ASSERT_FALSE(base::PathExists(user_desktop_shortcut_)); |
| base::win::ValidateShortcut(user_quick_launch_shortcut_, |
| expected_properties_); |
| base::win::ValidateShortcut(user_start_menu_shortcut_, |
| expected_start_menu_properties_); |
| } |
| |
| TEST_F(InstallShortcutTest, CreateAllShortcutsButQuickLaunchShortcut) { |
| scoped_ptr<installer::MasterPreferences> prefs_no_ql( |
| GetFakeMasterPrefs(false, true, false)); |
| installer::CreateOrUpdateShortcuts( |
| chrome_exe_, *product_, *prefs_no_ql, installer::CURRENT_USER, |
| installer::INSTALL_SHORTCUT_CREATE_ALL); |
| base::win::ValidateShortcut(user_desktop_shortcut_, expected_properties_); |
| ASSERT_FALSE(base::PathExists(user_quick_launch_shortcut_)); |
| base::win::ValidateShortcut(user_start_menu_shortcut_, |
| expected_start_menu_properties_); |
| } |
| |
| TEST_F(InstallShortcutTest, ReplaceAll) { |
| base::win::ShortcutProperties dummy_properties; |
| base::FilePath dummy_target; |
| ASSERT_TRUE(base::CreateTemporaryFileInDir(temp_dir_.path(), &dummy_target)); |
| dummy_properties.set_target(dummy_target); |
| dummy_properties.set_working_dir(fake_user_desktop_.path()); |
| dummy_properties.set_arguments(L"--dummy --args"); |
| dummy_properties.set_app_id(L"El.Dummiest"); |
| |
| ASSERT_TRUE(base::win::CreateOrUpdateShortcutLink( |
| user_desktop_shortcut_, dummy_properties, |
| base::win::SHORTCUT_CREATE_ALWAYS)); |
| ASSERT_TRUE(base::win::CreateOrUpdateShortcutLink( |
| user_quick_launch_shortcut_, dummy_properties, |
| base::win::SHORTCUT_CREATE_ALWAYS)); |
| ASSERT_TRUE(base::CreateDirectory(user_start_menu_shortcut_.DirName())); |
| ASSERT_TRUE(base::win::CreateOrUpdateShortcutLink( |
| user_start_menu_shortcut_, dummy_properties, |
| base::win::SHORTCUT_CREATE_ALWAYS)); |
| |
| installer::CreateOrUpdateShortcuts( |
| chrome_exe_, *product_, *prefs_, installer::CURRENT_USER, |
| installer::INSTALL_SHORTCUT_REPLACE_EXISTING); |
| base::win::ValidateShortcut(user_desktop_shortcut_, expected_properties_); |
| base::win::ValidateShortcut(user_quick_launch_shortcut_, |
| expected_properties_); |
| base::win::ValidateShortcut(user_start_menu_shortcut_, |
| expected_start_menu_properties_); |
| } |
| |
| TEST_F(InstallShortcutTest, ReplaceExisting) { |
| base::win::ShortcutProperties dummy_properties; |
| base::FilePath dummy_target; |
| ASSERT_TRUE(base::CreateTemporaryFileInDir(temp_dir_.path(), &dummy_target)); |
| dummy_properties.set_target(dummy_target); |
| dummy_properties.set_working_dir(fake_user_desktop_.path()); |
| dummy_properties.set_arguments(L"--dummy --args"); |
| dummy_properties.set_app_id(L"El.Dummiest"); |
| |
| ASSERT_TRUE(base::win::CreateOrUpdateShortcutLink( |
| user_desktop_shortcut_, dummy_properties, |
| base::win::SHORTCUT_CREATE_ALWAYS)); |
| ASSERT_TRUE(base::CreateDirectory(user_start_menu_shortcut_.DirName())); |
| |
| installer::CreateOrUpdateShortcuts( |
| chrome_exe_, *product_, *prefs_, installer::CURRENT_USER, |
| installer::INSTALL_SHORTCUT_REPLACE_EXISTING); |
| base::win::ValidateShortcut(user_desktop_shortcut_, expected_properties_); |
| ASSERT_FALSE(base::PathExists(user_quick_launch_shortcut_)); |
| ASSERT_FALSE(base::PathExists(user_start_menu_shortcut_)); |
| } |
| |
| TEST_F(InstallShortcutTest, CreateIfNoSystemLevelAllSystemShortcutsExist) { |
| base::win::ShortcutProperties dummy_properties; |
| base::FilePath dummy_target; |
| ASSERT_TRUE(base::CreateTemporaryFileInDir(temp_dir_.path(), &dummy_target)); |
| dummy_properties.set_target(dummy_target); |
| |
| ASSERT_TRUE(base::win::CreateOrUpdateShortcutLink( |
| system_desktop_shortcut_, dummy_properties, |
| base::win::SHORTCUT_CREATE_ALWAYS)); |
| ASSERT_TRUE(base::CreateDirectory( |
| system_start_menu_shortcut_.DirName())); |
| ASSERT_TRUE(base::win::CreateOrUpdateShortcutLink( |
| system_start_menu_shortcut_, dummy_properties, |
| base::win::SHORTCUT_CREATE_ALWAYS)); |
| |
| installer::CreateOrUpdateShortcuts( |
| chrome_exe_, *product_, *prefs_, installer::CURRENT_USER, |
| installer::INSTALL_SHORTCUT_CREATE_EACH_IF_NO_SYSTEM_LEVEL); |
| ASSERT_FALSE(base::PathExists(user_desktop_shortcut_)); |
| ASSERT_FALSE(base::PathExists(user_start_menu_shortcut_)); |
| // There is no system-level quick launch shortcut, so creating the user-level |
| // one should always succeed. |
| ASSERT_TRUE(base::PathExists(user_quick_launch_shortcut_)); |
| } |
| |
| TEST_F(InstallShortcutTest, CreateIfNoSystemLevelNoSystemShortcutsExist) { |
| installer::CreateOrUpdateShortcuts( |
| chrome_exe_, *product_, *prefs_, installer::CURRENT_USER, |
| installer::INSTALL_SHORTCUT_CREATE_EACH_IF_NO_SYSTEM_LEVEL); |
| base::win::ValidateShortcut(user_desktop_shortcut_, expected_properties_); |
| base::win::ValidateShortcut(user_quick_launch_shortcut_, |
| expected_properties_); |
| base::win::ValidateShortcut(user_start_menu_shortcut_, |
| expected_start_menu_properties_); |
| } |
| |
| TEST_F(InstallShortcutTest, CreateIfNoSystemLevelSomeSystemShortcutsExist) { |
| base::win::ShortcutProperties dummy_properties; |
| base::FilePath dummy_target; |
| ASSERT_TRUE(base::CreateTemporaryFileInDir(temp_dir_.path(), &dummy_target)); |
| dummy_properties.set_target(dummy_target); |
| |
| ASSERT_TRUE(base::win::CreateOrUpdateShortcutLink( |
| system_desktop_shortcut_, dummy_properties, |
| base::win::SHORTCUT_CREATE_ALWAYS)); |
| |
| installer::CreateOrUpdateShortcuts( |
| chrome_exe_, *product_, *prefs_, installer::CURRENT_USER, |
| installer::INSTALL_SHORTCUT_CREATE_EACH_IF_NO_SYSTEM_LEVEL); |
| ASSERT_FALSE(base::PathExists(user_desktop_shortcut_)); |
| base::win::ValidateShortcut(user_quick_launch_shortcut_, |
| expected_properties_); |
| base::win::ValidateShortcut(user_start_menu_shortcut_, |
| expected_start_menu_properties_); |
| } |
| |
| TEST(EscapeXmlAttributeValueTest, EscapeCrazyValue) { |
| base::string16 val(L"This has 'crazy' \"chars\" && < and > signs."); |
| static const wchar_t kExpectedEscapedVal[] = |
| L"This has 'crazy' \"chars\" && < and > signs."; |
| installer::EscapeXmlAttributeValueInSingleQuotes(&val); |
| ASSERT_STREQ(kExpectedEscapedVal, val.c_str()); |
| } |
| |
| TEST(EscapeXmlAttributeValueTest, DontEscapeNormalValue) { |
| base::string16 val(L"Google Chrome"); |
| static const wchar_t kExpectedEscapedVal[] = L"Google Chrome"; |
| installer::EscapeXmlAttributeValueInSingleQuotes(&val); |
| ASSERT_STREQ(kExpectedEscapedVal, val.c_str()); |
| } |
