// Copyright 2012 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "chrome/browser/ui/omnibox/clipboard_utils.h"

#include <string>

#include "base/strings/string_util.h"
#include "base/strings/utf_string_conversions.h"
#include "base/test/task_environment.h"
#include "base/test/test_future.h"
#include "build/build_config.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "testing/platform_test.h"
#include "ui/base/clipboard/clipboard.h"
#include "ui/base/clipboard/scoped_clipboard_writer.h"
#include "ui/base/clipboard/test/test_clipboard.h"

using base::ASCIIToUTF16;

namespace {

std::u16string GetClipboardTextSync(bool notify_if_restricted) {
  base::test::TestFuture<std::u16string> future;
  GetClipboardText(notify_if_restricted, future.GetCallback());
  return future.Take();
}

class ClipboardUtilsTest : public PlatformTest {
 public:
  ClipboardUtilsTest()
      : task_environment_(
            base::test::SingleThreadTaskEnvironment::MainThreadType::UI) {}

  void SetUp() override {
    PlatformTest::SetUp();
    ui::TestClipboard::CreateForCurrentThread();
  }

  void TearDown() override {
    PlatformTest::TearDown();
    ui::Clipboard::DestroyClipboardForCurrentThread();
  }

 private:
  // Windows requires a message loop for clipboard access.
  base::test::SingleThreadTaskEnvironment task_environment_;
};

TEST_F(ClipboardUtilsTest, GetClipboardText) {
  ui::Clipboard* clipboard = ui::Clipboard::GetForCurrentThread();

  const std::u16string kPlainText(u"test text");
  const std::string kURL("http://www.example.com/");

  // Can we pull straight text off the clipboard?
  {
    ui::ScopedClipboardWriter clipboard_writer(ui::ClipboardBuffer::kCopyPaste);
    clipboard_writer.WriteText(kPlainText);
  }
  EXPECT_EQ(kPlainText, GetClipboardTextSync(/*notify_if_restricted=*/false));

  // Can we pull a string consists of white-space?
  const std::u16string kSpace6(u"      ");
  const std::u16string kSpace1(u" ");
  {
    ui::ScopedClipboardWriter clipboard_writer(ui::ClipboardBuffer::kCopyPaste);
    clipboard_writer.WriteText(kSpace6);
  }
  EXPECT_EQ(kSpace1, GetClipboardTextSync(/*notify_if_restricted=*/false));

  // Does an empty clipboard get empty text?
  clipboard->Clear(ui::ClipboardBuffer::kCopyPaste);
  EXPECT_EQ(std::u16string(),
            GetClipboardTextSync(/*notify_if_restricted=*/false));

// Bookmark clipboard apparently not supported on Linux.
// See TODO on ClipboardText.BookmarkTest.
#if !BUILDFLAG(IS_POSIX) || BUILDFLAG(IS_MAC)
  const std::u16string kTitle(u"The Example Company");
  // Can we pull a bookmark off the clipboard?
  {
    ui::ScopedClipboardWriter clipboard_writer(ui::ClipboardBuffer::kCopyPaste);
    clipboard_writer.WriteURL(ui::ClipboardUrlInfo{
        .url = GURL(kURL), .title = std::u16string(kTitle)});
  }
  EXPECT_EQ(ASCIIToUTF16(kURL),
            GetClipboardTextSync(/*notify_if_restricted=*/false));

  // Do we pull text in preference to a bookmark?
  {
    ui::ScopedClipboardWriter clipboard_writer(ui::ClipboardBuffer::kCopyPaste);
    clipboard_writer.WriteText(kPlainText);
    clipboard_writer.WriteURL(ui::ClipboardUrlInfo{
        .url = GURL(kURL), .title = std::u16string(kTitle)});
  }
  EXPECT_EQ(kPlainText, GetClipboardTextSync(/*notify_if_restricted=*/false));
#endif

  // Do we get nothing if there is neither text nor a bookmark?
  {
    const std::u16string kMarkup(u"<strong>Hi!</string>");
    ui::ScopedClipboardWriter clipboard_writer(ui::ClipboardBuffer::kCopyPaste);
    clipboard_writer.WriteHTML(kMarkup, kURL);
  }
  EXPECT_TRUE(GetClipboardTextSync(/*notify_if_restricted=*/false).empty());
}

TEST_F(ClipboardUtilsTest, TruncateLongText) {
  const std::u16string almost_long_text =
      base::ASCIIToUTF16(std::string(kMaxClipboardTextLength, '.'));
  {
    ui::ScopedClipboardWriter clipboard_writer(ui::ClipboardBuffer::kCopyPaste);
    clipboard_writer.WriteText(almost_long_text);
  }
  EXPECT_EQ(almost_long_text,
            GetClipboardTextSync(/*notify_if_restricted=*/false));

  const std::u16string long_text =
      base::ASCIIToUTF16(std::string(kMaxClipboardTextLength + 1, '.'));
  {
    ui::ScopedClipboardWriter clipboard_writer(ui::ClipboardBuffer::kCopyPaste);
    clipboard_writer.WriteText(long_text);
  }
  EXPECT_EQ(almost_long_text,
            GetClipboardTextSync(/*notify_if_restricted=*/false));
}

}  // namespace