Handles users rotating screen too early

ash/test/shell_test_api.cc

Index: ash/test/shell_test_api.cc
diff --git a/ash/test/shell_test_api.cc b/ash/test/shell_test_api.cc
index 8c8293c4f9eb088522a151751330807de86fbe76..0533f62dc7823e8159192d8ce4802aacb73f1de6 100644
--- a/ash/test/shell_test_api.cc
+++ b/ash/test/shell_test_api.cc
@@ -38,6 +38,17 @@ void ShellTestApi::DisableDisplayAnimator() {
   shell_->display_configuration_controller()->ResetAnimatorForTest();
 }
 
+ScreenRotationAnimator* ShellTestApi::GetOrMakeAnimatorByDisplayId(
+    int64_t display_id) {
+  return shell_->display_configuration_controller()
+      ->GetOrMakeAnimatorByDisplayId(display_id);
+}
+
+int ShellTestApi::DisplayIdAnimatorMapSize() {
+  return shell_->display_configuration_controller()
+      ->display_animator_map_.size();
+}
+
 void ShellTestApi::SetSessionStateDelegate(
     SessionStateDelegate* session_state_delegate) {
   shell_->session_state_delegate_.reset(session_state_delegate);