chrome/browser/ui/ash/glanceables/glanceables_browsertest.cc

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

#include "ash/constants/ash_features.h"
#include "ash/constants/ash_switches.h"
#include "ash/glanceables/glanceables_controller.h"
#include "ash/shell.h"
#include "base/test/scoped_feature_list.h"
#include "chrome/browser/ash/profiles/profile_helper.h"
#include "chrome/browser/browser_process.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/profiles/profile_manager.h"
#include "chrome/browser/profiles/profile_test_util.h"
#include "chrome/browser/signin/identity_manager_factory.h"
#include "chrome/common/chrome_switches.h"
#include "chrome/test/base/in_process_browser_test.h"
#include "chrome/test/base/testing_profile.h"
#include "components/account_id/account_id.h"
#include "components/session_manager/core/session_manager.h"
#include "components/signin/public/base/consent_level.h"
#include "components/signin/public/identity_manager/identity_manager.h"
#include "components/signin/public/identity_manager/identity_test_utils.h"
#include "content/public/test/browser_test.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace {

constexpr char kTestUserName[] = "test@test.test";
constexpr char kTestUserGaiaId[] = "123456";

}  // namespace

// Tests for the glanceables feature, which adds a "welcome back" screen on
// some logins.
class GlanceablesBrowserTest : public InProcessBrowserTest {
 public:
  void SetUpDefaultCommandLine(base::CommandLine* command_line) override {
    InProcessBrowserTest::SetUpDefaultCommandLine(command_line);

    // The test harness adds --no-first-run. Remove it so glanceables show up.
    command_line->RemoveSwitch(switches::kNoFirstRun);

    // Don't open a browser window, because doing so would hide glanceables.
    // Note that InProcessBrowserTest::browser() will be null.
    command_line->AppendSwitch(switches::kNoStartupWindow);

    command_line->AppendSwitch(ash::switches::kLoginManager);
    command_line->AppendSwitchASCII(ash::switches::kLoginProfile,
                                    TestingProfile::kTestUserProfileDir);
    command_line->AppendSwitch(ash::switches::kAllowFailedPolicyFetchForTest);
  }

  void CreateAndStartUserSession() {
    const AccountId account_id =
        AccountId::FromUserEmailGaiaId(kTestUserName, kTestUserGaiaId);
    auto* session_manager = session_manager::SessionManager::Get();
    session_manager->CreateSession(account_id, kTestUserName, false);

    profile_ = profiles::testing::CreateProfileSync(
        g_browser_process->profile_manager(),
        ash::ProfileHelper::GetProfilePathByUserIdHash(
            user_manager::UserManager::Get()
                ->FindUser(account_id)
                ->username_hash()));

    session_manager->NotifyUserProfileLoaded(account_id);
    session_manager->SessionStarted();
  }

  ash::GlanceablesController* glanceables_controller() {
    return ash::Shell::Get()->glanceables_controller();
  }

  signin::IdentityManager* identity_manager() {
    return IdentityManagerFactory::GetForProfileIfExists(profile_);
  }

 protected:
  base::test::ScopedFeatureList features_{ash::features::kGlanceables};
  Profile* profile_ = nullptr;
};

IN_PROC_BROWSER_TEST_F(GlanceablesBrowserTest, ShowsAndHide) {
  // Not showing on the login screen.
  EXPECT_FALSE(glanceables_controller()->IsShowing());

  CreateAndStartUserSession();

  // Not showing right after login if there's no refresh token yet.
  EXPECT_FALSE(glanceables_controller()->IsShowing());

  // Makes the primary account available, generates a refresh token and runs
  // `IdentityManager` callbacks for signin success.
  signin::MakePrimaryAccountAvailable(identity_manager(), kTestUserName,
                                      signin::ConsentLevel::kSignin);

  // Showing once the refresh token is loaded.
  EXPECT_TRUE(glanceables_controller()->IsShowing());

  // Open a browser window.
  CreateBrowser(ProfileManager::GetLastUsedProfile());

  // Glanceables should close because a window opened.
  EXPECT_FALSE(glanceables_controller()->IsShowing());

  // Simulate re-showing glanceables from overview.
  glanceables_controller()->ShowFromOverview();

  // Glanceables are showing.
  EXPECT_TRUE(glanceables_controller()->IsShowing());
}