chrome/browser/extensions/api/identity/identity_api_unittest.cc

// Copyright 2018 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/extensions/api/identity/identity_api.h"

#include "base/test/scoped_feature_list.h"
#include "build/build_config.h"
#include "chrome/browser/signin/scoped_account_consistency.h"
#include "chrome/test/base/testing_profile.h"
#include "components/signin/public/base/signin_buildflags.h"
#include "content/public/test/test_browser_thread_bundle.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace extensions {

#if BUILDFLAG(ENABLE_DICE_SUPPORT)
// Tests that all accounts in extensions is only enabled when Dice is enabled.
TEST(IdentityApiTest, DiceAllAccountsExtensions) {
  content::TestBrowserThreadBundle test_thread_bundle;
  base::test::ScopedFeatureList feature_list;
  feature_list.InitAndEnableFeature(kExtensionsAllAccountsFeature);

  {
    ScopedAccountConsistencyDice scoped_dice;
    TestingProfile profile;
    IdentityAPI api(&profile);
    EXPECT_FALSE(api.AreExtensionsRestrictedToPrimaryAccount());
    api.Shutdown();
  }

  {
    ScopedAccountConsistencyDiceMigration scoped_dice_migration;
    TestingProfile::Builder profile_builder;
    // The profile is not a new profile to prevent automatic migration.
    profile_builder.OverrideIsNewProfile(false);
    std::unique_ptr<TestingProfile> profile = profile_builder.Build();
    IdentityAPI api(profile.get());
    EXPECT_TRUE(api.AreExtensionsRestrictedToPrimaryAccount());
    api.Shutdown();
  }
}
#endif

TEST(IdentityApiTest, AllAccountsExtensionDisabled) {
  content::TestBrowserThreadBundle test_thread_bundle;
#if BUILDFLAG(ENABLE_DICE_SUPPORT)
  base::test::ScopedFeatureList feature_list;
  feature_list.InitAndDisableFeature(kExtensionsAllAccountsFeature);
#endif
  TestingProfile profile;
  IdentityAPI api(&profile);
  EXPECT_TRUE(api.AreExtensionsRestrictedToPrimaryAccount());
  api.Shutdown();
}

}  // namespace extensions