Chromium Code Reviews| Index: chrome/browser/policy/device_token_fetcher_unittest.cc |
| diff --git a/chrome/browser/policy/device_token_fetcher_unittest.cc b/chrome/browser/policy/device_token_fetcher_unittest.cc |
| index 754628615a0a01aa7e63e646caadd32d8b2f1862..cd0f0d082db1e1ed3df63e7c76cc7464825a4a28 100644 |
| --- a/chrome/browser/policy/device_token_fetcher_unittest.cc |
| +++ b/chrome/browser/policy/device_token_fetcher_unittest.cc |
| @@ -23,15 +23,13 @@ const char kTestToken[] = "device_token_fetcher_test_auth_token"; |
| using testing::_; |
| using testing::Mock; |
| -class MockTokenAvailableObserver : public NotificationObserver { |
| +class MockTokenAvailableObserver : public DeviceTokenFetcher::Observer { |
| public: |
| MockTokenAvailableObserver() {} |
| virtual ~MockTokenAvailableObserver() {} |
| - MOCK_METHOD3(Observe, void( |
| - NotificationType type, |
| - const NotificationSource& source, |
| - const NotificationDetails& details)); |
| + MOCK_METHOD0(OnTokenSuccess, void()); |
| + MOCK_METHOD0(OnTokenError, void()); |
| private: |
| DISALLOW_COPY_AND_ASSIGN(MockTokenAvailableObserver); |
| @@ -142,12 +140,10 @@ TEST_F(DeviceTokenFetcherTest, SimpleFetchDoubleLogin) { |
| } |
| TEST_F(DeviceTokenFetcherTest, FetchBetweenBrowserLaunchAndNotify) { |
| - NotificationRegistrar registrar; |
| MockTokenAvailableObserver observer; |
| - registrar.Add(&observer, |
| - NotificationType::DEVICE_TOKEN_AVAILABLE, |
| - NotificationService::AllSources()); |
| - EXPECT_CALL(observer, Observe(_, _, _)).Times(1); |
| + fetcher_->AddObserver(&observer); |
|
danno
2010/11/22 13:47:50
Who removes the observer from the fetcher?
Jakob Kummerow (corp)
2010/11/22 16:56:08
Done.
|
| + EXPECT_CALL(observer, OnTokenSuccess()).Times(1); |
| + EXPECT_CALL(observer, OnTokenError()).Times(0); |
| backend_->AllShouldSucceed(); |
| EXPECT_CALL(*backend_, ProcessRegisterRequest(_, _, _, _)).Times(1); |
| SimulateSuccessfulLoginAndRunPending(); |
| @@ -170,7 +166,11 @@ TEST_F(DeviceTokenFetcherTest, FetchBetweenBrowserLaunchAndNotify) { |
| } |
| TEST_F(DeviceTokenFetcherTest, FailedServerRequest) { |
| - backend_->AllShouldFail(); |
| + MockTokenAvailableObserver observer; |
| + fetcher_->AddObserver(&observer); |
|
danno
2010/11/22 13:47:50
Who removes the observer from the fetcher?
Jakob Kummerow (corp)
2010/11/22 16:56:08
Done.
|
| + EXPECT_CALL(observer, OnTokenSuccess()).Times(0); |
| + EXPECT_CALL(observer, OnTokenError()).Times(1); |
| + backend_->AllShouldFail(); |
|
danno
2010/11/22 13:47:50
indentation above should be 2
Jakob Kummerow (corp)
2010/11/22 16:56:08
Done.
(Of course -- copy-paste artifact.)
|
| EXPECT_CALL(*backend_, ProcessRegisterRequest(_, _, _, _)).Times(1); |
| SimulateSuccessfulLoginAndRunPending(); |
| ASSERT_FALSE(fetcher_->IsTokenPending()); |
| @@ -179,14 +179,16 @@ TEST_F(DeviceTokenFetcherTest, FailedServerRequest) { |
| } |
| TEST_F(DeviceTokenFetcherTest, UnmanagedDevice) { |
| + FilePath token_path; |
| + GetDeviceTokenPath(fetcher_, &token_path); |
| + file_util::WriteFile(token_path, "foo", 3); |
| + ASSERT_TRUE(file_util::PathExists(token_path)); |
| backend_->UnmanagedDevice(); |
| EXPECT_CALL(*backend_, ProcessRegisterRequest(_, _, _, _)).Times(1); |
| SimulateSuccessfulLoginAndRunPending(); |
| ASSERT_FALSE(fetcher_->IsTokenPending()); |
| ASSERT_EQ("", fetcher_->GetDeviceToken()); |
| ASSERT_EQ("", device_id(fetcher_)); |
| - FilePath token_path; |
| - GetDeviceTokenPath(fetcher_, &token_path); |
| ASSERT_FALSE(file_util::PathExists(token_path)); |
| } |