Home | History | Annotate | Download | only in login 
      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
      3 // found in the LICENSE file.
      4 
      5 #include "chrome/browser/chromeos/login/login_manager_test.h"
      6 
      7 #include "base/prefs/scoped_user_pref_update.h"
      8 #include "chrome/browser/browser_process.h"
      9 #include "chrome/browser/chrome_notification_types.h"
     10 #include "chrome/browser/chromeos/login/existing_user_controller.h"
     11 #include "chrome/browser/chromeos/login/login_display_host_impl.h"
     12 #include "chrome/browser/chromeos/login/user_manager.h"
     13 #include "chrome/browser/chromeos/login/webui_login_view.h"
     14 #include "chrome/common/chrome_switches.h"
     15 #include "chromeos/chromeos_switches.h"
     16 #include "content/public/browser/notification_service.h"
     17 #include "content/public/browser/web_contents.h"
     18 #include "content/public/test/browser_test_utils.h"
     19 #include "content/public/test/test_utils.h"
     20 #include "testing/gtest/include/gtest/gtest.h"
     21 
     22 namespace chromeos {
     23 
     24 LoginManagerTest::LoginManagerTest(bool should_launch_browser)
     25     : should_launch_browser_(should_launch_browser),
     26       web_contents_(NULL) {
     27   set_exit_when_last_browser_closes(false);
     28 }
     29 
     30 void LoginManagerTest::CleanUpOnMainThread() {
     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(
     52       chrome::NOTIFICATION_LOGIN_OR_LOCK_WEBUI_VISIBLE,
     53       content::NotificationService::AllSources()).Wait();
     54   InitializeWebContents();
     55 }
     56 
     57 void LoginManagerTest::RegisterUser(const std::string& username) {
     58   ListPrefUpdate users_pref(g_browser_process->local_state(), "LoggedInUsers");
     59   users_pref->AppendIfNotPresent(new base::StringValue(username));
     60 }
     61 
     62 void LoginManagerTest::SetExpectedCredentials(const std::string& username,
     63                                               const std::string& password) {
     64   login_utils().GetFakeLoginUtils()->SetExpectedCredentials(username, password);
     65 }
     66 
     67 bool LoginManagerTest::TryToLogin(const std::string& username,
     68                                   const std::string& password) {
     69   if (!AddUserTosession(username, password))
     70     return false;
     71   if (const User* active_user = UserManager::Get()->GetActiveUser())
     72     return active_user->email() == username;
     73   return false;
     74 }
     75 
     76 bool LoginManagerTest::AddUserTosession(const std::string& username,
     77                                         const std::string& password) {
     78   ExistingUserController* controller =
     79       ExistingUserController::current_controller();
     80   if (!controller) {
     81     ADD_FAILURE();
     82     return false;
     83   }
     84   controller->Login(UserContext(username, password, std::string()));
     85   content::WindowedNotificationObserver(
     86       chrome::NOTIFICATION_SESSION_STARTED,
     87       content::NotificationService::AllSources()).Wait();
     88   const UserList& logged_users = UserManager::Get()->GetLoggedInUsers();
     89   for (UserList::const_iterator it = logged_users.begin();
     90        it != logged_users.end(); ++it) {
     91     if ((*it)->email() == username)
     92       return true;
     93   }
     94   return false;
     95 }
     96 
     97 void LoginManagerTest::LoginUser(const std::string& username) {
     98   SetExpectedCredentials(username, "password");
     99   EXPECT_TRUE(TryToLogin(username, "password"));
    100 }
    101 
    102 void LoginManagerTest::AddUser(const std::string& username) {
    103   SetExpectedCredentials(username, "password");
    104   EXPECT_TRUE(AddUserTosession(username, "password"));
    105 }
    106 
    107 void LoginManagerTest::JSExpect(const std::string& expression) {
    108   js_checker_.ExpectTrue(expression);
    109 }
    110 
    111 void LoginManagerTest::InitializeWebContents() {
    112     LoginDisplayHost* host = LoginDisplayHostImpl::default_host();
    113     EXPECT_TRUE(host != NULL);
    114 
    115     content::WebContents* web_contents =
    116         host->GetWebUILoginView()->GetWebContents();
    117     EXPECT_TRUE(web_contents != NULL);
    118     set_web_contents(web_contents);
    119     js_checker_.set_web_contents(web_contents);
    120   }
    121 
    122 }  // namespace chromeos
    123