OLD | NEW |
(Empty) | |
| 1 // Copyright (c) 2011 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 "base/memory/scoped_nsobject.h" |
| 6 #import "chrome/browser/ui/cocoa/bookmarks/bookmark_bar_folder_controller.h" |
| 7 #import "chrome/browser/ui/cocoa/bookmarks/bookmark_bar_folder_view.h" |
| 8 #import "chrome/browser/ui/cocoa/bookmarks/bookmark_button.h" |
| 9 #import "chrome/browser/ui/cocoa/bookmarks/bookmark_folder_target.h" |
| 10 #import "chrome/browser/ui/cocoa/cocoa_test_helper.h" |
| 11 #import "chrome/browser/ui/cocoa/url_drop_target.h" |
| 12 #include "testing/gtest/include/gtest/gtest.h" |
| 13 #include "testing/platform_test.h" |
| 14 #import "third_party/mozilla/NSPasteboard+Utils.h" |
| 15 #import "third_party/ocmock/OCMock/OCMock.h" |
| 16 #import "third_party/ocmock/ocmock_extensions.h" |
| 17 |
| 18 // Allows us to verify BookmarkBarFolderView. |
| 19 @interface BookmarkBarFolderView(TestingAPI) |
| 20 |
| 21 @property(readonly, nonatomic) BOOL dropIndicatorShown; |
| 22 @property(readonly, nonatomic) CGFloat dropIndicatorPosition; |
| 23 |
| 24 -(void)setController:(id<BookmarkButtonControllerProtocol>)controller; |
| 25 |
| 26 @end |
| 27 |
| 28 @implementation BookmarkBarFolderView(TestingAPI) |
| 29 |
| 30 -(void)setController:(id<BookmarkButtonControllerProtocol>)controller { |
| 31 controller_ = controller; |
| 32 } |
| 33 |
| 34 -(BOOL)dropIndicatorShown { |
| 35 return dropIndicatorShown_; |
| 36 } |
| 37 |
| 38 -(CGFloat)dropIndicatorPosition { |
| 39 return dropIndicatorPosition_; |
| 40 } |
| 41 |
| 42 @end |
| 43 |
| 44 namespace { |
| 45 |
| 46 // Some values used for mocks and fakes. |
| 47 const CGFloat kFakeIndicatorPos = 7.0; |
| 48 const NSPoint kPoint = {10, 10}; |
| 49 |
| 50 class BookmarkBarFolderViewTest : public CocoaTest { |
| 51 public: |
| 52 virtual void SetUp() { |
| 53 CocoaTest::SetUp(); |
| 54 view_.reset([[BookmarkBarFolderView alloc] init]); |
| 55 mock_controller_.reset([GetMockController(YES) retain]); |
| 56 [view_ awakeFromNib]; |
| 57 [view_ setController:mock_controller_]; |
| 58 } |
| 59 |
| 60 virtual void TearDown() { |
| 61 [mock_controller_ verify]; |
| 62 CocoaTest::TearDown(); |
| 63 } |
| 64 |
| 65 id GetFakePasteboardForType(NSString* dataType) { |
| 66 id pasteboard = [OCMockObject mockForClass:[NSPasteboard class]]; |
| 67 [[[pasteboard stub] andReturn:[NSData data]] dataForType:dataType]; |
| 68 [[[pasteboard stub] andReturn:nil] dataForType:OCMOCK_ANY]; |
| 69 [[[pasteboard stub] andReturnBool:YES] containsURLData]; |
| 70 [[pasteboard stub] getURLs:[OCMArg setTo:nil] |
| 71 andTitles:[OCMArg setTo:nil] |
| 72 convertingFilenames:YES]; |
| 73 return pasteboard; |
| 74 } |
| 75 |
| 76 id GetFakeDragInfoForType(NSString* dataType) { |
| 77 // Need something non-nil to return as the draggingSource. |
| 78 id source = [OCMockObject mockForClass:[NSObject class]]; |
| 79 id drag_info = [OCMockObject mockForProtocol:@protocol(NSDraggingInfo)]; |
| 80 id pasteboard = GetFakePasteboardForType(dataType); |
| 81 [[[drag_info stub] andReturn:pasteboard] draggingPasteboard]; |
| 82 [[[drag_info stub] andReturnNSPoint:kPoint] draggingLocation]; |
| 83 [[[drag_info stub] andReturn:source] draggingSource]; |
| 84 [[[drag_info stub] |
| 85 andReturnUnsignedInteger:NSDragOperationCopy | NSDragOperationMove] |
| 86 draggingSourceOperationMask]; |
| 87 return drag_info; |
| 88 } |
| 89 |
| 90 id GetMockController(BOOL show_indicator) { |
| 91 id mock_controller |
| 92 = [OCMockObject mockForClass:[BookmarkBarFolderController class]]; |
| 93 [[[mock_controller stub] andReturnBool:YES] |
| 94 draggingAllowed:OCMOCK_ANY]; |
| 95 [[[mock_controller stub] andReturnBool:show_indicator] |
| 96 shouldShowIndicatorShownForPoint:kPoint]; |
| 97 [[[mock_controller stub] andReturnFloat:kFakeIndicatorPos] |
| 98 indicatorPosForDragToPoint:kPoint]; |
| 99 return mock_controller; |
| 100 } |
| 101 |
| 102 scoped_nsobject<id> mock_controller_; |
| 103 scoped_nsobject<BookmarkBarFolderView> view_; |
| 104 }; |
| 105 |
| 106 TEST_F(BookmarkBarFolderViewTest, BookmarkButtonDragAndDrop) { |
| 107 id drag_info = GetFakeDragInfoForType(kBookmarkButtonDragType); |
| 108 [[[mock_controller_ expect] andReturnUnsignedInteger:NSDragOperationNone] |
| 109 draggingEntered:drag_info]; |
| 110 [[[mock_controller_ expect] andReturnBool:NO] dragBookmarkData:drag_info]; |
| 111 [[[mock_controller_ expect] andReturnBool:YES] dragButton:OCMOCK_ANY |
| 112 to:kPoint |
| 113 copy:NO]; |
| 114 |
| 115 EXPECT_EQ([view_ draggingEntered:drag_info], NSDragOperationMove); |
| 116 EXPECT_TRUE([view_ performDragOperation:drag_info]); |
| 117 } |
| 118 |
| 119 TEST_F(BookmarkBarFolderViewTest, URLDragAndDrop) { |
| 120 NSArray* dragTypes = [URLDropTargetHandler handledDragTypes]; |
| 121 for (NSString* type in dragTypes) { |
| 122 id drag_info = GetFakeDragInfoForType(type); |
| 123 [[[mock_controller_ expect] andReturnUnsignedInteger:NSDragOperationNone] |
| 124 draggingEntered:drag_info]; |
| 125 [[[mock_controller_ expect] andReturnBool:NO] dragBookmarkData:drag_info]; |
| 126 [[[mock_controller_ expect] andReturnBool:YES] addURLs:OCMOCK_ANY |
| 127 withTitles:OCMOCK_ANY |
| 128 at:kPoint]; |
| 129 EXPECT_EQ([view_ draggingEntered:drag_info], NSDragOperationMove); |
| 130 EXPECT_TRUE([view_ performDragOperation:drag_info]); |
| 131 [mock_controller_ verify]; |
| 132 } |
| 133 } |
| 134 |
| 135 TEST_F(BookmarkBarFolderViewTest, BookmarkButtonDropIndicator) { |
| 136 id drag_info = GetFakeDragInfoForType(kBookmarkButtonDragType); |
| 137 [[[mock_controller_ expect] andReturnUnsignedInteger:NSDragOperationNone] |
| 138 draggingEntered:drag_info]; |
| 139 EXPECT_EQ([view_ draggingEntered:drag_info], NSDragOperationMove); |
| 140 [mock_controller_ verify]; |
| 141 EXPECT_TRUE([view_ dropIndicatorShown]); |
| 142 EXPECT_EQ([view_ dropIndicatorPosition], kFakeIndicatorPos); |
| 143 mock_controller_.reset([GetMockController(NO) retain]); |
| 144 [view_ setController:mock_controller_]; |
| 145 [[[mock_controller_ expect] andReturnUnsignedInteger:NSDragOperationNone] |
| 146 draggingEntered:drag_info]; |
| 147 EXPECT_EQ([view_ draggingEntered:drag_info], NSDragOperationMove); |
| 148 EXPECT_FALSE([view_ dropIndicatorShown]); |
| 149 } |
| 150 |
| 151 } // namespace |
OLD | NEW |