Implement bookmark editor (no tree yet)

chrome/browser/cocoa/bookmark_bar_controller_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 <Cocoa/Cocoa.h>
 
+#include "base/basictypes.h"
 #include "base/scoped_nsobject.h"
 #import "chrome/browser/cocoa/bookmark_bar_controller.h"
 #include "chrome/browser/cocoa/browser_test_helper.h"
 #import "chrome/browser/cocoa/cocoa_test_helper.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 // Pretend BookmarkURLOpener delegate to keep track of requests
 @interface BookmarkURLOpenerPong : NSObject<BookmarkURLOpener> {
  @public
-  GURL url_;
-  WindowOpenDisposition disposition_;
+  std::vector<GURL> urls_;
+  std::vector<WindowOpenDisposition> dispositions_;
 }
 @end
 
 @implementation BookmarkURLOpenerPong
 - (void)openBookmarkURL:(const GURL&)url
             disposition:(WindowOpenDisposition)disposition {
-  url_ = url;
-  disposition_ = disposition;
+  urls_.push_back(url);
+  dispositions_.push_back(disposition);
+}
+- (void)clear {
+  urls_.clear();
+  dispositions_.clear();
 }
 @end
 
 
 // NSCell that is pre-provided with a desired size that becomes the
 // return value for -(NSSize)cellSize:.
 @interface CellWithDesiredSize : NSCell {
  @private
   NSSize cellSize_;
 }
@@ skipping 40 matching lines @@
     [bar_ view];  // force loading of the nib
 
     // Awkwardness to look like we've been installed.
     [parent_view_ addSubview:[bar_ view]];
     NSRect frame = [[[bar_ view] superview] frame];
     frame.origin.y = 100;
     [[[bar_ view] superview] setFrame:frame];
 
     // make sure it's open so certain things aren't no-ops
     [bar_ toggleBookmarkBar];
+
+    // Create a menu/item to act like a sender
+    menu_.reset([[NSMenu alloc] initWithTitle:@"I_dont_care"]);
+    menu_item_.reset([[NSMenuItem alloc]
+                       initWithTitle:@"still_dont_care"
+                              action:NULL
+                       keyEquivalent:@""]);
+    cell_.reset([[NSButtonCell alloc] init]);
+    [menu_item_ setMenu:menu_.get()];
+    [menu_ setDelegate:cell_.get()];
   }
 
+  // Return a menu item that points to the right URL.
+  NSMenuItem* ItemForBookmarkBarMenu(GURL& gurl) {
+    node_.reset(new BookmarkNode(gurl));
+    [cell_ setRepresentedObject:[NSValue valueWithPointer:node_.get()]];
+    return menu_item_;
+  }
+
+  // Does NOT take ownership of node.
+  NSMenuItem* ItemForBookmarkBarMenu(const BookmarkNode* node) {
+    [cell_ setRepresentedObject:[NSValue valueWithPointer:node]];
+    return menu_item_;
+  }
+
+
   CocoaTestHelper cocoa_helper_;  // Inits Cocoa, creates window, etc...
   scoped_nsobject<NSView> content_area_;
   scoped_nsobject<NSView> infobar_view_;
   scoped_nsobject<NSView> parent_view_;
   BrowserTestHelper helper_;
   scoped_nsobject<BookmarkBarController> bar_;
