Modal window in user session not blocks user adding screen.

ash/test/ash_test_base.cc

Index: ash/test/ash_test_base.cc
diff --git a/ash/test/ash_test_base.cc b/ash/test/ash_test_base.cc
index 6917e0ed92704172046f7e6f0a7b2f9e5a7de35f..7403638e004904101f8656b3608152dc6c539092 100644
--- a/ash/test/ash_test_base.cc
+++ b/ash/test/ash_test_base.cc
@@ -289,5 +289,37 @@ void AshTestBase::SetCanLockScreen(bool can_lock_screen) {
       SetCanLockScreen(can_lock_screen);
 }
 
+void AshTestBase::SetUserAddingScreenRunning(bool user_adding_screen_running) {
+  ash_test_helper_->test_shell_delegate()->test_session_state_delegate()->
+      SetUserAddingScreenRunning(user_adding_screen_running);
+}
+
+void AshTestBase::BlockUserSession(UserSessionBlockReason block_reason) {
+  switch (block_reason) {
+    case BLOCKED_BY_LOCK_SCREEN:
+      SetSessionStarted(true);
+      SetUserAddingScreenRunning(false);
+      Shell::GetInstance()->session_state_delegate()->LockScreen();
+      break;
+    case BLOCKED_BY_LOGIN_SCREEN:
+      SetUserAddingScreenRunning(false);
+      SetSessionStarted(false);
+      break;
+    case BLOCKED_BY_USER_ADDING_SCREEN:
+      SetUserAddingScreenRunning(true);
+      SetSessionStarted(true);
+    default:
+      NOTREACHED();
+      break;
+  }
+}
+
+void AshTestBase::UnblockUserSession() {
+  Shell::GetInstance()->session_state_delegate()->UnlockScreen();
+  SetSessionStarted(true);
+  SetUserAddingScreenRunning(false);
+}
+
+
 }  // namespace test
 }  // namespace ash