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/test/base/testing_browser_process.h"
14 #include "testing/gtest/include/gtest/gtest.h"
15
16 const std::string kGuestProfileName = "Guest";
17
18 namespace testing {
19
20 class ProfileManager : public ::ProfileManagerWithoutInit {
21 public:
ProfileManager(const base::FilePath & user_data_dir)22 explicit ProfileManager(const base::FilePath& user_data_dir)
23 : ::ProfileManagerWithoutInit(user_data_dir) {}
24
25 protected:
CreateProfileHelper(const base::FilePath & file_path)26 virtual Profile* CreateProfileHelper(
27 const base::FilePath& file_path) OVERRIDE {
28 return new TestingProfile(file_path);
29 }
30 };
31
32 } // namespace testing
33
TestingProfileManager(TestingBrowserProcess * process)34 TestingProfileManager::TestingProfileManager(TestingBrowserProcess* process)
35 : called_set_up_(false),
36 browser_process_(process),
37 local_state_(process) {
38 }
39
~TestingProfileManager()40 TestingProfileManager::~TestingProfileManager() {
41 // Destroying this class also destroys the LocalState, so make sure the
42 // associated ProfileManager is also destroyed.
43 browser_process_->SetProfileManager(NULL);
44 }
45
SetUp()46 bool TestingProfileManager::SetUp() {
47 SetUpInternal();
48 return called_set_up_;
49 }
50
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)51 TestingProfile* TestingProfileManager::CreateTestingProfile(
52 const std::string& profile_name,
53 scoped_ptr<PrefServiceSyncable> prefs,
54 const base::string16& user_name,
55 int avatar_id,
56 const std::string& supervised_user_id,
57 const TestingProfile::TestingFactories& factories) {
58 DCHECK(called_set_up_);
59
60 // Create a path for the profile based on the name.
61 base::FilePath profile_path(profiles_dir_.path());
62 profile_path = profile_path.AppendASCII(profile_name);
63
64 // Create the profile and register it.
65 TestingProfile::Builder builder;
66 builder.SetPath(profile_path);
67 builder.SetPrefService(prefs.Pass());
68 builder.SetSupervisedUserId(supervised_user_id);
69
70 for (TestingProfile::TestingFactories::const_iterator it = factories.begin();
71 it != factories.end(); ++it) {
72 builder.AddTestingFactory(it->first, it->second);
73 }
74
75 TestingProfile* profile = builder.Build().release();
76 profile->set_profile_name(profile_name);
77 profile_manager_->AddProfile(profile); // Takes ownership.
78
79 // Update the user metadata.
80 ProfileInfoCache& cache = profile_manager_->GetProfileInfoCache();
81 size_t index = cache.GetIndexOfProfileWithPath(profile_path);
82 cache.SetAvatarIconOfProfileAtIndex(index, avatar_id);
83 cache.SetSupervisedUserIdOfProfileAtIndex(index, supervised_user_id);
84 // SetNameOfProfileAtIndex may reshuffle the list of profiles, so we do it
85 // last.
86 cache.SetNameOfProfileAtIndex(index, user_name);
87
88 testing_profiles_.insert(std::make_pair(profile_name, profile));
89
90 return profile;
91 }
92
CreateTestingProfile(const std::string & name)93 TestingProfile* TestingProfileManager::CreateTestingProfile(
94 const std::string& name) {
95 DCHECK(called_set_up_);
96 return CreateTestingProfile(name, scoped_ptr<PrefServiceSyncable>(),
97 base::UTF8ToUTF16(name), 0, std::string(),
98 TestingProfile::TestingFactories());
99 }
100
CreateGuestProfile()101 TestingProfile* TestingProfileManager::CreateGuestProfile() {
102 DCHECK(called_set_up_);
103
104 // Set up a profile with an off the record profile.
105 TestingProfile::Builder otr_builder;
106 otr_builder.SetIncognito();
107 scoped_ptr<TestingProfile> otr_profile(otr_builder.Build());
108
109 // Create the profile and register it.
110 TestingProfile::Builder builder;
111 builder.SetGuestSession();
112 builder.SetPath(ProfileManager::GetGuestProfilePath());
113
114 // Add the guest profile to the profile manager, but not to the info cache.
115 TestingProfile* profile = builder.Build().release();
116 profile->set_profile_name(kGuestProfileName);
117
118 otr_profile->SetOriginalProfile(profile);
119 profile->SetOffTheRecordProfile(otr_profile.PassAs<Profile>());
120 profile_manager_->AddProfile(profile); // Takes ownership.
121 profile_manager_->SetGuestProfilePrefs(profile);
122
123 testing_profiles_.insert(std::make_pair(kGuestProfileName, profile));
124
125 return profile;
126 }
127
DeleteTestingProfile(const std::string & name)128 void TestingProfileManager::DeleteTestingProfile(const std::string& name) {
129 DCHECK(called_set_up_);
130
131 TestingProfilesMap::iterator it = testing_profiles_.find(name);
132 DCHECK(it != testing_profiles_.end());
133
134 TestingProfile* profile = it->second;
135
136 ProfileInfoCache& cache = profile_manager_->GetProfileInfoCache();
137 cache.DeleteProfileFromCache(profile->GetPath());
138
139 profile_manager_->profiles_info_.erase(profile->GetPath());
140 }
141
DeleteGuestProfile()142 void TestingProfileManager::DeleteGuestProfile() {
143 DCHECK(called_set_up_);
144
145 TestingProfilesMap::iterator it = testing_profiles_.find(kGuestProfileName);
146 DCHECK(it != testing_profiles_.end());
147
148 profile_manager_->profiles_info_.erase(ProfileManager::GetGuestProfilePath());
149 }
150
DeleteProfileInfoCache()151 void TestingProfileManager::DeleteProfileInfoCache() {
152 profile_manager_->profile_info_cache_.reset(NULL);
153 }
154
SetLoggedIn(bool logged_in)155 void TestingProfileManager::SetLoggedIn(bool logged_in) {
156 profile_manager_->logged_in_ = logged_in;
157 }
158
profiles_dir()159 const base::FilePath& TestingProfileManager::profiles_dir() {
160 DCHECK(called_set_up_);
161 return profiles_dir_.path();
162 }
163
profile_manager()164 ProfileManager* TestingProfileManager::profile_manager() {
165 DCHECK(called_set_up_);
166 return profile_manager_;
167 }
168
profile_info_cache()169 ProfileInfoCache* TestingProfileManager::profile_info_cache() {
170 DCHECK(called_set_up_);
171 return &profile_manager_->GetProfileInfoCache();
172 }
173
SetUpInternal()174 void TestingProfileManager::SetUpInternal() {
175 ASSERT_FALSE(browser_process_->profile_manager())
176 << "ProfileManager already exists";
177
178 // Set up the directory for profiles.
179 ASSERT_TRUE(profiles_dir_.CreateUniqueTempDir());
180
181 profile_manager_ = new testing::ProfileManager(profiles_dir_.path());
182 browser_process_->SetProfileManager(profile_manager_); // Takes ownership.
183
184 called_set_up_ = true;
185 }
186