src/chrome/browser/ui/cocoa/applescript/bookmark_applescript_utils_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 #import "chrome/browser/ui/cocoa/applescript/bookmark_applescript_utils_unittest.h"
   6
   7 #include "chrome/browser/bookmarks/bookmark_model_factory.h"
   8 #include "chrome/test/base/testing_profile.h"
   9 #include "components/bookmarks/core/browser/bookmark_model.h"
  10 #include "components/bookmarks/core/test/bookmark_test_helpers.h"
  11
  12 @implementation FakeAppDelegate
  13
  14 @synthesize test = test_;
  15
  16 - (Profile*)lastProfile {
  17   if (!test_)
  18     return NULL;
  19   return test_->profile();
  20 }
  21 @end
  22
  23 // Represents the current fake command that is executing.
  24 static FakeScriptCommand* kFakeCurrentCommand;
  25
  26 @implementation FakeScriptCommand
  27
  28 - (id)init {
  29   if ((self = [super init])) {
  30     originalMethod_ = class_getClassMethod([NSScriptCommand class],
  31                                            @selector(currentCommand));
  32     alternateMethod_ = class_getClassMethod([self class],
  33                                             @selector(currentCommand));
  34     method_exchangeImplementations(originalMethod_, alternateMethod_);
  35     kFakeCurrentCommand = self;
  36   }
  37   return self;
  38 }
  39
  40 + (NSScriptCommand*)currentCommand {
  41   return kFakeCurrentCommand;
  42 }
  43
  44 - (void)dealloc {
  45   method_exchangeImplementations(originalMethod_, alternateMethod_);
  46   kFakeCurrentCommand = nil;
  47   [super dealloc];
  48 }
  49
  50 @end
  51
  52 BookmarkAppleScriptTest::BookmarkAppleScriptTest() {
  53 }
  54
  55 BookmarkAppleScriptTest::~BookmarkAppleScriptTest() {
  56   [NSApp setDelegate:nil];
  57 }
  58
  59 void BookmarkAppleScriptTest::SetUp() {
  60   CocoaProfileTest::SetUp();
  61   ASSERT_TRUE(profile());
  62
  63   appDelegate_.reset([[FakeAppDelegate alloc] init]);
  64   [appDelegate_.get() setTest:this];
  65   DCHECK([NSApp delegate] == nil);
  66   [NSApp setDelegate:appDelegate_];
  67   BookmarkModel* model = BookmarkModelFactory::GetForProfile(profile());
  68   const BookmarkNode* root = model->bookmark_bar_node();
  69   const std::string modelString("a f1:[ b d c ] d f2:[ e f g ] h ");
  70   test::AddNodesFromModelString(model, root, modelString);
  71   bookmarkBar_.reset([[BookmarkFolderAppleScript alloc]
  72       initWithBookmarkNode:model->bookmark_bar_node()]);
  73 }