ash: Extract animator from PowerButtonController

ash/wm/power_button_controller_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/power_button_controller.h"
 
 #include "ash/shell.h"
 #include "ash/test/ash_test_base.h"
 #include "ash/wm/cursor_manager.h"
 #include "base/memory/scoped_ptr.h"
@@ skipping 66 matching lines @@
 TEST_F(PowerButtonControllerTest, LegacyLockAndShutDown) {
   controller_->set_has_legacy_power_button_for_test(true);
   controller_->OnLoginStateChanged(user::LOGGED_IN_USER);
   controller_->OnLockStateChanged(false);
 
   // We should request that the screen be locked immediately after seeing the
   // power button get pressed.
   controller_->OnPowerButtonEvent(true, base::TimeTicks::Now());
   EXPECT_TRUE(
       test_api_->ContainersAreAnimated(
-          PowerButtonController::NON_LOCK_SCREEN_CONTAINERS,
-          PowerButtonController::ANIMATION_SLOW_CLOSE));
+          SessionStateAnimator::NON_LOCK_SCREEN_CONTAINERS,
+          SessionStateAnimator::ANIMATION_SLOW_CLOSE));
   EXPECT_TRUE(test_api_->BlackLayerIsVisible());
   EXPECT_FALSE(test_api_->hide_black_layer_timer_is_running());
   EXPECT_FALSE(test_api_->lock_timer_is_running());
   EXPECT_EQ(1, delegate_->num_lock_requests());
 
   // Notify that we locked successfully.
   controller_->OnStartingLock();
   EXPECT_TRUE(
       test_api_->ContainersAreAnimated(
-          PowerButtonController::NON_LOCK_SCREEN_CONTAINERS,
-          PowerButtonController::ANIMATION_FAST_CLOSE));
+          SessionStateAnimator::NON_LOCK_SCREEN_CONTAINERS,
+          SessionStateAnimator::ANIMATION_FAST_CLOSE));
   EXPECT_TRUE(
       test_api_->ContainersAreAnimated(
-          PowerButtonController::LOCK_SCREEN_CONTAINERS,
-          PowerButtonController::ANIMATION_HIDE));
+          SessionStateAnimator::LOCK_SCREEN_CONTAINERS,
+          SessionStateAnimator::ANIMATION_HIDE));
 
   // Notify that the lock window is visible.  We should make it fade in.
   controller_->OnLockStateChanged(true);
   EXPECT_TRUE(
       test_api_->ContainersAreAnimated(
-          PowerButtonController::GetAllLockScreenContainersMask(),
-          PowerButtonController::ANIMATION_FADE_IN));
+          SessionStateAnimator::GetAllLockScreenContainersMask(),
+          SessionStateAnimator::ANIMATION_FADE_IN));
 
   // We shouldn't progress towards the shutdown state, however.
   EXPECT_FALSE(test_api_->lock_to_shutdown_timer_is_running());
   EXPECT_FALSE(test_api_->shutdown_timer_is_running());
   controller_->OnPowerButtonEvent(false, base::TimeTicks::Now());
 
   // Hold the button again and check that we start shutting down.
   controller_->OnPowerButtonEvent(true, base::TimeTicks::Now());
   EXPECT_EQ(0, delegate_->num_shutdown_requests());
 
   // Previously we're checking that the all containers group was animated which
   // was in fact checking that
   // 1. All user session containers have transform (including wallpaper).
   //    They're in this state after lock.
   // 2. Screen locker and related containers are in fact animating
   //    (as shutdown is in progress).
   // With http://crbug.com/144737 we no longer animate user session wallpaper
   // during lock so it makes sense only to check that screen lock and related
   // containers are animated during shutdown.
   EXPECT_TRUE(
       test_api_->ContainersAreAnimated(
-          PowerButtonController::GetAllLockScreenContainersMask(),
-          PowerButtonController::ANIMATION_FAST_CLOSE));
+          SessionStateAnimator::GetAllLockScreenContainersMask(),
+          SessionStateAnimator::ANIMATION_FAST_CLOSE));
   EXPECT_FALSE(cursor_visible());
   EXPECT_TRUE(test_api_->real_shutdown_timer_is_running());
   test_api_->trigger_real_shutdown_timeout();
   EXPECT_EQ(1, delegate_->num_shutdown_requests());
 }
 
 // Test that we start shutting down immediately if the power button is pressed
 // while we're not logged in on an unofficial system.
 TEST_F(PowerButtonControllerTest, LegacyNotLoggedIn) {
   controller_->set_has_legacy_power_button_for_test(true);
@@ skipping 20 matching lines @@
   controller_->OnLoginStateChanged(user::LOGGED_IN_NONE);
   controller_->OnLockStateChanged(false);
   EXPECT_FALSE(test_api_->BlackLayerIsVisible());
 
   // Press the power button and check that we start the shutdown timer.
   controller_->OnPowerButtonEvent(true, base::TimeTicks::Now());
   EXPECT_FALSE(test_api_->lock_timer_is_running());
   EXPECT_TRUE(test_api_->shutdown_timer_is_running());
   EXPECT_TRUE(
       test_api_->ContainersAreAnimated(
-          PowerButtonController::GetAllContainersMask(),
-          PowerButtonController::ANIMATION_SLOW_CLOSE));
+          SessionStateAnimator::GetAllContainersMask(),
+          SessionStateAnimator::ANIMATION_SLOW_CLOSE));
   EXPECT_TRUE(test_api_->BlackLayerIsVisible());
 
   // Release the power button before the shutdown timer fires.
   controller_->OnPowerButtonEvent(false, base::TimeTicks::Now());
   EXPECT_FALSE(test_api_->shutdown_timer_is_running());
   EXPECT_TRUE(
       test_api_->ContainersAreAnimated(
-          PowerButtonController::GetAllContainersMask(),
-          PowerButtonController::ANIMATION_UNDO_SLOW_CLOSE));
+          SessionStateAnimator::GetAllContainersMask(),
+          SessionStateAnimator::ANIMATION_UNDO_SLOW_CLOSE));
   EXPECT_TRUE(test_api_->BlackLayerIsVisible());
 
   // We should re-hide the black layer after waiting long enough for
   // the animation to finish.
   EXPECT_TRUE(test_api_->hide_black_layer_timer_is_running());
