[Mac] Refactor location-bar autocomplete cell classes.

chrome/browser/cocoa/autocomplete_text_field_unittest.mm

Index: chrome/browser/cocoa/autocomplete_text_field_unittest.mm
diff --git a/chrome/browser/cocoa/autocomplete_text_field_unittest.mm b/chrome/browser/cocoa/autocomplete_text_field_unittest.mm
new file mode 100644
index 0000000000000000000000000000000000000000..9e861a8e0473fe3411c860bfeab6308d53004a85
--- /dev/null
+++ b/chrome/browser/cocoa/autocomplete_text_field_unittest.mm
@@ -0,0 +1,46 @@
+// 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 <Cocoa/Cocoa.h>
+
+#include "base/scoped_nsobject.h"
+#import "chrome/browser/cocoa/autocomplete_text_field.h"
+#import "chrome/browser/cocoa/autocomplete_text_field_cell.h"
+#import "chrome/browser/cocoa/cocoa_test_helper.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace {
+
+class AutocompleteTextFieldTest : public testing::Test {
+ public:
+  AutocompleteTextFieldTest() {
+    NSRect frame = NSMakeRect(0, 0, 50, 30);
+    field_.reset([[AutocompleteTextField alloc] initWithFrame:frame]);
+    [field_ setStringValue:@"Testing"];
+    [cocoa_helper_.contentView() addSubview:field_.get()];
+  }
+
+  CocoaTestHelper cocoa_helper_;  // Inits Cocoa, creates window, etc...
+  scoped_nsobject<AutocompleteTextField> field_;
+};
+
+// Test that we have the right cell class.
+TEST_F(AutocompleteTextFieldTest, CellClass) {
+  EXPECT_TRUE([[field_ cell] isKindOfClass:[AutocompleteTextFieldCell class]]);
+}
+
+// Test adding/removing from the view hierarchy, mostly to ensure nothing
+// leaks or crashes.
+TEST_F(AutocompleteTextFieldTest, AddRemove) {
+  EXPECT_EQ(cocoa_helper_.contentView(), [field_ superview]);
+  [field_.get() removeFromSuperview];
+  EXPECT_FALSE([field_ superview]);
+}
+
+// Test drawing, mostly to ensure nothing leaks or crashes.
+TEST_F(AutocompleteTextFieldTest, Display) {
+  [field_ display];
+}
+
+}  // namespace