Add more tests to DataOfferTest

Bug: 750997
Test: exo_unittests --gtest_filter=DataOfferTest.*
Change-Id: Ic5d06494ae73d7b22e9463a5b303fb7c6a950259
Reviewed-on: https://chromium-review.googlesource.com/625513
Commit-Queue: Daichi Hirono <hirono@chromium.org>
Reviewed-by: David Reveman <reveman@chromium.org>
Cr-Commit-Position: refs/heads/master@{#501515}

components/exo/data_offer_unittest.cc

@@ -4,6 +4,9 @@
 
 #include "components/exo/data_offer.h"
 
+#include <fcntl.h>
+#include <stdio.h>
+
 #include <memory>
 #include <string>
 #include <vector>
@@ -15,8 +18,10 @@
 #include "base/strings/utf_string_conversions.h"
 #include "components/exo/data_device.h"
 #include "components/exo/data_offer_delegate.h"
+#include "components/exo/file_helper.h"
 #include "components/exo/test/exo_test_base.h"
 #include "ui/base/dragdrop/os_exchange_data.h"
+#include "url/gurl.h"
 
 namespace exo {
 namespace {
@@ -59,33 +64,120 @@ class TestDataOfferDelegate : public DataOfferDelegate {
   DISALLOW_COPY_AND_ASSIGN(TestDataOfferDelegate);
 };
 
-TEST_F(DataOfferTest, SendEvents) {
-  ui::OSExchangeData data;
-  data.SetString(base::string16(base::ASCIIToUTF16("Test data")));
+class TestFileHelper : public FileHelper {
+ public:
+  TestFileHelper() = default;
+
+  // Overridden from FileHelper:
+  std::string GetMimeTypeForUriList() const override { return "text/uri-list"; }
+  bool ConvertPathToUrl(const base::FilePath& path, GURL* out) override {
+    *out = GURL("file://" + path.AsUTF8Unsafe());
+    return true;
+  }
 
+ private:
+  DISALLOW_COPY_AND_ASSIGN(TestFileHelper);
+};
+
+void CreatePipe(base::ScopedFD* read_pipe, base::ScopedFD* write_pipe) {
+  int raw_pipe[2];
+  PCHECK(0 == pipe(raw_pipe));
+  read_pipe->reset(raw_pipe[0]);
+  write_pipe->reset(raw_pipe[1]);
+}
+
+bool ReadString16(base::ScopedFD fd, base::string16* out) {
+  std::array<char, 128> buffer;
+  char* it = buffer.begin();
+  while (it != buffer.end()) {
+    int result = read(fd.get(), it, buffer.end() - it);
+    PCHECK(-1 != result);
+    if (result == 0)
+      break;
+    it += result;
+  }
+  *out = base::string16(reinterpret_cast<base::char16*>(buffer.data()),
+                        (it - buffer.begin()) / sizeof(base::char16));
+  return true;
+}
+
+TEST_F(DataOfferTest, SetTextDropData) {
   base::flat_set<DndAction> source_actions;
   source_actions.insert(DndAction::kCopy);
   source_actions.insert(DndAction::kMove);
 
-  std::unique_ptr<TestDataOfferDelegate> delegate =
-      base::MakeUnique<TestDataOfferDelegate>();
-  std::unique_ptr<DataOffer> data_offer =
-      base::MakeUnique<DataOffer>(delegate.get());
-
-  EXPECT_EQ(0u, delegate->mime_types().size());
-  EXPECT_EQ(0u, delegate->source_actions().size());
-  EXPECT_EQ(DndAction::kNone, delegate->dnd_action());
-
-  data_offer->SetDropData(nullptr, data);
-  data_offer->SetSourceActions(source_actions);
-  data_offer->SetActions(base::flat_set<DndAction>(), DndAction::kMove);
-
-  EXPECT_EQ(1u, delegate->mime_types().size());
-  EXPECT_EQ("text/plain", delegate->mime_types()[0]);
-  EXPECT_EQ(2u, delegate->source_actions().size());
-  EXPECT_EQ(1u, delegate->source_actions().count(DndAction::kCopy));
-  EXPECT_EQ(1u, delegate->source_actions().count(DndAction::kMove));
-  EXPECT_EQ(DndAction::kMove, delegate->dnd_action());
+  ui::OSExchangeData data;
+  data.SetString(base::string16(base::ASCIIToUTF16("Test data")));
+
+  TestDataOfferDelegate delegate;
+  DataOffer data_offer(&delegate);
+
+  EXPECT_EQ(0u, delegate.mime_types().size());
+  EXPECT_EQ(0u, delegate.source_actions().size());
+  EXPECT_EQ(DndAction::kNone, delegate.dnd_action());
+
+  TestFileHelper file_helper;
+  data_offer.SetDropData(&file_helper, data);
+  data_offer.SetSourceActions(source_actions);
+  data_offer.SetActions(base::flat_set<DndAction>(), DndAction::kMove);
+
+  EXPECT_EQ(1u, delegate.mime_types().size());
+  EXPECT_EQ("text/plain", delegate.mime_types()[0]);
+  EXPECT_EQ(2u, delegate.source_actions().size());
+  EXPECT_EQ(1u, delegate.source_actions().count(DndAction::kCopy));
+  EXPECT_EQ(1u, delegate.source_actions().count(DndAction::kMove));
+  EXPECT_EQ(DndAction::kMove, delegate.dnd_action());
+}
+
+TEST_F(DataOfferTest, SetFileDropData) {
+  TestDataOfferDelegate delegate;
+  DataOffer data_offer(&delegate);
+
+  TestFileHelper file_helper;
+  ui::OSExchangeData data;
+  data.SetFilename(base::FilePath("/test/downloads/file"));
+  data_offer.SetDropData(&file_helper, data);
+
+  EXPECT_EQ(1u, delegate.mime_types().size());
+  EXPECT_EQ("text/uri-list", delegate.mime_types()[0]);
+}
+
+TEST_F(DataOfferTest, ReceiveString) {
+  TestDataOfferDelegate delegate;
+  DataOffer data_offer(&delegate);
+
+  TestFileHelper file_helper;
+  ui::OSExchangeData data;
+  data.SetString(base::ASCIIToUTF16("Test data"));
+  data_offer.SetDropData(&file_helper, data);
+
+  base::ScopedFD read_pipe;
+  base::ScopedFD write_pipe;
+  CreatePipe(&read_pipe, &write_pipe);
+
+  data_offer.Receive("text/plain", std::move(write_pipe));
+  base::string16 result;
+  ASSERT_TRUE(ReadString16(std::move(read_pipe), &result));
+  EXPECT_EQ(base::ASCIIToUTF16("Test data"), result);
+}
+
+TEST_F(DataOfferTest, ReceiveUriList) {
+  TestDataOfferDelegate delegate;
+  DataOffer data_offer(&delegate);
+
+  TestFileHelper file_helper;
+  ui::OSExchangeData data;
+  data.SetFilename(base::FilePath("/test/downloads/file"));
+  data_offer.SetDropData(&file_helper, data);
+
+  base::ScopedFD read_pipe;
+  base::ScopedFD write_pipe;
+  CreatePipe(&read_pipe, &write_pipe);
+
+  data_offer.Receive("text/uri-list", std::move(write_pipe));
+  base::string16 result;
+  ASSERT_TRUE(ReadString16(std::move(read_pipe), &result));
+  EXPECT_EQ(base::ASCIIToUTF16("file:///test/downloads/file"), result);
 }
 
 }  // namespace