| 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 f87265d95503ea3797826a626f50c56ce87979fa..ef809756f30eb86edef9fbd15f7e58f00b7e71e6 100644
|
| --- a/chrome/browser/policy/device_token_fetcher_unittest.cc
|
| +++ b/chrome/browser/policy/device_token_fetcher_unittest.cc
|
| @@ -11,6 +11,7 @@
|
| #include "chrome/browser/net/gaia/token_service.h"
|
| #include "chrome/browser/policy/cloud_policy_cache.h"
|
| #include "chrome/browser/policy/device_management_service.h"
|
| +#include "chrome/browser/policy/proto/device_management_backend.pb.h"
|
| #include "chrome/browser/policy/mock_device_management_backend.h"
|
| #include "chrome/common/net/gaia/gaia_constants.h"
|
| #include "chrome/test/testing_profile.h"
|
| @@ -85,14 +86,6 @@ class ProxyDeviceManagementBackend : public DeviceManagementBackend {
|
| backend_->ProcessPolicyRequest(device_management_token, device_id,
|
| request, delegate);
|
| }
|
| - virtual void ProcessCloudPolicyRequest(
|
| - const std::string& device_management_token,
|
| - const std::string& device_id,
|
| - const em::CloudPolicyRequest& request,
|
| - DevicePolicyResponseDelegate* delegate) {
|
| - backend_->ProcessCloudPolicyRequest(device_management_token, device_id,
|
| - request, delegate);
|
| - }
|
|
|
| private:
|
| DeviceManagementBackend* backend_; // weak
|
| @@ -142,7 +135,8 @@ TEST_F(DeviceTokenFetcherTest, FetchToken) {
|
| EXPECT_CALL(observer, OnDeviceTokenAvailable());
|
| fetcher.AddObserver(&observer);
|
| EXPECT_EQ("", fetcher.GetDeviceToken());
|
| - fetcher.FetchToken("fake_auth_token", "fake_device_id");
|
| + fetcher.FetchToken("fake_auth_token", "fake_device_id",
|
| + em::DeviceRegisterRequest::USER, "fake_machine_id");
|
| loop_.RunAllPending();
|
| Mock::VerifyAndClearExpectations(&observer);
|
| std::string token = fetcher.GetDeviceToken();
|
| @@ -152,7 +146,8 @@ TEST_F(DeviceTokenFetcherTest, FetchToken) {
|
| EXPECT_CALL(backend_, ProcessRegisterRequest(_, _, _, _)).WillOnce(
|
| MockDeviceManagementBackendSucceedRegister());
|
| EXPECT_CALL(observer, OnDeviceTokenAvailable());
|
| - fetcher.FetchToken("fake_auth_token", "fake_device_id");
|
| + fetcher.FetchToken("fake_auth_token", "fake_device_id",
|
| + em::DeviceRegisterRequest::USER, "fake_machine_id");
|
| loop_.RunAllPending();
|
| Mock::VerifyAndClearExpectations(&observer);
|
| std::string token2 = fetcher.GetDeviceToken();
|
| @@ -171,7 +166,8 @@ TEST_F(DeviceTokenFetcherTest, RetryOnError) {
|
| MockTokenAvailableObserver observer;
|
| EXPECT_CALL(observer, OnDeviceTokenAvailable());
|
| fetcher.AddObserver(&observer);
|
| - fetcher.FetchToken("fake_auth_token", "fake_device_id");
|
| + fetcher.FetchToken("fake_auth_token", "fake_device_id",
|
| + em::DeviceRegisterRequest::USER, "fake_machine_id");
|
| loop_.RunAllPending();
|
| Mock::VerifyAndClearExpectations(&observer);
|
| EXPECT_NE("", fetcher.GetDeviceToken());
|
| @@ -187,7 +183,8 @@ TEST_F(DeviceTokenFetcherTest, UnmanagedDevice) {
|
| MockTokenAvailableObserver observer;
|
| EXPECT_CALL(observer, OnDeviceTokenAvailable()).Times(0);
|
| fetcher.AddObserver(&observer);
|
| - fetcher.FetchToken("fake_auth_token", "fake_device_id");
|
| + fetcher.FetchToken("fake_auth_token", "fake_device_id",
|
| + em::DeviceRegisterRequest::USER, "fake_machine_id");
|
| loop_.RunAllPending();
|
| Mock::VerifyAndClearExpectations(&observer);
|
| EXPECT_EQ("", fetcher.GetDeviceToken());
|
|
|