| 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 |