Create/Delete windows profile shortcuts

chrome/browser/profiles/profile_manager_unittest.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 <string>
 
 #include "base/command_line.h"
 #include "base/file_util.h"
 #include "base/message_loop.h"
 #include "base/path_service.h"
 #include "base/scoped_temp_dir.h"
 #include "base/system_monitor/system_monitor.h"
 #include "base/utf_string_conversions.h"
 #include "base/values.h"
 #include "build/build_config.h"
 #include "chrome/browser/bookmarks/bookmark_model_factory.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/extensions/event_router_forwarder.h"
 #include "chrome/browser/history/history.h"
 #include "chrome/browser/history/history_service_factory.h"
 #include "chrome/browser/io_thread.h"
 #include "chrome/browser/prefs/browser_prefs.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/profiles/profile_info_cache.h"
 #include "chrome/browser/profiles/profile_manager.h"
+#include "chrome/browser/profiles/profile_shortcut_manager.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/common/chrome_constants.h"
 #include "chrome/common/chrome_notification_types.h"
 #include "chrome/common/chrome_paths.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/common/pref_names.h"
+#include "chrome/installer/util/browser_distribution.h"
+#include "chrome/installer/util/shell_util.h"
 #include "chrome/test/base/test_browser_window.h"
 #include "chrome/test/base/testing_browser_process.h"
 #include "chrome/test/base/testing_pref_service.h"
 #include "chrome/test/base/testing_profile.h"
 #include "content/public/browser/notification_service.h"
 #include "content/public/test/test_browser_thread.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
+#include "ui/base/resource/resource_bundle.h"
 
 #if defined(OS_CHROMEOS)
 #include "chrome/browser/chromeos/cros/cros_library.h"
 #endif
 
 using content::BrowserThread;
 
 namespace {
 
 // This global variable is used to check that value returned to different
 // observers is the same.
 Profile* g_created_profile;
 
+// Profile shortcut manager for desktop shortcuts
+ProfileShortcutManager* profile_shortcut_manager;

[sail, 2012/08/13 22:09:36]

you don't need a global for this, this can just be created inside the test

[Halli, 2012/08/14 18:25:57]

Kept as a global in the new test class profile_shortcut_manager_unittest since
all tests use it.
On 2012/08/13 22:09:36, sail wrote:

+
 }  // namespace
 
 namespace testing {
 
 class ProfileManager : public ::ProfileManagerWithoutInit {
  public:
   explicit ProfileManager(const FilePath& user_data_dir)
       : ::ProfileManagerWithoutInit(user_data_dir) {}
 
  protected:
   virtual Profile* CreateProfileHelper(const FilePath& file_path) OVERRIDE {
     if (!file_util::PathExists(file_path)) {
       if (!file_util::CreateDirectory(file_path))
         return NULL;
     }
     return new TestingProfile(file_path, NULL);
   }
 
   virtual Profile* CreateProfileAsyncHelper(const FilePath& path,
                                             Delegate* delegate) OVERRIDE {
     // This is safe while all file operations are done on the FILE thread.
     BrowserThread::PostTask(
         BrowserThread::FILE, FROM_HERE,
         base::Bind(base::IgnoreResult(&file_util::CreateDirectory), path));
 
     return new TestingProfile(path, this);
   }
-
-#if defined(OS_WIN)
-  virtual ProfileShortcutManagerWin* CreateShortcutManager() OVERRIDE {
-    // We should avoid creating shortcuts in these tests.
-    return NULL;
-  }
-#endif
 };
 
 }  // namespace testing
 
 class ProfileManagerTest : public testing::Test {
  protected:
   class MockObserver {
    public:
     MOCK_METHOD2(OnProfileCreated,
         void(Profile* profile, Profile::CreateStatus status));
@@ skipping 13 matching lines @@
     system_monitor_dummy_.reset(new base::SystemMonitor);
     static_cast<TestingBrowserProcess*>(g_browser_process)->SetIOThread(
         &io_thread_);
   }
 
   virtual void SetUp() {
     // Create a new temporary directory, and store the path
     ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
     static_cast<TestingBrowserProcess*>(g_browser_process)->SetProfileManager(
         new testing::ProfileManager(temp_dir_.path()));
+
+     // Profile shortcut manager will be NULL for non-windows platforms
+    profile_shortcut_manager = ProfileShortcutManager::Create();
+
 #if defined(OS_CHROMEOS)
-  CommandLine *cl = CommandLine::ForCurrentProcess();
+  CommandLine* cl = CommandLine::ForCurrentProcess();
   cl->AppendSwitch(switches::kTestType);
 #endif
   }
 
   virtual void TearDown() {
     static_cast<TestingBrowserProcess*>(g_browser_process)->SetProfileManager(
         NULL);
     message_loop_.RunAllPending();
   }
 
@@ skipping 148 matching lines @@
   EXPECT_CALL(mock_observer2, OnProfileCreated(
       SameNotNull(), NotFail())).Times(testing::AtLeast(1));
   MockObserver mock_observer3;
   EXPECT_CALL(mock_observer3, OnProfileCreated(
       SameNotNull(), NotFail())).Times(testing::AtLeast(1));
 
   ProfileManager* profile_manager = g_browser_process->profile_manager();
 
   profile_manager->CreateProfileAsync(dest_path,
       base::Bind(&MockObserver::OnProfileCreated,
-                 base::Unretained(&mock_observer1)), string16(), string16());
+                 base::Unretained(&mock_observer1)),
+                 string16(), string16());
   profile_manager->CreateProfileAsync(dest_path,
       base::Bind(&MockObserver::OnProfileCreated,
-                 base::Unretained(&mock_observer2)), string16(), string16());
+                 base::Unretained(&mock_observer2)),
+                 string16(), string16());
   profile_manager->CreateProfileAsync(dest_path,
       base::Bind(&MockObserver::OnProfileCreated,
-                 base::Unretained(&mock_observer3)), string16(), string16());
+                 base::Unretained(&mock_observer3)),
+                 string16(), string16());
 
   message_loop_.RunAllPending();
 }
 
 TEST_F(ProfileManagerTest, CreateProfilesAsync) {
   FilePath dest_path1 =
       temp_dir_.path().Append(FILE_PATH_LITERAL("New Profile 1"));
   FilePath dest_path2 =
       temp_dir_.path().Append(FILE_PATH_LITERAL("New Profile 2"));
 
@@ skipping 255 matching lines @@
   browser2b.reset();
   last_opened_profiles = profile_manager->GetLastOpenedProfiles();
   ASSERT_EQ(1U, last_opened_profiles.size());
   EXPECT_EQ(profile1, last_opened_profiles[0]);
 
   browser1.reset();
   last_opened_profiles = profile_manager->GetLastOpenedProfiles();
   ASSERT_EQ(0U, last_opened_profiles.size());
 }
 #endif  // !defined(OS_ANDROID)
