chrome: Use base::MessageLoop. (Part 3)

chrome/browser/policy/cloud/device_management_service_browsertest.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/message_loop.h"
 #include "base/stl_util.h"
 #include "chrome/browser/policy/cloud/cloud_policy_constants.h"
@@ skipping 89 matching lines @@
   }
 
  protected:
   void ExpectRequest() {
     if (interceptor_)
       interceptor_->PushJobCallback(base::Bind(&ResponseJob));
   }
 
   void PerformRegistration() {
     ExpectRequest();
-    EXPECT_CALL(*this, OnJobDone(DM_STATUS_SUCCESS, _))
-        .WillOnce(
-            DoAll(Invoke(this,
-                         &DeviceManagementServiceIntegrationTest::RecordToken),
-                  InvokeWithoutArgs(MessageLoop::current(),
-                                    &MessageLoop::Quit)));
+    EXPECT_CALL(*this, OnJobDone(DM_STATUS_SUCCESS, _)).WillOnce(DoAll(
+        Invoke(this, &DeviceManagementServiceIntegrationTest::RecordToken),
+        InvokeWithoutArgs(base::MessageLoop::current(),
+                          &base::MessageLoop::Quit)));
     scoped_ptr<DeviceManagementRequestJob> job(
         service_->CreateJob(DeviceManagementRequestJob::TYPE_REGISTRATION));
     job->SetGaiaToken("gaia_auth_token");
     job->SetOAuthToken("oauth_token");
     job->SetClientID("testid");
     job->GetRequest()->mutable_register_request();
     job->Start(base::Bind(&DeviceManagementServiceIntegrationTest::OnJobDone,
                           base::Unretained(this)));
-    MessageLoop::current()->Run();
+    base::MessageLoop::current()->Run();
   }
 
   virtual void SetUpOnMainThread() OVERRIDE {
     std::string service_url((this->*(GetParam()))());
     service_.reset(new DeviceManagementService(service_url));
     service_->ScheduleInitialization(0);
   }
 
   virtual void CleanUpOnMainThread() OVERRIDE {
     service_.reset();
@@ skipping 21 matching lines @@
 IN_PROC_BROWSER_TEST_P(DeviceManagementServiceIntegrationTest, Registration) {
   PerformRegistration();
   EXPECT_FALSE(token_.empty());
 }
 
 IN_PROC_BROWSER_TEST_P(DeviceManagementServiceIntegrationTest, PolicyFetch) {
   PerformRegistration();
 
   ExpectRequest();
   EXPECT_CALL(*this, OnJobDone(DM_STATUS_SUCCESS, _))
-      .WillOnce(InvokeWithoutArgs(MessageLoop::current(), &MessageLoop::Quit));
+      .WillOnce(InvokeWithoutArgs(base::MessageLoop::current(),
+                                  &base::MessageLoop::Quit));
   scoped_ptr<DeviceManagementRequestJob> job(
       service_->CreateJob(DeviceManagementRequestJob::TYPE_POLICY_FETCH));
   job->SetDMToken(token_);
   job->SetUserAffiliation(USER_AFFILIATION_NONE);
   job->SetClientID("testid");
   em::DevicePolicyRequest* request =
       job->GetRequest()->mutable_policy_request();
   request->add_request()->set_policy_type(dm_protocol::kChromeUserPolicyType);
   job->Start(base::Bind(&DeviceManagementServiceIntegrationTest::OnJobDone,
                         base::Unretained(this)));
-  MessageLoop::current()->Run();
+  base::MessageLoop::current()->Run();
 }
 
 IN_PROC_BROWSER_TEST_P(DeviceManagementServiceIntegrationTest, Unregistration) {
   PerformRegistration();
 
   ExpectRequest();
   EXPECT_CALL(*this, OnJobDone(DM_STATUS_SUCCESS, _))
-      .WillOnce(InvokeWithoutArgs(MessageLoop::current(), &MessageLoop::Quit));
+      .WillOnce(InvokeWithoutArgs(base::MessageLoop::current(),
+                                  &base::MessageLoop::Quit));
   scoped_ptr<DeviceManagementRequestJob> job(
       service_->CreateJob(DeviceManagementRequestJob::TYPE_UNREGISTRATION));
   job->SetDMToken(token_);
   job->SetClientID("testid");
   job->GetRequest()->mutable_unregister_request();
   job->Start(base::Bind(&DeviceManagementServiceIntegrationTest::OnJobDone,
                         base::Unretained(this)));
-  MessageLoop::current()->Run();
+  base::MessageLoop::current()->Run();
 }
 
 IN_PROC_BROWSER_TEST_P(DeviceManagementServiceIntegrationTest, AutoEnrollment) {
   ExpectRequest();
   EXPECT_CALL(*this, OnJobDone(DM_STATUS_SUCCESS, _))
-      .WillOnce(InvokeWithoutArgs(MessageLoop::current(), &MessageLoop::Quit));
+      .WillOnce(InvokeWithoutArgs(base::MessageLoop::current(),
+                                  &base::MessageLoop::Quit));
   scoped_ptr<DeviceManagementRequestJob> job(
       service_->CreateJob(DeviceManagementRequestJob::TYPE_AUTO_ENROLLMENT));
   job->SetClientID("testid");
   job->GetRequest()->mutable_auto_enrollment_request()->set_remainder(0);
   job->GetRequest()->mutable_auto_enrollment_request()->set_modulus(1);
   job->Start(base::Bind(&DeviceManagementServiceIntegrationTest::OnJobDone,
                         base::Unretained(this)));
-  MessageLoop::current()->Run();
+  base::MessageLoop::current()->Run();
 }
 
 INSTANTIATE_TEST_CASE_P(
     DeviceManagementServiceIntegrationTestInstance,
     DeviceManagementServiceIntegrationTest,
     testing::Values(&DeviceManagementServiceIntegrationTest::InitCannedResponse,
                     &DeviceManagementServiceIntegrationTest::InitTestServer));
 
 }  // namespace policy