components/send_tab_to_self/entry_point_display_reason_unittest.cc

// Copyright 2022 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 "components/send_tab_to_self/entry_point_display_reason.h"

#include <memory>
#include <utility>

#include "base/bind.h"
#include "base/test/scoped_feature_list.h"
#include "components/prefs/pref_registry_simple.h"
#include "components/prefs/testing_pref_service.h"
#include "components/send_tab_to_self/features.h"
#include "components/send_tab_to_self/send_tab_to_self_sync_service.h"
#include "components/send_tab_to_self/test_send_tab_to_self_model.h"
#include "components/signin/public/base/signin_pref_names.h"
#include "components/sync/driver/test_sync_service.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "url/gurl.h"

namespace send_tab_to_self {

namespace {

const char kHttpsUrl[] = "https://www.foo.com";
const char kHttpUrl[] = "http://www.foo.com";

class FakeSendTabToSelfModel : public TestSendTabToSelfModel {
 public:
  FakeSendTabToSelfModel() = default;
  ~FakeSendTabToSelfModel() override = default;

  void SetIsReady(bool is_ready) { is_ready_ = is_ready; }
  void SetHasValidTargetDevice(bool has_valid_target_device) {
    if (has_valid_target_device) {
      DCHECK(is_ready_) << "Target devices are only known if the model's ready";
    }
    has_valid_target_device_ = has_valid_target_device;
  }

  bool IsReady() override { return is_ready_; }
  bool HasValidTargetDevice() override { return has_valid_target_device_; }

 private:
  bool is_ready_ = false;
  bool has_valid_target_device_ = false;
};

class FakeSendTabToSelfSyncService : public SendTabToSelfSyncService {
 public:
  FakeSendTabToSelfSyncService() = default;
  ~FakeSendTabToSelfSyncService() override = default;

  FakeSendTabToSelfModel* GetSendTabToSelfModel() override { return &model_; }

 private:
  FakeSendTabToSelfModel model_;
};

class EntryPointDisplayReasonTest : public ::testing::Test {
 public:
  EntryPointDisplayReasonTest() {
    pref_service_.registry()->RegisterBooleanPref(prefs::kSigninAllowed, true);
  }

  syncer::TestSyncService* sync_service() { return &sync_service_; }
  FakeSendTabToSelfSyncService* send_tab_to_self_sync_service() {
    return &send_tab_to_self_sync_service_;
  }
  TestingPrefServiceSimple* pref_service() { return &pref_service_; }

  void SignIn() {
    CoreAccountInfo account;
    account.gaia = "gaia_id";
    account.email = "email@test.com";
    account.account_id = CoreAccountId::FromGaiaId(account.gaia);
    sync_service_.SetAccountInfo(account);
  }

