Properly collapse newlines (and leading/trailing whitespace) from pasted...

chrome/browser/cocoa/autocomplete_text_field_editor_unittest.mm

 // 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.
 
 #import "chrome/browser/cocoa/autocomplete_text_field_editor.h"
 
 #include "base/scoped_nsobject.h"
 #include "base/scoped_ptr.h"
 #include "base/string_util.h"
+#include "base/sys_string_conversions.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "testing/platform_test.h"
 
 namespace {
+
 int NumTypesOnPasteboard(NSPasteboard* pb) {
   return [[pb types] count];
 }
 
 void ClearPasteboard(NSPasteboard* pb) {
   [pb declareTypes:[NSArray array] owner:nil];
 }
 
 bool NoRichTextOnClipboard(NSPasteboard* pb) {
   return ([pb dataForType:NSRTFPboardType] == nil) &&
       ([pb dataForType:NSRTFDPboardType] == nil) &&
       ([pb dataForType:NSHTMLPboardType] == nil);
 }
 
 bool ClipboardContainsText(NSPasteboard* pb, NSString* cmp) {
   NSString* clipboard_text = [pb stringForType:NSStringPboardType];
   return [clipboard_text isEqualToString:cmp];
 }
 
+bool PutStringOnPasteboard(NSPasteboard* pb, NSString* string) {
+  return ([pb setString:string forType:NSStringPboardType] != NO);
+}
+
 } // namespace
 
 class AutocompleteTextFieldEditorTest : public PlatformTest {
  public:
   virtual void SetUp() {
     PlatformTest::SetUp();
     pb_ = [NSPasteboard pasteboardWithUniqueName];
   }
 
   virtual void TearDown() {
@@ skipping 36 matching lines @@
 
   // Check that cutting it works and we only get plain text.
   [field_editor.get() setString:test_string_2];
   [field_editor.get() selectAll:nil];
   [field_editor.get() alignLeft:nil];  // Add a rich text attribute.
   [field_editor.get() performCut:pb];
   ASSERT_TRUE(NoRichTextOnClipboard(clipboard()));
   ASSERT_TRUE(ClipboardContainsText(clipboard(), test_string_2));
   ASSERT_EQ([[field_editor.get() textStorage] length], 0U);
 }
+
+TEST_F(AutocompleteTextFieldEditorTest, PasteTest) {
+  const int num_tests = 5;
+  NSString *test_string_input[num_tests] = {
+    @"Has_no_spaces",
+    @"Has spaces  inside",
+    @" \t Has leading whitespace",
+    @"Has trailing whitespace \t ",
+    @"\rHas s\n cat\rtered new  \nlines and spaces\n\n\r"
+  };
+
+  // Clear the pasteboard is empty and double-check.
+  NSPasteboard* pb = clipboard();
+  ClearPasteboard(pb);
+  ASSERT_EQ(NumTypesOnPasteboard(pb), 0);
+
+  // Set up the pasteboard to take strings.
+  [pb declareTypes:[NSArray arrayWithObject:NSStringPboardType] owner:nil];
+
+  for(int i = 0; i < num_tests; i++) {
+    // Make a field editor and make sure it's empty.
+    scoped_nsobject<AutocompleteTextFieldEditor> field_editor(
+        [[AutocompleteTextFieldEditor alloc] init]);
+    ASSERT_EQ([[field_editor.get() textStorage] length], 0U);
+
+    // Put test_string_input[i] on pasteboard and make sure it's there.
+    ASSERT_TRUE(PutStringOnPasteboard(pb, test_string_input[i]));
+    ASSERT_TRUE(ClipboardContainsText(pb, test_string_input[i]));
+
+    // Paste and make sure we get the corresponding output from
+    // CollapseWhitespace(...,true). CollapseWhitespace() should be tested
+    // elsewhere and should yield the desired output.
+    [field_editor.get() performPaste:pb];
+    EXPECT_EQ(
+        CollapseWhitespace(base::SysNSStringToWide(test_string_input[i]),true),
+        base::SysNSStringToWide([[field_editor.get() textStorage] string]));
+  }
+}