Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(417)

Side by Side Diff: ui/base/dragdrop/os_exchange_data_win_unittest.cc

Issue 380553002: Add a unit test that filenames aren't unintentionally converted to URLs. (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src
Patch Set: Rebased Created 6 years, 5 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch | Annotate | Revision Log
« no previous file with comments | « ui/base/dragdrop/os_exchange_data_unittest.cc ('k') | ui/ui_unittests.gyp » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
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/memory/ref_counted.h" 5 #include "base/memory/ref_counted.h"
6 #include "base/memory/scoped_ptr.h" 6 #include "base/memory/scoped_ptr.h"
7 #include "base/strings/utf_string_conversions.h" 7 #include "base/strings/utf_string_conversions.h"
8 #include "base/win/scoped_hglobal.h" 8 #include "base/win/scoped_hglobal.h"
9 #include "testing/gtest/include/gtest/gtest.h" 9 #include "testing/gtest/include/gtest/gtest.h"
10 #include "ui/base/clipboard/clipboard.h" 10 #include "ui/base/clipboard/clipboard.h"
(...skipping 11 matching lines...) Expand all
22 base::win::ScopedComPtr<IDataObject> com_data( 22 base::win::ScopedComPtr<IDataObject> com_data(
23 OSExchangeDataProviderWin::GetIDataObject(data)); 23 OSExchangeDataProviderWin::GetIDataObject(data));
24 24
25 FORMATETC format_etc = 25 FORMATETC format_etc =
26 { CF_UNICODETEXT, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL }; 26 { CF_UNICODETEXT, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL };
27 EXPECT_EQ(S_OK, com_data->QueryGetData(&format_etc)); 27 EXPECT_EQ(S_OK, com_data->QueryGetData(&format_etc));
28 28
29 STGMEDIUM medium; 29 STGMEDIUM medium;
30 EXPECT_EQ(S_OK, com_data->GetData(&format_etc, &medium)); 30 EXPECT_EQ(S_OK, com_data->GetData(&format_etc, &medium));
31 std::wstring output = 31 std::wstring output =
32 base::win::ScopedHGlobal<wchar_t>(medium.hGlobal).get(); 32 base::win::ScopedHGlobal<wchar_t*>(medium.hGlobal).get();
33 EXPECT_EQ(input, output); 33 EXPECT_EQ(input, output);
34 ReleaseStgMedium(&medium); 34 ReleaseStgMedium(&medium);
35 } 35 }
36 36
37 // Test setting using the IDataObject COM API 37 // Test setting using the IDataObject COM API
38 TEST(OSExchangeDataWinTest, StringDataWritingViaCOM) { 38 TEST(OSExchangeDataWinTest, StringDataWritingViaCOM) {
39 OSExchangeData data; 39 OSExchangeData data;
40 std::wstring input = L"http://www.google.com/"; 40 std::wstring input = L"http://www.google.com/";
41 41
42 base::win::ScopedComPtr<IDataObject> com_data( 42 base::win::ScopedComPtr<IDataObject> com_data(
43 OSExchangeDataProviderWin::GetIDataObject(data)); 43 OSExchangeDataProviderWin::GetIDataObject(data));
44 44
45 // Store data in the object using the COM SetData API. 45 // Store data in the object using the COM SetData API.
46 CLIPFORMAT cfstr_ineturl = RegisterClipboardFormat(CFSTR_INETURL); 46 CLIPFORMAT cfstr_ineturl = RegisterClipboardFormat(CFSTR_INETURL);
47 FORMATETC format_etc = 47 FORMATETC format_etc =
48 { cfstr_ineturl, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL }; 48 { cfstr_ineturl, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL };
49 STGMEDIUM medium; 49 STGMEDIUM medium;
50 medium.tymed = TYMED_HGLOBAL; 50 medium.tymed = TYMED_HGLOBAL;
51 HGLOBAL glob = GlobalAlloc(GPTR, sizeof(wchar_t) * (input.size() + 1)); 51 HGLOBAL glob = GlobalAlloc(GPTR, sizeof(wchar_t) * (input.size() + 1));
52 size_t stringsz = input.size(); 52 size_t stringsz = input.size();
53 SIZE_T sz = GlobalSize(glob); 53 SIZE_T sz = GlobalSize(glob);
54 base::win::ScopedHGlobal<wchar_t> global_lock(glob); 54 base::win::ScopedHGlobal<wchar_t*> global_lock(glob);
55 wchar_t* buffer_handle = global_lock.get(); 55 wchar_t* buffer_handle = global_lock.get();
56 wcscpy_s(buffer_handle, input.size() + 1, input.c_str()); 56 wcscpy_s(buffer_handle, input.size() + 1, input.c_str());
57 medium.hGlobal = glob; 57 medium.hGlobal = glob;
58 medium.pUnkForRelease = NULL; 58 medium.pUnkForRelease = NULL;
59 EXPECT_EQ(S_OK, com_data->SetData(&format_etc, &medium, TRUE)); 59 EXPECT_EQ(S_OK, com_data->SetData(&format_etc, &medium, TRUE));
60 60
61 // Construct a new object with the old object so that we can use our access 61 // Construct a new object with the old object so that we can use our access
62 // APIs. 62 // APIs.
63 OSExchangeData data2(data.provider().Clone()); 63 OSExchangeData data2(data.provider().Clone());
64 EXPECT_TRUE(data2.HasURL(OSExchangeData::CONVERT_FILENAMES)); 64 EXPECT_TRUE(data2.HasURL(OSExchangeData::CONVERT_FILENAMES));
65 GURL url_from_data; 65 GURL url_from_data;
66 std::wstring title; 66 std::wstring title;
67 EXPECT_TRUE(data2.GetURLAndTitle( 67 EXPECT_TRUE(data2.GetURLAndTitle(
68 OSExchangeData::CONVERT_FILENAMES, &url_from_data, &title)); 68 OSExchangeData::CONVERT_FILENAMES, &url_from_data, &title));
69 GURL reference_url(input); 69 GURL reference_url(in put);
70 EXPECT_EQ(reference_url.spec(), url_from_data.spec()); 70 EXPECT_EQ(reference_url.spec(), url_from_data.spec());
71 } 71 }
72 72
73 // Verifies SetData invoked twice with the same data clobbers existing data. 73 // Verifies SetData invoked twice with the same data clobbers existing data.
74 TEST(OSExchangeDataWinTest, RemoveData) { 74 TEST(OSExchangeDataWinTest, RemoveData) {
75 OSExchangeData data; 75 OSExchangeData data;
76 std::wstring input = L"http://www.google.com/"; 76 std::wstring input = L"http://www.google.com/";
77 std::wstring input2 = L"http://www.google2.com/"; 77 std::wstring input2 = L"http://www.google2.com/";
78 78
79 base::win::ScopedComPtr<IDataObject> com_data( 79 base::win::ScopedComPtr<IDataObject> com_data(
80 OSExchangeDataProviderWin::GetIDataObject(data)); 80 OSExchangeDataProviderWin::GetIDataObject(data));
81 81
82 // Store data in the object using the COM SetData API. 82 // Store data in the object using the COM SetData API.
83 CLIPFORMAT cfstr_ineturl = RegisterClipboardFormat(CFSTR_INETURL); 83 CLIPFORMAT cfstr_ineturl = RegisterClipboardFormat(CFSTR_INETURL);
84 FORMATETC format_etc = 84 FORMATETC format_etc =
85 { cfstr_ineturl, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL }; 85 { cfstr_ineturl, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL };
86 STGMEDIUM medium; 86 STGMEDIUM medium;
87 medium.tymed = TYMED_HGLOBAL; 87 medium.tymed = TYMED_HGLOBAL;
88 { 88 {
89 HGLOBAL glob = GlobalAlloc(GPTR, sizeof(wchar_t) * (input.size() + 1)); 89 HGLOBAL glob = GlobalAlloc(GPTR, sizeof(wchar_t) * (input.size() + 1));
90 size_t stringsz = input.size(); 90 size_t stringsz = input.size();
91 SIZE_T sz = GlobalSize(glob); 91 SIZE_T sz = GlobalSize(glob);
92 base::win::ScopedHGlobal<wchar_t> global_lock(glob); 92 base::win::ScopedHGlobal<wchar_t*> global_lock(glob);
93 wchar_t* buffer_handle = global_lock.get(); 93 wchar_t* buffer_handle = global_lock.get();
94 wcscpy_s(buffer_handle, input.size() + 1, input.c_str()); 94 wcscpy_s(buffer_handle, input.size() + 1, input.c_str());
95 medium.hGlobal = glob; 95 medium.hGlobal = glob;
96 medium.pUnkForRelease = NULL; 96 medium.pUnkForRelease = NULL;
97 EXPECT_EQ(S_OK, com_data->SetData(&format_etc, &medium, TRUE)); 97 EXPECT_EQ(S_OK, com_data->SetData(&format_etc, &medium, TRUE));
98 } 98 }
99 // This should clobber the existing data. 99 // This should clobber the existing data.
100 { 100 {
101 HGLOBAL glob = GlobalAlloc(GPTR, sizeof(wchar_t) * (input2.size() + 1)); 101 HGLOBAL glob = GlobalAlloc(GPTR, sizeof(wchar_t) * (input2.size() + 1));
102 size_t stringsz = input2.size(); 102 size_t stringsz = input2.size();
103 SIZE_T sz = GlobalSize(glob); 103 SIZE_T sz = GlobalSize(glob);
104 base::win::ScopedHGlobal<wchar_t> global_lock(glob); 104 base::win::ScopedHGlobal<wchar_t*> global_lock(glob);
105 wchar_t* buffer_handle = global_lock.get(); 105 wchar_t* buffer_handle = global_lock.get();
106 wcscpy_s(buffer_handle, input2.size() + 1, input2.c_str()); 106 wcscpy_s(buffer_handle, input2.size() + 1, input2.c_str());
107 medium.hGlobal = glob; 107 medium.hGlobal = glob;
108 medium.pUnkForRelease = NULL; 108 medium.pUnkForRelease = NULL;
109 EXPECT_EQ(S_OK, com_data->SetData(&format_etc, &medium, TRUE)); 109 EXPECT_EQ(S_OK, com_data->SetData(&format_etc, &medium, TRUE));
110 } 110 }
111 EXPECT_EQ(1u, static_cast<DataObjectImpl*>(com_data.get())->size()); 111 EXPECT_EQ(1u, static_cast<DataObjectImpl*>(com_data.get())->size());
112 112
113 // Construct a new object with the old object so that we can use our access 113 // Construct a new object with the old object so that we can use our access
114 // APIs. 114 // APIs.
(...skipping 14 matching lines...) Expand all
129 OSExchangeDataProviderWin::GetIDataObject(data)); 129 OSExchangeDataProviderWin::GetIDataObject(data));
130 130
131 CLIPFORMAT cfstr_ineturl = RegisterClipboardFormat(CFSTR_INETURL); 131 CLIPFORMAT cfstr_ineturl = RegisterClipboardFormat(CFSTR_INETURL);
132 FORMATETC format_etc = 132 FORMATETC format_etc =
133 { cfstr_ineturl, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL }; 133 { cfstr_ineturl, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL };
134 EXPECT_EQ(S_OK, com_data->QueryGetData(&format_etc)); 134 EXPECT_EQ(S_OK, com_data->QueryGetData(&format_etc));
135 135
136 STGMEDIUM medium; 136 STGMEDIUM medium;
137 EXPECT_EQ(S_OK, com_data->GetData(&format_etc, &medium)); 137 EXPECT_EQ(S_OK, com_data->GetData(&format_etc, &medium));
138 std::wstring output = 138 std::wstring output =
139 base::win::ScopedHGlobal<wchar_t>(medium.hGlobal).get(); 139 base::win::ScopedHGlobal<wchar_t*>(medium.hGlobal).get();
140 EXPECT_EQ(url.spec(), base::WideToUTF8(output)); 140 EXPECT_EQ(url.spec(), base::WideToUTF8(output));
141 ReleaseStgMedium(&medium); 141 ReleaseStgMedium(&medium);
142 } 142 }
143 143
144 TEST(OSExchangeDataWinTest, MultipleFormatsViaCOM) { 144 TEST(OSExchangeDataWinTest, MultipleFormatsViaCOM) {
145 OSExchangeData data; 145 OSExchangeData data;
146 std::string url_spec = "http://www.google.com/"; 146 std::string url_spec = "http://www.google.com/";
147 GURL url(url_spec); 147 GURL url(url_spec);
148 std::wstring text = L"O hai googlz."; 148 std::wstring text = L"O hai googlz.";
149 data.SetURL(url, L"Google"); 149 data.SetURL(url, L"Google");
150 data.SetString(text); 150 data.SetString(text);
151 151
152 base::win::ScopedComPtr<IDataObject> com_data( 152 base::win::ScopedComPtr<IDataObject> com_data(
153 OSExchangeDataProviderWin::GetIDataObject(data)); 153 OSExchangeDataProviderWin::GetIDataObject(data));
154 154
155 CLIPFORMAT cfstr_ineturl = RegisterClipboardFormat(CFSTR_INETURL); 155 CLIPFORMAT cfstr_ineturl = RegisterClipboardFormat(CFSTR_INETURL);
156 FORMATETC url_format_etc = 156 FORMATETC url_format_etc =
157 { cfstr_ineturl, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL }; 157 { cfstr_ineturl, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL };
158 EXPECT_EQ(S_OK, com_data->QueryGetData(&url_format_etc)); 158 EXPECT_EQ(S_OK, com_data->QueryGetData(&url_format_etc));
159 FORMATETC text_format_etc = 159 FORMATETC text_format_etc =
160 { CF_UNICODETEXT, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL }; 160 { CF_UNICODETEXT, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL };
161 EXPECT_EQ(S_OK, com_data->QueryGetData(&text_format_etc)); 161 EXPECT_EQ(S_OK, com_data->QueryGetData(&text_format_etc));
162 162
163 STGMEDIUM medium; 163 STGMEDIUM medium;
164 EXPECT_EQ(S_OK, com_data->GetData(&url_format_etc, &medium)); 164 EXPECT_EQ(S_OK, com_data->GetData(&url_format_etc, &medium));
165 std::wstring output_url = 165 std::wstring output_url =
166 base::win::ScopedHGlobal<wchar_t>(medium.hGlobal).get(); 166 base::win::ScopedHGlobal<wchar_t*>(medium.hGlobal).get();
167 EXPECT_EQ(url.spec(), base::WideToUTF8(output_url)); 167 EXPECT_EQ(url.spec(), base::WideToUTF8(output_url));
168 ReleaseStgMedium(&medium); 168 ReleaseStgMedium(&medium);
169 169
170 // The text is supposed to be the raw text of the URL, _NOT_ the value of 170 // The text is supposed to be the raw text of the URL, _NOT_ the value of
171 // |text|! This is because the URL is added first and thus takes precedence! 171 // |text|! This is because the URL is added first and thus takes precedence!
172 EXPECT_EQ(S_OK, com_data->GetData(&text_format_etc, &medium)); 172 EXPECT_EQ(S_OK, com_data->GetData(&text_format_etc, &medium));
173 std::wstring output_text = 173 std::wstring output_text =
174 base::win::ScopedHGlobal<wchar_t>(medium.hGlobal).get(); 174 base::win::ScopedHGlobal<wchar_t*>(medium.hGlobal).get();
175 EXPECT_EQ(url_spec, base::WideToUTF8(output_text)); 175 EXPECT_EQ(url_spec, base::WideToUTF8(output_text));
176 ReleaseStgMedium(&medium); 176 ReleaseStgMedium(&medium);
177 } 177 }
178 178
179 TEST(OSExchangeDataWinTest, EnumerationViaCOM) { 179 TEST(OSExchangeDataWinTest, EnumerationViaCOM) {
180 OSExchangeData data; 180 OSExchangeData data;
181 data.SetURL(GURL("http://www.google.com/"), L""); 181 data.SetURL(GURL("http://www.google.com/"), L"");
182 data.SetString(L"O hai googlz."); 182 data.SetString(L"O hai googlz.");
183 183
184 CLIPFORMAT cfstr_file_group_descriptor = 184 CLIPFORMAT cfstr_file_group_descriptor =
(...skipping 92 matching lines...) Expand 10 before | Expand all | Expand 10 after
277 OSExchangeDataProviderWin::GetIDataObject(data)); 277 OSExchangeDataProviderWin::GetIDataObject(data));
278 { 278 {
279 CLIPFORMAT cfstr_file_contents = 279 CLIPFORMAT cfstr_file_contents =
280 RegisterClipboardFormat(CFSTR_FILECONTENTS); 280 RegisterClipboardFormat(CFSTR_FILECONTENTS);
281 FORMATETC format_etc = 281 FORMATETC format_etc =
282 { cfstr_file_contents, NULL, DVASPECT_CONTENT, 0, TYMED_HGLOBAL }; 282 { cfstr_file_contents, NULL, DVASPECT_CONTENT, 0, TYMED_HGLOBAL };
283 EXPECT_EQ(S_OK, com_data->QueryGetData(&format_etc)); 283 EXPECT_EQ(S_OK, com_data->QueryGetData(&format_etc));
284 284
285 STGMEDIUM medium; 285 STGMEDIUM medium;
286 EXPECT_EQ(S_OK, com_data->GetData(&format_etc, &medium)); 286 EXPECT_EQ(S_OK, com_data->GetData(&format_etc, &medium));
287 base::win::ScopedHGlobal<char> glob(medium.hGlobal); 287 base::win::ScopedHGlobal<char*> glob(medium.hGlobal);
288 std::string output(glob.get(), glob.Size()); 288 std::string output(glob.get(), glob.Size());
289 std::string file_contents = "[InternetShortcut]\r\nURL="; 289 std::string file_contents = "[InternetShortcut]\r\nURL=";
290 file_contents += url_spec; 290 file_contents += url_spec;
291 file_contents += "\r\n"; 291 file_contents += "\r\n";
292 EXPECT_EQ(file_contents, output); 292 EXPECT_EQ(file_contents, output);
293 ReleaseStgMedium(&medium); 293 ReleaseStgMedium(&medium);
294 } 294 }
295 } 295 }
296 296
297 TEST(OSExchangeDataWinTest, FileContents) { 297 TEST(OSExchangeDataWinTest, FileContents) {
(...skipping 24 matching lines...) Expand all
322 "StartFragment:0000000175\r\nEndFragment:0000000252\r\n" 322 "StartFragment:0000000175\r\nEndFragment:0000000252\r\n"
323 "SourceURL:http://www.google.com/\r\n<html>\r\n<body>\r\n" 323 "SourceURL:http://www.google.com/\r\n<html>\r\n<body>\r\n"
324 "<!--StartFragment-->"); 324 "<!--StartFragment-->");
325 expected_cf_html += base::WideToUTF8(html); 325 expected_cf_html += base::WideToUTF8(html);
326 expected_cf_html.append("<!--EndFragment-->\r\n</body>\r\n</html>"); 326 expected_cf_html.append("<!--EndFragment-->\r\n</body>\r\n</html>");
327 327
328 FORMATETC format = Clipboard::GetHtmlFormatType().ToFormatEtc(); 328 FORMATETC format = Clipboard::GetHtmlFormatType().ToFormatEtc();
329 STGMEDIUM medium; 329 STGMEDIUM medium;
330 IDataObject* data_object = OSExchangeDataProviderWin::GetIDataObject(data); 330 IDataObject* data_object = OSExchangeDataProviderWin::GetIDataObject(data);
331 EXPECT_EQ(S_OK, data_object->GetData(&format, &medium)); 331 EXPECT_EQ(S_OK, data_object->GetData(&format, &medium));
332 base::win::ScopedHGlobal<char> glob(medium.hGlobal); 332 base::win::ScopedHGlobal<char*> glob(medium.hGlobal);
333 std::string output(glob.get(), glob.Size()); 333 std::string output(glob.get(), glob.Size());
334 EXPECT_EQ(expected_cf_html, output); 334 EXPECT_EQ(expected_cf_html, output);
335 ReleaseStgMedium(&medium); 335 ReleaseStgMedium(&medium);
336 } 336 }
337 337
338 TEST(OSExchangeDataWinTest, SetURLWithMaxPath) { 338 TEST(OSExchangeDataWinTest, SetURLWithMaxPath) {
339 OSExchangeData data; 339 OSExchangeData data;
340 std::wstring long_title(L'a', MAX_PATH + 1); 340 std::wstring long_title(L'a', MAX_PATH + 1);
341 data.SetURL(GURL("http://google.com"), long_title); 341 data.SetURL(GURL("http://google.com"), long_title);
342 } 342 }
343 343
344 TEST(OSExchangeDataWinTest, ProvideURLForPlainTextURL) { 344 TEST(OSExchangeDataWinTest, ProvideURLForPlainTextURL) {
345 OSExchangeData data; 345 OSExchangeData data;
346 data.SetString(L"http://google.com"); 346 data.SetString(L"http://google.com");
347 347
348 OSExchangeData data2(data.provider().Clone()); 348 OSExchangeData data2(data.provider().Clone());
349 ASSERT_TRUE(data2.HasURL(OSExchangeData::CONVERT_FILENAMES)); 349 ASSERT_TRUE(data2.HasURL(OSExchangeData::CONVERT_FILENAMES));
350 GURL read_url; 350 GURL read_url;
351 std::wstring title; 351 std::wstring title;
352 EXPECT_TRUE(data2.GetURLAndTitle( 352 EXPECT_TRUE(data2.GetURLAndTitle(
353 OSExchangeData::CONVERT_FILENAMES, &read_url, &title)); 353 OSExchangeData::CONVERT_FILENAMES, &read_url, &title));
354 EXPECT_EQ(GURL("http://google.com"), read_url); 354 EXPECT_EQ(GURL("http://google.com"), read_url);
355 } 355 }
356 356
357 } // namespace ui 357 } // namespace ui
OLDNEW
« no previous file with comments | « ui/base/dragdrop/os_exchange_data_unittest.cc ('k') | ui/ui_unittests.gyp » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698