chromium/chrome/browser/ui/ash/multi_user/multi_user_util_chromeos_unittest.cc

// Copyright 2016 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include <memory>

#include "base/functional/bind.h"
#include "base/memory/ptr_util.h"
#include "base/memory/raw_ptr.h"
#include "chrome/browser/ash/login/users/fake_chrome_user_manager.h"
#include "chrome/browser/signin/identity_manager_factory.h"
#include "chrome/browser/signin/identity_test_environment_profile_adaptor.h"
#include "chrome/browser/ui/ash/multi_user/multi_user_util.h"
#include "chrome/test/base/chrome_ash_test_base.h"
#include "chrome/test/base/testing_profile.h"
#include "components/account_id/account_id.h"
#include "components/signin/public/identity_manager/account_info.h"
#include "components/signin/public/identity_manager/identity_manager.h"
#include "components/signin/public/identity_manager/identity_test_environment.h"
#include "components/user_manager/scoped_user_manager.h"
#include "components/user_manager/user.h"

namespace ash {

namespace {

const char kTestAccountEmail[] = "[email protected]";

}  // namespace

class MultiUserUtilTest : public ChromeAshTestBase {
 public:
  MultiUserUtilTest() {}

  MultiUserUtilTest(const MultiUserUtilTest&) = delete;
  MultiUserUtilTest& operator=(const MultiUserUtilTest&) = delete;

  ~MultiUserUtilTest() override {}

  void SetUp() override {
    ChromeAshTestBase::SetUp();

    fake_user_manager_ = new FakeChromeUserManager;
    user_manager_enabler_ = std::make_unique<user_manager::ScopedUserManager>(
        base::WrapUnique(fake_user_manager_.get()));

    profile_.reset(IdentityTestEnvironmentProfileAdaptor::
                       CreateProfileForIdentityTestEnvironment()
                           .release());

    identity_test_env_adaptor_ =
        std::make_unique<IdentityTestEnvironmentProfileAdaptor>(profile_.get());
  }

  void TearDown() override {
    identity_test_env_adaptor_.reset();
    profile_.reset();
    ChromeAshTestBase::TearDown();
  }

  // Add a user to the identity manager with given gaia_id and email.
  CoreAccountId AddUserAndSignIn(const std::string& email) {
    AccountInfo account_info = identity_test_env()->MakePrimaryAccountAvailable(
        email, signin::ConsentLevel::kSync);
    auto* user = fake_user_manager_->AddUser(
        multi_user_util::GetAccountIdFromEmail(account_info.email));
    fake_user_manager_->UserLoggedIn(
        user->GetAccountId(), user->username_hash(),
        false /* browser_restart */, false /* is_child */);

    return account_info.account_id;
  }

  void SimulateTokenRevoked(const CoreAccountId& account_id) {
    identity_test_env()->RemoveRefreshTokenForAccount(account_id);
  }

  TestingProfile* profile() { return profile_.get(); }

  signin::IdentityTestEnvironment* identity_test_env() {
    return identity_test_env_adaptor_->identity_test_env();
  }

 private:
  std::unique_ptr<TestingProfile> profile_;
  std::unique_ptr<IdentityTestEnvironmentProfileAdaptor>
      identity_test_env_adaptor_;
  // |fake_user_manager_| is owned by |user_manager_enabler_|.
  raw_ptr<FakeChromeUserManager, DanglingUntriaged> fake_user_manager_;
  std::unique_ptr<user_manager::ScopedUserManager> user_manager_enabler_;
};

// Test that during the session it will always return a valid account id if a
// valid profile is provided, even if this profile's refresh token has been
// revoked. (On Chrome OS we don't force to end the session in this case.)
TEST_F(MultiUserUtilTest, ReturnValidAccountIdIfTokenRevoked) {
  CoreAccountId account_id = AddUserAndSignIn(kTestAccountEmail);
  signin::IdentityManager* identity_manager =
      identity_test_env()->identity_manager();

  EXPECT_TRUE(identity_manager->HasAccountWithRefreshToken(account_id));
  EXPECT_EQ(kTestAccountEmail,
            multi_user_util::GetAccountIdFromProfile(profile()).GetUserEmail());

  SimulateTokenRevoked(account_id);

  EXPECT_FALSE(identity_manager->HasAccountWithRefreshToken(account_id));
  EXPECT_EQ(kTestAccountEmail,
            multi_user_util::GetAccountIdFromProfile(profile()).GetUserEmail());
}

}  // namespace ash