Stubs out OSExchangeData for views on gtk.

BUG=none
TEST=none

Review URL: http://codereview.chromium.org/113321

git-svn-id: svn://svn.chromium.org/chrome/trunk/src@16012 0039d316-1c4b-4281-b951-d872f2087c98

1 files changed, 365 insertions, 0 deletions

diff --git a/app/os_exchange_data_win_unittest.cc b/app/os_exchange_data_win_unittest.cc
new file mode 100644
index 0000000..13a28a9
--- /dev/null
+++ b/app/os_exchange_data_win_unittest.cc
@@ -0,0 +1,365 @@
+// Copyright (c) 2006-2008 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 <atlbase.h>
+#include <shlobj.h>
+
+#include "app/os_exchange_data.h"
+#include "base/clipboard_util.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;
+
+// Test setting/getting using the OSExchangeData API
+TEST(OSExchangeDataTest, StringDataGetAndSet) {
+  OSExchangeData* data = new OSExchangeData;
+  std::wstring input = L"I can has cheezburger?";
+  data->SetString(input);
+
+  OSExchangeData* data2 = new OSExchangeData(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());
+  // data gets freed when data2 releases the ref on it
+  delete data2;
+}
+
+// Test getting using the IDataObject COM API
+TEST(OSExchangeDataTest, StringDataAccessViaCOM) {
+  OSExchangeData* data = new OSExchangeData;
+  std::wstring input = L"O hai googlz.";
+  data->SetString(input);
+  CComPtr<IDataObject> com_data(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);
+
+  // data is freed automatically by CComPtr.
+}
+
+// Test setting using the IDataObject COM API
+TEST(OSExchangeDataTest, StringDataWritingViaCOM) {
+  OSExchangeData* data = new OSExchangeData;
+  std::wstring input = L"http://www.google.com/";
+
+  CComPtr<IDataObject> com_data(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 = new OSExchangeData(com_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());
+  // deleting data2 will free data because it holds a ref to it.
+  delete data2;
+}
+
+TEST(OSExchangeDataTest, URLDataAccessViaCOM) {
+  OSExchangeData* data = new OSExchangeData;
+  GURL url("http://www.google.com/");
+  data->SetURL(url, L"");
+  CComPtr<IDataObject> com_data(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 = new OSExchangeData;
+  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);
+
+  CComPtr<IDataObject> com_data(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 = new OSExchangeData;
+  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");
+
+  CComPtr<IDataObject> com_data(data);
+  CComPtr<IEnumFORMATETC> enumerator;
+  EXPECT_EQ(S_OK, com_data->EnumFormatEtc(DATADIR_GET, &enumerator));
+
+  // 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));
+    CComPtr<IEnumFORMATETC> cloned_enumerator;
+    EXPECT_EQ(S_OK, enumerator->Clone(&cloned_enumerator));
+    EXPECT_EQ(S_OK, enumerator->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 = new OSExchangeData;
+  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 = new OSExchangeData(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
+  CComPtr<IDataObject> com_data(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);
+  }
+
+  // Need to manually free data2 since we never stuff it into a COMPtr.
+  delete data2;
+}
+
+TEST(OSExchangeDataTest, TestPickledData) {
+  CLIPFORMAT test_cf = RegisterClipboardFormat(L"chrome/test");
+
+  Pickle saved_pickle;
+  saved_pickle.WriteInt(1);
+  saved_pickle.WriteInt(2);
+  scoped_refptr<OSExchangeData> data(new OSExchangeData());
+  data->SetPickledData(test_cf, saved_pickle);
+
+  scoped_refptr<OSExchangeData> copy(new OSExchangeData(data.get()));
+  EXPECT_TRUE(copy->HasFormat(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) {
+  scoped_refptr<OSExchangeData> data(new OSExchangeData);
+  std::string file_contents("data\0with\0nulls", 15);
+  data->SetFileContents(L"filename.txt", file_contents);
+
+  scoped_refptr<OSExchangeData> copy(new OSExchangeData(data.get()));
+  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) {
+  scoped_refptr<OSExchangeData> data(new OSExchangeData);
+  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);
+
+  scoped_refptr<OSExchangeData> copy(new OSExchangeData(data.get()));
+  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;
+  EXPECT_EQ(S_OK, data->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) {
+  scoped_refptr<OSExchangeData> data(new OSExchangeData);
+  std::wstring long_title(L'a', MAX_PATH + 1);
+  data->SetURL(GURL("http://google.com"), long_title);
+}
+
+TEST(OSExchangeDataTest, ProvideURLForPlainTextURL) {
+  scoped_refptr<OSExchangeData> data(new OSExchangeData);
+  data->SetString(L"http://google.com");
+
+  scoped_ptr<OSExchangeData> data2(new OSExchangeData(data.get()));
+  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);
+}