| // Copyright (c) 2011 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. |
| |
| #include "chrome/browser/ui/cocoa/history_menu_cocoa_controller.h" |
| |
| #include <memory> |
| |
| #include "base/mac/scoped_nsobject.h" |
| #include "base/strings/string_util.h" |
| #include "base/strings/sys_string_conversions.h" |
| #include "chrome/app/chrome_command_ids.h" |
| #include "chrome/browser/ui/cocoa/history_menu_bridge.h" |
| #include "chrome/browser/ui/cocoa/test/cocoa_profile_test.h" |
| #include "chrome/test/base/testing_profile.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| @interface FakeHistoryMenuController : HistoryMenuCocoaController { |
| @public |
| BOOL opened_[3]; |
| } |
| @end |
| |
| @implementation FakeHistoryMenuController |
| |
| - (id)initTest { |
| if ((self = [super init])) { |
| opened_[1] = NO; |
| opened_[2] = NO; |
| } |
| return self; |
| } |
| |
| - (void)openURLForItem:(const HistoryMenuBridge::HistoryItem*)item { |
| opened_[item->session_id.id()] = YES; |
| } |
| |
| @end // FakeHistoryMenuController |
| |
| class HistoryMenuCocoaControllerTest : public CocoaProfileTest { |
| public: |
| void SetUp() override { |
| CocoaProfileTest::SetUp(); |
| ASSERT_TRUE(profile()); |
| |
| bridge_.reset(new HistoryMenuBridge(profile())); |
| bridge_->controller_.reset( |
| [[FakeHistoryMenuController alloc] initWithBridge:bridge_.get()]); |
| [controller() initTest]; |
| } |
| |
| void CreateItems(NSMenu* menu) { |
| HistoryMenuBridge::HistoryItem* item = new HistoryMenuBridge::HistoryItem(); |
| item->url = GURL("http://google.com"); |
| item->session_id = SessionID::FromSerializedValue(1); |
| bridge_->AddItemToMenu(item, menu, HistoryMenuBridge::kVisited, 0); |
| |
| item = new HistoryMenuBridge::HistoryItem(); |
| item->url = GURL("http://apple.com"); |
| item->session_id = SessionID::FromSerializedValue(2); |
| bridge_->AddItemToMenu(item, menu, HistoryMenuBridge::kVisited, 1); |
| } |
| |
| std::map<NSMenuItem*, HistoryMenuBridge::HistoryItem*>& menu_item_map() { |
| return bridge_->menu_item_map_; |
| } |
| |
| FakeHistoryMenuController* controller() { |
| return static_cast<FakeHistoryMenuController*>(bridge_->controller_.get()); |
| } |
| |
| private: |
| std::unique_ptr<HistoryMenuBridge> bridge_; |
| }; |
| |
| TEST_F(HistoryMenuCocoaControllerTest, OpenURLForItem) { |
| base::scoped_nsobject<NSMenu> menu([[NSMenu alloc] initWithTitle:@"History"]); |
| CreateItems(menu.get()); |
| |
| std::map<NSMenuItem*, HistoryMenuBridge::HistoryItem*>& items = |
| menu_item_map(); |
| std::map<NSMenuItem*, HistoryMenuBridge::HistoryItem*>::iterator it = |
| items.begin(); |
| |
| for ( ; it != items.end(); ++it) { |
| HistoryMenuBridge::HistoryItem* item = it->second; |
| EXPECT_FALSE(controller()->opened_[item->session_id.id()]); |
| [controller() openHistoryMenuItem:it->first]; |
| EXPECT_TRUE(controller()->opened_[item->session_id.id()]); |
| } |
| } |