chrome/browser/ui/omnibox/omnibox_view_unittest.cc

   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
   3 // found in the LICENSE file.
   4
   5 #include "chrome/browser/ui/omnibox/omnibox_view.h"
   6 #include "base/message_loop/message_loop.h"
   7 #include "base/strings/string16.h"
   8 #include "base/strings/string_util.h"
   9 #include "base/strings/utf_string_conversions.h"
  10 #include "testing/gtest/include/gtest/gtest.h"
  11 #include "testing/platform_test.h"
  12 #include "ui/base/clipboard/clipboard.h"
  13 #include "ui/base/clipboard/scoped_clipboard_writer.h"
  14
  15 using base::ASCIIToUTF16;
  16
  17 namespace {
  18
  19 class OmniboxViewTest : public PlatformTest {
  20  public:
  21   virtual void TearDown() OVERRIDE {
  22     ui::Clipboard::DestroyClipboardForCurrentThread();
  23   }
  24
  25  private:
  26   // Windows requires a message loop for clipboard access.
  27   base::MessageLoopForUI message_loop_;
  28 };
  29
  30 TEST_F(OmniboxViewTest, TestStripSchemasUnsafeForPaste) {
  31   const char* urls[] = {
  32     "http://www.google.com?q=javascript:alert(0)",  // Safe URL.
  33     "javAscript:alert(0)",                          // Unsafe JS URL.
  34     "jaVascript:\njavaScript: alert(0)"             // Single strip unsafe.
  35   };
  36
  37   const char* expecteds[] = {
  38     "http://www.google.com?q=javascript:alert(0)",  // Safe URL.
  39     "alert(0)",                                     // Unsafe JS URL.
  40     "alert(0)"                                      // Single strip unsafe.
  41   };
  42
  43   for (size_t i = 0; i < arraysize(urls); i++) {
  44     EXPECT_EQ(ASCIIToUTF16(expecteds[i]),
  45               OmniboxView::StripJavascriptSchemas(ASCIIToUTF16(urls[i])));
  46   }
  47 }
  48
  49 TEST_F(OmniboxViewTest, SanitizeTextForPaste) {
  50   // Broken URL has newlines stripped.
  51   const base::string16 kWrappedURL(ASCIIToUTF16(
  52       "http://www.chromium.org/developers/testing/chromium-\n"
  53       "build-infrastructure/tour-of-the-chromium-buildbot"));
  54
  55   const base::string16 kFixedURL(ASCIIToUTF16(
  56       "http://www.chromium.org/developers/testing/chromium-"
  57       "build-infrastructure/tour-of-the-chromium-buildbot"));
  58   EXPECT_EQ(kFixedURL, OmniboxView::SanitizeTextForPaste(kWrappedURL));
  59
  60   // Multi-line address is converted to a single-line address.
  61   const base::string16 kWrappedAddress(ASCIIToUTF16(
  62       "1600 Amphitheatre Parkway\nMountain View, CA"));
  63
  64   const base::string16 kFixedAddress(ASCIIToUTF16(
  65       "1600 Amphitheatre Parkway Mountain View, CA"));
  66   EXPECT_EQ(kFixedAddress, OmniboxView::SanitizeTextForPaste(kWrappedAddress));
  67 }
  68
  69 TEST_F(OmniboxViewTest, GetClipboardText) {
  70   ui::Clipboard* clipboard = ui::Clipboard::GetForCurrentThread();
  71
  72   const base::string16 kPlainText(ASCIIToUTF16("test text"));
  73   const std::string kURL("http://www.example.com/");
  74
  75   // Can we pull straight text off the clipboard?
  76   {
  77     ui::ScopedClipboardWriter clipboard_writer(clipboard,
  78                                                ui::CLIPBOARD_TYPE_COPY_PASTE);
  79     clipboard_writer.WriteText(kPlainText);
  80   }
  81   EXPECT_EQ(kPlainText, OmniboxView::GetClipboardText());
  82
  83   // Can we pull a string consists of white-space?
  84   const base::string16 kSpace6(ASCIIToUTF16("      "));
  85   const base::string16 kSpace1(ASCIIToUTF16(" "));
  86   {
  87     ui::ScopedClipboardWriter clipboard_writer(clipboard,
  88                                                ui::CLIPBOARD_TYPE_COPY_PASTE);
  89     clipboard_writer.WriteText(kSpace6);
  90   }
  91   EXPECT_EQ(kSpace1, OmniboxView::GetClipboardText());
  92
  93   // Does an empty clipboard get empty text?
  94   clipboard->Clear(ui::CLIPBOARD_TYPE_COPY_PASTE);
  95   EXPECT_EQ(base::string16(), OmniboxView::GetClipboardText());
  96
  97   // Bookmark clipboard apparently not supported on Linux.
  98   // See TODO on ClipboardText.BookmarkTest.
  99 #if !defined(OS_POSIX) || defined(OS_MACOSX)
 100   const base::string16 kTitle(ASCIIToUTF16("The Example Company"));
 101   // Can we pull a bookmark off the clipboard?
 102   {
 103     ui::ScopedClipboardWriter clipboard_writer(clipboard,
 104                                                ui::CLIPBOARD_TYPE_COPY_PASTE);
 105     clipboard_writer.WriteBookmark(kTitle, kURL);
 106   }
 107   EXPECT_EQ(ASCIIToUTF16(kURL), OmniboxView::GetClipboardText());
 108
 109   // Do we pull text in preference to a bookmark?
 110   {
 111     ui::ScopedClipboardWriter clipboard_writer(clipboard,
 112                                                ui::CLIPBOARD_TYPE_COPY_PASTE);
 113     clipboard_writer.WriteText(kPlainText);
 114     clipboard_writer.WriteBookmark(kTitle, kURL);
 115   }
 116   EXPECT_EQ(kPlainText, OmniboxView::GetClipboardText());
 117 #endif
 118
 119   // Do we get nothing if there is neither text nor a bookmark?
 120   {
 121     const base::string16 kMarkup(ASCIIToUTF16("<strong>Hi!</string>"));
 122     ui::ScopedClipboardWriter clipboard_writer(clipboard,
 123                                                ui::CLIPBOARD_TYPE_COPY_PASTE);
 124     clipboard_writer.WriteHTML(kMarkup, kURL);
 125   }
 126   EXPECT_TRUE(OmniboxView::GetClipboardText().empty());
 127 }
 128
 129 }  // namespace