Revert of Use a standard message box for DesktopCastingWarningView.

chrome/browser/ui/ash/multi_user/user_switch_util_unittest.cc

Index: chrome/browser/ui/ash/multi_user/user_switch_util_unittest.cc
diff --git a/chrome/browser/ui/ash/multi_user/user_switch_util_unittest.cc b/chrome/browser/ui/ash/multi_user/user_switch_util_unittest.cc
index 0c5acab85a2711c57aa8a808e18dff043d6f643e..623e65db3f6829f58f91af99b1b6265b814794d4 100644
--- a/chrome/browser/ui/ash/multi_user/user_switch_util_unittest.cc
+++ b/chrome/browser/ui/ash/multi_user/user_switch_util_unittest.cc
@@ -7,10 +7,7 @@
 #include "ash/common/wm/overview/window_selector_controller.h"
 #include "ash/common/wm_shell.h"
 #include "ash/test/ash_test_base.h"
-#include "base/run_loop.h"
-#include "base/threading/thread_task_runner_handle.h"
 #include "chrome/browser/ui/ash/multi_user/user_switch_util.h"
-#include "chrome/browser/ui/simple_message_box.h"
 #include "ui/aura/window.h"
 
 namespace ash {
@@ -71,11 +68,19 @@
   // The passed |action| type parameter defines the outcome (which will be
   // checked) and the action the user will choose.
   void SwitchUser(ActionType action) {
-    base::ThreadTaskRunnerHandle::Get()->PostTask(
-        FROM_HERE, base::Bind(&CloseMessageBox, action));
     TrySwitchingActiveUser(base::Bind(&TrySwitchingUserTest::SwitchCallback,
                                       base::Unretained(this)));
-    base::RunLoop().RunUntilIdle();
+    switch (action) {
+      case NO_DIALOG:
+        EXPECT_TRUE(!TestAndTerminateDesktopCastingWarningForTest(true));
+        return;
+      case ACCEPT_DIALOG:
+        EXPECT_TRUE(TestAndTerminateDesktopCastingWarningForTest(true));
+        return;
+      case DECLINE_DIALOG:
+        EXPECT_TRUE(TestAndTerminateDesktopCastingWarningForTest(false));
+        return;
+    }
   }
 
   // Called when the user will get actually switched.
@@ -105,20 +110,6 @@
   int switch_callback_hit_count() const { return switch_callback_hit_count_; }
 
  private:
-  static void CloseMessageBox(ActionType action) {
-    switch (action) {
-      case NO_DIALOG:
-        EXPECT_FALSE(chrome::CloseMessageBoxForTest(true));
-        return;
-      case ACCEPT_DIALOG:
-        EXPECT_TRUE(chrome::CloseMessageBoxForTest(true));
-        return;
-      case DECLINE_DIALOG:
-        EXPECT_TRUE(chrome::CloseMessageBoxForTest(false));
-        return;
-    }
-  }
-
   // The two items from the SystemTray for the screen capture / share
   // functionality.
   ScreenTrayItem* capture_item_;