15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/win/shortcut.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <string> 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 92a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/files/file_path.h" 106e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles)#include "base/files/file_util.h" 112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/files/scoped_temp_dir.h" 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/test/test_file_util.h" 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/test/test_shortcut_win.h" 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/win/scoped_com_initializer.h" 155d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)#include "base/win/windows_version.h" 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "testing/gtest/include/gtest/gtest.h" 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)namespace base { 192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)namespace win { 202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace { 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)static const char kFileContents[] = "This is a target."; 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)static const char kFileContents2[] = "This is another target."; 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class ShortcutTest : public testing::Test { 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) protected: 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void SetUp() OVERRIDE { 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_TRUE(temp_dir_2_.CreateUniqueTempDir()); 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) link_file_ = temp_dir_.path().Append(L"My Link.lnk"); 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Shortcut 1's properties 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) { 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const FilePath target_file(temp_dir_.path().Append(L"Target 1.txt")); 37a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) WriteFile(target_file, kFileContents, arraysize(kFileContents)); 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) link_properties_.set_target(target_file); 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) link_properties_.set_working_dir(temp_dir_.path()); 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) link_properties_.set_arguments(L"--magic --awesome"); 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) link_properties_.set_description(L"Chrome is awesome."); 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) link_properties_.set_icon(link_properties_.target, 4); 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) link_properties_.set_app_id(L"Chrome"); 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) link_properties_.set_dual_mode(false); 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Shortcut 2's properties (all different from properties of shortcut 1). 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) { 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const FilePath target_file_2(temp_dir_.path().Append(L"Target 2.txt")); 51a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) WriteFile(target_file_2, kFileContents2, arraysize(kFileContents2)); 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) FilePath icon_path_2; 54a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) CreateTemporaryFileInDir(temp_dir_.path(), &icon_path_2); 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) link_properties_2_.set_target(target_file_2); 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) link_properties_2_.set_working_dir(temp_dir_2_.path()); 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) link_properties_2_.set_arguments(L"--super --crazy"); 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) link_properties_2_.set_description(L"The best in the west."); 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) link_properties_2_.set_icon(icon_path_2, 0); 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) link_properties_2_.set_app_id(L"Chrome.UserLevelCrazySuffix"); 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) link_properties_2_.set_dual_mode(true); 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ScopedCOMInitializer com_initializer_; 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ScopedTempDir temp_dir_; 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ScopedTempDir temp_dir_2_; 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The link file to be created/updated in the shortcut tests below. 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) FilePath link_file_; 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Properties for the created shortcut. 742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ShortcutProperties link_properties_; 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Properties for the updated shortcut. 772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ShortcutProperties link_properties_2_; 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 825d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)TEST_F(ShortcutTest, CreateAndResolveShortcutProperties) { 835d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) uint32 valid_properties = ShortcutProperties::PROPERTIES_BASIC; 845d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) if (GetVersion() >= VERSION_WIN7) 855d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) valid_properties |= ShortcutProperties::PROPERTIES_WIN7; 865d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 875d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) // Test all properties. 885d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) FilePath file_1(temp_dir_.path().Append(L"Link1.lnk")); 895d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) ASSERT_TRUE(CreateOrUpdateShortcutLink( 905d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) file_1, link_properties_, SHORTCUT_CREATE_ALWAYS)); 915d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 925d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) ShortcutProperties properties_read_1; 935d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) ASSERT_TRUE(ResolveShortcutProperties( 945d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) file_1, ShortcutProperties::PROPERTIES_ALL, &properties_read_1)); 955d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) EXPECT_EQ(valid_properties, properties_read_1.options); 965d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) ValidatePathsAreEqual(link_properties_.target, properties_read_1.target); 975d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) ValidatePathsAreEqual(link_properties_.working_dir, 985d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) properties_read_1.working_dir); 995d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) EXPECT_EQ(link_properties_.arguments, properties_read_1.arguments); 1005d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) EXPECT_EQ(link_properties_.description, properties_read_1.description); 1015d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) ValidatePathsAreEqual(link_properties_.icon, properties_read_1.icon); 1025d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) EXPECT_EQ(link_properties_.icon_index, properties_read_1.icon_index); 1035d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) if (GetVersion() >= VERSION_WIN7) { 1045d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) EXPECT_EQ(link_properties_.app_id, properties_read_1.app_id); 1055d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) EXPECT_EQ(link_properties_.dual_mode, properties_read_1.dual_mode); 1065d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) } 1075d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 1085d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) // Test simple shortcut with no special properties set. 1095d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) FilePath file_2(temp_dir_.path().Append(L"Link2.lnk")); 1105d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) ShortcutProperties only_target_properties; 1115d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) only_target_properties.set_target(link_properties_.target); 1125d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) ASSERT_TRUE(CreateOrUpdateShortcutLink( 1135d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) file_2, only_target_properties, SHORTCUT_CREATE_ALWAYS)); 1145d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 1155d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) ShortcutProperties properties_read_2; 1165d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) ASSERT_TRUE(ResolveShortcutProperties( 1175d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) file_2, ShortcutProperties::PROPERTIES_ALL, &properties_read_2)); 1185d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) EXPECT_EQ(valid_properties, properties_read_2.options); 1195d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) ValidatePathsAreEqual(only_target_properties.target, 1205d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) properties_read_2.target); 1215d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) ValidatePathsAreEqual(FilePath(), properties_read_2.working_dir); 1225d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) EXPECT_EQ(L"", properties_read_2.arguments); 1235d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) EXPECT_EQ(L"", properties_read_2.description); 1245d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) ValidatePathsAreEqual(FilePath(), properties_read_2.icon); 1255d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) EXPECT_EQ(0, properties_read_2.icon_index); 1265d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) if (GetVersion() >= VERSION_WIN7) { 1275d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) EXPECT_EQ(L"", properties_read_2.app_id); 1285d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) EXPECT_FALSE(properties_read_2.dual_mode); 1295d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) } 1305d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)} 1315d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 1325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(ShortcutTest, CreateAndResolveShortcut) { 1332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ShortcutProperties only_target_properties; 1345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) only_target_properties.set_target(link_properties_.target); 1355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ASSERT_TRUE(CreateOrUpdateShortcutLink( 1372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) link_file_, only_target_properties, SHORTCUT_CREATE_ALWAYS)); 1385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) FilePath resolved_name; 1402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_TRUE(ResolveShortcut(link_file_, &resolved_name, NULL)); 1415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) char read_contents[arraysize(kFileContents)]; 143a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) base::ReadFile(resolved_name, read_contents, arraysize(read_contents)); 1445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_STREQ(kFileContents, read_contents); 1455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(ShortcutTest, ResolveShortcutWithArgs) { 1482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ASSERT_TRUE(CreateOrUpdateShortcutLink( 1492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) link_file_, link_properties_, SHORTCUT_CREATE_ALWAYS)); 1505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) FilePath resolved_name; 1525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) string16 args; 1532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_TRUE(ResolveShortcut(link_file_, &resolved_name, &args)); 1545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) char read_contents[arraysize(kFileContents)]; 156a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) base::ReadFile(resolved_name, read_contents, arraysize(read_contents)); 1575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_STREQ(kFileContents, read_contents); 1585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(link_properties_.arguments, args); 1595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(ShortcutTest, CreateShortcutWithOnlySomeProperties) { 1622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ShortcutProperties target_and_args_properties; 1635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) target_and_args_properties.set_target(link_properties_.target); 1645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) target_and_args_properties.set_arguments(link_properties_.arguments); 1655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ASSERT_TRUE(CreateOrUpdateShortcutLink( 1675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) link_file_, target_and_args_properties, 1682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) SHORTCUT_CREATE_ALWAYS)); 1695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ValidateShortcut(link_file_, target_and_args_properties); 1715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(ShortcutTest, CreateShortcutVerifyProperties) { 1742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ASSERT_TRUE(CreateOrUpdateShortcutLink( 1752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) link_file_, link_properties_, SHORTCUT_CREATE_ALWAYS)); 1765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ValidateShortcut(link_file_, link_properties_); 1785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(ShortcutTest, UpdateShortcutVerifyProperties) { 1812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ASSERT_TRUE(CreateOrUpdateShortcutLink( 1822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) link_file_, link_properties_, SHORTCUT_CREATE_ALWAYS)); 1835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ASSERT_TRUE(CreateOrUpdateShortcutLink( 1852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) link_file_, link_properties_2_, SHORTCUT_UPDATE_EXISTING)); 1865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ValidateShortcut(link_file_, link_properties_2_); 1885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(ShortcutTest, UpdateShortcutUpdateOnlyTargetAndResolve) { 1912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ASSERT_TRUE(CreateOrUpdateShortcutLink( 1922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) link_file_, link_properties_, SHORTCUT_CREATE_ALWAYS)); 1935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ShortcutProperties update_only_target_properties; 1955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) update_only_target_properties.set_target(link_properties_2_.target); 1965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ASSERT_TRUE(CreateOrUpdateShortcutLink( 1985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) link_file_, update_only_target_properties, 1992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) SHORTCUT_UPDATE_EXISTING)); 2005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ShortcutProperties expected_properties = link_properties_; 2025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) expected_properties.set_target(link_properties_2_.target); 2032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ValidateShortcut(link_file_, expected_properties); 2045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) FilePath resolved_name; 2062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_TRUE(ResolveShortcut(link_file_, &resolved_name, NULL)); 2075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) char read_contents[arraysize(kFileContents2)]; 209a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) base::ReadFile(resolved_name, read_contents, arraysize(read_contents)); 2105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_STREQ(kFileContents2, read_contents); 2115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(ShortcutTest, UpdateShortcutMakeDualMode) { 2142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ASSERT_TRUE(CreateOrUpdateShortcutLink( 2152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) link_file_, link_properties_, SHORTCUT_CREATE_ALWAYS)); 2165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ShortcutProperties make_dual_mode_properties; 2185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) make_dual_mode_properties.set_dual_mode(true); 2195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ASSERT_TRUE(CreateOrUpdateShortcutLink( 2215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) link_file_, make_dual_mode_properties, 2222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) SHORTCUT_UPDATE_EXISTING)); 2235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ShortcutProperties expected_properties = link_properties_; 2255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) expected_properties.set_dual_mode(true); 2262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ValidateShortcut(link_file_, expected_properties); 2275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(ShortcutTest, UpdateShortcutRemoveDualMode) { 2302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ASSERT_TRUE(CreateOrUpdateShortcutLink( 2312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) link_file_, link_properties_2_, SHORTCUT_CREATE_ALWAYS)); 2325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ShortcutProperties remove_dual_mode_properties; 2345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) remove_dual_mode_properties.set_dual_mode(false); 2355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ASSERT_TRUE(CreateOrUpdateShortcutLink( 2375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) link_file_, remove_dual_mode_properties, 2382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) SHORTCUT_UPDATE_EXISTING)); 2395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ShortcutProperties expected_properties = link_properties_2_; 2415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) expected_properties.set_dual_mode(false); 2422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ValidateShortcut(link_file_, expected_properties); 2435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(ShortcutTest, UpdateShortcutClearArguments) { 2462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ASSERT_TRUE(CreateOrUpdateShortcutLink( 2472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) link_file_, link_properties_, SHORTCUT_CREATE_ALWAYS)); 2485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ShortcutProperties clear_arguments_properties; 2505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) clear_arguments_properties.set_arguments(string16()); 2515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ASSERT_TRUE(CreateOrUpdateShortcutLink( 2535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) link_file_, clear_arguments_properties, 2542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) SHORTCUT_UPDATE_EXISTING)); 2555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ShortcutProperties expected_properties = link_properties_; 2575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) expected_properties.set_arguments(string16()); 2582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ValidateShortcut(link_file_, expected_properties); 2595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(ShortcutTest, FailUpdateShortcutThatDoesNotExist) { 2622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ASSERT_FALSE(CreateOrUpdateShortcutLink( 2632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) link_file_, link_properties_, SHORTCUT_UPDATE_EXISTING)); 2647dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch ASSERT_FALSE(PathExists(link_file_)); 2655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)TEST_F(ShortcutTest, ReplaceShortcutAllProperties) { 2682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ASSERT_TRUE(CreateOrUpdateShortcutLink( 2692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) link_file_, link_properties_, SHORTCUT_CREATE_ALWAYS)); 2705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ASSERT_TRUE(CreateOrUpdateShortcutLink( 2722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) link_file_, link_properties_2_, SHORTCUT_REPLACE_EXISTING)); 2735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ValidateShortcut(link_file_, link_properties_2_); 2755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)TEST_F(ShortcutTest, ReplaceShortcutSomeProperties) { 2782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ASSERT_TRUE(CreateOrUpdateShortcutLink( 2792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) link_file_, link_properties_, SHORTCUT_CREATE_ALWAYS)); 2805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ShortcutProperties new_properties; 2825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) new_properties.set_target(link_properties_2_.target); 2832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) new_properties.set_arguments(link_properties_2_.arguments); 2845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) new_properties.set_description(link_properties_2_.description); 2852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ASSERT_TRUE(CreateOrUpdateShortcutLink( 2862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) link_file_, new_properties, SHORTCUT_REPLACE_EXISTING)); 2875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Expect only properties in |new_properties| to be set, all other properties 2895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // should have been overwritten. 2902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ShortcutProperties expected_properties(new_properties); 2915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) expected_properties.set_working_dir(FilePath()); 2925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) expected_properties.set_icon(FilePath(), 0); 2935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) expected_properties.set_app_id(string16()); 2945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) expected_properties.set_dual_mode(false); 2952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ValidateShortcut(link_file_, expected_properties); 2965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)TEST_F(ShortcutTest, FailReplaceShortcutThatDoesNotExist) { 2992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ASSERT_FALSE(CreateOrUpdateShortcutLink( 3002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) link_file_, link_properties_, SHORTCUT_REPLACE_EXISTING)); 3017dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch ASSERT_FALSE(PathExists(link_file_)); 3025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Test that the old arguments remain on the replaced shortcut when not 3052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// otherwise specified. 3062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)TEST_F(ShortcutTest, ReplaceShortcutKeepOldArguments) { 3072a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ASSERT_TRUE(CreateOrUpdateShortcutLink( 3082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) link_file_, link_properties_, SHORTCUT_CREATE_ALWAYS)); 3092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Do not explicitly set the arguments. 3112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) link_properties_2_.options &= 3122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ~ShortcutProperties::PROPERTIES_ARGUMENTS; 3132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ASSERT_TRUE(CreateOrUpdateShortcutLink( 3142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) link_file_, link_properties_2_, SHORTCUT_REPLACE_EXISTING)); 3152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ShortcutProperties expected_properties(link_properties_2_); 3172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) expected_properties.set_arguments(link_properties_.arguments); 3182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ValidateShortcut(link_file_, expected_properties); 3192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 3202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} // namespace win 3222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} // namespace base 323