Return correct window when user is logged in but before session is started.

chromeos/login/login_state_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 "chromeos/login/login_state.h"
 
 #include "base/command_line.h"
 #include "base/compiler_specific.h"
 #include "base/message_loop/message_loop.h"
 #include "chromeos/chromeos_switches.h"
@@ skipping 41 matching lines @@
  private:
   DISALLOW_COPY_AND_ASSIGN(LoginStateTest);
 };
 
 TEST_F(LoginStateTest, TestLoginState) {
   EXPECT_EQ(LoginState::LOGGED_IN_OOBE, logged_in_state_);
   EXPECT_EQ(LoginState::LOGGED_IN_OOBE, LoginState::Get()->GetLoggedInState());
   EXPECT_EQ(LoginState::LOGGED_IN_USER_NONE, logged_in_user_type_);
   EXPECT_EQ(LoginState::LOGGED_IN_USER_NONE,
             LoginState::Get()->GetLoggedInUserType());
+  EXPECT_FALSE(LoginState::Get()->IsSessionStarted());
   // Setting login state to ACTIVE.
   LoginState::Get()->SetLoggedInState(LoginState::LOGGED_IN_ACTIVE,
                                       LoginState::LOGGED_IN_USER_REGULAR);
   EXPECT_EQ(LoginState::LOGGED_IN_ACTIVE,
             LoginState::Get()->GetLoggedInState());
   EXPECT_EQ(LoginState::LOGGED_IN_USER_REGULAR,
             LoginState::Get()->GetLoggedInUserType());
   EXPECT_TRUE(LoginState::Get()->IsUserLoggedIn());
+  // Set session started.
+  LoginState::Get()->SetSessionStarted(true);
+  EXPECT_TRUE(LoginState::Get()->IsSessionStarted());
   // Run the message loop, observer should update members.
   message_loop_.RunUntilIdle();
   EXPECT_EQ(LoginState::LOGGED_IN_ACTIVE, logged_in_state_);
   EXPECT_EQ(LoginState::LOGGED_IN_USER_REGULAR, logged_in_user_type_);
 }
 
-}  // namespace
+}  // namespace chromeos