+  scoped_nsobject<NSMenu> menu_;
+  scoped_nsobject<NSMenuItem> menu_item_;
+  scoped_nsobject<NSButtonCell> cell_;
+  scoped_ptr<BookmarkNode> node_;
 };
 
 TEST_F(BookmarkBarControllerTest, ShowHide) {
   // The test class opens the bar by default since many actions are
   // no-ops with it closed.  Set back to closed as a baseline.
   if ([bar_ isBookmarkBarVisible])
     [bar_ toggleBookmarkBar];
 
   // Start hidden.
   EXPECT_FALSE([bar_ isBookmarkBarVisible]);
@@ skipping 28 matching lines @@
   scoped_nsobject<BookmarkURLOpenerPong> pong([[BookmarkURLOpenerPong alloc]
                                                 init]);
   [bar_ setDelegate:pong.get()];
 
   scoped_nsobject<NSButtonCell> cell([[NSButtonCell alloc] init]);
   scoped_nsobject<NSButton> button([[NSButton alloc] init]);
   [button setCell:cell.get()];
   [cell setRepresentedObject:[NSValue valueWithPointer:node.get()]];
 
   [bar_ openBookmark:button];
-  EXPECT_EQ(pong.get()->url_, node->GetURL());
-  EXPECT_EQ(pong.get()->disposition_, CURRENT_TAB);
+  EXPECT_EQ(pong.get()->urls_[0], node->GetURL());
+  EXPECT_EQ(pong.get()->dispositions_[0], CURRENT_TAB);
 
   [bar_ setDelegate:nil];
 }
 
 // Confirm opening of bookmarks works from the menus (different
 // dispositions than clicking on the button).
 TEST_F(BookmarkBarControllerTest, OpenBookmarkFromMenus) {
   scoped_nsobject<BookmarkURLOpenerPong> pong([[BookmarkURLOpenerPong alloc]
                                                 init]);
   [bar_ setDelegate:pong.get()];
 
-  scoped_nsobject<NSMenu> menu([[NSMenu alloc] initWithTitle:@"I_dont_care"]);
-  scoped_nsobject<NSMenuItem> item([[NSMenuItem alloc]
-                                     initWithTitle:@"still_dont_care"
-                                            action:NULL
-                                     keyEquivalent:@""]);
-  scoped_nsobject<NSButtonCell> cell([[NSButtonCell alloc] init]);
-  [item setMenu:menu.get()];
-  [menu setDelegate:cell];
-
   const char* urls[] = { "http://walla.walla.ding.dong.com",
                          "http://i_dont_know.com",
                          "http://cee.enn.enn.dot.com" };
   SEL selectors[] = { @selector(openBookmarkInNewForegroundTab:),
                       @selector(openBookmarkInNewWindow:),
                       @selector(openBookmarkInIncognitoWindow:) };
   WindowOpenDisposition dispositions[] = { NEW_FOREGROUND_TAB,
                                            NEW_WINDOW,
                                            OFF_THE_RECORD };
   for (unsigned int i = 0;
        i < sizeof(dispositions)/sizeof(dispositions[0]);
        i++) {
-    scoped_ptr<BookmarkNode> node(new BookmarkNode(GURL(urls[i])));
-    [cell setRepresentedObject:[NSValue valueWithPointer:node.get()]];
-    [bar_ performSelector:selectors[i] withObject:item.get()];
-    EXPECT_EQ(pong.get()->url_, node->GetURL());
-    EXPECT_EQ(pong.get()->disposition_, dispositions[i]);
-    [cell setRepresentedObject:nil];
+    GURL gurl(urls[i]);
+    [bar_ performSelector:selectors[i]
+               withObject:ItemForBookmarkBarMenu(gurl)];
+    EXPECT_EQ(pong.get()->urls_[0], gurl);
+    EXPECT_EQ(pong.get()->dispositions_[0], dispositions[i]);
+    [pong clear];
   }
+  [bar_ setDelegate:nil];
 }
 
 TEST_F(BookmarkBarControllerTest, TestAddRemoveAndClear) {
   BookmarkModel* model = helper_.profile()->GetBookmarkModel();
 
   EXPECT_EQ(0U, [[bar_ buttons] count]);
   unsigned int initial_subview_count = [[[bar_ view] subviews] count];
 
   // Make sure a redundant call doesn't choke
   [bar_ clearBookmarkBar];
@@ skipping 78 matching lines @@
 
   // Make sure the 1st button is now wider, the 2nd one is moved over,
   // and they don't overlap.
   NSRect frame_1 = [[[bar_ buttons] objectAtIndex:0] frame];
   NSRect frame_2 = [[[bar_ buttons] objectAtIndex:1] frame];
   EXPECT_GT(frame_1.size.width, width_1);
   EXPECT_GT(frame_2.origin.x, x_2);
   EXPECT_GE(frame_2.origin.x, frame_1.origin.x + frame_1.size.width);
 }
 
