chrome/browser/invalidation/profile_invalidation_provider_factory_browsertest.cc

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

#include "chrome/browser/invalidation/profile_invalidation_provider_factory.h"

#include <stdint.h>

#include "ash/constants/ash_switches.h"
#include "base/command_line.h"
#include "base/compiler_specific.h"
#include "chrome/browser/ash/profiles/profile_helper.h"
#include "chrome/browser/profiles/profile.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/invalidation/invalidation_constants.h"
#include "components/invalidation/invalidation_listener.h"
#include "components/invalidation/profile_invalidation_provider.h"
#include "components/invalidation/public/invalidation_service.h"
#include "components/user_manager/user_manager.h"
#include "components/user_manager/user_names.h"
#include "content/public/test/browser_test.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace invalidation {

namespace {
constexpr int64_t kFakeProjectNumber = 1234567890;
}  // namespace

class ProfileInvalidationProviderFactoryTestBase : public InProcessBrowserTest {
 public:
  ProfileInvalidationProviderFactoryTestBase(
      const ProfileInvalidationProviderFactoryTestBase&) = delete;
  ProfileInvalidationProviderFactoryTestBase& operator=(
      const ProfileInvalidationProviderFactoryTestBase&) = delete;

 protected:
  ProfileInvalidationProviderFactoryTestBase();
  ~ProfileInvalidationProviderFactoryTestBase() override;

  bool CanConstructProfileInvalidationProvider(Profile* profile);
};

ProfileInvalidationProviderFactoryTestBase::
    ProfileInvalidationProviderFactoryTestBase() = default;

ProfileInvalidationProviderFactoryTestBase::
    ~ProfileInvalidationProviderFactoryTestBase() = default;

bool ProfileInvalidationProviderFactoryTestBase::
    CanConstructProfileInvalidationProvider(Profile* profile) {
  return static_cast<bool>(ProfileInvalidationProviderFactory::GetInstance()
                               ->GetServiceForBrowserContext(profile, false));
}

class ProfileInvalidationProviderFactoryLoginScreenBrowserTest
    : public ProfileInvalidationProviderFactoryTestBase {
 public:
  ProfileInvalidationProviderFactoryLoginScreenBrowserTest(
      const ProfileInvalidationProviderFactoryLoginScreenBrowserTest&) = delete;
  ProfileInvalidationProviderFactoryLoginScreenBrowserTest& operator=(
      const ProfileInvalidationProviderFactoryLoginScreenBrowserTest&) = delete;

 protected:
  ProfileInvalidationProviderFactoryLoginScreenBrowserTest();
  ~ProfileInvalidationProviderFactoryLoginScreenBrowserTest() override;

  void SetUpCommandLine(base::CommandLine* command_line) override;
};

ProfileInvalidationProviderFactoryLoginScreenBrowserTest::
    ProfileInvalidationProviderFactoryLoginScreenBrowserTest() = default;

ProfileInvalidationProviderFactoryLoginScreenBrowserTest::
    ~ProfileInvalidationProviderFactoryLoginScreenBrowserTest() = default;

void ProfileInvalidationProviderFactoryLoginScreenBrowserTest::SetUpCommandLine(
    base::CommandLine* command_line) {
  command_line->AppendSwitch(ash::switches::kLoginManager);
  command_line->AppendSwitchASCII(ash::switches::kLoginProfile, "user");
}

// Verify that no InvalidationService is instantiated for the login profile on
// the login screen.
IN_PROC_BROWSER_TEST_F(ProfileInvalidationProviderFactoryLoginScreenBrowserTest,
                       NoInvalidationService) {
  Profile* login_profile =
      ash::ProfileHelper::GetSigninProfile()->GetOriginalProfile();
  EXPECT_FALSE(CanConstructProfileInvalidationProvider(login_profile));
}

class ProfileInvalidationProviderFactoryGuestBrowserTest
    : public ProfileInvalidationProviderFactoryTestBase {
 public:
  ProfileInvalidationProviderFactoryGuestBrowserTest(
      const ProfileInvalidationProviderFactoryGuestBrowserTest&) = delete;
  ProfileInvalidationProviderFactoryGuestBrowserTest& operator=(
      const ProfileInvalidationProviderFactoryGuestBrowserTest&) = delete;

 protected:
  ProfileInvalidationProviderFactoryGuestBrowserTest();
  ~ProfileInvalidationProviderFactoryGuestBrowserTest() override;

  void SetUpCommandLine(base::CommandLine* command_line) override;
};

