Try to enable software mirror mode in UI message loop after Init()

ash/display/mirror_window_controller_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/mirror_window_controller.h"
 
+#include "ash/ash_switches.h"
 #include "ash/display/display_manager.h"
 #include "ash/shell.h"
 #include "ash/test/ash_test_base.h"
 #include "ash/test/mirror_window_test_api.h"
+#include "base/command_line.h"
 #include "base/strings/stringprintf.h"
 #include "ui/aura/root_window.h"
 #include "ui/aura/test/event_generator.h"
 #include "ui/aura/test/test_window_delegate.h"
 #include "ui/aura/test/test_windows.h"
 #include "ui/aura/window.h"
 #include "ui/base/hit_test.h"
 
 namespace ash {
 namespace internal {
 
 namespace {
 DisplayInfo CreateDisplayInfo(int64 id, const gfx::Rect& bounds) {
   DisplayInfo info(id, base::StringPrintf("x-%d", static_cast<int>(id)), false);
   info.SetBounds(bounds);
   return info;
 }
 
+class MirrorOnBootTest : public test::AshTestBase {
+ public:
+  MirrorOnBootTest() {}
+  virtual ~MirrorOnBootTest() {}
+
+  virtual void SetUp() OVERRIDE {
+    CommandLine::ForCurrentProcess()->AppendSwitchASCII(
+        switches::kAshHostWindowBounds, "1+1-300x300,1+301-300x300");
+    CommandLine::ForCurrentProcess()->AppendSwitch(
+        switches::kAshEnableSoftwareMirroring);
+    test::AshTestBase::SetUp();
+  }
+  virtual void TearDown() OVERRIDE {
+    test::AshTestBase::TearDown();
+  }
+
+ private:
+  DISALLOW_COPY_AND_ASSIGN(MirrorOnBootTest);
+};
+
 }
 
 typedef test::AshTestBase MirrorWindowControllerTest;
 
 #if defined(OS_WIN)
 // Software mirroring does not work on win.
 #define MAYBE_MirrorCursorBasic DISABLED_MirrorCursorBasic
 #define MAYBE_MirrorCursorLocations DISABLED_MirrorCursorLocations
 #define MAYBE_MirrorCursorRotate DISABLED_MirrorCursorRotate
 #define MAYBE_DockMode DISABLED_DockMode
+#define MAYBE_MirrorOnBoot DISABLED_MirrorOnBoot
 #else
 #define MAYBE_MirrorCursorBasic MirrorCursorBasic
 #define MAYBE_MirrorCursorLocations MirrorCursorLocations
 #define MAYBE_MirrorCursorRotate MirrorCursorRotate
 #define MAYBE_DockMode DockMode
+#define MAYBE_MirrorOnBoot MirrorOnBoot
 #endif
 
 TEST_F(MirrorWindowControllerTest, MAYBE_MirrorCursorBasic) {
   test::MirrorWindowTestApi test_api;
   aura::test::TestWindowDelegate test_window_delegate;
   test_window_delegate.set_window_component(HTTOP);
 
   DisplayManager* display_manager = Shell::GetInstance()->display_manager();
   display_manager->SetSoftwareMirroring(true);
   UpdateDisplay("400x400,400x400");
@@ skipping 190 matching lines @@
   display_info_list.clear();
   display_info_list.push_back(internal_display_info);
   display_info_list.push_back(external_display_info);
   display_manager->SetSoftwareMirroring(true);
   display_manager->UpdateDisplays(display_info_list);
   EXPECT_EQ(1U, display_manager->GetNumDisplays());
   EXPECT_TRUE(display_manager->IsMirrored());
   EXPECT_EQ(external_id, display_manager->mirrored_display().id());
 }
 
+TEST_F(MirrorOnBootTest, MAYBE_MirrorOnBoot) {
+  DisplayManager* display_manager = Shell::GetInstance()->display_manager();
+  EXPECT_TRUE(display_manager->IsMirrored());
+  RunAllPendingInMessageLoop();
+  test::MirrorWindowTestApi test_api;
+  EXPECT_TRUE(test_api.GetRootWindow());
+}
+
 }  // namsspace internal
 }  // namespace ash