text: // Copyright 2019 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/wm/work_area_insets.h"
  
  #include "ash/root_window_controller.h"
  #include "ash/shelf/shelf.h"
  #include "ash/shell.h"
  #include "ash/test/ash_test_base.h"
  #include "ui/aura/window.h"
  #include "ui/gfx/geometry/rect.h"
  
  namespace ash {
  
  using WorkAreaInsetsTest = AshTestBase;
  
  TEST_F(WorkAreaInsetsTest, WorkAreaBoundsForTwoScreens) {
    UpdateDisplay("800x600,800x600");
  
    // The work area for the default configuration is simply the window bounds
    // minus the shelf bounds.
    gfx::Rect shelf_bounds =
        Shell::GetPrimaryRootWindowController()->shelf()->GetIdealBounds();
    gfx::Size expected_work_area_size =
        gfx::Size(800, 600 - shelf_bounds.height());
  
    aura::Window::Windows root_windows = Shell::GetAllRootWindows();
    gfx::Rect work_area1 =
        WorkAreaInsets::ForWindow(root_windows[0])->user_work_area_bounds();
    EXPECT_EQ(gfx::Point(0, 0), work_area1.origin());
    EXPECT_EQ(expected_work_area_size, work_area1.size());
  
    gfx::Rect work_area2 =
        WorkAreaInsets::ForWindow(root_windows[1])->user_work_area_bounds();
    EXPECT_EQ(gfx::Point(800, 0), work_area2.origin());
    EXPECT_EQ(expected_work_area_size, work_area2.size());
  }
  
  TEST_F(WorkAreaInsetsTest, ComputeStableWorkAreaForTwoScreens) {
    UpdateDisplay("800x600,800x600");
  
    // The work area for the default configuration is simply the window bounds
    // minus the shelf bounds.
    gfx::Rect shelf_bounds =
        Shell::GetPrimaryRootWindowController()->shelf()->GetIdealBounds();
    gfx::Size expected_work_area_size =
        gfx::Size(800, 600 - shelf_bounds.height());
  
    aura::Window::Windows root_windows = Shell::GetAllRootWindows();
    gfx::Rect stable_work_area1 =
        WorkAreaInsets::ForWindow(root_windows[0])->ComputeStableWorkArea();
    EXPECT_EQ(gfx::Point(0, 0), stable_work_area1.origin());
    EXPECT_EQ(expected_work_area_size, stable_work_area1.size());
  
    gfx::Rect stable_work_area2 =
        WorkAreaInsets::ForWindow(root_windows[1])->ComputeStableWorkArea();
    EXPECT_EQ(gfx::Point(800, 0), stable_work_area2.origin());
    EXPECT_EQ(expected_work_area_size, stable_work_area2.size());
  }
  
  }  // namespace ash
  