src/chrome/browser/ui/webui/chromeos/login/signin_userlist_unittest.cc

   1 // Copyright 2014 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 "base/compiler_specific.h"
   6 #include "chrome/browser/chromeos/login/screens/user_selection_screen.h"
   7 #include "chrome/browser/chromeos/login/users/fake_user_manager.h"
   8 #include "chrome/browser/chromeos/login/users/multi_profile_user_controller.h"
   9 #include "chrome/browser/chromeos/login/users/multi_profile_user_controller_delegate.h"
  10 #include "chrome/browser/chromeos/login/users/scoped_user_manager_enabler.h"
  11 #include "chrome/browser/signin/screenlock_bridge.h"
  12 #include "chrome/test/base/testing_browser_process.h"
  13 #include "chrome/test/base/testing_profile_manager.h"
  14 #include "components/user_manager/user.h"
  15 #include "testing/gtest/include/gtest/gtest.h"
  16
  17 namespace {
  18
  19 const size_t kMaxUsers = 18; // same as in user_selection_screen.cc
  20 const char* kOwner = "owner@gmail.com";
  21 const char* kUsersPublic[] = {"public0@gmail.com", "public1@gmail.com"};
  22 const char* kUsers[] = {
  23     "a0@gmail.com", "a1@gmail.com", "a2@gmail.com", "a3@gmail.com",
  24     "a4@gmail.com", "a5@gmail.com", "a6@gmail.com", "a7@gmail.com",
  25     "a8@gmail.com", "a9@gmail.com", "a10@gmail.com", "a11@gmail.com",
  26     "a12@gmail.com", "a13@gmail.com", "a14@gmail.com", "a15@gmail.com",
  27     "a16@gmail.com", "a17@gmail.com", kOwner, "a18@gmail.com"};
  28
  29 }  // namespace
  30
  31 namespace chromeos {
  32
  33 class SigninPrepareUserListTest
  34     : public testing::Test,
  35       public MultiProfileUserControllerDelegate {
  36  public:
  37   SigninPrepareUserListTest()
  38       : fake_user_manager_(new FakeUserManager()),
  39         user_manager_enabler_(fake_user_manager_) {
  40   }
  41
  42   virtual ~SigninPrepareUserListTest() {
  43   }
  44
  45   virtual void SetUp() OVERRIDE {
  46     profile_manager_.reset(
  47         new TestingProfileManager(TestingBrowserProcess::GetGlobal()));
  48     ASSERT_TRUE(profile_manager_->SetUp());
  49     controller_.reset(new MultiProfileUserController(
  50         this, TestingBrowserProcess::GetGlobal()->local_state()));
  51     fake_user_manager_->set_multi_profile_user_controller(controller_.get());
  52
  53     for (size_t i = 0; i < arraysize(kUsersPublic); ++i)
  54       fake_user_manager_->AddPublicAccountUser(kUsersPublic[i]);
  55
  56     for (size_t i = 0; i < arraysize(kUsers); ++i)
  57       fake_user_manager_->AddUser(kUsers[i]);
  58
  59     fake_user_manager_->set_owner_email(kOwner);
  60   }
  61
  62   virtual void TearDown() OVERRIDE {
  63     controller_.reset();
  64     profile_manager_.reset();
  65   }
  66
  67   // MultiProfileUserControllerDelegate overrides:
  68   virtual void OnUserNotAllowed(const std::string& user_email) OVERRIDE {
  69   }
  70
  71   FakeUserManager* fake_user_manager_;
  72   ScopedUserManagerEnabler user_manager_enabler_;
  73   scoped_ptr<TestingProfileManager> profile_manager_;
  74   std::map<std::string,
  75            ScreenlockBridge::LockHandler::AuthType> user_auth_type_map;
  76   scoped_ptr<MultiProfileUserController> controller_;
  77
  78   DISALLOW_COPY_AND_ASSIGN(SigninPrepareUserListTest);
  79 };
  80
  81 TEST_F(SigninPrepareUserListTest, AlwaysKeepOwnerInList) {
  82   EXPECT_LT(kMaxUsers, fake_user_manager_->GetUsers().size());
  83   user_manager::UserList users_to_send =
  84       UserSelectionScreen::PrepareUserListForSending(
  85           fake_user_manager_->GetUsers(), kOwner, true /* is signin to add */);
  86
  87   EXPECT_EQ(kMaxUsers, users_to_send.size());
  88   EXPECT_EQ(kOwner, users_to_send.back()->email());
  89
  90   fake_user_manager_->RemoveUserFromList("a16@gmail.com");
  91   fake_user_manager_->RemoveUserFromList("a17@gmail.com");
  92   users_to_send = UserSelectionScreen::PrepareUserListForSending(
  93       fake_user_manager_->GetUsers(),
  94       kOwner,
  95       true /* is signin to add */);
  96
  97   EXPECT_EQ(kMaxUsers, users_to_send.size());
  98   EXPECT_EQ("a18@gmail.com", users_to_send.back()->email());
  99   EXPECT_EQ(kOwner, users_to_send[kMaxUsers-2]->email());
 100 }
 101
 102 TEST_F(SigninPrepareUserListTest, PublicAccounts) {
 103   user_manager::UserList users_to_send =
 104       UserSelectionScreen::PrepareUserListForSending(
 105           fake_user_manager_->GetUsers(), kOwner, true /* is signin to add */);
 106
 107   EXPECT_EQ(kMaxUsers, users_to_send.size());
 108   EXPECT_EQ("a0@gmail.com", users_to_send.front()->email());
 109
 110   users_to_send = UserSelectionScreen::PrepareUserListForSending(
 111       fake_user_manager_->GetUsers(),
 112       kOwner,
 113       false /* is signin to add */);
 114
 115   EXPECT_EQ(kMaxUsers, users_to_send.size());
 116   EXPECT_EQ("public0@gmail.com", users_to_send.front()->email());
 117 }
 118
 119 }  // namespace chromeos