[Autofill] Fix MessageLoop in test.

chrome/browser/ui/autofill/autofill_dialog_controller_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/memory/ref_counted.h"
 #include "base/message_loop.h"
 #include "base/time.h"
 #include "base/utf_string_conversions.h"
 #include "chrome/browser/ui/autofill/autofill_dialog_controller_impl.h"
 #include "chrome/browser/ui/autofill/autofill_dialog_view.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/tabs/tab_strip_model.h"
 #include "chrome/test/base/in_process_browser_test.h"
 #include "components/autofill/browser/autofill_metrics.h"
 #include "components/autofill/common/form_data.h"
@@ skipping 57 matching lines @@
  public:
   TestAutofillDialogController(content::WebContents* contents,
                                const FormData& form_data,
                                const AutofillMetrics& metric_logger,
                                const DialogType dialog_type)
       : AutofillDialogControllerImpl(contents,
                                      form_data,
                                      GURL(),
                                      dialog_type,
                                      base::Bind(&MockCallback)),
-        metric_logger_(metric_logger) {
+        metric_logger_(metric_logger),
+        message_loop_runner_(new content::MessageLoopRunner) {
     DisableWallet();
   }
 
   virtual ~TestAutofillDialogController() {}
 
   virtual void ViewClosed() OVERRIDE {
     AutofillDialogControllerImpl::ViewClosed();
-    MessageLoop::current()->Quit();
+    message_loop_runner_->Quit();
   }
 
   virtual bool InputIsValid(AutofillFieldType type,
                             const string16& value) OVERRIDE {
     return true;
   }
 
   virtual std::vector<AutofillFieldType> InputsAreValid(
       const DetailOutputMap& inputs, ValidationType validation_type) OVERRIDE {
     return std::vector<AutofillFieldType>();
   }
 
   // Increase visibility for testing.
   AutofillDialogView* view() { return AutofillDialogControllerImpl::view(); }
 
+  void RunMessageLoop(void) {

[Ilya Sherman, 2013/04/17 00:03:55]

nit: Docs

[Ilya Sherman, 2013/04/17 00:03:55]

nit: Omit the second "void"

+    message_loop_runner_->Run();
+  }
+
  private:
   // To specify our own metric logger.
   virtual const AutofillMetrics& GetMetricLogger() const OVERRIDE {
     return metric_logger_;
   }
 
   const AutofillMetrics& metric_logger_;
+  scoped_refptr<content::MessageLoopRunner> message_loop_runner_;
 
   DISALLOW_COPY_AND_ASSIGN(TestAutofillDialogController);
 };
 
 }  // namespace
 
 class AutofillDialogControllerTest : public InProcessBrowserTest {
  public:
   AutofillDialogControllerTest() {}
   virtual ~AutofillDialogControllerTest() {}
@@ skipping 31 matching lines @@
   // Submit the form data.
   {
     MockAutofillMetrics metric_logger;
     TestAutofillDialogController* dialog_controller =
         new TestAutofillDialogController(
             GetActiveWebContents(), form, metric_logger,
             DIALOG_TYPE_REQUEST_AUTOCOMPLETE);
     dialog_controller->Show();
     dialog_controller->view()->SubmitForTesting();
 
-    content::RunMessageLoop();
+    dialog_controller->RunMessageLoop();
 
     EXPECT_EQ(AutofillMetrics::DIALOG_ACCEPTED,
               metric_logger.dialog_dismissal_action());
     EXPECT_EQ(DIALOG_TYPE_REQUEST_AUTOCOMPLETE, metric_logger.dialog_type());
   }
 
   // Cancel out of the dialog.
   {
     MockAutofillMetrics metric_logger;
     TestAutofillDialogController* dialog_controller =
         new TestAutofillDialogController(
             GetActiveWebContents(), form, metric_logger,
             DIALOG_TYPE_AUTOCHECKOUT);
     dialog_controller->Show();
     dialog_controller->view()->CancelForTesting();
 
-    content::RunMessageLoop();
+    dialog_controller->RunMessageLoop();
 
     EXPECT_EQ(AutofillMetrics::DIALOG_CANCELED,
               metric_logger.dialog_dismissal_action());
     EXPECT_EQ(DIALOG_TYPE_AUTOCHECKOUT, metric_logger.dialog_type());
   }
 
   // Take some other action that dismisses the dialog.
   {
     MockAutofillMetrics metric_logger;
     TestAutofillDialogController* dialog_controller =
         new TestAutofillDialogController(
             GetActiveWebContents(), form, metric_logger,
             DIALOG_TYPE_AUTOCHECKOUT);
     dialog_controller->Show();
     dialog_controller->Hide();
 
-    content::RunMessageLoop();
+    dialog_controller->RunMessageLoop();
 
     EXPECT_EQ(AutofillMetrics::DIALOG_CANCELED,
               metric_logger.dialog_dismissal_action());
     EXPECT_EQ(DIALOG_TYPE_AUTOCHECKOUT, metric_logger.dialog_type());
   }
 
   // Test Autocheckout success metrics.
   {
     MockAutofillMetrics metric_logger;
     TestAutofillDialogController* dialog_controller =
         new TestAutofillDialogController(
             GetActiveWebContents(), form, metric_logger,
             DIALOG_TYPE_AUTOCHECKOUT);
     dialog_controller->Show();
     dialog_controller->view()->SubmitForTesting();
 
     EXPECT_EQ(AutofillMetrics::DIALOG_ACCEPTED,
               metric_logger.dialog_dismissal_action());
     EXPECT_EQ(DIALOG_TYPE_AUTOCHECKOUT, metric_logger.dialog_type());
 
     dialog_controller->Hide();
 
-    content::RunMessageLoop();
+    dialog_controller->RunMessageLoop();
 
     EXPECT_EQ(AutofillMetrics::AUTOCHECKOUT_SUCCEEDED,
               metric_logger.autocheckout_status());
   }
 
   // Test Autocheckout failure metric.
   {
     MockAutofillMetrics metric_logger;
     TestAutofillDialogController* dialog_controller =
         new TestAutofillDialogController(
             GetActiveWebContents(), form, metric_logger,
             DIALOG_TYPE_AUTOCHECKOUT);
     dialog_controller->Show();
     dialog_controller->view()->SubmitForTesting();
 
     EXPECT_EQ(AutofillMetrics::DIALOG_ACCEPTED,
               metric_logger.dialog_dismissal_action());
     EXPECT_EQ(DIALOG_TYPE_AUTOCHECKOUT, metric_logger.dialog_type());
 
     dialog_controller->OnAutocheckoutError();
     dialog_controller->view()->CancelForTesting();
 
-    content::RunMessageLoop();
+    dialog_controller->RunMessageLoop();
 
     EXPECT_EQ(AutofillMetrics::AUTOCHECKOUT_FAILED,
               metric_logger.autocheckout_status());
   }
 }
 
 }  // namespace autofill