| Index: chrome/installer/util/shell_util_unittest.cc
|
| diff --git a/chrome/installer/util/shell_util_unittest.cc b/chrome/installer/util/shell_util_unittest.cc
|
| index 647ec28ba02f03f9500a4216cdae25f009e2ee9d..400081312733e9244da4ca958b45901883df073c 100644
|
| --- a/chrome/installer/util/shell_util_unittest.cc
|
| +++ b/chrome/installer/util/shell_util_unittest.cc
|
| @@ -104,7 +104,9 @@ class ShellUtilShortcutTest : public testing::Test {
|
| case ShellUtil::SHORTCUT_LOCATION_START_MENU:
|
| expected_path = (properties.level == ShellUtil::CURRENT_USER) ?
|
| fake_start_menu_.path() : fake_common_start_menu_.path();
|
| - expected_path = expected_path.Append(dist_->GetAppShortCutName());
|
| + expected_path = expected_path.Append(
|
| + dist_->GetStartMenuShortcutSubfolder(
|
| + BrowserDistribution::SUBFOLDER_CHROME));
|
| break;
|
| default:
|
| ADD_FAILURE() << "Unknown location";
|
| @@ -112,10 +114,12 @@ class ShellUtilShortcutTest : public testing::Test {
|
| }
|
|
|
| string16 shortcut_name;
|
| - if (properties.has_shortcut_name())
|
| + if (properties.has_shortcut_name()) {
|
| shortcut_name = properties.shortcut_name;
|
| - else
|
| - shortcut_name = dist_->GetAppShortCutName();
|
| + } else {
|
| + shortcut_name =
|
| + dist_->GetShortcutName(BrowserDistribution::SHORTCUT_CHROME);
|
| + }
|
| shortcut_name.append(installer::kLnkExt);
|
| expected_path = expected_path.Append(shortcut_name);
|
|
|
| @@ -141,7 +145,7 @@ class ShellUtilShortcutTest : public testing::Test {
|
| if (properties.has_icon()) {
|
| expected_properties.set_icon(properties.icon, 0);
|
| } else {
|
| - int icon_index = dist->GetIconIndex();
|
| + int icon_index = dist->GetIconIndex(BrowserDistribution::SHORTCUT_CHROME);
|
| expected_properties.set_icon(chrome_exe_, icon_index);
|
| }
|
|
|
| @@ -200,12 +204,16 @@ TEST_F(ShellUtilShortcutTest, GetShortcutPath) {
|
| ShellUtil::GetShortcutPath(ShellUtil::SHORTCUT_LOCATION_QUICK_LAUNCH, dist_,
|
| ShellUtil::SYSTEM_LEVEL, &path);
|
| EXPECT_EQ(fake_default_user_quick_launch_.path(), path);
|
| + string16 start_menu_subfolder =
|
| + dist_->GetStartMenuShortcutSubfolder(
|
| + BrowserDistribution::SUBFOLDER_CHROME);
|
| ShellUtil::GetShortcutPath(ShellUtil::SHORTCUT_LOCATION_START_MENU, dist_,
|
| ShellUtil::CURRENT_USER, &path);
|
| - EXPECT_EQ(fake_start_menu_.path().Append(dist_->GetAppShortCutName()), path);
|
| + EXPECT_EQ(fake_start_menu_.path().Append(start_menu_subfolder),
|
| + path);
|
| ShellUtil::GetShortcutPath(ShellUtil::SHORTCUT_LOCATION_START_MENU, dist_,
|
| ShellUtil::SYSTEM_LEVEL, &path);
|
| - EXPECT_EQ(fake_common_start_menu_.path().Append(dist_->GetAppShortCutName()),
|
| + EXPECT_EQ(fake_common_start_menu_.path().Append(start_menu_subfolder),
|
| path);
|
| }
|
|
|
| @@ -308,7 +316,9 @@ TEST_F(ShellUtilShortcutTest, CreateIfNoSystemLevel) {
|
| }
|
|
|
| TEST_F(ShellUtilShortcutTest, CreateIfNoSystemLevelWithSystemLevelPresent) {
|
| - string16 shortcut_name(dist_->GetAppShortCutName() + installer::kLnkExt);
|
| + string16 shortcut_name(
|
| + dist_->GetShortcutName(BrowserDistribution::SHORTCUT_CHROME) +
|
| + installer::kLnkExt);
|
|
|
| test_properties_.level = ShellUtil::SYSTEM_LEVEL;
|
| ASSERT_TRUE(ShellUtil::CreateOrUpdateShortcut(
|
| @@ -335,7 +345,9 @@ TEST_F(ShellUtilShortcutTest, CreateIfNoSystemLevelStartMenu) {
|
| }
|
|
|
| TEST_F(ShellUtilShortcutTest, CreateAlwaysUserWithSystemLevelPresent) {
|
| - string16 shortcut_name(dist_->GetAppShortCutName() + installer::kLnkExt);
|
| + string16 shortcut_name(
|
| + dist_->GetShortcutName(BrowserDistribution::SHORTCUT_CHROME) +
|
| + installer::kLnkExt);
|
|
|
| test_properties_.level = ShellUtil::SYSTEM_LEVEL;
|
| ASSERT_TRUE(ShellUtil::CreateOrUpdateShortcut(
|
| @@ -357,7 +369,9 @@ TEST_F(ShellUtilShortcutTest, RemoveChromeShortcut) {
|
| ShellUtil::SHORTCUT_LOCATION_DESKTOP, dist_, test_properties_,
|
| ShellUtil::SHELL_SHORTCUT_CREATE_ALWAYS));
|
|
|
| - string16 shortcut_name(dist_->GetAppShortCutName() + installer::kLnkExt);
|
| + string16 shortcut_name(
|
| + dist_->GetShortcutName(BrowserDistribution::SHORTCUT_CHROME) +
|
| + installer::kLnkExt);
|
| base::FilePath shortcut_path(fake_user_desktop_.path().Append(shortcut_name));
|
| ASSERT_TRUE(base::PathExists(shortcut_path));
|
|
|
| @@ -374,7 +388,9 @@ TEST_F(ShellUtilShortcutTest, RemoveSystemLevelChromeShortcut) {
|
| ShellUtil::SHORTCUT_LOCATION_DESKTOP, dist_, test_properties_,
|
| ShellUtil::SHELL_SHORTCUT_CREATE_ALWAYS));
|
|
|
| - string16 shortcut_name(dist_->GetAppShortCutName() + installer::kLnkExt);
|
| + string16 shortcut_name(
|
| + dist_->GetShortcutName(BrowserDistribution::SHORTCUT_CHROME) +
|
| + installer::kLnkExt);
|
| base::FilePath shortcut_path(
|
| fake_common_desktop_.path().Append(shortcut_name));
|
| ASSERT_TRUE(base::PathExists(shortcut_path));
|
| @@ -423,7 +439,9 @@ TEST_F(ShellUtilShortcutTest, UpdateChromeShortcut) {
|
| ShellUtil::SHORTCUT_LOCATION_DESKTOP, dist_, test_properties_,
|
| ShellUtil::SHELL_SHORTCUT_CREATE_ALWAYS));
|
|
|
| - string16 shortcut_name(dist_->GetAppShortCutName() + installer::kLnkExt);
|
| + string16 shortcut_name(
|
| + dist_->GetShortcutName(BrowserDistribution::SHORTCUT_CHROME) +
|
| + installer::kLnkExt);
|
| base::FilePath shortcut_path(fake_user_desktop_.path().Append(shortcut_name));
|
| ASSERT_TRUE(base::PathExists(shortcut_path));
|
|
|
| @@ -447,7 +465,9 @@ TEST_F(ShellUtilShortcutTest, UpdateSystemLevelChromeShortcut) {
|
| ShellUtil::SHORTCUT_LOCATION_DESKTOP, dist_, test_properties_,
|
| ShellUtil::SHELL_SHORTCUT_CREATE_ALWAYS));
|
|
|
| - string16 shortcut_name(dist_->GetAppShortCutName() + installer::kLnkExt);
|
| + string16 shortcut_name(
|
| + dist_->GetShortcutName(BrowserDistribution::SHORTCUT_CHROME) +
|
| + installer::kLnkExt);
|
| base::FilePath shortcut_path(
|
| fake_common_desktop_.path().Append(shortcut_name));
|
| ASSERT_TRUE(base::PathExists(shortcut_path));
|
| @@ -525,7 +545,9 @@ TEST_F(ShellUtilShortcutTest, CreateMultipleStartMenuShortcutsAndRemoveFolder) {
|
| ShellUtil::SHELL_SHORTCUT_CREATE_ALWAYS));
|
|
|
| base::FilePath shortcut_folder(
|
| - fake_start_menu_.path().Append(dist_->GetAppShortCutName()));
|
| + fake_start_menu_.path().Append(
|
| + dist_->GetStartMenuShortcutSubfolder(
|
| + BrowserDistribution::SUBFOLDER_CHROME)));
|
| base::FileEnumerator file_counter(shortcut_folder, false,
|
| base::FileEnumerator::FILES);
|
| int count = 0;
|
| @@ -552,7 +574,9 @@ TEST_F(ShellUtilShortcutTest, DontRemoveChromeShortcutIfPointsToAnotherChrome) {
|
| ShellUtil::SHORTCUT_LOCATION_DESKTOP, dist_, test_properties_,
|
| ShellUtil::SHELL_SHORTCUT_CREATE_ALWAYS));
|
|
|
| - string16 shortcut_name(dist_->GetAppShortCutName() + installer::kLnkExt);
|
| + string16 shortcut_name(
|
| + dist_->GetShortcutName(BrowserDistribution::SHORTCUT_CHROME) +
|
| + installer::kLnkExt);
|
| base::FilePath shortcut_path(fake_user_desktop_.path().Append(shortcut_name));
|
| ASSERT_TRUE(base::PathExists(shortcut_path));
|
|
|
|
|