-  test_api_->trigger_hide_black_layer_timeout();
+  test_api_->TriggerHideBlackLayerTimeout();
   EXPECT_FALSE(test_api_->BlackLayerIsVisible());
 
   // Press the button again and make the shutdown timeout fire this time.
   // Check that we start the timer for actually requesting the shutdown.
   controller_->OnPowerButtonEvent(true, base::TimeTicks::Now());
   EXPECT_TRUE(test_api_->shutdown_timer_is_running());
   test_api_->trigger_shutdown_timeout();
   EXPECT_TRUE(test_api_->real_shutdown_timer_is_running());
   EXPECT_EQ(0, delegate_->num_shutdown_requests());
   EXPECT_TRUE(
       test_api_->ContainersAreAnimated(
-          PowerButtonController::NON_LOCK_SCREEN_CONTAINERS,
-          PowerButtonController::ANIMATION_HIDE));
+          SessionStateAnimator::NON_LOCK_SCREEN_CONTAINERS,
+          SessionStateAnimator::ANIMATION_HIDE));
   EXPECT_TRUE(
       test_api_->ContainersAreAnimated(
-          PowerButtonController::GetAllLockScreenContainersMask(),
-          PowerButtonController::ANIMATION_FAST_CLOSE));
+          SessionStateAnimator::GetAllLockScreenContainersMask(),
+          SessionStateAnimator::ANIMATION_FAST_CLOSE));
 
   // When the timout fires, we should request a shutdown.
   test_api_->trigger_real_shutdown_timeout();
   EXPECT_EQ(1, delegate_->num_shutdown_requests());
 }
 
 // Test that we lock the screen and deal with unlocking correctly.
 TEST_F(PowerButtonControllerTest, LockAndUnlock) {
   controller_->set_has_legacy_power_button_for_test(false);
   controller_->OnLoginStateChanged(user::LOGGED_IN_USER);
   controller_->OnLockStateChanged(false);
   EXPECT_FALSE(test_api_->BlackLayerIsVisible());
 
   // We should initially be showing the screen locker containers, since they
   // also contain login-related windows that we want to show during the
   // logging-in animation.
   EXPECT_TRUE(
       test_api_->ContainersAreAnimated(
-          PowerButtonController::LOCK_SCREEN_CONTAINERS,
-          PowerButtonController::ANIMATION_RESTORE));
+          SessionStateAnimator::LOCK_SCREEN_CONTAINERS,
+          SessionStateAnimator::ANIMATION_RESTORE));
 
   // Press the power button and check that the lock timer is started and that we
   // start scaling the non-screen-locker containers.
   controller_->OnPowerButtonEvent(true, base::TimeTicks::Now());
   EXPECT_TRUE(test_api_->lock_timer_is_running());
   EXPECT_FALSE(test_api_->shutdown_timer_is_running());
   EXPECT_TRUE(
       test_api_->ContainersAreAnimated(
-          PowerButtonController::NON_LOCK_SCREEN_CONTAINERS,
-          PowerButtonController::ANIMATION_SLOW_CLOSE));
+          SessionStateAnimator::NON_LOCK_SCREEN_CONTAINERS,
+          SessionStateAnimator::ANIMATION_SLOW_CLOSE));
   EXPECT_TRUE(test_api_->BlackLayerIsVisible());
 
   // Release the button before the lock timer fires.
   controller_->OnPowerButtonEvent(false, base::TimeTicks::Now());
   EXPECT_FALSE(test_api_->lock_timer_is_running());
   EXPECT_TRUE(
       test_api_->ContainersAreAnimated(
-          PowerButtonController::NON_LOCK_SCREEN_CONTAINERS,
-          PowerButtonController::ANIMATION_UNDO_SLOW_CLOSE));
+          SessionStateAnimator::NON_LOCK_SCREEN_CONTAINERS,
+          SessionStateAnimator::ANIMATION_UNDO_SLOW_CLOSE));
   EXPECT_TRUE(test_api_->BlackLayerIsVisible());
   EXPECT_TRUE(test_api_->hide_black_layer_timer_is_running());
   test_api_->trigger_hide_black_layer_timeout();
   EXPECT_FALSE(test_api_->BlackLayerIsVisible());
 
   // Press the button and fire the lock timer.  We should request that the
   // screen be locked, but we should still be in the slow-close animation.
   controller_->OnPowerButtonEvent(true, base::TimeTicks::Now());
   EXPECT_TRUE(test_api_->lock_timer_is_running());
   EXPECT_EQ(0, delegate_->num_lock_requests());
   test_api_->trigger_lock_timeout();
   EXPECT_EQ(1, delegate_->num_lock_requests());
   EXPECT_TRUE(
       test_api_->ContainersAreAnimated(
-          PowerButtonController::NON_LOCK_SCREEN_CONTAINERS,
-          PowerButtonController::ANIMATION_SLOW_CLOSE));
+          SessionStateAnimator::NON_LOCK_SCREEN_CONTAINERS,
+          SessionStateAnimator::ANIMATION_SLOW_CLOSE));
   EXPECT_TRUE(test_api_->BlackLayerIsVisible());
 
   // Notify that we locked successfully.
   controller_->OnStartingLock();
   EXPECT_TRUE(
       test_api_->ContainersAreAnimated(
-          PowerButtonController::NON_LOCK_SCREEN_CONTAINERS,
-          PowerButtonController::ANIMATION_FAST_CLOSE));
+          SessionStateAnimator::NON_LOCK_SCREEN_CONTAINERS,
+          SessionStateAnimator::ANIMATION_FAST_CLOSE));
   EXPECT_TRUE(
       test_api_->ContainersAreAnimated(
-          PowerButtonController::LOCK_SCREEN_CONTAINERS,
-          PowerButtonController::ANIMATION_HIDE));
+          SessionStateAnimator::LOCK_SCREEN_CONTAINERS,
+          SessionStateAnimator::ANIMATION_HIDE));
 
   // Notify that the lock window is visible.  We should make it fade in.
   controller_->OnLockStateChanged(true);
   EXPECT_TRUE(
       test_api_->ContainersAreAnimated(
-          PowerButtonController::GetAllLockScreenContainersMask(),
-          PowerButtonController::ANIMATION_FADE_IN));
+          SessionStateAnimator::GetAllLockScreenContainersMask(),
+          SessionStateAnimator::ANIMATION_FADE_IN));
 
   // When we release the power button, the lock-to-shutdown timer should be
   // stopped.
   EXPECT_TRUE(test_api_->lock_to_shutdown_timer_is_running());
   controller_->OnPowerButtonEvent(false, base::TimeTicks::Now());
   EXPECT_FALSE(test_api_->lock_to_shutdown_timer_is_running());
 
   // Notify that the screen has been unlocked.  We should show the
   // non-screen-locker windows and hide the black layer.
   controller_->OnLockStateChanged(false);
   EXPECT_TRUE(
       test_api_->ContainersAreAnimated(
-          PowerButtonController::DESKTOP_BACKGROUND |
-          PowerButtonController::NON_LOCK_SCREEN_CONTAINERS,
-          PowerButtonController::ANIMATION_RESTORE));
+          SessionStateAnimator::DESKTOP_BACKGROUND |
+          SessionStateAnimator::NON_LOCK_SCREEN_CONTAINERS,
+          SessionStateAnimator::ANIMATION_RESTORE));
   EXPECT_FALSE(test_api_->BlackLayerIsVisible());
 }
 
 // Hold the power button down from the unlocked state to eventual shutdown.
 TEST_F(PowerButtonControllerTest, LockToShutdown) {
   controller_->set_has_legacy_power_button_for_test(false);
   controller_->OnLoginStateChanged(user::LOGGED_IN_USER);
   controller_->OnLockStateChanged(false);
 
   // Hold the power button and lock the screen.
   controller_->OnPowerButtonEvent(true, base::TimeTicks::Now());
   EXPECT_TRUE(test_api_->lock_timer_is_running());
   test_api_->trigger_lock_timeout();
   controller_->OnStartingLock();
   controller_->OnLockStateChanged(true);
   EXPECT_TRUE(test_api_->BlackLayerIsVisible());
 
   // When the lock-to-shutdown timeout fires, we should start the shutdown
   // timer.
   EXPECT_TRUE(test_api_->lock_to_shutdown_timer_is_running());
   test_api_->trigger_lock_to_shutdown_timeout();
   EXPECT_TRUE(test_api_->shutdown_timer_is_running());
   EXPECT_TRUE(
       test_api_->ContainersAreAnimated(
-          PowerButtonController::GetAllContainersMask(),
-          PowerButtonController::ANIMATION_SLOW_CLOSE));
+          SessionStateAnimator::GetAllContainersMask(),
+          SessionStateAnimator::ANIMATION_SLOW_CLOSE));
 
   // Fire the shutdown timeout and check that we request shutdown.
   test_api_->trigger_shutdown_timeout();
   EXPECT_TRUE(test_api_->real_shutdown_timer_is_running());
   EXPECT_EQ(0, delegate_->num_shutdown_requests());
   test_api_->trigger_real_shutdown_timeout();
   EXPECT_EQ(1, delegate_->num_shutdown_requests());
   EXPECT_TRUE(test_api_->BlackLayerIsVisible());
 }
 
 // Test that we handle the case where lock requests are ignored.
 TEST_F(PowerButtonControllerTest, LockFail) {
   // We require animations to have a duration for this test.
   ui::LayerAnimator::set_disable_animations_for_test(false);
 
   controller_->set_has_legacy_power_button_for_test(false);
   controller_->OnLoginStateChanged(user::LOGGED_IN_USER);
   controller_->OnLockStateChanged(false);
 
   // Hold the power button and lock the screen.
   controller_->OnPowerButtonEvent(true, base::TimeTicks::Now());
   EXPECT_TRUE(test_api_->lock_timer_is_running());
   EXPECT_TRUE(
       test_api_->ContainersAreAnimated(
-          PowerButtonController::NON_LOCK_SCREEN_CONTAINERS,
-          PowerButtonController::ANIMATION_RESTORE));
+          SessionStateAnimator::NON_LOCK_SCREEN_CONTAINERS,
+          SessionStateAnimator::ANIMATION_RESTORE));
   EXPECT_TRUE(test_api_->BlackLayerIsVisible());
   test_api_->trigger_lock_timeout();
   EXPECT_EQ(1, delegate_->num_lock_requests());
   EXPECT_TRUE(test_api_->lock_fail_timer_is_running());
 
   // We shouldn't start the lock-to-shutdown timer until the screen has actually
   // been locked.
   EXPECT_FALSE(test_api_->lock_to_shutdown_timer_is_running());
 
   // Act as if the request timed out.  We should restore the windows.
   test_api_->trigger_lock_fail_timeout();
   EXPECT_TRUE(
       test_api_->ContainersAreAnimated(
-          PowerButtonController::NON_LOCK_SCREEN_CONTAINERS,
-          PowerButtonController::ANIMATION_RESTORE));
+          SessionStateAnimator::NON_LOCK_SCREEN_CONTAINERS,
+          SessionStateAnimator::ANIMATION_RESTORE));
   EXPECT_FALSE(test_api_->BlackLayerIsVisible());
 }
 
 // Test that we start the timer to hide the black layer when the power
 // button is released, but that we cancel the timer if the button is pressed
 // again before the timer has fired.
 TEST_F(PowerButtonControllerTest, CancelHideBackground) {
   controller_->set_has_legacy_power_button_for_test(false);
   controller_->OnLoginStateChanged(user::LOGGED_IN_NONE);
   controller_->OnLockStateChanged(false);
@@ skipping 25 matching lines @@
   controller_->OnLockButtonEvent(false, base::TimeTicks::Now());
   EXPECT_EQ(0, delegate_->num_lock_requests());
 
   // If we're logged in as a regular user, we should start the lock timer and
   // the pre-lock animation.
   controller_->OnLoginStateChanged(user::LOGGED_IN_USER);
   controller_->OnLockButtonEvent(true, base::TimeTicks::Now());
   EXPECT_TRUE(test_api_->lock_timer_is_running());
   EXPECT_TRUE(
       test_api_->ContainersAreAnimated(
-          PowerButtonController::NON_LOCK_SCREEN_CONTAINERS,
-          PowerButtonController::ANIMATION_SLOW_CLOSE));
+          SessionStateAnimator::NON_LOCK_SCREEN_CONTAINERS,
+          SessionStateAnimator::ANIMATION_SLOW_CLOSE));
   EXPECT_TRUE(test_api_->BlackLayerIsVisible());
 
   // If the button is released immediately, we shouldn't lock the screen.
   controller_->OnLockButtonEvent(false, base::TimeTicks::Now());
   EXPECT_FALSE(test_api_->lock_timer_is_running());
   EXPECT_TRUE(
       test_api_->ContainersAreAnimated(
-          PowerButtonController::NON_LOCK_SCREEN_CONTAINERS,
-          PowerButtonController::ANIMATION_UNDO_SLOW_CLOSE));
+          SessionStateAnimator::NON_LOCK_SCREEN_CONTAINERS,
+          SessionStateAnimator::ANIMATION_UNDO_SLOW_CLOSE));
   EXPECT_TRUE(test_api_->BlackLayerIsVisible());
   EXPECT_TRUE(test_api_->hide_black_layer_timer_is_running());
   test_api_->trigger_hide_black_layer_timeout();
   EXPECT_FALSE(test_api_->BlackLayerIsVisible());
   EXPECT_EQ(0, delegate_->num_lock_requests());
 
   // Press the button again and let the lock timeout fire.  We should request
   // that the screen be locked.
   controller_->OnLockButtonEvent(true, base::TimeTicks::Now());
   EXPECT_TRUE(test_api_->lock_timer_is_running());
