Add Get/SetDisplayLayout to chrome.system.display extension API

chrome/browser/extensions/display_info_provider_chromeos_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 "extensions/browser/api/system_display/display_info_provider.h"
 
 #include <stdint.h>
 
 #include "ash/common/ash_switches.h"
 #include "ash/display/display_manager.h"
 #include "ash/display/screen_orientation_controller_chromeos.h"
 #include "ash/screen_util.h"
 #include "ash/shell.h"
 #include "ash/test/ash_test_base.h"
 #include "ash/test/display_manager_test_api.h"
 #include "ash/wm/maximize_mode/maximize_mode_controller.h"
 #include "base/command_line.h"
 #include "base/macros.h"
 #include "base/strings/string_number_conversions.h"
 #include "base/strings/stringprintf.h"
 #include "extensions/common/api/system_display.h"
 #include "ui/display/display.h"
 #include "ui/display/manager/display_layout.h"
 #include "ui/gfx/geometry/rect.h"
 
 namespace extensions {
 namespace {
 
+using DisplayUnitInfoList = DisplayInfoProvider::DisplayUnitInfoList;
+using DisplayLayoutList = DisplayInfoProvider::DisplayLayoutList;
+
 class DisplayInfoProviderChromeosTest : public ash::test::AshTestBase {
  public:
   DisplayInfoProviderChromeosTest() {}
 
   ~DisplayInfoProviderChromeosTest() override {}
 
   void SetUp() override {
     base::CommandLine::ForCurrentProcess()->AppendSwitch(
         ash::switches::kAshUseFirstDisplayAsInternal);
     ash::test::AshTestBase::SetUp();
@@ skipping 388 matching lines @@
   GetDisplayManager()->SetLayoutForCurrentDisplays(
       ash::test::CreateDisplayLayout(display::DisplayPlacement::BOTTOM, 80));
 
   result = DisplayInfoProvider::Get()->GetAllDisplaysInfo();
   ASSERT_EQ(2u, result.size());
   EXPECT_EQ("0,0 600x600", SystemInfoDisplayBoundsToString(result[0].bounds));
   EXPECT_EQ("80,600 400x520",
             SystemInfoDisplayBoundsToString(result[1].bounds));
 }
 
+TEST_F(DisplayInfoProviderChromeosTest, Layout) {
+  UpdateDisplay("500x400,500x400,500x400");
+
+  DisplayUnitInfoList displays =
+      DisplayInfoProvider::Get()->GetAllDisplaysInfo();
+  std::string primary_id = displays[0].id;
+  ASSERT_EQ(3u, displays.size());
+
+  DisplayLayoutList layout = DisplayInfoProvider::Get()->GetDisplayLayout();
+
+  ASSERT_EQ(2u, layout.size());
+
+  // Confirm layout.
+  EXPECT_EQ(displays[1].id, layout[0].id);
+  EXPECT_EQ(primary_id, layout[0].parent_id);
+  EXPECT_EQ(api::system_display::LAYOUT_POSITION_RIGHT, layout[0].position);
+  EXPECT_EQ(0, layout[0].offset);
+
+  EXPECT_EQ(displays[2].id, layout[1].id);
+  EXPECT_EQ(layout[0].id, layout[1].parent_id);
+  EXPECT_EQ(api::system_display::LAYOUT_POSITION_RIGHT, layout[1].position);
+  EXPECT_EQ(0, layout[1].offset);
+
+  // Modify layout.
+  layout[0].offset = 100;
+  layout[1].parent_id = primary_id;
+  layout[1].position = api::system_display::LAYOUT_POSITION_BOTTOM;
+  layout[1].offset = -100;
+
+  // Update with modified layout.
+  ASSERT_TRUE(DisplayInfoProvider::Get()->SetDisplayLayout(layout));
+
+  // Get updated layout.
+  layout = DisplayInfoProvider::Get()->GetDisplayLayout();
+
+  // Confirm modified layout.
+  EXPECT_EQ(displays[1].id, layout[0].id);
+  EXPECT_EQ(primary_id, layout[0].parent_id);
+  EXPECT_EQ(api::system_display::LAYOUT_POSITION_RIGHT, layout[0].position);
+  EXPECT_EQ(100, layout[0].offset);
+
+  EXPECT_EQ(displays[2].id, layout[1].id);
+  EXPECT_EQ(primary_id, layout[1].parent_id);
+  EXPECT_EQ(api::system_display::LAYOUT_POSITION_BOTTOM, layout[1].position);
+  EXPECT_EQ(-100, layout[1].offset);
+
+  // TODO(stevenjb/oshima): Implement and test illegal layout prevention.
+
+  // Test setting invalid layout fails.
+  layout[0].parent_id = displays[2].id;
+  layout[1].parent_id = displays[1].id;
+  EXPECT_FALSE(DisplayInfoProvider::Get()->SetDisplayLayout(layout));
+}
+
 TEST_F(DisplayInfoProviderChromeosTest, SetBoundsOriginLeftExact) {
   UpdateDisplay("1200x600,520x400");
 
   const display::Display& secondary = ash::ScreenUtil::GetSecondaryDisplay();
   api::system_display::DisplayProperties info;
   info.bounds_origin_x.reset(new int(-520));
   info.bounds_origin_y.reset(new int(50));
 
   bool success = false;
   std::string error;
@@ skipping 679 matching lines @@
   CallSetDisplayUnitInfo(base::Int64ToString(id), info, &success, &error);
   ASSERT_TRUE(success);
 
   // Verify that other_mode now matches the active mode.
   active_mode = GetDisplayManager()->GetActiveModeForDisplayId(id);
   EXPECT_TRUE(active_mode.IsEquivalent(other_mode_ash));
 }
 
 }  // namespace
 }  // namespace extensions