chrome/browser/ui/cocoa/bookmarks/bookmark_bar_bridge_unittest.mm

   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/bookmarks/bookmark_model_factory.h"
   6 #include "chrome/browser/ui/cocoa/bookmarks/bookmark_bar_bridge.h"
   7 #include "chrome/browser/ui/cocoa/bookmarks/bookmark_bar_controller.h"
   8 #include "chrome/browser/ui/cocoa/cocoa_profile_test.h"
   9 #include "chrome/test/base/testing_profile.h"
  10 #include "testing/gtest/include/gtest/gtest.h"
  11 #import "testing/gtest_mac.h"
  12 #include "testing/platform_test.h"
  13 #include "url/gurl.h"
  14
  15 using bookmarks::BookmarkModel;
  16 using bookmarks::BookmarkNode;
  17
  18 // TODO(jrg): use OCMock.
  19
  20 namespace {
  21
  22 // Information needed to open a URL, as passed to the
  23 // BookmarkBarController's delegate.
  24 typedef std::pair<GURL,WindowOpenDisposition> OpenInfo;
  25
  26 }  // The namespace must end here -- I need to use OpenInfo in
  27    // FakeBookmarkBarController but can't place
  28    // FakeBookmarkBarController itself in the namespace ("error:
  29    // Objective-C declarations may only appear in global scope")
  30
  31 // Oddly, we are our own delegate.
  32 @interface FakeBookmarkBarController : BookmarkBarController {
  33  @public
  34   base::scoped_nsobject<NSMutableArray> callbacks_;
  35   std::vector<OpenInfo> opens_;
  36 }
  37 @end
  38
  39 @implementation FakeBookmarkBarController
  40
  41 - (id)initWithBrowser:(Browser*)browser {
  42   if ((self = [super initWithBrowser:browser
  43                         initialWidth:100  // arbitrary
  44                             delegate:nil
  45                       resizeDelegate:nil])) {
  46     callbacks_.reset([[NSMutableArray alloc] init]);
  47   }
  48   return self;
  49 }
  50
  51 - (void)loaded:(BookmarkModel*)model {
  52   [callbacks_ addObject:[NSNumber numberWithInt:0]];
  53 }
  54
  55 - (void)beingDeleted:(BookmarkModel*)model {
  56   [callbacks_ addObject:[NSNumber numberWithInt:1]];
  57 }
  58
  59 - (void)nodeMoved:(BookmarkModel*)model
  60         oldParent:(const BookmarkNode*)oldParent oldIndex:(int)oldIndex
  61         newParent:(const BookmarkNode*)newParent newIndex:(int)newIndex {
  62   [callbacks_ addObject:[NSNumber numberWithInt:2]];
  63 }
  64
  65 - (void)nodeAdded:(BookmarkModel*)model
  66            parent:(const BookmarkNode*)oldParent index:(int)index {
  67   [callbacks_ addObject:[NSNumber numberWithInt:3]];
  68 }
  69
  70 - (void)nodeChanged:(BookmarkModel*)model
  71                node:(const BookmarkNode*)node {
  72   [callbacks_ addObject:[NSNumber numberWithInt:4]];
  73 }
  74
  75 - (void)nodeFaviconLoaded:(BookmarkModel*)model
  76                      node:(const BookmarkNode*)node {
  77   [callbacks_ addObject:[NSNumber numberWithInt:5]];
  78 }
  79
  80 - (void)nodeChildrenReordered:(BookmarkModel*)model
  81                          node:(const BookmarkNode*)node {
  82   [callbacks_ addObject:[NSNumber numberWithInt:6]];
  83 }
  84
  85 - (void)nodeRemoved:(BookmarkModel*)model
  86              parent:(const BookmarkNode*)oldParent index:(int)index {
  87   [callbacks_ addObject:[NSNumber numberWithInt:7]];
  88 }
  89
  90 // Save the request.
  91 - (void)openBookmarkURL:(const GURL&)url
  92             disposition:(WindowOpenDisposition)disposition {
  93   opens_.push_back(OpenInfo(url, disposition));
  94 }
  95
  96 @end
  97
  98
  99 class BookmarkBarBridgeTest : public CocoaProfileTest {
 100 };
 101
 102 // Call all the callbacks; make sure they are all redirected to the objc object.
 103 TEST_F(BookmarkBarBridgeTest, TestRedirect) {
 104   BookmarkModel* model = BookmarkModelFactory::GetForProfile(profile());
 105
 106   base::scoped_nsobject<NSView> parentView(
 107       [[NSView alloc] initWithFrame:NSMakeRect(0, 0, 100, 100)]);
 108   base::scoped_nsobject<NSView> webView(
 109       [[NSView alloc] initWithFrame:NSMakeRect(0, 0, 100, 100)]);
 110   base::scoped_nsobject<NSView> infoBarsView(
 111       [[NSView alloc] initWithFrame:NSMakeRect(0, 0, 100, 100)]);
 112
 113   base::scoped_nsobject<FakeBookmarkBarController> controller(
 114       [[FakeBookmarkBarController alloc] initWithBrowser:browser()]);
 115   EXPECT_TRUE(controller.get());
 116   scoped_ptr<BookmarkBarBridge> bridge(new BookmarkBarBridge(profile(),
 117                                                              controller.get(),
 118                                                              model));
 119   EXPECT_TRUE(bridge.get());
 120
 121   bridge->BookmarkModelLoaded(NULL, false);
 122   bridge->BookmarkModelBeingDeleted(NULL);
 123   bridge->BookmarkNodeMoved(NULL, NULL, 0, NULL, 0);
 124   bridge->BookmarkNodeAdded(NULL, NULL, 0);
 125   bridge->BookmarkNodeChanged(NULL, NULL);
 126   bridge->BookmarkNodeFaviconChanged(NULL, NULL);
 127   bridge->BookmarkNodeChildrenReordered(NULL, NULL);
 128   bridge->BookmarkNodeRemoved(NULL, NULL, 0, NULL, std::set<GURL>());
 129
 130   // 8 calls above plus an initial Loaded() in init routine makes 9
 131   EXPECT_TRUE([controller.get()->callbacks_ count] == 9);
 132
 133   for (int x = 1; x < 9; x++) {
 134     NSNumber* num = [NSNumber numberWithInt:x-1];
 135     EXPECT_NSEQ(num, [controller.get()->callbacks_ objectAtIndex:x]);
 136   }
 137 }