// Copyright (c) 2010 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 "chrome/browser/ui/cocoa/bookmarks/bookmark_bar_unittest_helper.h"

@interface NSArray (BookmarkBarUnitTestHelper)

// A helper function for scanning an array of buttons looking for the
// button with the given |title|.
- (BookmarkButton*)buttonWithTitleEqualTo:(NSString*)title;

@end


@implementation NSArray (BookmarkBarUnitTestHelper)

- (BookmarkButton*)buttonWithTitleEqualTo:(NSString*)title {
  for (BookmarkButton* button in self) {
    if ([[button title] isEqualToString:title])
      return button;
  }
  return nil;
}

@end

@implementation BookmarkBarController (BookmarkBarUnitTestHelper)

- (BookmarkButton*)buttonWithTitleEqualTo:(NSString*)title {
  return [[self buttons] buttonWithTitleEqualTo:title];
}

@end

@implementation BookmarkBarFolderController(BookmarkBarUnitTestHelper)

- (BookmarkButton*)buttonWithTitleEqualTo:(NSString*)title {
  return [[self buttons] buttonWithTitleEqualTo:title];
}

@end

@implementation BookmarkButton(BookmarkBarUnitTestHelper)

- (NSPoint)center {
  NSRect frame = [self frame];
  NSPoint center = NSMakePoint(NSMidX(frame), NSMidY(frame));
  center = [[self superview] convertPoint:center toView:nil];
  return center;
}

- (NSPoint)top {
  NSRect frame = [self frame];
  NSPoint top = NSMakePoint(NSMidX(frame), NSMaxY(frame));
  top = [[self superview] convertPoint:top toView:nil];
  return top;
}

- (NSPoint)bottom {
  NSRect frame = [self frame];
  NSPoint bottom = NSMakePoint(NSMidX(frame), NSMinY(frame));
  bottom = [[self superview] convertPoint:bottom toView:nil];
  return bottom;
}

- (NSPoint)left {
  NSRect frame = [self frame];
  NSPoint left = NSMakePoint(NSMinX(frame), NSMidY(frame));
  left = [[self superview] convertPoint:left toView:nil];
  return left;
}

- (NSPoint)right {
  NSRect frame = [self frame];
  NSPoint right = NSMakePoint(NSMaxX(frame), NSMidY(frame));
  right = [[self superview] convertPoint:right toView:nil];
  return right;
}

@end