Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(236)

Side by Side Diff: chrome/browser/chromeos/login/login_manager_test_helper.cc

Issue 270563002: Componentize LoginManagerTest. (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src
Patch Set: Move parameter from ctor to fn Created 6 years, 7 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch | Annotate | Revision Log
OLDNEW
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
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698