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
|
// 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/optimization_guide/prediction/chrome_profile_download_service_tracker.h"
#include "base/test/test_file_util.h"
#include "build/build_config.h"
#include "chrome/browser/download/background_download_service_factory.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/profiles/profile_key.h"
#include "chrome/test/base/fake_profile_manager.h"
#include "chrome/test/base/testing_browser_process.h"
#include "chrome/test/base/testing_profile_manager.h"
#include "content/public/test/browser_task_environment.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace {
constexpr char kProfileFoo[] = "foo";
constexpr char kProfileBar[] = "bar";
download::BackgroundDownloadService* GetBackgroundDownloadServiceForProfile(
Profile* profile) {
return BackgroundDownloadServiceFactory::GetForKey(profile->GetProfileKey());
}
} // namespace
namespace optimization_guide {
class ChromeProfileDownloadServiceTrackerTest : public testing::Test {
public:
ChromeProfileDownloadServiceTrackerTest()
: profile_manager_(TestingBrowserProcess::GetGlobal()) {}
ChromeProfileDownloadServiceTrackerTest(
const ChromeProfileDownloadServiceTrackerTest&) = delete;
ChromeProfileDownloadServiceTrackerTest& operator=(
const ChromeProfileDownloadServiceTrackerTest&) = delete;
void SetUp() override { ASSERT_TRUE(profile_manager_.SetUp()); }
protected:
content::BrowserTaskEnvironment task_environment_;
TestingProfileManager profile_manager_;
};
TEST_F(ChromeProfileDownloadServiceTrackerTest, OneProfile) {
auto service_tracker =
std::make_unique<ChromeProfileDownloadServiceTracker>();
auto* foo_profile = profile_manager_.CreateTestingProfile(kProfileFoo);
base::RunLoop().RunUntilIdle();
EXPECT_EQ(service_tracker->GetBackgroundDownloadService(),
GetBackgroundDownloadServiceForProfile(foo_profile));
}
// ChromeOS does not support deletion of profiles. So, skip the tests involving
// it.
#if BUILDFLAG(IS_CHROMEOS)
#define MAYBE_TwoProfiles DISABLED_TwoProfiles
#else
#define MAYBE_TwoProfiles TwoProfiles
#endif
TEST_F(ChromeProfileDownloadServiceTrackerTest, MAYBE_TwoProfiles) {
auto service_tracker =
std::make_unique<ChromeProfileDownloadServiceTracker>();
auto* foo_profile = profile_manager_.CreateTestingProfile(kProfileFoo);
auto* bar_profile = profile_manager_.CreateTestingProfile(kProfileBar);
base::RunLoop().RunUntilIdle();
EXPECT_EQ(service_tracker->GetBackgroundDownloadService(),
GetBackgroundDownloadServiceForProfile(foo_profile));
// When foo profile is deleted, the download manager should be picked from
// bar profile.
profile_manager_.DeleteTestingProfile(kProfileFoo);
base::RunLoop().RunUntilIdle();
EXPECT_EQ(service_tracker->GetBackgroundDownloadService(),
GetBackgroundDownloadServiceForProfile(bar_profile));
// When another profile is created, its still picked from the bar profile
profile_manager_.CreateTestingProfile(kProfileFoo);
base::RunLoop().RunUntilIdle();
EXPECT_EQ(service_tracker->GetBackgroundDownloadService(),
GetBackgroundDownloadServiceForProfile(bar_profile));
}
TEST_F(ChromeProfileDownloadServiceTrackerTest,
ServiceTrackerCreatedAfterProfile) {
auto* foo_profile = profile_manager_.CreateTestingProfile(kProfileFoo);
base::RunLoop().RunUntilIdle();
auto service_tracker =
std::make_unique<ChromeProfileDownloadServiceTracker>();
EXPECT_EQ(service_tracker->GetBackgroundDownloadService(),
GetBackgroundDownloadServiceForProfile(foo_profile));
}
} // namespace optimization_guide
|
