Add context to gfx::Screen calls in support of simultaneous desktop+ash

ash/wm/base_layout_manager_unittest.cc

 // Copyright (c) 2012 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/base_layout_manager.h"
 
 #include "ash/screen_ash.h"
 #include "ash/shell.h"
 #include "ash/shell_window_ids.h"
 #include "ash/test/ash_test_base.h"
@@ skipping 146 matching lines @@
   gfx::Rect old_bounds = window->bounds();
   Shell::GetPrimaryRootWindow()->SetHostSize(gfx::Size(800, 600));
   EXPECT_EQ(old_bounds.width(), window->bounds().width());
   EXPECT_EQ(old_bounds.height(), window->bounds().height());
 }
 
 // Tests that a maximized window with too-large restore bounds will be restored
 // to smaller than the full work area.
 TEST_F(BaseLayoutManagerTest, BoundsWithScreenEdgeVisible) {
   // Create a window with bounds that fill the screen.
-  gfx::Rect bounds = gfx::Screen::GetPrimaryDisplay().bounds();
+  gfx::Rect bounds = gfx::Screen::GetPrimaryDisplay(
+      ash::Shell::GetRootNativeView()).bounds();
   scoped_ptr<aura::Window> window(CreateTestWindow(bounds));
   // Maximize it, which writes the old bounds to restore bounds.
   window->SetProperty(aura::client::kShowStateKey, ui::SHOW_STATE_MAXIMIZED);
   // Restore it.
   window->SetProperty(aura::client::kShowStateKey, ui::SHOW_STATE_NORMAL);
   // It should have the default maximized window bounds, inset by the grid size.
   int grid_size = internal::WorkspaceWindowResizer::kScreenEdgeInset;
   gfx::Rect max_bounds =
       ash::ScreenAsh::GetMaximizedWindowBoundsInParent(window.get());
   max_bounds.Inset(grid_size, grid_size);
@@ skipping 49 matching lines @@
             GetRestoreBoundsInParent(window.get()).ToString());
   EXPECT_TRUE(wm::IsWindowMaximized(window.get()));
 
   wm::RestoreWindow(window.get());
   EXPECT_EQ(bounds.ToString(), window->bounds().ToString());
 }
 
 }  // namespace
 
 }  // namespace ash