+TEST_F(BookmarkBarControllerTest, DeleteBookmark) {
+  BookmarkModel* model = helper_.profile()->GetBookmarkModel();
+
+  const char* urls[] = { "https://secret.url.com",
+                         "http://super.duper.web.site.for.doodz.gov",
+                         "http://www.foo-bar-baz.com/" };
+  const BookmarkNode* parent = model->GetBookmarkBarNode();
+  for (unsigned int i = 0; i < arraysize(urls); i++) {
+    model->AddURL(parent, parent->GetChildCount(),
+                  L"title", GURL(urls[i]));
+  }
+  EXPECT_EQ(3, parent->GetChildCount());
+  const BookmarkNode* middle_node = parent->GetChild(1);
+
+  NSMenuItem* item = ItemForBookmarkBarMenu(middle_node);
+  [bar_ deleteBookmark:item];
+  EXPECT_EQ(2, parent->GetChildCount());
+  EXPECT_EQ(parent->GetChild(0)->GetURL(), GURL(urls[0]));
+  // node 2 moved into spot 1
+  EXPECT_EQ(parent->GetChild(1)->GetURL(), GURL(urls[2]));
+}
+
+TEST_F(BookmarkBarControllerTest, OpenAllBookmarks) {
+  scoped_nsobject<BookmarkURLOpenerPong> pong([[BookmarkURLOpenerPong alloc]
+                                                init]);
+  [bar_ setDelegate:pong.get()];
+
+  BookmarkModel* model = helper_.profile()->GetBookmarkModel();
+  const BookmarkNode* parent = model->GetBookmarkBarNode();
+  // { one, { two-one, two-two }, three }
+  model->AddURL(parent, parent->GetChildCount(),
+                L"title", GURL("http://one.com"));
+  const BookmarkNode* folder = model->AddGroup(parent,
+                                               parent->GetChildCount(),
+                                               L"group");
+  model->AddURL(folder, folder->GetChildCount(),
+                L"title", GURL("http://two-one.com"));
+  model->AddURL(folder, folder->GetChildCount(),
+                L"title", GURL("http://two-two.com"));
+  model->AddURL(parent, parent->GetChildCount(),
+                L"title", GURL("https://three.com"));
+  [bar_ openAllBookmarks:nil];
+
+  EXPECT_EQ(pong.get()->urls_.size(), 4U);
+  EXPECT_EQ(pong.get()->dispositions_.size(), 4U);
+
+  // I can't use EXPECT_EQ() here since the macro can't expand
+  // properly (no way to print the value of an iterator).
+  std::vector<GURL>::iterator i;
+  std::vector<GURL>::iterator begin = pong.get()->urls_.begin();
+  std::vector<GURL>::iterator end = pong.get()->urls_.end();
+  i = find(begin, end, GURL("http://two-one.com"));
+  EXPECT_FALSE(i == end);
+  i = find(begin, end, GURL("https://three.com"));
+  EXPECT_FALSE(i == end);
+  i = find(begin, end, GURL("https://will-not-be-found.com"));
+  EXPECT_TRUE(i == end);
+
+  EXPECT_EQ(pong.get()->dispositions_[3], NEW_BACKGROUND_TAB);
+
+  [bar_ setDelegate:nil];
+}
+
 // TODO(jrg): write a test to confirm that nodeFavIconLoaded calls
 // checkForBookmarkButtonGrowth:.
 
 // TODO(jrg): Make sure showing the bookmark bar calls loaded: (to
 // process bookmarks)
 TEST_F(BookmarkBarControllerTest, ShowAndLoad) {
 }
 
 // TODO(jrg): Test cellForBookmarkNode:
 TEST_F(BookmarkBarControllerTest, Cell) {
 }
 
 TEST_F(BookmarkBarControllerTest, Contents) {
   // TODO(jrg): addNodesToBar has a LOT of TODOs; when flushed out, write
   // appropriate tests.
 }
 
 // Test drawing, mostly to ensure nothing leaks or crashes.
 TEST_F(BookmarkBarControllerTest, Display) {
   [[bar_ view] display];
 }
 
+// Cannot test these methods since they simply call a single static
+// method, BookmarkEditor::Show(), which is impossible to mock.
+// editBookmark:, addPage:
+
+
 }  // namespace