Uniquify profile shortcut name.

chrome/browser/profiles/profile_shortcut_manager_unittest_win.cc

 // 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>  // For CoInitialize().
 
 #include "base/base_paths.h"
 #include "base/files/file_util.h"
 #include "base/location.h"
 #include "base/message_loop/message_loop.h"
@@ skipping 220 matching lines @@
                 installer::kLnkExt);
     EXPECT_TRUE(base::PathExists(system_level_shortcut_path))
         << location.ToString();
     return system_level_shortcut_path;
   }
 
   void RenameProfile(const tracked_objects::Location& location,
                      const base::FilePath& profile_path,
                      const base::string16& new_profile_name) {
     const size_t profile_index =
-        profile_info_cache_->GetIndexOfProfileWithPath(profile_2_path_);
+        profile_info_cache_->GetIndexOfProfileWithPath(profile_path);
     ASSERT_NE(std::string::npos, profile_index);
     ASSERT_NE(profile_info_cache_->GetNameOfProfileAtIndex(profile_index),
               new_profile_name);
     profile_info_cache_->SetNameOfProfileAtIndex(profile_index,
                                                  new_profile_name);
     RunPendingTasks();
   }
 
   BrowserDistribution* GetDistribution() {
     return BrowserDistribution::GetDistribution();
@@ skipping 618 matching lines @@
   EXPECT_NE(unbadged_icon_1, new_badged_icon_1);
 
   // Ensure the icon doesn't change on avatar change without 2 profiles.
   profile_info_cache_->SetAvatarIconOfProfileAtIndex(profile_index_1, 1);
   RunPendingTasks();
 
   std::string unbadged_icon_1_a;
   EXPECT_TRUE(base::ReadFileToString(icon_path_1, &unbadged_icon_1_a));
   EXPECT_EQ(unbadged_icon_1, unbadged_icon_1_a);
 }
