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
|
// Copyright 2024 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/ash/birch/birch_test_util.h"
#include "ash/birch/birch_item_remover.h"
#include "ash/birch/birch_model.h"
#include "ash/constants/ash_pref_names.h"
#include "ash/session/session_controller_impl.h"
#include "ash/shell.h"
#include "ash/wm/overview/birch/birch_chip_button_base.h"
#include "ash/wm/overview/overview_grid_test_api.h"
#include "base/run_loop.h"
#include "components/prefs/pref_service.h"
namespace ash {
void EnsureItemRemoverInitialized() {
BirchItemRemover* remover =
Shell::Get()->birch_model()->GetItemRemoverForTest();
if (!remover->Initialized()) {
base::RunLoop run_loop;
remover->SetProtoInitCallbackForTest(run_loop.QuitClosure());
run_loop.Run();
}
}
BirchChipButtonBase* GetBirchChipButton() {
aura::Window* root = Shell::GetPrimaryRootWindow();
OverviewGridTestApi test_api(root);
CHECK(test_api.birch_bar_view());
if (!test_api.GetBirchChips().size()) {
return nullptr;
}
return test_api.GetBirchChips()[0];
}
size_t GetBirchChipsNum() {
aura::Window* root = Shell::GetPrimaryRootWindow();
return OverviewGridTestApi(root).GetBirchChips().size();
}
void DisableAllDataTypePrefsExcept(std::vector<std::string_view> exceptions) {
PrefService* pref_service =
Shell::Get()->session_controller()->GetPrimaryUserPrefService();
CHECK(pref_service);
const std::string_view kDataPrefs[] = {
prefs::kBirchUseCalendar, prefs::kBirchUseFileSuggest,
prefs::kBirchUseChromeTabs, prefs::kBirchUseLostMedia,
prefs::kBirchUseReleaseNotes, prefs::kBirchUseWeather,
prefs::kBirchUseCoral,
};
for (const std::string_view pref : kDataPrefs) {
const bool enable = base::Contains(exceptions, pref);
pref_service->SetBoolean(pref, enable);
}
}
} // namespace ash
|
