OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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 "chrome/browser/chromeos/login/login_manager_test.h" | 5 #include "chrome/browser/chromeos/login/login_manager_test_helper.h" |
6 | 6 |
7 #include "base/prefs/scoped_user_pref_update.h" | 7 #include "base/prefs/scoped_user_pref_update.h" |
8 #include "chrome/browser/browser_process.h" | 8 #include "chrome/browser/browser_process.h" |
9 #include "chrome/browser/chrome_notification_types.h" | 9 #include "chrome/browser/chrome_notification_types.h" |
10 #include "chrome/browser/chromeos/login/existing_user_controller.h" | 10 #include "chrome/browser/chromeos/login/existing_user_controller.h" |
11 #include "chrome/browser/chromeos/login/login_display_host_impl.h" | 11 #include "chrome/browser/chromeos/login/login_display_host_impl.h" |
12 #include "chrome/browser/chromeos/login/user_manager.h" | 12 #include "chrome/browser/chromeos/login/user_manager.h" |
13 #include "chrome/browser/chromeos/login/webui_login_view.h" | 13 #include "chrome/browser/chromeos/login/webui_login_view.h" |
14 #include "chrome/common/chrome_switches.h" | 14 #include "chrome/common/chrome_switches.h" |
15 #include "chromeos/chromeos_switches.h" | 15 #include "chromeos/chromeos_switches.h" |
16 #include "content/public/browser/notification_service.h" | 16 #include "content/public/browser/notification_service.h" |
17 #include "content/public/browser/web_contents.h" | 17 #include "content/public/browser/web_contents.h" |
18 #include "content/public/test/browser_test_utils.h" | 18 #include "content/public/test/browser_test_utils.h" |
19 #include "content/public/test/test_utils.h" | 19 #include "content/public/test/test_utils.h" |
20 #include "testing/gtest/include/gtest/gtest.h" | 20 #include "testing/gtest/include/gtest/gtest.h" |
21 | 21 |
22 namespace chromeos { | 22 namespace chromeos { |
23 | 23 |
24 LoginManagerTest::LoginManagerTest(bool should_launch_browser) | 24 LoginManagerTestHelper::LoginManagerTestHelper() : web_contents_(NULL) {} |
25 : should_launch_browser_(should_launch_browser), | |
26 web_contents_(NULL) { | |
27 set_exit_when_last_browser_closes(false); | |
28 } | |
29 | 25 |
30 void LoginManagerTest::CleanUpOnMainThread() { | 26 void LoginManagerTestHelper::SetUp() { |
31 if (LoginDisplayHostImpl::default_host()) | |
32 LoginDisplayHostImpl::default_host()->Finalize(); | |
33 base::MessageLoop::current()->RunUntilIdle(); | |
34 } | |
35 | |
36 void LoginManagerTest::SetUpCommandLine(CommandLine* command_line) { | |
37 command_line->AppendSwitch(chromeos::switches::kLoginManager); | |
38 command_line->AppendSwitch(chromeos::switches::kForceLoginManagerInTests); | |
39 command_line->AppendSwitch(::switches::kMultiProfiles); | |
40 } | |
41 | |
42 void LoginManagerTest::SetUpInProcessBrowserTestFixture() { | |
43 mock_login_utils_ = new testing::NiceMock<MockLoginUtils>(); | |
44 mock_login_utils_->DelegateToFake(); | |
45 mock_login_utils_->GetFakeLoginUtils()->set_should_launch_browser( | |
46 should_launch_browser_); | |
47 LoginUtils::Set(mock_login_utils_); | |
48 } | |
49 | |
50 void LoginManagerTest::SetUpOnMainThread() { | |
51 content::WindowedNotificationObserver( | 27 content::WindowedNotificationObserver( |
52 chrome::NOTIFICATION_LOGIN_OR_LOCK_WEBUI_VISIBLE, | 28 chrome::NOTIFICATION_LOGIN_OR_LOCK_WEBUI_VISIBLE, |
53 content::NotificationService::AllSources()).Wait(); | 29 content::NotificationService::AllSources()).Wait(); |
54 InitializeWebContents(); | 30 InitializeWebContents(); |
55 } | 31 } |
56 | 32 |
57 void LoginManagerTest::RegisterUser(const std::string& username) { | 33 void LoginManagerTestHelper::SetUpCommandLine(CommandLine* command_line) { |
| 34 command_line->AppendSwitch(chromeos::switches::kLoginManager); |
| 35 command_line->AppendSwitch(chromeos::switches::kForceLoginManagerInTests); |
| 36 command_line->AppendSwitch(::switches::kMultiProfiles); |
| 37 } |
| 38 |
| 39 void LoginManagerTestHelper::SetUpLoginUtils(bool should_launch_browser) { |
| 40 mock_login_utils_ = new testing::NiceMock<MockLoginUtils>(); |
| 41 mock_login_utils_->DelegateToFake(); |
| 42 mock_login_utils_->GetFakeLoginUtils()->set_should_launch_browser( |
| 43 should_launch_browser); |
| 44 LoginUtils::Set(mock_login_utils_); |
| 45 } |
| 46 |
| 47 void LoginManagerTestHelper::CleanUp() { |
| 48 if (LoginDisplayHostImpl::default_host()) |
| 49 LoginDisplayHostImpl::default_host()->Finalize(); |
| 50 base::MessageLoop::current()->RunUntilIdle(); |
| 51 } |
| 52 |
| 53 void LoginManagerTestHelper::RegisterUser(const std::string& username) { |
58 ListPrefUpdate users_pref(g_browser_process->local_state(), "LoggedInUsers"); | 54 ListPrefUpdate users_pref(g_browser_process->local_state(), "LoggedInUsers"); |
59 users_pref->AppendIfNotPresent(new base::StringValue(username)); | 55 users_pref->AppendIfNotPresent(new base::StringValue(username)); |
60 } | 56 } |
61 | 57 |
62 void LoginManagerTest::SetExpectedCredentials(const std::string& username, | 58 void LoginManagerTestHelper::SetExpectedCredentials(const std::string& username, |
63 const std::string& password) { | 59 const std::string& password) { |
64 login_utils().GetFakeLoginUtils()->SetExpectedCredentials(username, password); | 60 login_utils().GetFakeLoginUtils()->SetExpectedCredentials(username, password); |
65 } | 61 } |
66 | 62 |
67 bool LoginManagerTest::TryToLogin(const std::string& username, | 63 bool LoginManagerTestHelper::TryToLogIn(const std::string& username, |
68 const std::string& password) { | 64 const std::string& password) { |
69 if (!AddUserTosession(username, password)) | 65 if (!AddUserToSession(username, password)) |
70 return false; | 66 return false; |
71 if (const User* active_user = UserManager::Get()->GetActiveUser()) | 67 if (const User* active_user = UserManager::Get()->GetActiveUser()) |
72 return active_user->email() == username; | 68 return active_user->email() == username; |
73 return false; | 69 return false; |
74 } | 70 } |
75 | 71 |
76 bool LoginManagerTest::AddUserTosession(const std::string& username, | 72 bool LoginManagerTestHelper::AddUserToSession(const std::string& username, |
77 const std::string& password) { | 73 const std::string& password) { |
78 ExistingUserController* controller = | 74 ExistingUserController* controller = |
79 ExistingUserController::current_controller(); | 75 ExistingUserController::current_controller(); |
80 if (!controller) { | 76 if (!controller) { |
81 ADD_FAILURE(); | 77 ADD_FAILURE(); |
82 return false; | 78 return false; |
83 } | 79 } |
84 controller->Login(UserContext(username, password, std::string())); | 80 controller->Login(UserContext(username, password, std::string())); |
85 content::WindowedNotificationObserver( | 81 content::WindowedNotificationObserver( |
86 chrome::NOTIFICATION_SESSION_STARTED, | 82 chrome::NOTIFICATION_SESSION_STARTED, |
87 content::NotificationService::AllSources()).Wait(); | 83 content::NotificationService::AllSources()).Wait(); |
88 const UserList& logged_users = UserManager::Get()->GetLoggedInUsers(); | 84 const UserList& logged_users = UserManager::Get()->GetLoggedInUsers(); |
89 for (UserList::const_iterator it = logged_users.begin(); | 85 for (UserList::const_iterator it = logged_users.begin(); |
90 it != logged_users.end(); ++it) { | 86 it != logged_users.end(); ++it) { |
91 if ((*it)->email() == username) | 87 if ((*it)->email() == username) |
92 return true; | 88 return true; |
93 } | 89 } |
94 return false; | 90 return false; |
95 } | 91 } |
96 | 92 |
97 void LoginManagerTest::LoginUser(const std::string& username) { | 93 void LoginManagerTestHelper::LogInUser(const std::string& username) { |
98 SetExpectedCredentials(username, "password"); | 94 SetExpectedCredentials(username, "password"); |
99 EXPECT_TRUE(TryToLogin(username, "password")); | 95 EXPECT_TRUE(TryToLogIn(username, "password")); |
100 } | 96 } |
101 | 97 |
102 void LoginManagerTest::AddUser(const std::string& username) { | 98 void LoginManagerTestHelper::AddUser(const std::string& username) { |
103 SetExpectedCredentials(username, "password"); | 99 SetExpectedCredentials(username, "password"); |
104 EXPECT_TRUE(AddUserTosession(username, "password")); | 100 EXPECT_TRUE(AddUserToSession(username, "password")); |
105 } | 101 } |
106 | 102 |
107 void LoginManagerTest::JSExpect(const std::string& expression) { | 103 void LoginManagerTestHelper::JSExpect(const std::string& expression) { |
108 js_checker_.ExpectTrue(expression); | 104 js_checker_.ExpectTrue(expression); |
109 } | 105 } |
110 | 106 |
111 void LoginManagerTest::InitializeWebContents() { | 107 void LoginManagerTestHelper::InitializeWebContents() { |
112 LoginDisplayHost* host = LoginDisplayHostImpl::default_host(); | 108 LoginDisplayHost* host = LoginDisplayHostImpl::default_host(); |
113 EXPECT_TRUE(host != NULL); | 109 EXPECT_TRUE(host != NULL); |
114 | 110 |
115 content::WebContents* web_contents = | 111 content::WebContents* web_contents = |
116 host->GetWebUILoginView()->GetWebContents(); | 112 host->GetWebUILoginView()->GetWebContents(); |
117 EXPECT_TRUE(web_contents != NULL); | 113 EXPECT_TRUE(web_contents != NULL); |
118 set_web_contents(web_contents); | 114 set_web_contents(web_contents); |
119 js_checker_.set_web_contents(web_contents); | 115 js_checker_.set_web_contents(web_contents); |
120 } | 116 } |
121 | 117 |
122 } // namespace chromeos | 118 } // namespace chromeos |
OLD | NEW |