@@ skipping 47 matching lines @@
 }
 
 // When the screen is locked without going through the usual power-button
 // slow-close path (e.g. via the wrench menu), test that we still show the
 // fast-close animation and display the black layer.
 TEST_F(PowerButtonControllerTest, LockWithoutButton) {
   controller_->OnLoginStateChanged(user::LOGGED_IN_USER);
   controller_->OnStartingLock();
   EXPECT_TRUE(
       test_api_->ContainersAreAnimated(
-          PowerButtonController::NON_LOCK_SCREEN_CONTAINERS,
-          PowerButtonController::ANIMATION_FAST_CLOSE));
+          SessionStateAnimator::NON_LOCK_SCREEN_CONTAINERS,
+          SessionStateAnimator::ANIMATION_FAST_CLOSE));
   EXPECT_TRUE(test_api_->BlackLayerIsVisible());
 }
 
 // When we hear that the process is exiting but we haven't had a chance to
 // display an animation, we should just blank the screen.
 TEST_F(PowerButtonControllerTest, ShutdownWithoutButton) {
   controller_->OnLoginStateChanged(user::LOGGED_IN_USER);
   controller_->OnAppTerminating();
   EXPECT_TRUE(
       test_api_->ContainersAreAnimated(
-          PowerButtonController::GetAllContainersMask(),
-          PowerButtonController::ANIMATION_HIDE));
+          SessionStateAnimator::GetAllContainersMask(),
+          SessionStateAnimator::ANIMATION_HIDE));
   EXPECT_TRUE(test_api_->BlackLayerIsVisible());
   EXPECT_FALSE(cursor_visible());
 }
 
 // Test that we display the fast-close animation and shut down when we get an
 // outside request to shut down (e.g. from the login or lock screen).
 TEST_F(PowerButtonControllerTest, RequestShutdownFromLoginScreen) {
   controller_->OnLoginStateChanged(user::LOGGED_IN_NONE);
   controller_->RequestShutdown();
   EXPECT_TRUE(
       test_api_->ContainersAreAnimated(
-          PowerButtonController::NON_LOCK_SCREEN_CONTAINERS,
-          PowerButtonController::ANIMATION_HIDE));
+          SessionStateAnimator::NON_LOCK_SCREEN_CONTAINERS,
+          SessionStateAnimator::ANIMATION_HIDE));
   EXPECT_TRUE(
       test_api_->ContainersAreAnimated(
-          PowerButtonController::GetAllLockScreenContainersMask(),
-          PowerButtonController::ANIMATION_FAST_CLOSE));
+          SessionStateAnimator::GetAllLockScreenContainersMask(),
+          SessionStateAnimator::ANIMATION_FAST_CLOSE));
   EXPECT_TRUE(test_api_->BlackLayerIsVisible());
   EXPECT_FALSE(cursor_visible());
 
   EXPECT_EQ(0, delegate_->num_shutdown_requests());
   EXPECT_TRUE(test_api_->real_shutdown_timer_is_running());
   test_api_->trigger_real_shutdown_timeout();
   EXPECT_EQ(1, delegate_->num_shutdown_requests());
 }
 
 TEST_F(PowerButtonControllerTest, RequestShutdownFromLockScreen) {
   controller_->OnLoginStateChanged(user::LOGGED_IN_USER);
   controller_->OnLockStateChanged(true);
   controller_->RequestShutdown();
   EXPECT_TRUE(
       test_api_->ContainersAreAnimated(
-          PowerButtonController::NON_LOCK_SCREEN_CONTAINERS,
-          PowerButtonController::ANIMATION_HIDE));
+          SessionStateAnimator::NON_LOCK_SCREEN_CONTAINERS,
+          SessionStateAnimator::ANIMATION_HIDE));
   EXPECT_TRUE(
       test_api_->ContainersAreAnimated(
-          PowerButtonController::GetAllLockScreenContainersMask(),
-          PowerButtonController::ANIMATION_FAST_CLOSE));
+          SessionStateAnimator::GetAllLockScreenContainersMask(),
+          SessionStateAnimator::ANIMATION_FAST_CLOSE));
   EXPECT_TRUE(test_api_->BlackLayerIsVisible());
   EXPECT_FALSE(cursor_visible());
 
   EXPECT_EQ(0, delegate_->num_shutdown_requests());
   EXPECT_TRUE(test_api_->real_shutdown_timer_is_running());
   test_api_->trigger_real_shutdown_timeout();
   EXPECT_EQ(1, delegate_->num_shutdown_requests());
 }
 
 TEST_F(PowerButtonControllerTest, RequestAndCancelShutdownFromLockScreen) {
   controller_->OnLoginStateChanged(user::LOGGED_IN_USER);
   controller_->OnLockStateChanged(true);
 
   // Press the power button and check that we start the shutdown timer.
   controller_->OnPowerButtonEvent(true, base::TimeTicks::Now());
   EXPECT_FALSE(test_api_->lock_timer_is_running());
   EXPECT_TRUE(test_api_->shutdown_timer_is_running());
   EXPECT_TRUE(
       test_api_->ContainersAreAnimated(
-          PowerButtonController::GetAllContainersMask(),
-          PowerButtonController::ANIMATION_SLOW_CLOSE));
+          SessionStateAnimator::GetAllContainersMask(),
+          SessionStateAnimator::ANIMATION_SLOW_CLOSE));
   EXPECT_TRUE(test_api_->BlackLayerIsVisible());
 
   // Release the power button before the shutdown timer fires.
   controller_->OnPowerButtonEvent(false, base::TimeTicks::Now());
   EXPECT_FALSE(test_api_->shutdown_timer_is_running());
   EXPECT_TRUE(
       test_api_->ContainersAreAnimated(
-          PowerButtonController::GetAllLockScreenContainersMask(),
-          PowerButtonController::ANIMATION_UNDO_SLOW_CLOSE));
+          SessionStateAnimator::GetAllLockScreenContainersMask(),
+          SessionStateAnimator::ANIMATION_UNDO_SLOW_CLOSE));
   EXPECT_TRUE(
       test_api_->ContainersAreAnimated(
-          PowerButtonController::DESKTOP_BACKGROUND,
-          PowerButtonController::ANIMATION_RESTORE));
+          SessionStateAnimator::DESKTOP_BACKGROUND,
+          SessionStateAnimator::ANIMATION_RESTORE));
   EXPECT_TRUE(test_api_->BlackLayerIsVisible());
 }
 
 // Test that the black layer is resized in response to root window resizes.
 TEST_F(PowerButtonControllerTest, ResizeBlackLayer) {
   controller_->OnPowerButtonEvent(true, base::TimeTicks::Now());
   EXPECT_EQ(Shell::GetPrimaryRootWindow()->bounds().ToString(),
             test_api_->GetBlackLayerBounds().ToString());
 
   const gfx::Size kNewSize(400, 300);
@@ skipping 14 matching lines @@
 
   // After increasing the brightness to 10%, we should start the timer like
   // usual.
   controller_->OnScreenBrightnessChanged(10.0);
   controller_->OnPowerButtonEvent(true, base::TimeTicks::Now());
   EXPECT_TRUE(test_api_->lock_timer_is_running());
 }
 
 }  // namespace test
 }  // namespace ash