Modal window in user session not blocks user adding screen.

ash/test/ash_test_base.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/test/ash_test_base.h"
 
 #include <string>
 #include <vector>
 
 #include "ash/ash_switches.h"
@@ skipping 271 matching lines @@
 void AshTestBase::SetUserLoggedIn(bool user_logged_in) {
   ash_test_helper_->test_shell_delegate()->test_session_state_delegate()->
       SetHasActiveUser(user_logged_in);
 }
 
 void AshTestBase::SetCanLockScreen(bool can_lock_screen) {
   ash_test_helper_->test_shell_delegate()->test_session_state_delegate()->
       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