+
+TEST_F(ProfileShortcutManagerTest, ShortcutFilenameUniquified) {
+  const auto suffix =
+      GetDistribution()->GetShortcutName(BrowserDistribution::SHORTCUT_CHROME);
+  std::set<base::FilePath> excludes;
+
+  auto shortcut_filename =
+      profiles::internal::GetUniqueShortcutFilenameForProfile(
+          L"Carrie", excludes, GetDistribution());
+  EXPECT_EQ(L"Carrie - " + suffix + L".lnk", shortcut_filename);
+  excludes.insert(GetUserShortcutsDirectory().Append(shortcut_filename));
+
+  shortcut_filename = profiles::internal::GetUniqueShortcutFilenameForProfile(
+      L"Carrie", excludes, GetDistribution());
+  EXPECT_EQ(L"Carrie - " + suffix + L" (1).lnk", shortcut_filename);
+  excludes.insert(GetUserShortcutsDirectory().Append(shortcut_filename));
+
+  shortcut_filename = profiles::internal::GetUniqueShortcutFilenameForProfile(
+      L"Carrie", excludes, GetDistribution());
+  EXPECT_EQ(L"Carrie - " + suffix + L" (2).lnk", shortcut_filename);
+  excludes.insert(GetUserShortcutsDirectory().Append(shortcut_filename));
+
+  shortcut_filename = profiles::internal::GetUniqueShortcutFilenameForProfile(
+      L"Steven", excludes, GetDistribution());
+  EXPECT_EQ(L"Steven - " + suffix + L".lnk", shortcut_filename);
+  excludes.insert(GetUserShortcutsDirectory().Append(shortcut_filename));
+
+  shortcut_filename = profiles::internal::GetUniqueShortcutFilenameForProfile(
+      L"Steven", excludes, GetDistribution());
+  EXPECT_EQ(L"Steven - " + suffix + L" (1).lnk", shortcut_filename);
+  excludes.insert(GetUserShortcutsDirectory().Append(shortcut_filename));
+
+  shortcut_filename = profiles::internal::GetUniqueShortcutFilenameForProfile(
+      L"Carrie", excludes, GetDistribution());
+  EXPECT_EQ(L"Carrie - " + suffix + L" (3).lnk", shortcut_filename);
+  excludes.insert(GetUserShortcutsDirectory().Append(shortcut_filename));
+
+  excludes.erase(
+      GetUserShortcutsDirectory().Append(L"Carrie - " + suffix + L".lnk"));
+  shortcut_filename = profiles::internal::GetUniqueShortcutFilenameForProfile(
+      L"Carrie", excludes, GetDistribution());
+  EXPECT_EQ(L"Carrie - " + suffix + L".lnk", shortcut_filename);
+}
+
+TEST_F(ProfileShortcutManagerTest, ShortcutFilenameMatcher) {
+  profiles::internal::ShortcutFilenameMatcher matcher(L"Carrie",
+                                                      GetDistribution());
+  const auto suffix =
+      GetDistribution()->GetShortcutName(BrowserDistribution::SHORTCUT_CHROME);
+  EXPECT_TRUE(matcher.IsCanonical(L"Carrie - " + suffix + L" (2).lnk"));
+  EXPECT_FALSE(matcher.IsCanonical(L"Carrie - " + suffix + L"(2).lnk"));
+  EXPECT_FALSE(matcher.IsCanonical(L"Carrie - " + suffix + L" 2.lnk"));
+  EXPECT_FALSE(matcher.IsCanonical(L"Carrie - " + suffix + L"2.lnk"));
+  EXPECT_FALSE(matcher.IsCanonical(L"Carrie - " + suffix + L" - 2.lnk"));
+  EXPECT_FALSE(matcher.IsCanonical(L"Carrie - " + suffix + L" (a).lnk"));
+  EXPECT_TRUE(matcher.IsCanonical(L"Carrie - " + suffix + L" (11).lnk"));
+  EXPECT_TRUE(matcher.IsCanonical(L"Carrie - " + suffix + L" (03).lnk"));
+  EXPECT_TRUE(matcher.IsCanonical(L"Carrie - " + suffix + L" (999).lnk"));
+  EXPECT_FALSE(matcher.IsCanonical(L"Carrie - " + suffix + L" (999).lin"));
+  EXPECT_FALSE(matcher.IsCanonical(L"ABC Carrie - " + suffix + L" DEF.lnk"));
+  EXPECT_FALSE(matcher.IsCanonical(L"ABC Carrie DEF.lnk"));
+  EXPECT_FALSE(matcher.IsCanonical(L"Carrie.lnk"));
+}
+
+TEST_F(ProfileShortcutManagerTest, ShortcutsForProfilesWithIdenticalNames) {
+  SetupDefaultProfileShortcut(FROM_HERE);
+
+  // Create new profile - profile2.
+  CreateProfileWithShortcut(FROM_HERE, profile_2_name_, profile_2_path_);
+  // Check that nothing is changed for profile1.
+  ValidateProfileShortcut(FROM_HERE, profile_1_name_, profile_1_path_);
+
+  // Give to profile1 the same name as profile2.
+  base::string16 new_profile_1_name = profile_2_name_;
+  RenameProfile(FROM_HERE, profile_1_path_, new_profile_1_name);
+  const auto profile_1_shortcut_name = new_profile_1_name + L" - " +
+      GetDistribution()->GetShortcutName(BrowserDistribution::SHORTCUT_CHROME) +
+      L" (1)";
+  const auto profile_1_shortcut_path = GetUserShortcutsDirectory()
+      .Append(profile_1_shortcut_name + installer::kLnkExt);
+  ValidateProfileShortcutAtPath(FROM_HERE,
+                                profile_1_shortcut_path,
+                                profile_1_path_);
+  // Check that nothing is changed for profile2.
+  ValidateProfileShortcut(FROM_HERE, profile_2_name_, profile_2_path_);
+
+  // Create new profile - profile3.
+  CreateProfileWithShortcut(FROM_HERE, profile_3_name_, profile_3_path_);
+  // Check that nothing is changed for profile1 and profile2.
+  ValidateProfileShortcutAtPath(FROM_HERE,
+                                profile_1_shortcut_path,
+                                profile_1_path_);
+  ValidateProfileShortcut(FROM_HERE, profile_2_name_, profile_2_path_);
+
+  // Give to profile3 the same name as profile2.
+  const base::string16 new_profile_3_name = profile_2_name_;
+  RenameProfile(FROM_HERE, profile_3_path_, new_profile_3_name);
+  const auto profile_3_shortcut_name = new_profile_3_name + L" - " +
+      GetDistribution()->GetShortcutName(BrowserDistribution::SHORTCUT_CHROME) +
+      L" (2)";
+  const auto profile_3_shortcut_path = GetUserShortcutsDirectory()
+      .Append(profile_3_shortcut_name + installer::kLnkExt);
+  ValidateProfileShortcutAtPath(FROM_HERE,
+                                profile_3_shortcut_path,
+                                profile_3_path_);
+  // Check that nothing is changed for profile1 and profile2.
+  ValidateProfileShortcut(FROM_HERE, profile_2_name_, profile_2_path_);
+  ValidateProfileShortcutAtPath(FROM_HERE,
+                                profile_1_shortcut_path,
+                                profile_1_path_);
+
+  // Rename profile1 again.
+  new_profile_1_name = L"Carrie";
+  RenameProfile(FROM_HERE, profile_1_path_, new_profile_1_name);
+  ValidateProfileShortcut(FROM_HERE, new_profile_1_name, profile_1_path_);
+  // Check that nothing is changed for profile2 and profile3.
+  ValidateProfileShortcut(FROM_HERE, profile_2_name_, profile_2_path_);
+  ValidateProfileShortcutAtPath(FROM_HERE,
+                                profile_3_shortcut_path,
+                                profile_3_path_);
+
+  // Delete profile1.
+  profile_info_cache_->DeleteProfileFromCache(profile_1_path_);
+  RunPendingTasks();
+  EXPECT_FALSE(ProfileShortcutExistsAtDefaultPath(new_profile_1_name));
+  // Check that nothing is changed for profile2 and profile3.
+  ValidateProfileShortcut(FROM_HERE, profile_2_name_, profile_2_path_);
+  ValidateProfileShortcutAtPath(FROM_HERE,
+                                profile_3_shortcut_path,
+                                profile_3_path_);
+
+  // Delete profile2.
+  EXPECT_TRUE(base::PathExists(
+      GetDefaultShortcutPathForProfile(profile_2_name_)));
+  EXPECT_TRUE(base::PathExists(profile_3_shortcut_path));
+  profile_info_cache_->DeleteProfileFromCache(profile_2_path_);
+  RunPendingTasks();
+  EXPECT_FALSE(base::PathExists(
+      GetDefaultShortcutPathForProfile(profile_2_name_)));
+  // Only profile3 exists. There should be non-profile shortcut only.
+  EXPECT_FALSE(base::PathExists(profile_3_shortcut_path));
+  ValidateNonProfileShortcut(FROM_HERE);
+}