Add GetDisplayWithDisplayId to display::Screen.

ash/test/ash_test_base.h

Index: ash/test/ash_test_base.h
diff --git a/ash/test/ash_test_base.h b/ash/test/ash_test_base.h
index 8e11877c3363689b984cff198d7c860053a0e68f..2c53e04786cc522dd688bc22319edc6950b6c8ac 100644
--- a/ash/test/ash_test_base.h
+++ b/ash/test/ash_test_base.h
@@ -76,6 +76,12 @@ class AshTestBase : public testing::Test {
   // See ash::test::DisplayManagerTestApi::UpdateDisplay for more details.
   void UpdateDisplay(const std::string& display_specs);
 
+  // Set the 1st display as an internal display and returns the display Id for
+  // the internal display.
+  // See ash::test::DisplayManagerTestApi::SetFirstDisplayAsInternalDisplay for
+  // more details.
+  int64_t SetFirstDisplayAsInternalDisplay();
+
   // Returns a root Window. Usually this is the active root Window, but that
   // method can return NULL sometimes, and in those cases, we fall back on the
   // primary root Window.