OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "base/win/shortcut.h" | 5 #include "base/win/shortcut.h" |
6 | 6 |
7 #include <stdint.h> | 7 #include <stdint.h> |
8 | 8 |
9 #include <string> | 9 #include <string> |
10 | 10 |
11 #include "base/files/file_path.h" | 11 #include "base/files/file_path.h" |
12 #include "base/files/file_util.h" | 12 #include "base/files/file_util.h" |
13 #include "base/files/scoped_temp_dir.h" | 13 #include "base/files/scoped_temp_dir.h" |
14 #include "base/macros.h" | 14 #include "base/macros.h" |
15 #include "base/test/test_file_util.h" | 15 #include "base/test/test_file_util.h" |
16 #include "base/test/test_shortcut_win.h" | 16 #include "base/test/test_shortcut_win.h" |
17 #include "base/win/scoped_com_initializer.h" | 17 #include "base/win/scoped_com_initializer.h" |
18 #include "base/win/windows_version.h" | |
19 #include "testing/gtest/include/gtest/gtest.h" | 18 #include "testing/gtest/include/gtest/gtest.h" |
20 | 19 |
21 namespace base { | 20 namespace base { |
22 namespace win { | 21 namespace win { |
23 | 22 |
24 namespace { | 23 namespace { |
25 | 24 |
26 static const char kFileContents[] = "This is a target."; | 25 static const char kFileContents[] = "This is a target."; |
27 static const char kFileContents2[] = "This is another target."; | 26 static const char kFileContents2[] = "This is another target."; |
28 | 27 |
(...skipping 48 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
77 ShortcutProperties link_properties_; | 76 ShortcutProperties link_properties_; |
78 | 77 |
79 // Properties for the updated shortcut. | 78 // Properties for the updated shortcut. |
80 ShortcutProperties link_properties_2_; | 79 ShortcutProperties link_properties_2_; |
81 }; | 80 }; |
82 | 81 |
83 } // namespace | 82 } // namespace |
84 | 83 |
85 TEST_F(ShortcutTest, CreateAndResolveShortcutProperties) { | 84 TEST_F(ShortcutTest, CreateAndResolveShortcutProperties) { |
86 uint32_t valid_properties = ShortcutProperties::PROPERTIES_BASIC; | 85 uint32_t valid_properties = ShortcutProperties::PROPERTIES_BASIC; |
87 if (GetVersion() >= VERSION_WIN7) | 86 valid_properties |= ShortcutProperties::PROPERTIES_WIN7; |
88 valid_properties |= ShortcutProperties::PROPERTIES_WIN7; | |
89 | 87 |
90 // Test all properties. | 88 // Test all properties. |
91 FilePath file_1(temp_dir_.GetPath().Append(L"Link1.lnk")); | 89 FilePath file_1(temp_dir_.GetPath().Append(L"Link1.lnk")); |
92 ASSERT_TRUE(CreateOrUpdateShortcutLink( | 90 ASSERT_TRUE(CreateOrUpdateShortcutLink( |
93 file_1, link_properties_, SHORTCUT_CREATE_ALWAYS)); | 91 file_1, link_properties_, SHORTCUT_CREATE_ALWAYS)); |
94 | 92 |
95 ShortcutProperties properties_read_1; | 93 ShortcutProperties properties_read_1; |
96 ASSERT_TRUE(ResolveShortcutProperties( | 94 ASSERT_TRUE(ResolveShortcutProperties( |
97 file_1, ShortcutProperties::PROPERTIES_ALL, &properties_read_1)); | 95 file_1, ShortcutProperties::PROPERTIES_ALL, &properties_read_1)); |
98 EXPECT_EQ(valid_properties, properties_read_1.options); | 96 EXPECT_EQ(valid_properties, properties_read_1.options); |
99 ValidatePathsAreEqual(link_properties_.target, properties_read_1.target); | 97 ValidatePathsAreEqual(link_properties_.target, properties_read_1.target); |
100 ValidatePathsAreEqual(link_properties_.working_dir, | 98 ValidatePathsAreEqual(link_properties_.working_dir, |
101 properties_read_1.working_dir); | 99 properties_read_1.working_dir); |
102 EXPECT_EQ(link_properties_.arguments, properties_read_1.arguments); | 100 EXPECT_EQ(link_properties_.arguments, properties_read_1.arguments); |
103 EXPECT_EQ(link_properties_.description, properties_read_1.description); | 101 EXPECT_EQ(link_properties_.description, properties_read_1.description); |
104 ValidatePathsAreEqual(link_properties_.icon, properties_read_1.icon); | 102 ValidatePathsAreEqual(link_properties_.icon, properties_read_1.icon); |
105 EXPECT_EQ(link_properties_.icon_index, properties_read_1.icon_index); | 103 EXPECT_EQ(link_properties_.icon_index, properties_read_1.icon_index); |
106 if (GetVersion() >= VERSION_WIN7) { | 104 EXPECT_EQ(link_properties_.app_id, properties_read_1.app_id); |
107 EXPECT_EQ(link_properties_.app_id, properties_read_1.app_id); | 105 EXPECT_EQ(link_properties_.dual_mode, properties_read_1.dual_mode); |
108 EXPECT_EQ(link_properties_.dual_mode, properties_read_1.dual_mode); | |
109 } | |
110 | 106 |
111 // Test simple shortcut with no special properties set. | 107 // Test simple shortcut with no special properties set. |
112 FilePath file_2(temp_dir_.GetPath().Append(L"Link2.lnk")); | 108 FilePath file_2(temp_dir_.GetPath().Append(L"Link2.lnk")); |
113 ShortcutProperties only_target_properties; | 109 ShortcutProperties only_target_properties; |
114 only_target_properties.set_target(link_properties_.target); | 110 only_target_properties.set_target(link_properties_.target); |
115 ASSERT_TRUE(CreateOrUpdateShortcutLink( | 111 ASSERT_TRUE(CreateOrUpdateShortcutLink( |
116 file_2, only_target_properties, SHORTCUT_CREATE_ALWAYS)); | 112 file_2, only_target_properties, SHORTCUT_CREATE_ALWAYS)); |
117 | 113 |
118 ShortcutProperties properties_read_2; | 114 ShortcutProperties properties_read_2; |
119 ASSERT_TRUE(ResolveShortcutProperties( | 115 ASSERT_TRUE(ResolveShortcutProperties( |
120 file_2, ShortcutProperties::PROPERTIES_ALL, &properties_read_2)); | 116 file_2, ShortcutProperties::PROPERTIES_ALL, &properties_read_2)); |
121 EXPECT_EQ(valid_properties, properties_read_2.options); | 117 EXPECT_EQ(valid_properties, properties_read_2.options); |
122 ValidatePathsAreEqual(only_target_properties.target, | 118 ValidatePathsAreEqual(only_target_properties.target, |
123 properties_read_2.target); | 119 properties_read_2.target); |
124 ValidatePathsAreEqual(FilePath(), properties_read_2.working_dir); | 120 ValidatePathsAreEqual(FilePath(), properties_read_2.working_dir); |
125 EXPECT_EQ(L"", properties_read_2.arguments); | 121 EXPECT_EQ(L"", properties_read_2.arguments); |
126 EXPECT_EQ(L"", properties_read_2.description); | 122 EXPECT_EQ(L"", properties_read_2.description); |
127 ValidatePathsAreEqual(FilePath(), properties_read_2.icon); | 123 ValidatePathsAreEqual(FilePath(), properties_read_2.icon); |
128 EXPECT_EQ(0, properties_read_2.icon_index); | 124 EXPECT_EQ(0, properties_read_2.icon_index); |
129 if (GetVersion() >= VERSION_WIN7) { | 125 EXPECT_EQ(L"", properties_read_2.app_id); |
130 EXPECT_EQ(L"", properties_read_2.app_id); | 126 EXPECT_FALSE(properties_read_2.dual_mode); |
131 EXPECT_FALSE(properties_read_2.dual_mode); | |
132 } | |
133 } | 127 } |
134 | 128 |
135 TEST_F(ShortcutTest, CreateAndResolveShortcut) { | 129 TEST_F(ShortcutTest, CreateAndResolveShortcut) { |
136 ShortcutProperties only_target_properties; | 130 ShortcutProperties only_target_properties; |
137 only_target_properties.set_target(link_properties_.target); | 131 only_target_properties.set_target(link_properties_.target); |
138 | 132 |
139 ASSERT_TRUE(CreateOrUpdateShortcutLink( | 133 ASSERT_TRUE(CreateOrUpdateShortcutLink( |
140 link_file_, only_target_properties, SHORTCUT_CREATE_ALWAYS)); | 134 link_file_, only_target_properties, SHORTCUT_CREATE_ALWAYS)); |
141 | 135 |
142 FilePath resolved_name; | 136 FilePath resolved_name; |
(...skipping 173 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
316 ASSERT_TRUE(CreateOrUpdateShortcutLink( | 310 ASSERT_TRUE(CreateOrUpdateShortcutLink( |
317 link_file_, link_properties_2_, SHORTCUT_REPLACE_EXISTING)); | 311 link_file_, link_properties_2_, SHORTCUT_REPLACE_EXISTING)); |
318 | 312 |
319 ShortcutProperties expected_properties(link_properties_2_); | 313 ShortcutProperties expected_properties(link_properties_2_); |
320 expected_properties.set_arguments(link_properties_.arguments); | 314 expected_properties.set_arguments(link_properties_.arguments); |
321 ValidateShortcut(link_file_, expected_properties); | 315 ValidateShortcut(link_file_, expected_properties); |
322 } | 316 } |
323 | 317 |
324 } // namespace win | 318 } // namespace win |
325 } // namespace base | 319 } // namespace base |
OLD | NEW |