 private:
  syncer::TestSyncService sync_service_;
  FakeSendTabToSelfSyncService send_tab_to_self_sync_service_;
  TestingPrefServiceSimple pref_service_;
};

TEST_F(EntryPointDisplayReasonTest,
       ShouldHideEntryPointIfSignedOutAndPromoFeatureDisabled) {
  base::test::ScopedFeatureList feature_list;
  feature_list.InitAndDisableFeature(kSendTabToSelfSigninPromo);

  EXPECT_FALSE(GetEntryPointDisplayReason(GURL(kHttpsUrl), sync_service(),
                                          send_tab_to_self_sync_service(),
                                          pref_service()));
}

TEST_F(EntryPointDisplayReasonTest,
       ShouldShowPromoIfSignedOutAndPromoFeatureEnabled) {
  base::test::ScopedFeatureList feature_list;
  feature_list.InitAndEnableFeature(kSendTabToSelfSigninPromo);

  EXPECT_EQ(EntryPointDisplayReason::kOfferSignIn,
            GetEntryPointDisplayReason(GURL(kHttpsUrl), sync_service(),
                                       send_tab_to_self_sync_service(),
                                       pref_service()));
}

TEST_F(EntryPointDisplayReasonTest, ShouldHideEntryPointIfModelNotReady) {
  SignIn();
  send_tab_to_self_sync_service()->GetSendTabToSelfModel()->SetIsReady(false);
  send_tab_to_self_sync_service()
      ->GetSendTabToSelfModel()
      ->SetHasValidTargetDevice(false);

  EXPECT_FALSE(GetEntryPointDisplayReason(GURL(kHttpsUrl), sync_service(),
                                          send_tab_to_self_sync_service(),
                                          pref_service()));
}

TEST_F(EntryPointDisplayReasonTest,
       ShouldHideEntryPointIfHasNoValidTargetDeviceAndPromoFeatureDisabled) {
  base::test::ScopedFeatureList feature_list;
  feature_list.InitAndDisableFeature(kSendTabToSelfSigninPromo);

  SignIn();
  send_tab_to_self_sync_service()->GetSendTabToSelfModel()->SetIsReady(true);
  send_tab_to_self_sync_service()
      ->GetSendTabToSelfModel()
      ->SetHasValidTargetDevice(false);

  EXPECT_FALSE(GetEntryPointDisplayReason(GURL(kHttpsUrl), sync_service(),
                                          send_tab_to_self_sync_service(),
                                          pref_service()));
}

TEST_F(EntryPointDisplayReasonTest,
       ShouldShowPromoIfHasNoValidTargetDeviceAndPromoFeatureEnabled) {
  base::test::ScopedFeatureList feature_list;
  feature_list.InitAndEnableFeature(kSendTabToSelfSigninPromo);

  SignIn();
  send_tab_to_self_sync_service()->GetSendTabToSelfModel()->SetIsReady(true);
  send_tab_to_self_sync_service()
      ->GetSendTabToSelfModel()
      ->SetHasValidTargetDevice(false);

  EXPECT_EQ(EntryPointDisplayReason::kInformNoTargetDevice,
            GetEntryPointDisplayReason(GURL(kHttpsUrl), sync_service(),
                                       send_tab_to_self_sync_service(),
                                       pref_service()));
}

TEST_F(EntryPointDisplayReasonTest, ShouldOnlyOfferFeatureIfHttpOrHttps) {
  SignIn();
  send_tab_to_self_sync_service()->GetSendTabToSelfModel()->SetIsReady(true);
  send_tab_to_self_sync_service()
      ->GetSendTabToSelfModel()
      ->SetHasValidTargetDevice(true);

  EXPECT_EQ(EntryPointDisplayReason::kOfferFeature,
            GetEntryPointDisplayReason(GURL(kHttpsUrl), sync_service(),
                                       send_tab_to_self_sync_service(),
                                       pref_service()));

  EXPECT_EQ(EntryPointDisplayReason::kOfferFeature,
            GetEntryPointDisplayReason(GURL(kHttpUrl), sync_service(),
                                       send_tab_to_self_sync_service(),
                                       pref_service()));

  EXPECT_FALSE(GetEntryPointDisplayReason(GURL("192.168.0.0"), sync_service(),
                                          send_tab_to_self_sync_service(),
                                          pref_service()));

  EXPECT_FALSE(GetEntryPointDisplayReason(
      GURL("chrome-untrusted://url"), sync_service(),
      send_tab_to_self_sync_service(), pref_service()));

  EXPECT_FALSE(GetEntryPointDisplayReason(
      GURL("chrome://flags"), sync_service(), send_tab_to_self_sync_service(),
      pref_service()));

  EXPECT_FALSE(GetEntryPointDisplayReason(
      GURL("tel:07399999999"), sync_service(), send_tab_to_self_sync_service(),
      pref_service()));
}

TEST_F(EntryPointDisplayReasonTest, ShouldHideEntryPointInIncognitoMode) {
  // Note: if changing this, audit profile-finding logic in the feature.
  // For example, NotificationManager.java in the Android code assumes
  // incognito is not supported.
  EXPECT_FALSE(GetEntryPointDisplayReason(GURL(kHttpsUrl), nullptr, nullptr,
                                          pref_service()));
}

}  // namespace

}  // namespace send_tab_to_self