blob: 690b8c8b64a1843a4b568409803571b0707cee5b [file] [log] [blame]
// 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_[2];
}
@end
@implementation FakeHistoryMenuController
- (id)initTest {
if ((self = [super init])) {
opened_[0] = NO;
opened_[1] = NO;
}
return self;
}
- (void)openURLForItem:(const HistoryMenuBridge::HistoryItem*)item {
opened_[item->session_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 = 0;
bridge_->AddItemToMenu(item, menu, HistoryMenuBridge::kVisited, 0);
item = new HistoryMenuBridge::HistoryItem();
item->url = GURL("http://apple.com");
item->session_id = 1;
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]);
[controller() openHistoryMenuItem:it->first];
EXPECT_TRUE(controller()->opened_[item->session_id]);
}
}