Fix the crash of DisplayErrorDialog.

ash/display/display_error_dialog_unittest.cc

+// Copyright (c) 2013 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 "ash/display/display_error_dialog.h"
+
+#include "ash/shell.h"
+#include "ash/test/ash_test_base.h"
+#include "ui/aura/window.h"
+#include "ui/views/widget/widget.h"
+
+namespace ash {
+namespace internal {
+
+typedef test::AshTestBase DisplayErrorDialogTest;
+
+// The test cases in this file usually check if the showing dialog doesn't
+// cause any crashes, and the code doesn't cause any memory leaks.
+TEST_F(DisplayErrorDialogTest, Normal) {
+  UpdateDisplay("200x200,300x300");
+  DisplayErrorDialog::ShowDialog();
+  DisplayErrorDialog* dialog = DisplayErrorDialog::GetInstanceForTest();
+  EXPECT_TRUE(dialog);
+  EXPECT_TRUE(dialog->GetWidget()->IsVisible());
+  EXPECT_EQ(Shell::GetAllRootWindows()[1],
+            dialog->GetWidget()->GetNativeView()->GetRootWindow());
+}
+
+TEST_F(DisplayErrorDialogTest, CallTwice) {
+  UpdateDisplay("200x200,300x300");
+  DisplayErrorDialog::ShowDialog();
+  DisplayErrorDialog* dialog = DisplayErrorDialog::GetInstanceForTest();
+  EXPECT_TRUE(dialog);
+  DisplayErrorDialog::ShowDialog();
+  EXPECT_EQ(dialog, DisplayErrorDialog::GetInstanceForTest());
+}
+
+TEST_F(DisplayErrorDialogTest, SingleDisplay) {
+  UpdateDisplay("200x200");
+  DisplayErrorDialog::ShowDialog();
+  DisplayErrorDialog* dialog = DisplayErrorDialog::GetInstanceForTest();
+  EXPECT_TRUE(dialog);
+  EXPECT_TRUE(dialog->GetWidget()->IsVisible());
+  EXPECT_EQ(Shell::GetInstance()->GetPrimaryRootWindow(),
+            dialog->GetWidget()->GetNativeView()->GetRootWindow());
+}
+
+TEST_F(DisplayErrorDialogTest, DisplayDisconnected) {
+  UpdateDisplay("200x200,300x300");
+  DisplayErrorDialog::ShowDialog();
+  DisplayErrorDialog* dialog = DisplayErrorDialog::GetInstanceForTest();
+  EXPECT_TRUE(dialog);
+
+  UpdateDisplay("200x200");
+  // Disconnection will close the dialog but we have to run all pending tasks
+  // to make the effect of the close.
+  RunAllPendingInMessageLoop();
+  EXPECT_FALSE(DisplayErrorDialog::GetInstanceForTest());
+}
+
+}  // namespace internal
+}  // namespace ash