1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
|
// Copyright 2011 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifdef UNSAFE_BUFFERS_BUILD
// TODO(crbug.com/40285824): Remove this and convert code to safer constructs.
#pragma allow_unsafe_buffers
#endif
#include "chrome/browser/ui/cocoa/history_menu_cocoa_controller.h"
#include <memory>
#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_test_helper.h"
#include "chrome/test/base/browser_with_test_window_test.h"
#include "chrome/test/base/testing_profile.h"
#include "testing/gtest/include/gtest/gtest.h"
@interface FakeHistoryMenuController : HistoryMenuCocoaController {
@public
// ivars are initialized to zero, so these all start out as NO.
BOOL _opened[3];
}
@end
@implementation FakeHistoryMenuController
- (void)openURLForItem:(const HistoryMenuBridge::HistoryItem*)item {
_opened[item->session_id.id()] = YES;
}
@end // FakeHistoryMenuController
class HistoryMenuCocoaControllerTest : public BrowserWithTestWindowTest {
public:
void SetUp() override {
BrowserWithTestWindowTest::SetUp();
ASSERT_TRUE(profile());
bridge_ = std::make_unique<HistoryMenuBridge>(profile());
bridge_->controller_ =
[[FakeHistoryMenuController alloc] initWithBridge:bridge_.get()];
}
void TearDown() override {
bridge_.reset();
BrowserWithTestWindowTest::TearDown();
}
void CreateItems(NSMenu* menu) {
auto item = std::make_unique<HistoryMenuBridge::HistoryItem>();
item->url = GURL("http://google.com");
item->session_id = SessionID::FromSerializedValue(1);
bridge_->AddItemToMenu(std::move(item), menu, HistoryMenuBridge::kVisited,
0);
item = std::make_unique<HistoryMenuBridge::HistoryItem>();
item->url = GURL("http://apple.com");
item->session_id = SessionID::FromSerializedValue(2);
bridge_->AddItemToMenu(std::move(item), menu, HistoryMenuBridge::kVisited,
1);
}
std::map<NSMenuItem*, std::unique_ptr<HistoryMenuBridge::HistoryItem>>&
menu_item_map() {
return bridge_->menu_item_map_;
}
FakeHistoryMenuController* controller() {
return static_cast<FakeHistoryMenuController*>(bridge_->controller_);
}
private:
CocoaTestHelper cocoa_test_helper_;
std::unique_ptr<HistoryMenuBridge> bridge_;
};
TEST_F(HistoryMenuCocoaControllerTest, OpenURLForItem) {
NSMenu* menu = [[NSMenu alloc] initWithTitle:@"History"];
CreateItems(menu);
std::map<NSMenuItem*, std::unique_ptr<HistoryMenuBridge::HistoryItem>>&
items = menu_item_map();
for (const auto& pair : items) {
HistoryMenuBridge::HistoryItem* item = pair.second.get();
EXPECT_FALSE(controller()->_opened[item->session_id.id()]);
[controller() openHistoryMenuItem:pair.first];
EXPECT_TRUE(controller()->_opened[item->session_id.id()]);
}
}
|