+
+TEST_F(ProfileManagerTest, WinDesktopShortcutsIconExists) {

[sail, 2012/08/13 22:09:36]

This isn't related to profile manager so it should go inside
profile_shortcut_manager_unittest.cc

Also, most of this isn't windows specific so you don't need Win in the test
name.

The windows specific can probably be put into a #ifdef

[Halli, 2012/08/14 18:25:57]

Done.

+  if (!profile_shortcut_manager)
+    return;
+
+  FilePath dest_path = temp_dir_.path();
+  string16 profile_name = ASCIIToUTF16("My Profile");
+
+  ProfileManager* profile_manager = g_browser_process->profile_manager();
+
+  gfx::Image& avatar = ResourceBundle::GetSharedInstance().
+      GetNativeImageNamed(profile_manager->GetProfileInfoCache().
+      GetDefaultAvatarIconResourceIDAtIndex(0));
+
+  profile_shortcut_manager->CreateChromeDesktopShortcut(dest_path,
+                                                        profile_name, avatar);
+
+  ASSERT_TRUE(file_util::ContainsPath(temp_dir_.path(), dest_path.Append(

[sail, 2012/08/13 22:14:35]

maybe file_util::PathExists() instead of ContainsPath() ?

[Halli, 2012/08/14 18:25:57]

Done.

+      (FILE_PATH_LITERAL("Google Profile.ico")))));
+
+  profile_shortcut_manager->DeleteChromeDesktopShortcut(dest_path);

[sail, 2012/08/13 22:14:35]

maybe test that the file no longer exists?

[Halli, 2012/08/14 18:25:57]

TODO in separate CL

+}
+
+TEST_F(ProfileManagerTest, WinDesktopShortcutsLnk) {
+  if (!profile_shortcut_manager)
+    return;
+
+  FilePath dest_path = temp_dir_.path();
+  dest_path = dest_path.Append(FILE_PATH_LITERAL("New Profile 1"));
+
+  ProfileManager* profile_manager = g_browser_process->profile_manager();
+
+  // Successfully create the profile.
+  TestingProfile* profile =
+      static_cast<TestingProfile*>(profile_manager->GetProfile(dest_path));
+  ASSERT_TRUE(profile);
+
+  profile->SetName(ASCIIToWide("My Profile"));
+
+  gfx::Image& avatar = ResourceBundle::GetSharedInstance().
+      GetNativeImageNamed(profile_manager->GetProfileInfoCache().
+      GetDefaultAvatarIconResourceIDAtIndex(0));
+
+  profile_shortcut_manager->CreateChromeDesktopShortcut(dest_path,
+      profile->GetName(), avatar);
+
+  FilePath exe_path;
+  ASSERT_TRUE(PathService::Get(base::FILE_EXE, &exe_path));
+
+  FilePath shortcut;
+  string16 shortcut_name;
+  BrowserDistribution* dist = BrowserDistribution::GetDistribution();
+
+  // Get the desktop path of the current user
+  ShellUtil::GetDesktopPath(false, &shortcut);
+  // Get the name of the shortcut with profile attached
+  ShellUtil::GetChromeShortcutName(dist, false, profile->GetName(),
+      &shortcut_name);
+  shortcut = shortcut.Append(shortcut_name);
+
+  EXPECT_EQ(ShellUtil::VERIFY_SHORTCUT_SUCCESS,
+      ShellUtil::VerifyChromeShortcut(exe_path.value(),
+      shortcut.value(), dist->GetAppDescription(), 0));
+
+  profile_shortcut_manager->DeleteChromeDesktopShortcut(dest_path);
+}
+