ProfileInvalidationProviderFactoryGuestBrowserTest::
    ProfileInvalidationProviderFactoryGuestBrowserTest() = default;

ProfileInvalidationProviderFactoryGuestBrowserTest::
    ~ProfileInvalidationProviderFactoryGuestBrowserTest() = default;

void ProfileInvalidationProviderFactoryGuestBrowserTest::SetUpCommandLine(
    base::CommandLine* command_line) {
  command_line->AppendSwitch(ash::switches::kGuestSession);
  command_line->AppendSwitch(::switches::kIncognito);
  command_line->AppendSwitchASCII(ash::switches::kLoginProfile, "user");
  command_line->AppendSwitchASCII(
      ash::switches::kLoginUser, user_manager::GuestAccountId().GetUserEmail());
}

// Verify that no InvalidationService is instantiated for the login profile or
// the guest profile while a guest session is in progress.
IN_PROC_BROWSER_TEST_F(ProfileInvalidationProviderFactoryGuestBrowserTest,
                       NoInvalidationService) {
  user_manager::UserManager* user_manager = user_manager::UserManager::Get();
  EXPECT_TRUE(user_manager->IsLoggedInAsGuest());
  Profile* guest_profile = ash::ProfileHelper::Get()
                               ->GetProfileByUser(user_manager->GetActiveUser())
                               ->GetOriginalProfile();
  Profile* login_profile =
      ash::ProfileHelper::GetSigninProfile()->GetOriginalProfile();
  EXPECT_FALSE(CanConstructProfileInvalidationProvider(guest_profile));
  EXPECT_FALSE(CanConstructProfileInvalidationProvider(login_profile));
}

using ProfileInvalidationProviderFactoryBrowserTest =
    ProfileInvalidationProviderFactoryTestBase;

IN_PROC_BROWSER_TEST_F(
    ProfileInvalidationProviderFactoryBrowserTest,
    CreatesInvalidationServiceForRegularProfileWhenDirectInvalidationsFeatureDisabled) {
  std::unique_ptr<TestingProfile> testing_profile =
      TestingProfile::Builder().Build();
  ProfileInvalidationProvider* provider =
      ProfileInvalidationProviderFactory::GetForProfile(testing_profile.get());

  ASSERT_TRUE(provider);

  auto service_or_listener =
      provider->GetInvalidationServiceOrListener(kFakeProjectNumber);

  EXPECT_TRUE(
      std::holds_alternative<InvalidationService*>(service_or_listener));
}

class ProfileInvalidationProviderFactoryWithDirectInvalidationsBrowserTest
    : public ProfileInvalidationProviderFactoryBrowserTest,
      public testing::WithParamInterface<int64_t> {
 protected:
  const auto& GetProjectNumber() const { return GetParam(); }
};

IN_PROC_BROWSER_TEST_P(
    ProfileInvalidationProviderFactoryWithDirectInvalidationsBrowserTest,
    CreatesInvalidationListenerForRegularProfileWhenDirectInvalidationsFeatureEnabled) {
  std::unique_ptr<TestingProfile> testing_profile =
      TestingProfile::Builder().Build();

  ProfileInvalidationProvider* provider =
      ProfileInvalidationProviderFactory::GetForProfile(testing_profile.get());

  ASSERT_TRUE(provider);

  auto service_or_listener =
      provider->GetInvalidationServiceOrListener(GetProjectNumber());

  ASSERT_TRUE(
      std::holds_alternative<InvalidationListener*>(service_or_listener));
  EXPECT_TRUE(std::get<InvalidationListener*>(service_or_listener));
}

INSTANTIATE_TEST_SUITE_P(
    All,
    ProfileInvalidationProviderFactoryWithDirectInvalidationsBrowserTest,
    testing::Values(kCriticalInvalidationsProjectNumber,
                    kNonCriticalInvalidationsProjectNumber));

}  // namespace invalidation