chrome/browser/ui/promos/ios_promos_utils_browsertest.cc

// Copyright 2024 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/ui/promos/ios_promos_utils.h"

#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/promos/promos_types.h"
#include "chrome/browser/signin/identity_manager_factory.h"
#include "chrome/browser/sync/sync_service_factory.h"
#include "chrome/browser/sync/test/integration/sync_service_impl_harness.h"
#include "chrome/browser/sync/test/integration/sync_test.h"
#include "chrome/browser/ui/browser.h"
#include "chrome/browser/ui/views/frame/browser_view.h"
#include "chrome/test/base/chrome_test_utils.h"
#include "components/signin/public/base/consent_level.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_utils.h"
#include "components/signin/public/identity_manager/signin_constants.h"
#include "components/sync/service/sync_service.h"
#include "components/sync/test/fake_server_network_resources.h"
#include "content/public/test/browser_test.h"

using signin::constants::kNoHostedDomainFound;

class IOSPromosUtilsTest : public SyncTest {
 public:
  IOSPromosUtilsTest() : SyncTest(SINGLE_CLIENT) {
    scoped_iph_feature_list_.InitAndEnableFeatures(
        {feature_engagement::kIPHiOSPasswordPromoDesktopFeature,
         feature_engagement::kIPHiOSPaymentPromoDesktopFeature,
         feature_engagement::kIPHiOSAddressPromoDesktopFeature});
  }

  IOSPromosUtilsTest(const IOSPromosUtilsTest&) = delete;

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

  void SetupSyncForAccount() {
    ASSERT_TRUE(SetupClients());

    const signin::ConsentLevel consent_level = signin::ConsentLevel::kSync;

    // Sign the profile in.
    ASSERT_TRUE(GetClient(0)->SignInPrimaryAccount(consent_level));

    CoreAccountInfo current_info =
        IdentityManagerFactory::GetForProfile(GetProfile(0))
            ->GetPrimaryAccountInfo(consent_level);
    // Need to update hosted domain since it is not populated.
    AccountInfo account_info;
    account_info.account_id = current_info.account_id;
    account_info.gaia = current_info.gaia;
    account_info.email = current_info.email;
    account_info.hosted_domain = kNoHostedDomainFound;
    signin::UpdateAccountInfoForAccount(
        IdentityManagerFactory::GetForProfile(GetProfile(0)), account_info);

    // TODO(crbug.com/417921582): Use SetupClientsAndSignIn() once it exists,
    // and switch to ConsentLevel::kSignin above.
    ASSERT_TRUE(SetupSync());
  }

  feature_engagement::test::ScopedIphFeatureList scoped_iph_feature_list_;
};

IN_PROC_BROWSER_TEST_F(IOSPromosUtilsTest, InvokeUi_passwords) {
  SetupSyncForAccount();

  ios_promos_utils::VerifyIOSPromoEligibility(IOSPromoType::kPassword,
                                              GetBrowser(0));
}

IN_PROC_BROWSER_TEST_F(IOSPromosUtilsTest, InvokeUi_addresses) {
  SetupSyncForAccount();

  ios_promos_utils::VerifyIOSPromoEligibility(IOSPromoType::kAddress,
                                              browser());
}

IN_PROC_BROWSER_TEST_F(IOSPromosUtilsTest, InvokeUi_payments) {
  SetupSyncForAccount();

  ios_promos_utils::VerifyIOSPromoEligibility(IOSPromoType::kPayment,
                                              browser());
}