Refactor modality-specific behavior from ConstrainedWindowViews to WebContentsModalDialogManager

chrome/browser/ui/web_contents_modal_dialog_manager_unittest.cc

Index: chrome/browser/ui/web_contents_modal_dialog_manager_unittest.cc
diff --git a/chrome/browser/ui/web_contents_modal_dialog_manager_unittest.cc b/chrome/browser/ui/web_contents_modal_dialog_manager_unittest.cc
index 1532136fc71102f1b442beac2e1eb3fcf3d46eaf..369b55c229f663259bf13cd263e5444508caf96d 100644
--- a/chrome/browser/ui/web_contents_modal_dialog_manager_unittest.cc
+++ b/chrome/browser/ui/web_contents_modal_dialog_manager_unittest.cc
@@ -52,19 +52,21 @@ class WebContentsModalDialogCloseTest : public WebContentsModalDialog {
   content::WebContents* web_contents_;
 };
 
-TEST_F(WebContentsModalDialogManagerTest, ConstrainedWindows) {
+TEST_F(WebContentsModalDialogManagerTest, WebContentsModalDialogs) {
   WebContentsModalDialogCloseTest window(web_contents());
   window.close_count = 0;
   WebContentsModalDialogManager* web_contents_modal_dialog_manager =
       WebContentsModalDialogManager::FromWebContents(web_contents());
+  WebContentsModalDialogManager::TestApi test_api(
+      web_contents_modal_dialog_manager);
+
+  test_api.ResetUIDelegate(NULL);
 
   const int kWindowCount = 4;
   for (int i = 0; i < kWindowCount; i++)
     web_contents_modal_dialog_manager->AddDialog(&window);
   EXPECT_EQ(window.close_count, 0);
 
-  WebContentsModalDialogManager::TestApi test_api(
-      web_contents_modal_dialog_manager);
   test_api.CloseAllDialogs();
   EXPECT_EQ(window.close_count, kWindowCount);
 }