Badge Windows profile shortcuts with multi-user avatar.

chrome/installer/util/shell_util_unittest.cc

 // Copyright (c) 2011 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 <windows.h>
 #include <shellapi.h>
 #include <shlobj.h>
 
 #include <fstream>
 
@@ skipping 96 matching lines @@
   FilePath exe_path = temp_dir_.path().AppendASCII("setup.exe");
   EXPECT_TRUE(file_util::CopyFile(exe_full_path, exe_path));
 
   FilePath shortcut_path = temp_dir_.path().AppendASCII("shortcut.lnk");
   const std::wstring description(L"dummy description");
   EXPECT_TRUE(ShellUtil::UpdateChromeShortcut(dist,
                                               exe_path.value(),
                                               shortcut_path.value(),
                                               L"",
                                               description,
+                                              exe_path.value(),
+                                              dist->GetIconIndex(),
                                               true));
   EXPECT_TRUE(VerifyChromeShortcut(exe_path.value(),
                                    shortcut_path.value(),
                                    description, 0));
 
   // Now specify an icon index in master prefs and make sure it works.
   FilePath prefs_path = temp_dir_.path().AppendASCII(
       installer::kDefaultMasterPrefs);
   std::ofstream file;
   file.open(prefs_path.value().c_str());
   ASSERT_TRUE(file.is_open());
   file <<
 "{"
 " \"distribution\":{"
 "   \"chrome_shortcut_icon_index\" : 1"
 " }"
 "}";
   file.close();
   ASSERT_TRUE(file_util::Delete(shortcut_path, false));
   EXPECT_TRUE(ShellUtil::UpdateChromeShortcut(dist,
                                               exe_path.value(),
                                               shortcut_path.value(),
                                               L"",
                                               description,
+                                              exe_path.value(),
+                                              dist->GetIconIndex(),
                                               true));
   EXPECT_TRUE(VerifyChromeShortcut(exe_path.value(),
                                    shortcut_path.value(),
                                    description, 1));
 
   // Now change only description to update shortcut and make sure icon index
   // doesn't change.
   const std::wstring description2(L"dummy description 2");
   EXPECT_TRUE(ShellUtil::UpdateChromeShortcut(dist,
                                               exe_path.value(),
                                               shortcut_path.value(),
                                               L"",
                                               description2,
+                                              exe_path.value(),
+                                              dist->GetIconIndex(),
                                               false));
   EXPECT_TRUE(VerifyChromeShortcut(exe_path.value(),
                                    shortcut_path.value(),
                                    description2, 1));
 }
 
 TEST_F(ShellUtilTest, CreateChromeDesktopShortcutTest) {
   // Run this test on Vista+ only if we are running elevated.
   if (base::win::GetVersion() > base::win::VERSION_XP && !IsUserAnAdmin()) {
     LOG(ERROR) << "Must be admin to run this test on Vista+";
@@ skipping 39 matching lines @@
       default_profile_shortcut_name);
   FilePath second_profile_shortcut_path = user_desktop_path.Append(
       second_profile_shortcut_name);
 
   // Test simple creation of a user-level shortcut.
   EXPECT_TRUE(ShellUtil::CreateChromeDesktopShortcut(dist,
                                                      exe_path.value(),
                                                      description,
                                                      L"",
                                                      L"",
+                                                     exe_path.value(),
+                                                     dist->GetIconIndex(),
                                                      ShellUtil::CURRENT_USER,
                                                      false,
                                                      true));
   EXPECT_TRUE(VerifyChromeShortcut(exe_path.value(),
                                    user_shortcut_path.value(),
                                    description,
                                    0));
   EXPECT_TRUE(ShellUtil::RemoveChromeDesktopShortcut(dist,
                                                      ShellUtil::CURRENT_USER,
                                                      false));
 
   // Test simple creation of a system-level shortcut.
   EXPECT_TRUE(ShellUtil::CreateChromeDesktopShortcut(dist,
                                                      exe_path.value(),
                                                      description,
                                                      L"",
                                                      L"",
+                                                     exe_path.value(),
+                                                     dist->GetIconIndex(),
                                                      ShellUtil::SYSTEM_LEVEL,
                                                      false,
                                                      true));
   EXPECT_TRUE(VerifyChromeShortcut(exe_path.value(),
                                    system_shortcut_path.value(),
                                    description,
                                    0));
   EXPECT_TRUE(ShellUtil::RemoveChromeDesktopShortcut(dist,
                                                      ShellUtil::SYSTEM_LEVEL,
                                                      false));
 
   // Test creation of a user-level shortcut when a system-level shortcut
   // is already present (should fail).
   EXPECT_TRUE(ShellUtil::CreateChromeDesktopShortcut(dist,
                                                      exe_path.value(),
                                                      description,
                                                      L"",
                                                      L"",
+                                                     exe_path.value(),
+                                                     dist->GetIconIndex(),
                                                      ShellUtil::SYSTEM_LEVEL,
                                                      false,
                                                      true));
   EXPECT_FALSE(ShellUtil::CreateChromeDesktopShortcut(dist,
                                                       exe_path.value(),
                                                       description,
                                                       L"",
                                                       L"",
+                                                      exe_path.value(),
+                                                      dist->GetIconIndex(),
                                                       ShellUtil::CURRENT_USER,
                                                       false,
                                                       true));
   EXPECT_TRUE(VerifyChromeShortcut(exe_path.value(),
                                    system_shortcut_path.value(),
                                    description,
                                    0));
   EXPECT_FALSE(file_util::PathExists(user_shortcut_path));
   EXPECT_TRUE(ShellUtil::RemoveChromeDesktopShortcut(dist,
                                                      ShellUtil::SYSTEM_LEVEL,
                                                      false));
 
   // Test creation of a system-level shortcut when a user-level shortcut
   // is already present (should succeed).
   EXPECT_TRUE(ShellUtil::CreateChromeDesktopShortcut(dist,
                                                      exe_path.value(),
                                                      description,
                                                      L"",
                                                      L"",
+                                                     exe_path.value(),
+                                                     dist->GetIconIndex(),
                                                      ShellUtil::CURRENT_USER,
                                                      false,
                                                      true));
   EXPECT_TRUE(ShellUtil::CreateChromeDesktopShortcut(dist,
                                                      exe_path.value(),
                                                      description,
                                                      L"",
                                                      L"",
+                                                     exe_path.value(),
+                                                     dist->GetIconIndex(),
                                                      ShellUtil::SYSTEM_LEVEL,
                                                      false,
                                                      true));
   EXPECT_TRUE(VerifyChromeShortcut(exe_path.value(),
                                    user_shortcut_path.value(),
                                    description,
                                    0));
   EXPECT_TRUE(VerifyChromeShortcut(exe_path.value(),
                                    system_shortcut_path.value(),
                                    description,
                                    0));
   EXPECT_TRUE(ShellUtil::RemoveChromeDesktopShortcut(dist,
                                                      ShellUtil::CURRENT_USER,
                                                      false));
   EXPECT_TRUE(ShellUtil::RemoveChromeDesktopShortcut(dist,
                                                      ShellUtil::SYSTEM_LEVEL,
                                                      false));
 
   // Test creation of two profile-specific shortcuts (these are always
   // user-level).
   EXPECT_TRUE(ShellUtil::CreateChromeDesktopShortcut(
       dist,
       exe_path.value(),
       description,
       default_profile_user_name,
       L"--profile-directory=\"Default\"",
+      exe_path.value(),
+      dist->GetIconIndex(),
       ShellUtil::CURRENT_USER,
       false,
       true));
   EXPECT_TRUE(VerifyChromeShortcut(exe_path.value(),
                                    default_profile_shortcut_path.value(),
                                    description,
                                    0));
   EXPECT_TRUE(ShellUtil::CreateChromeDesktopShortcut(
       dist,
       exe_path.value(),
       description,
       second_profile_user_name,
       L"--profile-directory=\"Profile 1\"",
+      exe_path.value(),
+      dist->GetIconIndex(),
       ShellUtil::CURRENT_USER,
       false,
       true));
   EXPECT_TRUE(VerifyChromeShortcut(exe_path.value(),
                                    second_profile_shortcut_path.value(),
                                    description,
                                    0));
   std::vector<string16> profile_names;
   profile_names.push_back(default_profile_shortcut_name);
   profile_names.push_back(second_profile_shortcut_name);
   EXPECT_TRUE(ShellUtil::RemoveChromeDesktopShortcutsWithAppendedNames(
       profile_names));
 }