Add a Key class

chrome/browser/chromeos/login/login_manager_test.cc

 // Copyright 2013 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chrome/browser/chromeos/login/login_manager_test.h"
 
 #include "base/prefs/scoped_user_pref_update.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/chrome_notification_types.h"
+#include "chrome/browser/chromeos/login/auth/key.h"
 #include "chrome/browser/chromeos/login/auth/user_context.h"
 #include "chrome/browser/chromeos/login/existing_user_controller.h"
 #include "chrome/browser/chromeos/login/ui/login_display_host_impl.h"
 #include "chrome/browser/chromeos/login/ui/webui_login_view.h"
 #include "chrome/browser/chromeos/login/users/user.h"
 #include "chrome/browser/chromeos/login/users/user_manager.h"
 #include "chromeos/chromeos_switches.h"
 #include "content/public/browser/notification_service.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/test/browser_test_utils.h"
@@ skipping 27 matching lines @@
   LoginUtils::Set(mock_login_utils_);
 }
 
 void LoginManagerTest::SetUpOnMainThread() {
   content::WindowedNotificationObserver(
       chrome::NOTIFICATION_LOGIN_OR_LOCK_WEBUI_VISIBLE,
       content::NotificationService::AllSources()).Wait();
   InitializeWebContents();
 }
 
-void LoginManagerTest::RegisterUser(const std::string& username) {
+void LoginManagerTest::RegisterUser(const std::string& user_id) {
   ListPrefUpdate users_pref(g_browser_process->local_state(), "LoggedInUsers");
-  users_pref->AppendIfNotPresent(new base::StringValue(username));
+  users_pref->AppendIfNotPresent(new base::StringValue(user_id));
 }
 
-void LoginManagerTest::SetExpectedCredentials(const std::string& username,
-                                              const std::string& password) {
-  login_utils().GetFakeLoginUtils()->SetExpectedCredentials(username, password);
+void LoginManagerTest::SetExpectedCredentials(const UserContext& user_context) {
+  login_utils().GetFakeLoginUtils()->SetExpectedCredentials(user_context);
 }
 
-bool LoginManagerTest::TryToLogin(const std::string& username,
-                                  const std::string& password) {
-  if (!AddUserToSession(username, password))
+bool LoginManagerTest::TryToLogin(const UserContext& user_context) {
+  if (!AddUserToSession(user_context))
     return false;
   if (const User* active_user = UserManager::Get()->GetActiveUser())
-    return active_user->email() == username;
+    return active_user->email() == user_context.GetUserID();
   return false;
 }
 
-bool LoginManagerTest::AddUserToSession(const std::string& username,
-                                        const std::string& password) {
+bool LoginManagerTest::AddUserToSession(const UserContext& user_context) {
   ExistingUserController* controller =
       ExistingUserController::current_controller();
   if (!controller) {
     ADD_FAILURE();
     return false;
   }
-  UserContext user_context(username);
-  user_context.SetPassword(password);
   controller->Login(user_context);
   content::WindowedNotificationObserver(
       chrome::NOTIFICATION_SESSION_STARTED,
       content::NotificationService::AllSources()).Wait();
   const UserList& logged_users = UserManager::Get()->GetLoggedInUsers();
   for (UserList::const_iterator it = logged_users.begin();
        it != logged_users.end(); ++it) {
-    if ((*it)->email() == username)
+    if ((*it)->email() == user_context.GetUserID())
       return true;
   }
   return false;
 }
 
-void LoginManagerTest::LoginUser(const std::string& username) {
-  SetExpectedCredentials(username, "password");
-  EXPECT_TRUE(TryToLogin(username, "password"));
+void LoginManagerTest::LoginUser(const std::string& user_id) {
+  UserContext user_context(user_id);
+  user_context.SetKey(Key("password"));
+  SetExpectedCredentials(user_context);
+  EXPECT_TRUE(TryToLogin(user_context));
 }
 
-void LoginManagerTest::AddUser(const std::string& username) {
-  SetExpectedCredentials(username, "password");
-  EXPECT_TRUE(AddUserToSession(username, "password"));
+void LoginManagerTest::AddUser(const std::string& user_id) {
+  UserContext user_context(user_id);
+  user_context.SetKey(Key("password"));
+  SetExpectedCredentials(user_context);
+  EXPECT_TRUE(AddUserToSession(user_context));
 }
 
 void LoginManagerTest::JSExpect(const std::string& expression) {
   js_checker_.ExpectTrue(expression);
 }
 
 void LoginManagerTest::InitializeWebContents() {
     LoginDisplayHost* host = LoginDisplayHostImpl::default_host();
     EXPECT_TRUE(host != NULL);
 
     content::WebContents* web_contents =
         host->GetWebUILoginView()->GetWebContents();
     EXPECT_TRUE(web_contents != NULL);
     set_web_contents(web_contents);
     js_checker_.set_web_contents(web_contents);
   }
 
 }  // namespace chromeos