blob: a237a2fff03d6e957076fbf3f3495948a4267729 [file] [log] [blame]
// Copyright (c) 2009 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 "app/clipboard/clipboard_util_win.h"
#include "app/os_exchange_data.h"
#include "app/os_exchange_data_provider_win.h"
#include "base/pickle.h"
#include "base/ref_counted.h"
#include "base/scoped_handle.h"
#include "base/scoped_ptr.h"
#include "base/string_util.h"
#include "googleurl/src/gurl.h"
#include "testing/gtest/include/gtest/gtest.h"
typedef testing::Test OSExchangeDataTest;
namespace {
OSExchangeData::Provider* CloneProvider(const OSExchangeData& data) {
return new OSExchangeDataProviderWin(
OSExchangeDataProviderWin::GetIDataObject(data));
}
} // namespace
// Test setting/getting using the OSExchangeData API
TEST(OSExchangeDataTest, StringDataGetAndSet) {
OSExchangeData data;
std::wstring input = L"I can has cheezburger?";
data.SetString(input);
OSExchangeData data2(CloneProvider(data));
std::wstring output;
EXPECT_TRUE(data2.GetString(&output));
EXPECT_EQ(input, output);
std::string url_spec = "http://www.goats.com/";
GURL url(url_spec);
std::wstring title;
EXPECT_FALSE(data2.GetURLAndTitle(&url, &title));
// No URLs in |data|, so url should be untouched.
EXPECT_EQ(url_spec, url.spec());
}
// Test getting using the IDataObject COM API
TEST(OSExchangeDataTest, StringDataAccessViaCOM) {
OSExchangeData data;
std::wstring input = L"O hai googlz.";
data.SetString(input);
ScopedComPtr<IDataObject> com_data(
OSExchangeDataProviderWin::GetIDataObject(data));
FORMATETC format_etc =
{ CF_UNICODETEXT, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL };
EXPECT_EQ(S_OK, com_data->QueryGetData(&format_etc));
STGMEDIUM medium;
EXPECT_EQ(S_OK, com_data->GetData(&format_etc, &medium));
std::wstring output =
ScopedHGlobal<wchar_t>(medium.hGlobal).get();
EXPECT_EQ(input, output);
ReleaseStgMedium(&medium);
}
// Test setting using the IDataObject COM API
TEST(OSExchangeDataTest, StringDataWritingViaCOM) {
OSExchangeData data;
std::wstring input = L"http://www.google.com/";
ScopedComPtr<IDataObject> com_data(
OSExchangeDataProviderWin::GetIDataObject(data));
// Store data in the object using the COM SetData API.
CLIPFORMAT cfstr_ineturl = RegisterClipboardFormat(CFSTR_INETURL);
FORMATETC format_etc =
{ cfstr_ineturl, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL };
STGMEDIUM medium;
medium.tymed = TYMED_HGLOBAL;
HGLOBAL glob = GlobalAlloc(GPTR, sizeof(wchar_t) * (input.size() + 1));
size_t stringsz = input.size();
SIZE_T sz = GlobalSize(glob);
ScopedHGlobal<wchar_t> global_lock(glob);
wchar_t* buffer_handle = global_lock.get();
wcscpy_s(buffer_handle, input.size() + 1, input.c_str());
medium.hGlobal = glob;
medium.pUnkForRelease = NULL;
EXPECT_EQ(S_OK, com_data->SetData(&format_etc, &medium, TRUE));
// Construct a new object with the old object so that we can use our access
// APIs.
OSExchangeData data2(CloneProvider(data));
EXPECT_TRUE(data2.HasURL());
GURL url_from_data;
std::wstring title;
EXPECT_TRUE(data2.GetURLAndTitle(&url_from_data, &title));
GURL reference_url(input);
EXPECT_EQ(reference_url.spec(), url_from_data.spec());
}
TEST(OSExchangeDataTest, URLDataAccessViaCOM) {
OSExchangeData data;
GURL url("http://www.google.com/");
data.SetURL(url, L"");
ScopedComPtr<IDataObject> com_data(
OSExchangeDataProviderWin::GetIDataObject(data));
CLIPFORMAT cfstr_ineturl = RegisterClipboardFormat(CFSTR_INETURL);
FORMATETC format_etc =
{ cfstr_ineturl, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL };
EXPECT_EQ(S_OK, com_data->QueryGetData(&format_etc));
STGMEDIUM medium;
EXPECT_EQ(S_OK, com_data->GetData(&format_etc, &medium));
std::wstring output =
ScopedHGlobal<wchar_t>(medium.hGlobal).get();
EXPECT_EQ(url.spec(), WideToUTF8(output));
ReleaseStgMedium(&medium);
}
TEST(OSExchangeDataTest, MultipleFormatsViaCOM) {
OSExchangeData data;
std::string url_spec = "http://www.google.com/";
GURL url(url_spec);
std::wstring text = L"O hai googlz.";
data.SetURL(url, L"Google");
data.SetString(text);
ScopedComPtr<IDataObject> com_data(
OSExchangeDataProviderWin::GetIDataObject(data));
CLIPFORMAT cfstr_ineturl = RegisterClipboardFormat(CFSTR_INETURL);
FORMATETC url_format_etc =
{ cfstr_ineturl, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL };
EXPECT_EQ(S_OK, com_data->QueryGetData(&url_format_etc));
FORMATETC text_format_etc =
{ CF_UNICODETEXT, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL };
EXPECT_EQ(S_OK, com_data->QueryGetData(&text_format_etc));
STGMEDIUM medium;
EXPECT_EQ(S_OK, com_data->GetData(&url_format_etc, &medium));
std::wstring output_url =
ScopedHGlobal<wchar_t>(medium.hGlobal).get();
EXPECT_EQ(url.spec(), WideToUTF8(output_url));
ReleaseStgMedium(&medium);
// The text is supposed to be the raw text of the URL, _NOT_ the value of
// |text|! This is because the URL is added first and thus takes precedence!
EXPECT_EQ(S_OK, com_data->GetData(&text_format_etc, &medium));
std::wstring output_text =
ScopedHGlobal<wchar_t>(medium.hGlobal).get();
EXPECT_EQ(url_spec, WideToUTF8(output_text));
ReleaseStgMedium(&medium);
}
TEST(OSExchangeDataTest, EnumerationViaCOM) {
OSExchangeData data;
data.SetURL(GURL("http://www.google.com/"), L"");
data.SetString(L"O hai googlz.");
CLIPFORMAT cfstr_file_group_descriptor =
RegisterClipboardFormat(CFSTR_FILEDESCRIPTOR);
CLIPFORMAT text_x_moz_url = RegisterClipboardFormat(L"text/x-moz-url");
ScopedComPtr<IDataObject> com_data(
OSExchangeDataProviderWin::GetIDataObject(data));
ScopedComPtr<IEnumFORMATETC> enumerator;
EXPECT_EQ(S_OK, com_data.get()->EnumFormatEtc(DATADIR_GET,
enumerator.Receive()));
// Test that we can get one item.
{
// Explictly don't reset the first time, to verify the creation state is
// OK.
ULONG retrieved = 0;
FORMATETC elements_array[1];
EXPECT_EQ(S_OK, enumerator->Next(1,
reinterpret_cast<FORMATETC*>(&elements_array), &retrieved));
EXPECT_EQ(1, retrieved);
EXPECT_EQ(text_x_moz_url, elements_array[0].cfFormat);
}
// Test that we can get one item with a NULL retrieved value.
{
EXPECT_EQ(S_OK, enumerator->Reset());
FORMATETC elements_array[1];
EXPECT_EQ(S_OK, enumerator->Next(1,
reinterpret_cast<FORMATETC*>(&elements_array), NULL));
EXPECT_EQ(text_x_moz_url, elements_array[0].cfFormat);
}
// Test that we can get two items.
{
EXPECT_EQ(S_OK, enumerator->Reset());
ULONG retrieved = 0;
FORMATETC elements_array[2];
EXPECT_EQ(S_OK, enumerator->Next(2,
reinterpret_cast<FORMATETC*>(&elements_array), &retrieved));
EXPECT_EQ(2, retrieved);
EXPECT_EQ(text_x_moz_url, elements_array[0].cfFormat);
EXPECT_EQ(cfstr_file_group_descriptor, elements_array[1].cfFormat);
}
// Test that we can skip the first item.
{
EXPECT_EQ(S_OK, enumerator->Reset());
EXPECT_EQ(S_OK, enumerator->Skip(1));
ULONG retrieved = 0;
FORMATETC elements_array[1];
EXPECT_EQ(S_OK, enumerator->Next(1,
reinterpret_cast<FORMATETC*>(&elements_array), &retrieved));
EXPECT_EQ(1, retrieved);
EXPECT_EQ(cfstr_file_group_descriptor, elements_array[0].cfFormat);
}
// Test that we can skip the first item, and create a clone that matches in
// this state, and modify the original without affecting the clone.
{
EXPECT_EQ(S_OK, enumerator->Reset());
EXPECT_EQ(S_OK, enumerator->Skip(1));
ScopedComPtr<IEnumFORMATETC> cloned_enumerator;
EXPECT_EQ(S_OK, enumerator.get()->Clone(cloned_enumerator.Receive()));
EXPECT_EQ(S_OK, enumerator.get()->Reset());
{
ULONG retrieved = 0;
FORMATETC elements_array[1];
EXPECT_EQ(S_OK, cloned_enumerator->Next(1,
reinterpret_cast<FORMATETC*>(&elements_array), &retrieved));
EXPECT_EQ(1, retrieved);
EXPECT_EQ(cfstr_file_group_descriptor, elements_array[0].cfFormat);
}
{
ULONG retrieved = 0;
FORMATETC elements_array[1];
EXPECT_EQ(S_OK, enumerator->Next(1,
reinterpret_cast<FORMATETC*>(&elements_array), &retrieved));
EXPECT_EQ(1, retrieved);
EXPECT_EQ(text_x_moz_url, elements_array[0].cfFormat);
}
}
}
TEST(OSExchangeDataTest, TestURLExchangeFormats) {
OSExchangeData data;
std::string url_spec = "http://www.google.com/";
GURL url(url_spec);
std::wstring url_title = L"Google";
data.SetURL(url, url_title);
std::wstring output;
OSExchangeData data2(CloneProvider(data));
// URL spec and title should match
GURL output_url;
std::wstring output_title;
EXPECT_TRUE(data2.GetURLAndTitle(&output_url, &output_title));
EXPECT_EQ(url_spec, output_url.spec());
EXPECT_EQ(url_title, output_title);
std::wstring output_string;
// URL should be the raw text response
EXPECT_TRUE(data2.GetString(&output_string));
EXPECT_EQ(url_spec, WideToUTF8(output_string));
// File contents access via COM
ScopedComPtr<IDataObject> com_data(
OSExchangeDataProviderWin::GetIDataObject(data));
{
CLIPFORMAT cfstr_file_contents =
RegisterClipboardFormat(CFSTR_FILECONTENTS);
FORMATETC format_etc =
{ cfstr_file_contents, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL };
EXPECT_EQ(S_OK, com_data->QueryGetData(&format_etc));
STGMEDIUM medium;
EXPECT_EQ(S_OK, com_data->GetData(&format_etc, &medium));
ScopedHGlobal<char> glob(medium.hGlobal);
std::string output(glob.get(), glob.Size());
std::string file_contents = "[InternetShortcut]\r\nURL=";
file_contents += url_spec;
file_contents += "\r\n";
EXPECT_EQ(file_contents, output);
ReleaseStgMedium(&medium);
}
}
TEST(OSExchangeDataTest, TestPickledData) {
CLIPFORMAT test_cf = RegisterClipboardFormat(L"chrome/test");
Pickle saved_pickle;
saved_pickle.WriteInt(1);
saved_pickle.WriteInt(2);
OSExchangeData data;
data.SetPickledData(test_cf, saved_pickle);
OSExchangeData copy(CloneProvider(data));
EXPECT_TRUE(copy.HasCustomFormat(test_cf));
Pickle restored_pickle;
EXPECT_TRUE(copy.GetPickledData(test_cf, &restored_pickle));
void* p_iterator = NULL;
int value;
EXPECT_TRUE(restored_pickle.ReadInt(&p_iterator, &value));
EXPECT_EQ(1, value);
EXPECT_TRUE(restored_pickle.ReadInt(&p_iterator, &value));
EXPECT_EQ(2, value);
}
TEST(OSExchangeDataTest, FileContents) {
OSExchangeData data;
std::string file_contents("data\0with\0nulls", 15);
data.SetFileContents(L"filename.txt", file_contents);
OSExchangeData copy(CloneProvider(data));
std::wstring filename;
std::string read_contents;
EXPECT_TRUE(copy.GetFileContents(&filename, &read_contents));
EXPECT_EQ(L"filename.txt", filename);
EXPECT_EQ(file_contents, read_contents);
}
TEST(OSExchangeDataTest, Html) {
OSExchangeData data;
GURL url("http://www.google.com/");
std::wstring html(
L"<HTML>\n<BODY>\n"
L"<b>bold.</b> <i><b>This is bold italic.</b></i>\n"
L"</BODY>\n</HTML>");
data.SetHtml(html, url);
OSExchangeData copy(CloneProvider(data));
std::wstring read_html;
EXPECT_TRUE(copy.GetHtml(&read_html, &url));
EXPECT_EQ(html, read_html);
// Check the CF_HTML too.
std::string expected_cf_html(
"Version:0.9\r\nStartHTML:0000000139\r\nEndHTML:0000000292\r\n"
"StartFragment:0000000177\r\nEndFragment:0000000254\r\n"
"SourceURL:http://www.google.com/\r\n<html>\r\n<body>\r\n"
"<!--StartFragment-->\r\n");
expected_cf_html += WideToUTF8(html);
expected_cf_html.append("\r\n<!--EndFragment-->\r\n</body>\r\n</html>");
STGMEDIUM medium;
IDataObject* data_object = OSExchangeDataProviderWin::GetIDataObject(data);
EXPECT_EQ(S_OK,
data_object->GetData(ClipboardUtil::GetHtmlFormat(), &medium));
ScopedHGlobal<char> glob(medium.hGlobal);
std::string output(glob.get(), glob.Size());
EXPECT_EQ(expected_cf_html, output);
ReleaseStgMedium(&medium);
}
TEST(OSExchangeDataTest, SetURLWithMaxPath) {
OSExchangeData data;
std::wstring long_title(L'a', MAX_PATH + 1);
data.SetURL(GURL("http://google.com"), long_title);
}
TEST(OSExchangeDataTest, ProvideURLForPlainTextURL) {
OSExchangeData data;
data.SetString(L"http://google.com");
OSExchangeData data2(CloneProvider(data));
ASSERT_TRUE(data2.HasURL());
GURL read_url;
std::wstring title;
EXPECT_TRUE(data2.GetURLAndTitle(&read_url, &title));
EXPECT_EQ(GURL("http://google.com"), read_url);
}