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/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 Loading... |
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 Loading... |
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 Loading... |
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 Loading... |
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 |
OLD | NEW |