1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "chrome/test/base/testing_profile_manager.h"
6
7 #include "base/memory/ref_counted.h"
8 #include "base/strings/utf_string_conversions.h"
9 #include "chrome/browser/extensions/extension_special_storage_policy.h"
10 #include "chrome/browser/prefs/pref_service_syncable.h"
11 #include "chrome/browser/profiles/profile_info_cache.h"
12 #include "chrome/browser/profiles/profile_manager.h"
13 #include "chrome/common/chrome_constants.h"
14 #include "chrome/test/base/testing_browser_process.h"
15 #include "testing/gtest/include/gtest/gtest.h"
16
17 #if defined(OS_CHROMEOS)
18 #include "chrome/browser/chromeos/profiles/profile_helper.h"
19 #endif
20
21 const std::string kGuestProfileName = "Guest";
22
23 namespace testing {
24
25 class ProfileManager : public ::ProfileManagerWithoutInit {
26 public:
ProfileManager(const base::FilePath & user_data_dir)27 explicit ProfileManager(const base::FilePath& user_data_dir)
28 : ::ProfileManagerWithoutInit(user_data_dir) {}
29
30 protected:
CreateProfileHelper(const base::FilePath & file_path)31 virtual Profile* CreateProfileHelper(
32 const base::FilePath& file_path) OVERRIDE {
33 return new TestingProfile(file_path);
34 }
35 };
36
37 } // namespace testing
38
TestingProfileManager(TestingBrowserProcess * process)39 TestingProfileManager::TestingProfileManager(TestingBrowserProcess* process)
40 : called_set_up_(false),
41 browser_process_(process),
42 local_state_(process),
43 profile_manager_(NULL) {
44 }
45
~TestingProfileManager()46 TestingProfileManager::~TestingProfileManager() {
47 // Destroying this class also destroys the LocalState, so make sure the
48 // associated ProfileManager is also destroyed.
49 browser_process_->SetProfileManager(NULL);
50 }
51
SetUp()52 bool TestingProfileManager::SetUp() {
53 SetUpInternal();
54 return called_set_up_;
55 }
56
CreateTestingProfile(const std::string & profile_name,scoped_ptr<PrefServiceSyncable> prefs,const base::string16 & user_name,int avatar_id,const std::string & supervised_user_id,const TestingProfile::TestingFactories & factories)57 TestingProfile* TestingProfileManager::CreateTestingProfile(
58 const std::string& profile_name,
59 scoped_ptr<PrefServiceSyncable> prefs,
60 const base::string16& user_name,
61 int avatar_id,
62 const std::string& supervised_user_id,
63 const TestingProfile::TestingFactories& factories) {
64 DCHECK(called_set_up_);
65
66 // Create a path for the profile based on the name.
67 base::FilePath profile_path(profiles_dir_.path());
68 #if defined(OS_CHROMEOS)
69 if (profile_name != chrome::kInitialProfile) {
70 profile_path =
71 profile_path.Append(chromeos::ProfileHelper::Get()->GetUserProfileDir(
72 chromeos::ProfileHelper::GetUserIdHashByUserIdForTesting(
73 profile_name)));
74 } else {
75 profile_path = profile_path.AppendASCII(profile_name);
76 }
77 #else
78 profile_path = profile_path.AppendASCII(profile_name);
79 #endif
80
81 // Create the profile and register it.
82 TestingProfile::Builder builder;
83 builder.SetPath(profile_path);
84 builder.SetPrefService(prefs.Pass());
85 builder.SetSupervisedUserId(supervised_user_id);
86
87 for (TestingProfile::TestingFactories::const_iterator it = factories.begin();
88 it != factories.end(); ++it) {
89 builder.AddTestingFactory(it->first, it->second);
90 }
91
92 TestingProfile* profile = builder.Build().release();
93 profile->set_profile_name(profile_name);
94 profile_manager_->AddProfile(profile); // Takes ownership.
95
96 // Update the user metadata.
97 ProfileInfoCache& cache = profile_manager_->GetProfileInfoCache();
98 size_t index = cache.GetIndexOfProfileWithPath(profile_path);
99 cache.SetAvatarIconOfProfileAtIndex(index, avatar_id);
100 cache.SetSupervisedUserIdOfProfileAtIndex(index, supervised_user_id);
101 // SetNameOfProfileAtIndex may reshuffle the list of profiles, so we do it
102 // last.
103 cache.SetNameOfProfileAtIndex(index, user_name);
104
105 testing_profiles_.insert(std::make_pair(profile_name, profile));
106
107 return profile;
108 }
109
CreateTestingProfile(const std::string & name)110 TestingProfile* TestingProfileManager::CreateTestingProfile(
111 const std::string& name) {
112 DCHECK(called_set_up_);
113 return CreateTestingProfile(name, scoped_ptr<PrefServiceSyncable>(),
114 base::UTF8ToUTF16(name), 0, std::string(),
115 TestingProfile::TestingFactories());
116 }
117
CreateGuestProfile()118 TestingProfile* TestingProfileManager::CreateGuestProfile() {
119 DCHECK(called_set_up_);
120
121 // Create the profile and register it.
122 TestingProfile::Builder builder;
123 builder.SetGuestSession();
124 builder.SetPath(ProfileManager::GetGuestProfilePath());
125
126 // Add the guest profile to the profile manager, but not to the info cache.
127 TestingProfile* profile = builder.Build().release();
128 profile->set_profile_name(kGuestProfileName);
129
130 // Set up a profile with an off the record profile.
131 TestingProfile::Builder().BuildIncognito(profile);
132
133 profile_manager_->AddProfile(profile); // Takes ownership.
134 profile_manager_->SetGuestProfilePrefs(profile);
135
136 testing_profiles_.insert(std::make_pair(kGuestProfileName, profile));
137
138 return profile;
139 }
140
DeleteTestingProfile(const std::string & name)141 void TestingProfileManager::DeleteTestingProfile(const std::string& name) {
142 DCHECK(called_set_up_);
143
144 TestingProfilesMap::iterator it = testing_profiles_.find(name);
145 DCHECK(it != testing_profiles_.end());
146
147 TestingProfile* profile = it->second;
148
149 ProfileInfoCache& cache = profile_manager_->GetProfileInfoCache();
150 cache.DeleteProfileFromCache(profile->GetPath());
151
152 profile_manager_->profiles_info_.erase(profile->GetPath());
153 }
154
DeleteAllTestingProfiles()155 void TestingProfileManager::DeleteAllTestingProfiles() {
156 for (TestingProfilesMap::iterator it = testing_profiles_.begin();
157 it != testing_profiles_.end(); ++it) {
158 TestingProfile* profile = it->second;
159 ProfileInfoCache& cache = profile_manager_->GetProfileInfoCache();
160 cache.DeleteProfileFromCache(profile->GetPath());
161 }
162 testing_profiles_.clear();
163 }
164
165
DeleteGuestProfile()166 void TestingProfileManager::DeleteGuestProfile() {
167 DCHECK(called_set_up_);
168
169 TestingProfilesMap::iterator it = testing_profiles_.find(kGuestProfileName);
170 DCHECK(it != testing_profiles_.end());
171
172 profile_manager_->profiles_info_.erase(ProfileManager::GetGuestProfilePath());
173 }
174
DeleteProfileInfoCache()175 void TestingProfileManager::DeleteProfileInfoCache() {
176 profile_manager_->profile_info_cache_.reset(NULL);
177 }
178
SetLoggedIn(bool logged_in)179 void TestingProfileManager::SetLoggedIn(bool logged_in) {
180 profile_manager_->logged_in_ = logged_in;
181 }
182
profiles_dir()183 const base::FilePath& TestingProfileManager::profiles_dir() {
184 DCHECK(called_set_up_);
185 return profiles_dir_.path();
186 }
187
profile_manager()188 ProfileManager* TestingProfileManager::profile_manager() {
189 DCHECK(called_set_up_);
190 return profile_manager_;
191 }
192
profile_info_cache()193 ProfileInfoCache* TestingProfileManager::profile_info_cache() {
194 DCHECK(called_set_up_);
195 return &profile_manager_->GetProfileInfoCache();
196 }
197
SetUpInternal()198 void TestingProfileManager::SetUpInternal() {
199 ASSERT_FALSE(browser_process_->profile_manager())
200 << "ProfileManager already exists";
201
202 // Set up the directory for profiles.
203 ASSERT_TRUE(profiles_dir_.CreateUniqueTempDir());
204
205 profile_manager_ = new testing::ProfileManager(profiles_dir_.path());
206 browser_process_->SetProfileManager(profile_manager_); // Takes ownership.
207
208 called_set_up_ = true;
209 }
210