Layout login/lock screen, shelf and app list inside the first display in Unified Desktop mode

ash/system/web_notification/ash_popup_alignment_delegate_unittest.cc

 // Copyright 2014 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/system/web_notification/ash_popup_alignment_delegate.h"
 
 #include <vector>
 
 #include "ash/display/display_manager.h"
 #include "ash/shelf/shelf_layout_manager.h"
 #include "ash/shelf/shelf_types.h"
 #include "ash/shell.h"
 #include "ash/shell_window_ids.h"
 #include "ash/test/ash_test_base.h"
+#include "ash/test/display_manager_test_api.h"
 #include "ui/gfx/screen.h"
 #include "ui/message_center/message_center_style.h"
 
 namespace ash {
 
 class AshPopupAlignmentDelegateTest : public test::AshTestBase {
  public:
   AshPopupAlignmentDelegateTest() {}
   ~AshPopupAlignmentDelegateTest() override {}
 
   void SetUp() override {
     test::AshTestBase::SetUp();
-    alignment_delegate_.reset(new AshPopupAlignmentDelegate());
-    alignment_delegate_->StartObserving(
-        Shell::GetScreen(), Shell::GetScreen()->GetPrimaryDisplay());
+    SetAlignmentDelegate(make_scoped_ptr(new AshPopupAlignmentDelegate()));
   }
 
   void TearDown() override {
     alignment_delegate_.reset();
     test::AshTestBase::TearDown();
   }
 
  protected:
   enum Position {
     TOP_LEFT,
     TOP_RIGHT,
     BOTTOM_LEFT,
     BOTTOM_RIGHT,
     OUTSIDE
   };
 
   AshPopupAlignmentDelegate* alignment_delegate() {
     return alignment_delegate_.get();
   }
 
+  void SetAlignmentDelegate(scoped_ptr<AshPopupAlignmentDelegate> delegate) {
+    if (!delegate.get()) {
+      alignment_delegate_.reset();
+      return;
+    }
+    alignment_delegate_ = delegate.Pass();
+    alignment_delegate_->StartObserving(
+        Shell::GetScreen(), Shell::GetScreen()->GetPrimaryDisplay());
+    // Update the layout
+    alignment_delegate_->OnDisplayWorkAreaInsetsChanged();
+  }
+
   Position GetPositionInDisplay(const gfx::Point& point) {
     const gfx::Rect& work_area =
         Shell::GetScreen()->GetPrimaryDisplay().work_area();
     const gfx::Point center_point = work_area.CenterPoint();
     if (work_area.x() > point.x() || work_area.y() > point.y() ||
         work_area.right() < point.x() || work_area.bottom() < point.y()) {
       return OUTSIDE;
     }
 
     if (center_point.x() < point.x())
@@ skipping 151 matching lines @@
   int baseline = alignment_delegate()->GetBaseLine();
 
   const int kTrayHeight = 100;
   alignment_delegate()->SetSystemTrayHeight(kTrayHeight);
 
   EXPECT_EQ(origin_x, alignment_delegate()->GetToastOriginX(toast_size));
   EXPECT_EQ(baseline - kTrayHeight - message_center::kMarginBetweenItems,
             alignment_delegate()->GetBaseLine());
 }
 
+TEST_F(AshPopupAlignmentDelegateTest, Unified) {
+  if (!SupportsMultipleDisplays())
+    return;
+  DisplayManager* display_manager = Shell::GetInstance()->display_manager();
+  test::DisplayManagerTestApi test_api(display_manager);
+  test_api.SetDefaultMultiDisplayMode(DisplayManager::UNIFIED);
+  display_manager->SetMultiDisplayMode(DisplayManager::UNIFIED);
+
+  // Reset the delegate as the primary display's shelf will be destroyed during
+  // transition.
+  SetAlignmentDelegate(scoped_ptr<AshPopupAlignmentDelegate>());
+
+  UpdateDisplay("600x600,800x800");
+  SetAlignmentDelegate(make_scoped_ptr(new AshPopupAlignmentDelegate()));
+
+  EXPECT_GT(600,
+            alignment_delegate()->GetToastOriginX(gfx::Rect(0, 0, 10, 10)));
+}
+
 }  // namespace ash