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
|
// 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_model_test_utils.h"
#include <unordered_map>
#include "base/strings/string_number_conversions.h"
#include "chrome/browser/ui/tabs/tab_strip_model.h"
#include "content/public/browser/web_contents.h"
void SetID(WebContents* contents, int id) {
contents->SetUserData(&kTabStripModelTestIDUserDataKey,
std::make_unique<TabStripModelTestIDUserData>(id));
}
int GetID(WebContents* contents) {
TabStripModelTestIDUserData* user_data =
static_cast<TabStripModelTestIDUserData*>(
contents->GetUserData(&kTabStripModelTestIDUserDataKey));
return user_data ? user_data->id() : -1;
}
void PrepareTabstripForSelectionTest(
base::OnceCallback<void(int)> add_tabs_callback,
TabStripModel* model,
int tab_count,
int pinned_count,
const std::vector<int> selected_tabs) {
std::move(add_tabs_callback).Run(tab_count);
for (int i = 0; i < pinned_count; ++i) {
model->SetTabPinned(i, true);
}
ui::ListSelectionModel selection_model;
for (const int selected_tab : selected_tabs) {
selection_model.AddIndexToSelection(selected_tab);
}
selection_model.set_active(*selection_model.selected_indices().begin());
model->SetSelectionFromModel(selection_model);
}
std::string GetTabStripStateString(const TabStripModel* model,
bool annotate_groups) {
std::unordered_map<tab_groups::TabGroupId, size_t, tab_groups::TabGroupIdHash>
groups;
std::string actual;
for (int i = 0; i < model->count(); ++i) {
if (i > 0) {
actual += " ";
}
actual += base::NumberToString(GetID(model->GetWebContentsAt(i)));
if (model->IsTabPinned(i)) {
actual += "p";
}
std::optional<tab_groups::TabGroupId> group = model->GetTabGroupForTab(i);
if (annotate_groups && group.has_value()) {
if (!groups.contains(group.value())) {
groups.emplace(group.value(), groups.size());
}
actual += "g" + base::NumberToString(groups[group.value()]);
}
if (model->GetSplitForTab(i).has_value()) {
actual += "s";
}
}
return actual;
}
|
