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
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
|
// Copyright 2025 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/browser/ui/tabs/tab_strip_api/testing/toy_tab_strip.h"
#include "base/logging.h"
namespace tabs_api::testing {
void ToyTabStrip::AddTab(ToyTab tab) {
root_.tabs.push_back(tab);
}
std::vector<tabs::TabHandle> ToyTabStrip::GetTabs() {
std::vector<tabs::TabHandle> result;
for (auto& tab : root_.tabs) {
result.push_back(tab.tab_handle);
}
return result;
}
void ToyTabStrip::CloseTab(size_t index) {
if (index < 0 || index >= root_.tabs.size()) {
LOG(FATAL) << "invalid idx passed in: " << index
<< ", tab size is: " << root_.tabs.size();
}
root_.tabs.erase(root_.tabs.begin() + index);
}
std::optional<int> ToyTabStrip::GetIndexForHandle(tabs::TabHandle tab_handle) {
for (size_t i = 0; i < root_.tabs.size(); ++i) {
if (root_.tabs.at(i).tab_handle == tab_handle) {
return i;
}
}
return std::nullopt;
}
tabs::TabHandle ToyTabStrip::AddTabAt(const GURL& url,
std::optional<int> index) {
auto tab = ToyTab{
tabs::TabHandle(GetNextId()),
url,
};
if (index.has_value()) {
root_.tabs.insert(root_.tabs.begin() + index.value(), tab);
} else {
root_.tabs.push_back(tab);
}
return tab.tab_handle;
}
void ToyTabStrip::ActivateTab(tabs::TabHandle handle) {
for (auto& tab : root_.tabs) {
tab.active = tab.tab_handle == handle;
}
}
tabs::TabHandle ToyTabStrip::FindActiveTab() {
for (auto& tab : root_.tabs) {
if (tab.active) {
return tab.tab_handle;
}
}
NOTREACHED() << "toy tab strip does not guarantee one tab is always active, "
"did you forget to activate a tab beforehand?";
}
void ToyTabStrip::MoveTab(tabs::TabHandle handle, size_t to) {
auto idx = GetIndexForHandle(handle).value();
auto tab = root_.tabs.at(idx);
root_.tabs.erase(root_.tabs.begin() + idx);
root_.tabs.insert(root_.tabs.begin() + to, tab);
}
int ToyTabStrip::GetNextId() {
static int id = 0;
return id++;
}
std::optional<tab_groups::TabGroupId> ToyTabStrip::GetGroupIdFor(
const tabs::TabCollectionHandle& handle) const {
for (const auto& group : groups_with_visuals_) {
if (group.handle == handle) {
return group.id;
}
}
return std::nullopt;
}
tabs::TabCollectionHandle ToyTabStrip::AddGroup(
const tab_groups::TabGroupVisualData& visual_data) {
ToyTabGroupData new_group{tab_groups::TabGroupId::GenerateNew(),
tabs::TabCollectionHandle(GetNextId()),
visual_data};
auto handle = new_group.handle;
groups_with_visuals_.push_back(std::move(new_group));
return handle;
}
const tab_groups::TabGroupVisualData* ToyTabStrip::GetGroupVisualData(
const tabs::TabCollectionHandle& handle) const {
for (const auto& group : groups_with_visuals_) {
if (group.handle == handle) {
return &group.visuals;
}
}
return nullptr;
}
void ToyTabStrip::UpdateGroupVisuals(
const tab_groups::TabGroupId& group_id,
const tab_groups::TabGroupVisualData& new_visuals) {
for (auto& group : groups_with_visuals_) {
if (group.id == group_id) {
group.visuals = new_visuals;
return;
}
}
}
} // namespace tabs_api::testing
|
