OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "chromeos/login/login_state.h" | 5 #include "chromeos/login/login_state.h" |
6 | 6 |
7 #include "base/command_line.h" | 7 #include "base/command_line.h" |
8 #include "base/compiler_specific.h" | 8 #include "base/compiler_specific.h" |
9 #include "base/message_loop/message_loop.h" | 9 #include "base/message_loop/message_loop.h" |
10 #include "chromeos/chromeos_switches.h" | 10 #include "chromeos/chromeos_switches.h" |
11 #include "chromeos/dbus/dbus_thread_manager.h" | 11 #include "chromeos/dbus/dbus_thread_manager.h" |
12 #include "testing/gtest/include/gtest/gtest.h" | 12 #include "testing/gtest/include/gtest/gtest.h" |
13 | 13 |
14 namespace chromeos { | 14 namespace chromeos { |
15 | 15 |
16 class LoginStateTest : public testing::Test, | 16 class LoginStateTest : public testing::Test, |
17 public LoginState::Observer { | 17 public LoginState::Observer { |
18 public: | 18 public: |
19 LoginStateTest() | 19 LoginStateTest() : logged_in_user_type_(LoginState::LOGGED_IN_USER_NONE), |
20 : logged_in_state_(LoginState::LOGGED_IN_OOBE), | 20 login_state_changes_count_(0) { |
21 logged_in_user_type_(LoginState::LOGGED_IN_USER_NONE) { | |
22 } | 21 } |
23 virtual ~LoginStateTest() { | 22 virtual ~LoginStateTest() { |
24 } | 23 } |
25 | 24 |
26 // testing::Test | 25 // testing::Test |
27 virtual void SetUp() OVERRIDE { | 26 virtual void SetUp() OVERRIDE { |
28 CommandLine::ForCurrentProcess()->AppendSwitch(switches::kLoginManager); | 27 CommandLine::ForCurrentProcess()->AppendSwitch(switches::kLoginManager); |
29 // Initialize DBusThreadManager with a stub implementation. | 28 // Initialize DBusThreadManager with a stub implementation. |
30 DBusThreadManager::InitializeWithStub(); | 29 DBusThreadManager::InitializeWithStub(); |
31 LoginState::Initialize(); | 30 LoginState::Initialize(); |
32 LoginState::Get()->AddObserver(this); | 31 LoginState::Get()->AddObserver(this); |
33 } | 32 } |
34 | 33 |
35 virtual void TearDown() OVERRIDE { | 34 virtual void TearDown() OVERRIDE { |
36 LoginState::Get()->RemoveObserver(this); | 35 LoginState::Get()->RemoveObserver(this); |
37 LoginState::Shutdown(); | 36 LoginState::Shutdown(); |
38 DBusThreadManager::Shutdown(); | 37 DBusThreadManager::Shutdown(); |
39 } | 38 } |
40 | 39 |
41 // LoginState::Observer | 40 // LoginState::Observer |
42 virtual void LoggedInStateChanged(LoginState::LoggedInState state) OVERRIDE { | 41 virtual void LoggedInStateChanged() OVERRIDE { |
43 logged_in_state_ = state; | 42 ++login_state_changes_count_; |
44 logged_in_user_type_ = LoginState::Get()->GetLoggedInUserType(); | 43 logged_in_user_type_ = LoginState::Get()->GetLoggedInUserType(); |
45 } | 44 } |
46 | 45 |
47 protected: | 46 protected: |
| 47 // Returns number of times the login state changed since the last call to |
| 48 // this method. |
| 49 unsigned int GetNewLoginStateChangesCount() { |
| 50 unsigned int result = login_state_changes_count_; |
| 51 login_state_changes_count_ = 0; |
| 52 return result; |
| 53 } |
| 54 |
48 base::MessageLoopForUI message_loop_; | 55 base::MessageLoopForUI message_loop_; |
49 LoginState::LoggedInState logged_in_state_; | |
50 LoginState::LoggedInUserType logged_in_user_type_; | 56 LoginState::LoggedInUserType logged_in_user_type_; |
51 | 57 |
52 private: | 58 private: |
| 59 unsigned int login_state_changes_count_; |
| 60 |
53 DISALLOW_COPY_AND_ASSIGN(LoginStateTest); | 61 DISALLOW_COPY_AND_ASSIGN(LoginStateTest); |
54 }; | 62 }; |
55 | 63 |
56 TEST_F(LoginStateTest, TestLoginState) { | 64 TEST_F(LoginStateTest, TestLoginState) { |
57 EXPECT_EQ(LoginState::LOGGED_IN_OOBE, logged_in_state_); | 65 EXPECT_FALSE(LoginState::Get()->IsUserLoggedIn()); |
58 EXPECT_EQ(LoginState::LOGGED_IN_OOBE, LoginState::Get()->GetLoggedInState()); | 66 EXPECT_FALSE(LoginState::Get()->IsInSafeMode()); |
59 EXPECT_EQ(LoginState::LOGGED_IN_USER_NONE, logged_in_user_type_); | 67 EXPECT_EQ(LoginState::LOGGED_IN_USER_NONE, logged_in_user_type_); |
60 EXPECT_EQ(LoginState::LOGGED_IN_USER_NONE, | 68 EXPECT_EQ(LoginState::LOGGED_IN_USER_NONE, |
61 LoginState::Get()->GetLoggedInUserType()); | 69 LoginState::Get()->GetLoggedInUserType()); |
| 70 |
62 // Setting login state to ACTIVE. | 71 // Setting login state to ACTIVE. |
63 LoginState::Get()->SetLoggedInState(LoginState::LOGGED_IN_ACTIVE, | 72 LoginState::Get()->SetLoggedInState(LoginState::LOGGED_IN_ACTIVE, |
64 LoginState::LOGGED_IN_USER_REGULAR); | 73 LoginState::LOGGED_IN_USER_REGULAR); |
65 EXPECT_EQ(LoginState::LOGGED_IN_ACTIVE, | |
66 LoginState::Get()->GetLoggedInState()); | |
67 EXPECT_EQ(LoginState::LOGGED_IN_USER_REGULAR, | 74 EXPECT_EQ(LoginState::LOGGED_IN_USER_REGULAR, |
68 LoginState::Get()->GetLoggedInUserType()); | 75 LoginState::Get()->GetLoggedInUserType()); |
69 EXPECT_TRUE(LoginState::Get()->IsUserLoggedIn()); | 76 EXPECT_TRUE(LoginState::Get()->IsUserLoggedIn()); |
| 77 EXPECT_FALSE(LoginState::Get()->IsInSafeMode()); |
| 78 |
70 // Run the message loop, observer should update members. | 79 // Run the message loop, observer should update members. |
71 message_loop_.RunUntilIdle(); | 80 message_loop_.RunUntilIdle(); |
72 EXPECT_EQ(LoginState::LOGGED_IN_ACTIVE, logged_in_state_); | 81 EXPECT_EQ(1U, GetNewLoginStateChangesCount()); |
73 EXPECT_EQ(LoginState::LOGGED_IN_USER_REGULAR, logged_in_user_type_); | 82 EXPECT_EQ(LoginState::LOGGED_IN_USER_REGULAR, logged_in_user_type_); |
74 } | 83 } |
75 | 84 |
76 } // namespace | 85 TEST_F(LoginStateTest, TestSafeModeLoginState) { |
| 86 EXPECT_FALSE(LoginState::Get()->IsUserLoggedIn()); |
| 87 EXPECT_FALSE(LoginState::Get()->IsInSafeMode()); |
| 88 EXPECT_EQ(LoginState::LOGGED_IN_USER_NONE, logged_in_user_type_); |
| 89 EXPECT_EQ(LoginState::LOGGED_IN_USER_NONE, |
| 90 LoginState::Get()->GetLoggedInUserType()); |
| 91 // Setting login state to SAFE MODE. |
| 92 LoginState::Get()->SetLoggedInState(LoginState::LOGGED_IN_SAFE_MODE, |
| 93 LoginState::LOGGED_IN_USER_NONE); |
| 94 EXPECT_EQ(LoginState::LOGGED_IN_USER_NONE, |
| 95 LoginState::Get()->GetLoggedInUserType()); |
| 96 EXPECT_FALSE(LoginState::Get()->IsUserLoggedIn()); |
| 97 EXPECT_TRUE(LoginState::Get()->IsInSafeMode()); |
| 98 |
| 99 // Run the message loop, observer should update members. |
| 100 message_loop_.RunUntilIdle(); |
| 101 EXPECT_EQ(1U, GetNewLoginStateChangesCount()); |
| 102 EXPECT_EQ(LoginState::LOGGED_IN_USER_NONE, logged_in_user_type_); |
| 103 |
| 104 // Setting login state to ACTIVE. |
| 105 LoginState::Get()->SetLoggedInState(LoginState::LOGGED_IN_ACTIVE, |
| 106 LoginState::LOGGED_IN_USER_OWNER); |
| 107 EXPECT_EQ(LoginState::LOGGED_IN_USER_OWNER, |
| 108 LoginState::Get()->GetLoggedInUserType()); |
| 109 EXPECT_TRUE(LoginState::Get()->IsUserLoggedIn()); |
| 110 EXPECT_FALSE(LoginState::Get()->IsInSafeMode()); |
| 111 |
| 112 // Run the message loop, observer should update members. |
| 113 message_loop_.RunUntilIdle(); |
| 114 EXPECT_EQ(1U, GetNewLoginStateChangesCount()); |
| 115 EXPECT_EQ(LoginState::LOGGED_IN_USER_OWNER, logged_in_user_type_); |
| 116 } |
| 117 |
| 118 } // namespace chromeos |
OLD | NEW |