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), | |
21 logged_in_user_type_(LoginState::LOGGED_IN_USER_NONE) { | |
22 } | 20 } |
23 virtual ~LoginStateTest() { | 21 virtual ~LoginStateTest() { |
24 } | 22 } |
25 | 23 |
26 // testing::Test | 24 // testing::Test |
27 virtual void SetUp() OVERRIDE { | 25 virtual void SetUp() OVERRIDE { |
28 CommandLine::ForCurrentProcess()->AppendSwitch(switches::kLoginManager); | 26 CommandLine::ForCurrentProcess()->AppendSwitch(switches::kLoginManager); |
29 // Initialize DBusThreadManager with a stub implementation. | 27 // Initialize DBusThreadManager with a stub implementation. |
30 DBusThreadManager::InitializeWithStub(); | 28 DBusThreadManager::InitializeWithStub(); |
31 LoginState::Initialize(); | 29 LoginState::Initialize(); |
32 LoginState::Get()->AddObserver(this); | 30 LoginState::Get()->AddObserver(this); |
33 } | 31 } |
34 | 32 |
35 virtual void TearDown() OVERRIDE { | 33 virtual void TearDown() OVERRIDE { |
36 LoginState::Get()->RemoveObserver(this); | 34 LoginState::Get()->RemoveObserver(this); |
37 LoginState::Shutdown(); | 35 LoginState::Shutdown(); |
38 DBusThreadManager::Shutdown(); | 36 DBusThreadManager::Shutdown(); |
39 } | 37 } |
40 | 38 |
41 // LoginState::Observer | 39 // LoginState::Observer |
42 virtual void LoggedInStateChanged(LoginState::LoggedInState state) OVERRIDE { | 40 virtual void LoggedInStateChanged() OVERRIDE { |
stevenjb
2013/09/12 00:48:01
We should explicitly track that the observer gets
tbarzic
2013/09/12 01:18:12
Done.
| |
43 logged_in_state_ = state; | |
44 logged_in_user_type_ = LoginState::Get()->GetLoggedInUserType(); | 41 logged_in_user_type_ = LoginState::Get()->GetLoggedInUserType(); |
45 } | 42 } |
46 | 43 |
47 protected: | 44 protected: |
48 base::MessageLoopForUI message_loop_; | 45 base::MessageLoopForUI message_loop_; |
49 LoginState::LoggedInState logged_in_state_; | |
50 LoginState::LoggedInUserType logged_in_user_type_; | 46 LoginState::LoggedInUserType logged_in_user_type_; |
51 | 47 |
52 private: | 48 private: |
53 DISALLOW_COPY_AND_ASSIGN(LoginStateTest); | 49 DISALLOW_COPY_AND_ASSIGN(LoginStateTest); |
54 }; | 50 }; |
55 | 51 |
56 TEST_F(LoginStateTest, TestLoginState) { | 52 TEST_F(LoginStateTest, TestLoginState) { |
57 EXPECT_EQ(LoginState::LOGGED_IN_OOBE, logged_in_state_); | 53 EXPECT_FALSE(LoginState::Get()->IsUserLoggedIn()); |
58 EXPECT_EQ(LoginState::LOGGED_IN_OOBE, LoginState::Get()->GetLoggedInState()); | 54 EXPECT_FALSE(LoginState::Get()->IsInSafeMode()); |
59 EXPECT_EQ(LoginState::LOGGED_IN_USER_NONE, logged_in_user_type_); | 55 EXPECT_EQ(LoginState::LOGGED_IN_USER_NONE, logged_in_user_type_); |
60 EXPECT_EQ(LoginState::LOGGED_IN_USER_NONE, | 56 EXPECT_EQ(LoginState::LOGGED_IN_USER_NONE, |
61 LoginState::Get()->GetLoggedInUserType()); | 57 LoginState::Get()->GetLoggedInUserType()); |
58 // Setting login state to SAFE MODE. | |
stevenjb
2013/09/12 00:48:01
We should probably test safe mode separately, sinc
tbarzic
2013/09/12 01:18:12
Done.
| |
59 LoginState::Get()->SetLoggedInState(LoginState::LOGGED_IN_SAFE_MODE, | |
60 LoginState::LOGGED_IN_USER_NONE); | |
61 EXPECT_EQ(LoginState::LOGGED_IN_USER_NONE, | |
62 LoginState::Get()->GetLoggedInUserType()); | |
63 EXPECT_FALSE(LoginState::Get()->IsUserLoggedIn()); | |
64 EXPECT_TRUE(LoginState::Get()->IsInSafeMode()); | |
65 | |
66 // Run the message loop, observer should update members. | |
67 message_loop_.RunUntilIdle(); | |
68 EXPECT_EQ(LoginState::LOGGED_IN_USER_NONE, logged_in_user_type_); | |
69 | |
62 // Setting login state to ACTIVE. | 70 // Setting login state to ACTIVE. |
63 LoginState::Get()->SetLoggedInState(LoginState::LOGGED_IN_ACTIVE, | 71 LoginState::Get()->SetLoggedInState(LoginState::LOGGED_IN_ACTIVE, |
64 LoginState::LOGGED_IN_USER_REGULAR); | 72 LoginState::LOGGED_IN_USER_OWNER); |
65 EXPECT_EQ(LoginState::LOGGED_IN_ACTIVE, | 73 EXPECT_EQ(LoginState::LOGGED_IN_USER_OWNER, |
66 LoginState::Get()->GetLoggedInState()); | |
67 EXPECT_EQ(LoginState::LOGGED_IN_USER_REGULAR, | |
68 LoginState::Get()->GetLoggedInUserType()); | 74 LoginState::Get()->GetLoggedInUserType()); |
69 EXPECT_TRUE(LoginState::Get()->IsUserLoggedIn()); | 75 EXPECT_TRUE(LoginState::Get()->IsUserLoggedIn()); |
76 EXPECT_FALSE(LoginState::Get()->IsInSafeMode()); | |
77 | |
70 // Run the message loop, observer should update members. | 78 // Run the message loop, observer should update members. |
71 message_loop_.RunUntilIdle(); | 79 message_loop_.RunUntilIdle(); |
72 EXPECT_EQ(LoginState::LOGGED_IN_ACTIVE, logged_in_state_); | 80 EXPECT_EQ(LoginState::LOGGED_IN_USER_OWNER, logged_in_user_type_); |
73 EXPECT_EQ(LoginState::LOGGED_IN_USER_REGULAR, logged_in_user_type_); | |
74 } | 81 } |
75 | 82 |
76 } // namespace | 83 } // namespace chromeos |
OLD | NEW |