Converts most of WorkspaceLayoutManager tests to use common code

ash/test/test_session_state_delegate.cc

 // Copyright (c) 2013 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/test_session_state_delegate.h"
 
 #include <algorithm>
 #include <string>
 
 #include "ash/common/login_status.h"
+#include "ash/common/wm_shell.h"
 #include "ash/shell.h"
 #include "base/memory/ptr_util.h"
 #include "base/strings/string16.h"
 #include "base/strings/utf_string_conversions.h"
 #include "components/signin/core/account_id/account_id.h"
 #include "components/user_manager/user_info.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace ash {
 namespace test {
@@ skipping 154 matching lines @@
     user_manager_->SessionStarted();
     session_state_ = SESSION_STATE_ACTIVE;
     Shell::GetInstance()->CreateShelf();
     Shell::GetInstance()->UpdateAfterLoginStatusChange(LoginStatus::USER);
   } else {
     session_state_ = SESSION_STATE_LOGIN_PRIMARY;
     user_manager_.reset(new TestUserManager());
   }
 }
 
+// static
 void TestSessionStateDelegate::SetCanLockScreen(bool can_lock_screen) {
-  can_lock_screen_ = can_lock_screen;
+  CHECK(WmShell::HasInstance());
+  static_cast<ash::test::TestSessionStateDelegate*>(
+      WmShell::Get()->GetSessionStateDelegate())
+      ->can_lock_screen_ = can_lock_screen;
 }
 
 void TestSessionStateDelegate::SetShouldLockScreenBeforeSuspending(
     bool should_lock) {
   should_lock_screen_before_suspending_ = should_lock;
 }
 
 void TestSessionStateDelegate::SetUserAddingScreenRunning(
     bool user_adding_screen_running) {
   user_adding_screen_running_ = user_adding_screen_running;
@@ skipping 46 matching lines @@
 }
 
 void TestSessionStateDelegate::AddSessionStateObserver(
     SessionStateObserver* observer) {}
 
 void TestSessionStateDelegate::RemoveSessionStateObserver(
     SessionStateObserver* observer) {}
 
 }  // namespace test
 }  // namespace ash