OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "base/command_line.h" | |
6 #include "base/memory/scoped_ptr.h" | 5 #include "base/memory/scoped_ptr.h" |
7 #include "base/run_loop.h" | 6 #include "base/run_loop.h" |
8 #include "base/strings/utf_string_conversions.h" | |
9 #include "base/time/time.h" | 7 #include "base/time/time.h" |
10 #include "chrome/browser/prefs/pref_service_syncable.h" | |
11 #include "chrome/browser/signin/account_reconcilor_factory.h" | 8 #include "chrome/browser/signin/account_reconcilor_factory.h" |
12 #include "chrome/browser/signin/chrome_signin_client_factory.h" | 9 #include "chrome/browser/signin/chrome_signin_client_factory.h" |
13 #include "chrome/browser/signin/fake_profile_oauth2_token_service.h" | 10 #include "chrome/browser/signin/fake_profile_oauth2_token_service.h" |
14 #include "chrome/browser/signin/fake_profile_oauth2_token_service_builder.h" | 11 #include "chrome/browser/signin/fake_profile_oauth2_token_service_builder.h" |
15 #include "chrome/browser/signin/fake_signin_manager.h" | 12 #include "chrome/browser/signin/fake_signin_manager.h" |
16 #include "chrome/browser/signin/profile_oauth2_token_service_factory.h" | 13 #include "chrome/browser/signin/profile_oauth2_token_service_factory.h" |
17 #include "chrome/browser/signin/signin_manager_factory.h" | 14 #include "chrome/browser/signin/signin_manager_factory.h" |
18 #include "chrome/test/base/testing_browser_process.h" | |
19 #include "chrome/test/base/testing_profile.h" | 15 #include "chrome/test/base/testing_profile.h" |
20 #include "chrome/test/base/testing_profile_manager.h" | |
21 #include "components/signin/core/browser/account_reconcilor.h" | 16 #include "components/signin/core/browser/account_reconcilor.h" |
22 #include "components/signin/core/browser/profile_oauth2_token_service.h" | 17 #include "components/signin/core/browser/profile_oauth2_token_service.h" |
23 #include "components/signin/core/browser/signin_manager.h" | 18 #include "components/signin/core/browser/signin_manager.h" |
24 #include "components/signin/core/common/signin_switches.h" | |
25 #include "content/public/test/test_browser_thread_bundle.h" | 19 #include "content/public/test/test_browser_thread_bundle.h" |
26 #include "google_apis/gaia/gaia_urls.h" | 20 #include "google_apis/gaia/gaia_urls.h" |
27 #include "net/url_request/test_url_fetcher_factory.h" | 21 #include "net/url_request/test_url_fetcher_factory.h" |
28 #include "testing/gmock/include/gmock/gmock.h" | 22 #include "testing/gmock/include/gmock/gmock.h" |
29 #include "testing/gtest/include/gtest/gtest.h" | 23 #include "testing/gtest/include/gtest/gtest.h" |
30 | 24 |
31 namespace { | 25 namespace { |
32 | 26 |
33 const char kTestEmail[] = "user@gmail.com"; | 27 const char kTestEmail[] = "user@gmail.com"; |
34 | 28 |
35 class MockAccountReconcilor : public testing::StrictMock<AccountReconcilor> { | 29 class MockAccountReconcilor : public testing::StrictMock<AccountReconcilor> { |
36 public: | 30 public: |
37 static KeyedService* Build(content::BrowserContext* context); | 31 static KeyedService* Build(content::BrowserContext* context); |
38 | 32 |
39 MockAccountReconcilor(ProfileOAuth2TokenService* token_service, | 33 MockAccountReconcilor(ProfileOAuth2TokenService* token_service, |
40 SigninManagerBase* signin_manager, | 34 SigninManagerBase* signin_manager, |
41 SigninClient* client); | 35 SigninClient* client); |
42 virtual ~MockAccountReconcilor() {} | 36 virtual ~MockAccountReconcilor() {} |
43 | 37 |
44 MOCK_METHOD1(PerformMergeAction, void(const std::string& account_id)); | 38 MOCK_METHOD1(PerformMergeAction, void(const std::string& account_id)); |
45 MOCK_METHOD1(PerformStartRemoveAction, void(const std::string& account_id)); | 39 MOCK_METHOD1(StartRemoveAction, void(const std::string& account_id)); |
46 MOCK_METHOD3( | 40 MOCK_METHOD3( |
47 PerformFinishRemoveAction, | 41 FinishRemoveAction, |
48 void(const std::string& account_id, | 42 void(const std::string& account_id, |
49 const GoogleServiceAuthError& error, | 43 const GoogleServiceAuthError& error, |
50 const std::vector<std::pair<std::string, bool> >& accounts)); | 44 const std::vector<std::pair<std::string, bool> >& accounts)); |
51 MOCK_METHOD2(PerformAddToChromeAction, void(const std::string& account_id, | 45 MOCK_METHOD2(PerformAddToChromeAction, void(const std::string& account_id, |
52 int session_index)); | 46 int session_index)); |
53 MOCK_METHOD0(PerformLogoutAllAccountsAction, void()); | 47 MOCK_METHOD0(PerformLogoutAllAccountsAction, void()); |
54 }; | 48 }; |
55 | 49 |
56 // static | 50 // static |
57 KeyedService* MockAccountReconcilor::Build(content::BrowserContext* context) { | 51 KeyedService* MockAccountReconcilor::Build(content::BrowserContext* context) { |
(...skipping 15 matching lines...) Expand all Loading... |
73 client) {} | 67 client) {} |
74 | 68 |
75 } // namespace | 69 } // namespace |
76 | 70 |
77 class AccountReconcilorTest : public testing::Test { | 71 class AccountReconcilorTest : public testing::Test { |
78 public: | 72 public: |
79 AccountReconcilorTest(); | 73 AccountReconcilorTest(); |
80 virtual void SetUp() OVERRIDE; | 74 virtual void SetUp() OVERRIDE; |
81 virtual void TearDown() OVERRIDE; | 75 virtual void TearDown() OVERRIDE; |
82 | 76 |
83 TestingProfile* profile() { return profile_; } | 77 TestingProfile* profile() { return profile_.get(); } |
84 FakeSigninManagerForTesting* signin_manager() { return signin_manager_; } | 78 FakeSigninManagerForTesting* signin_manager() { return signin_manager_; } |
85 FakeProfileOAuth2TokenService* token_service() { return token_service_; } | 79 FakeProfileOAuth2TokenService* token_service() { return token_service_; } |
86 | 80 |
87 void SetFakeResponse(const std::string& url, | 81 void SetFakeResponse(const std::string& url, |
88 const std::string& data, | 82 const std::string& data, |
89 net::HttpStatusCode code, | 83 net::HttpStatusCode code, |
90 net::URLRequestStatus::Status status) { | 84 net::URLRequestStatus::Status status) { |
91 url_fetcher_factory_.SetFakeResponse(GURL(url), data, code, status); | 85 url_fetcher_factory_.SetFakeResponse(GURL(url), data, code, status); |
92 } | 86 } |
93 | 87 |
94 MockAccountReconcilor* GetMockReconcilor(); | 88 MockAccountReconcilor* GetMockReconcilor(); |
95 | 89 |
96 void SimulateMergeSessionCompleted( | 90 void SimulateMergeSessionCompleted( |
97 MergeSessionHelper::Observer* observer, | 91 MergeSessionHelper::Observer* observer, |
98 const std::string& account_id, | 92 const std::string& account_id, |
99 const GoogleServiceAuthError& error); | 93 const GoogleServiceAuthError& error); |
100 | 94 |
101 void SimulateRefreshTokenFetched( | 95 void SimulateRefreshTokenFetched( |
102 AccountReconcilor* reconcilor, | 96 AccountReconcilor* reconcilor, |
103 const std::string& account_id, | 97 const std::string& account_id, |
104 const std::string& refresh_token); | 98 const std::string& refresh_token); |
105 | 99 |
106 private: | 100 private: |
107 content::TestBrowserThreadBundle bundle_; | 101 content::TestBrowserThreadBundle bundle_; |
108 TestingProfile* profile_; | 102 scoped_ptr<TestingProfile> profile_; |
109 FakeSigninManagerForTesting* signin_manager_; | 103 FakeSigninManagerForTesting* signin_manager_; |
110 FakeProfileOAuth2TokenService* token_service_; | 104 FakeProfileOAuth2TokenService* token_service_; |
111 MockAccountReconcilor* mock_reconcilor_; | 105 MockAccountReconcilor* mock_reconcilor_; |
112 net::FakeURLFetcherFactory url_fetcher_factory_; | 106 net::FakeURLFetcherFactory url_fetcher_factory_; |
113 TestingProfileManager* testing_profile_manager_; | |
114 }; | 107 }; |
115 | 108 |
116 AccountReconcilorTest::AccountReconcilorTest() | 109 AccountReconcilorTest::AccountReconcilorTest() |
117 : signin_manager_(NULL), | 110 : signin_manager_(NULL), |
118 token_service_(NULL), | 111 token_service_(NULL), |
119 mock_reconcilor_(NULL), | 112 mock_reconcilor_(NULL), |
120 url_fetcher_factory_(NULL) {} | 113 url_fetcher_factory_(NULL) {} |
121 | 114 |
122 void AccountReconcilorTest::SetUp() { | 115 void AccountReconcilorTest::SetUp() { |
123 CommandLine::ForCurrentProcess()->AppendSwitch( | 116 TestingProfile::Builder builder; |
124 switches::kNewProfileManagement); | 117 builder.AddTestingFactory(ProfileOAuth2TokenServiceFactory::GetInstance(), |
125 | 118 BuildFakeProfileOAuth2TokenService); |
126 testing_profile_manager_ = | 119 builder.AddTestingFactory(SigninManagerFactory::GetInstance(), |
127 new TestingProfileManager(TestingBrowserProcess::GetGlobal()); | 120 FakeSigninManagerBase::Build); |
128 ASSERT_TRUE(testing_profile_manager_->SetUp()); | 121 builder.AddTestingFactory(AccountReconcilorFactory::GetInstance(), |
129 | 122 MockAccountReconcilor::Build); |
130 TestingProfile::TestingFactories factories; | 123 profile_ = builder.Build(); |
131 factories.push_back(std::make_pair( | |
132 ProfileOAuth2TokenServiceFactory::GetInstance(), | |
133 BuildFakeProfileOAuth2TokenService)); | |
134 factories.push_back(std::make_pair(SigninManagerFactory::GetInstance(), | |
135 FakeSigninManagerBase::Build)); | |
136 factories.push_back(std::make_pair(AccountReconcilorFactory::GetInstance(), | |
137 MockAccountReconcilor::Build)); | |
138 | |
139 profile_ = testing_profile_manager_->CreateTestingProfile("name", | |
140 scoped_ptr<PrefServiceSyncable>(), | |
141 base::UTF8ToUTF16("name"), 0, std::string(), | |
142 factories); | |
143 | 124 |
144 signin_manager_ = | 125 signin_manager_ = |
145 static_cast<FakeSigninManagerForTesting*>( | 126 static_cast<FakeSigninManagerForTesting*>( |
146 SigninManagerFactory::GetForProfile(profile())); | 127 SigninManagerFactory::GetForProfile(profile())); |
147 | 128 |
148 token_service_ = | 129 token_service_ = |
149 static_cast<FakeProfileOAuth2TokenService*>( | 130 static_cast<FakeProfileOAuth2TokenService*>( |
150 ProfileOAuth2TokenServiceFactory::GetForProfile(profile())); | 131 ProfileOAuth2TokenServiceFactory::GetForProfile(profile())); |
151 } | 132 } |
152 | 133 |
153 void AccountReconcilorTest::TearDown() { | 134 void AccountReconcilorTest::TearDown() { |
154 // The |testing_profile_manager_| will handle destroying the profile. | 135 // Destroy the profile before all threads are torn down. |
155 delete testing_profile_manager_; | 136 profile_.reset(); |
156 } | 137 } |
157 | 138 |
158 MockAccountReconcilor* AccountReconcilorTest::GetMockReconcilor() { | 139 MockAccountReconcilor* AccountReconcilorTest::GetMockReconcilor() { |
159 if (!mock_reconcilor_) { | 140 if (!mock_reconcilor_) { |
160 mock_reconcilor_ = | 141 mock_reconcilor_ = |
161 static_cast<MockAccountReconcilor*>( | 142 static_cast<MockAccountReconcilor*>( |
162 AccountReconcilorFactory::GetForProfile(profile())); | 143 AccountReconcilorFactory::GetForProfile(profile())); |
163 } | 144 } |
164 | 145 |
165 return mock_reconcilor_; | 146 return mock_reconcilor_; |
(...skipping 21 matching lines...) Expand all Loading... |
187 } | 168 } |
188 | 169 |
189 #if !defined(OS_CHROMEOS) | 170 #if !defined(OS_CHROMEOS) |
190 | 171 |
191 TEST_F(AccountReconcilorTest, SigninManagerRegistration) { | 172 TEST_F(AccountReconcilorTest, SigninManagerRegistration) { |
192 AccountReconcilor* reconcilor = | 173 AccountReconcilor* reconcilor = |
193 AccountReconcilorFactory::GetForProfile(profile()); | 174 AccountReconcilorFactory::GetForProfile(profile()); |
194 ASSERT_TRUE(reconcilor); | 175 ASSERT_TRUE(reconcilor); |
195 ASSERT_FALSE(reconcilor->IsRegisteredWithTokenService()); | 176 ASSERT_FALSE(reconcilor->IsRegisteredWithTokenService()); |
196 | 177 |
197 signin_manager()->set_password("password"); | |
198 signin_manager()->OnExternalSigninCompleted(kTestEmail); | 178 signin_manager()->OnExternalSigninCompleted(kTestEmail); |
199 ASSERT_TRUE(reconcilor->IsRegisteredWithTokenService()); | 179 ASSERT_TRUE(reconcilor->IsRegisteredWithTokenService()); |
200 | 180 |
201 EXPECT_CALL(*GetMockReconcilor(), PerformLogoutAllAccountsAction()); | 181 EXPECT_CALL(*GetMockReconcilor(), PerformLogoutAllAccountsAction()); |
202 | 182 |
203 signin_manager()->SignOut(); | 183 signin_manager()->SignOut(); |
204 ASSERT_FALSE(reconcilor->IsRegisteredWithTokenService()); | 184 ASSERT_FALSE(reconcilor->IsRegisteredWithTokenService()); |
205 } | 185 } |
206 | 186 |
207 TEST_F(AccountReconcilorTest, Reauth) { | 187 TEST_F(AccountReconcilorTest, Reauth) { |
208 signin_manager()->SetAuthenticatedUsername(kTestEmail); | 188 signin_manager()->SetAuthenticatedUsername(kTestEmail); |
209 signin_manager()->set_password("password"); | |
210 | 189 |
211 AccountReconcilor* reconcilor = | 190 AccountReconcilor* reconcilor = |
212 AccountReconcilorFactory::GetForProfile(profile()); | 191 AccountReconcilorFactory::GetForProfile(profile()); |
213 ASSERT_TRUE(reconcilor); | 192 ASSERT_TRUE(reconcilor); |
214 ASSERT_TRUE(reconcilor->IsRegisteredWithTokenService()); | 193 ASSERT_TRUE(reconcilor->IsRegisteredWithTokenService()); |
215 | 194 |
216 // Simulate reauth. The state of the reconcilor should not change. | 195 // Simulate reauth. The state of the reconcilor should not change. |
217 signin_manager()->OnExternalSigninCompleted(kTestEmail); | 196 signin_manager()->OnExternalSigninCompleted(kTestEmail); |
218 ASSERT_TRUE(reconcilor->IsRegisteredWithTokenService()); | 197 ASSERT_TRUE(reconcilor->IsRegisteredWithTokenService()); |
219 } | 198 } |
(...skipping 216 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
436 ASSERT_FALSE(reconcilor->AreAllRefreshTokensChecked()); | 415 ASSERT_FALSE(reconcilor->AreAllRefreshTokensChecked()); |
437 | 416 |
438 token_service()->IssueAllTokensForAccount("user@gmail.com", "access_token", | 417 token_service()->IssueAllTokensForAccount("user@gmail.com", "access_token", |
439 base::Time::Now() + base::TimeDelta::FromHours(1)); | 418 base::Time::Now() + base::TimeDelta::FromHours(1)); |
440 | 419 |
441 base::RunLoop().RunUntilIdle(); | 420 base::RunLoop().RunUntilIdle(); |
442 ASSERT_TRUE(reconcilor->AreAllRefreshTokensChecked()); | 421 ASSERT_TRUE(reconcilor->AreAllRefreshTokensChecked()); |
443 ASSERT_FALSE(reconcilor->is_reconcile_started_); | 422 ASSERT_FALSE(reconcilor->is_reconcile_started_); |
444 } | 423 } |
445 | 424 |
446 | |
447 TEST_F(AccountReconcilorTest, StartReconcileAddToCookie) { | 425 TEST_F(AccountReconcilorTest, StartReconcileAddToCookie) { |
448 signin_manager()->SetAuthenticatedUsername("user@gmail.com"); | 426 signin_manager()->SetAuthenticatedUsername("user@gmail.com"); |
449 token_service()->UpdateCredentials("user@gmail.com", "refresh_token"); | 427 token_service()->UpdateCredentials("user@gmail.com", "refresh_token"); |
450 token_service()->UpdateCredentials("other@gmail.com", "refresh_token"); | 428 token_service()->UpdateCredentials("other@gmail.com", "refresh_token"); |
451 | 429 |
452 EXPECT_CALL(*GetMockReconcilor(), PerformMergeAction("other@gmail.com")); | 430 EXPECT_CALL(*GetMockReconcilor(), PerformMergeAction("other@gmail.com")); |
453 | 431 |
454 SetFakeResponse(GaiaUrls::GetInstance()->list_accounts_url().spec(), | 432 SetFakeResponse(GaiaUrls::GetInstance()->list_accounts_url().spec(), |
455 "[\"f\", [[\"b\", 0, \"n\", \"user@gmail.com\", \"p\", 0, 0, 0, 0, 1]]]", | 433 "[\"f\", [[\"b\", 0, \"n\", \"user@gmail.com\", \"p\", 0, 0, 0, 0, 1]]]", |
456 net::HTTP_OK, net::URLRequestStatus::SUCCESS); | 434 net::HTTP_OK, net::URLRequestStatus::SUCCESS); |
(...skipping 122 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
579 token_service()->IssueAllTokensForAccount("user@gmail.com", "access_token", | 557 token_service()->IssueAllTokensForAccount("user@gmail.com", "access_token", |
580 base::Time::Now() + base::TimeDelta::FromHours(1)); | 558 base::Time::Now() + base::TimeDelta::FromHours(1)); |
581 token_service()->IssueAllTokensForAccount("other@gmail.com", "access_token", | 559 token_service()->IssueAllTokensForAccount("other@gmail.com", "access_token", |
582 base::Time::Now() + base::TimeDelta::FromHours(1)); | 560 base::Time::Now() + base::TimeDelta::FromHours(1)); |
583 | 561 |
584 base::RunLoop().RunUntilIdle(); | 562 base::RunLoop().RunUntilIdle(); |
585 SimulateMergeSessionCompleted(reconcilor, "user@gmail.com", | 563 SimulateMergeSessionCompleted(reconcilor, "user@gmail.com", |
586 GoogleServiceAuthError::AuthErrorNone()); | 564 GoogleServiceAuthError::AuthErrorNone()); |
587 ASSERT_FALSE(reconcilor->is_reconcile_started_); | 565 ASSERT_FALSE(reconcilor->is_reconcile_started_); |
588 } | 566 } |
OLD | NEW |