chromium/services/network/ip_protection/ip_protection_proxy_list_manager_impl_unittest.cc

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

#include "services/network/ip_protection/ip_protection_proxy_list_manager_impl.h"

#include <deque>
#include <map>
#include <optional>
#include <string>
#include <utility>
#include <vector>

#include "base/notreached.h"
#include "base/strings/stringprintf.h"
#include "base/test/metrics/histogram_tester.h"
#include "base/test/scoped_feature_list.h"
#include "base/test/task_environment.h"
#include "components/ip_protection/common/ip_protection_data_types.h"
#include "components/ip_protection/common/ip_protection_telemetry.h"
#include "net/base/features.h"
#include "net/base/proxy_chain.h"
#include "services/network/ip_protection/ip_protection_geo_utils.h"
#include "services/network/ip_protection/ip_protection_proxy_list_manager.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace network {

namespace {

constexpr char kGetProxyListResultHistogram[] = …;
constexpr char kProxyListRefreshTimeHistogram[] = …;

constexpr char kDefaultGeoId[] = …;
constexpr char kMountainViewGeoId[] = …;
constexpr char kSunnyvaleGeoId[] = …;

constexpr bool kEnableTokenCacheByGeo = …;
constexpr bool kDisableTokenCacheByGeo = …;

struct GetProxyListCall { … };

class MockIpProtectionConfigGetter : public IpProtectionConfigGetter { … };

class MockIpProtectionConfigCache : public IpProtectionConfigCache { … };

class IpProtectionProxyListManagerImplTest : public testing::Test { … };

// The manager gets the proxy list on startup and once again on schedule.
TEST_F(IpProtectionProxyListManagerImplTest,
       ProxyListOnStartupGeoCachingDisabled) { … }

// The manager gets the proxy list on startup and once again on schedule.
TEST_F(IpProtectionProxyListManagerImplTest,
       ProxyListOnStartupGeoCachingEnabled) { … }

// The manager should continue scheduling refreshes even if the most recent
// fails.
TEST_F(IpProtectionProxyListManagerImplTest,
       ProxyListRefreshScheduledIfRefreshFails) { … }

// The manager refreshes the proxy list on demand, but only once even if
// `RequestRefreshProxyList()` is called repeatedly.
TEST_F(IpProtectionProxyListManagerImplTest,
       ProxyListRefreshGeoCachingDisabled) { … }

// The manager refreshes the proxy list on demand, but only once even if
// `RequestRefreshProxyList()` is called repeatedly.
TEST_F(IpProtectionProxyListManagerImplTest,
       ProxyListRefreshGeoCachingEnabled) { … }

TEST_F(IpProtectionProxyListManagerImplTest,
       IsProxyListAvailableEvenIfEmptyGeoCachingDisabled) { … }

TEST_F(IpProtectionProxyListManagerImplTest,
       IsProxyListAvailableEvenIfEmptyGeoCachingEnabled) { … }

// The manager keeps its existing proxy list if it fails to fetch a new one.
TEST_F(IpProtectionProxyListManagerImplTest,
       ProxyListKeptAfterFailureGeoCachingDisabled) { … }

// The manager keeps its existing proxy list if it fails to fetch a new one.
TEST_F(IpProtectionProxyListManagerImplTest,
       ProxyListKeptAfterFailureGeoCachingEnabled) { … }

TEST_F(IpProtectionProxyListManagerImplTest, GetProxyListFailureRecorded) { … }

TEST_F(IpProtectionProxyListManagerImplTest, GotEmptyProxyListRecorded) { … }

TEST_F(IpProtectionProxyListManagerImplTest, GotPopulatedProxyListRecorded) { … }

TEST_F(IpProtectionProxyListManagerImplTest,
       CurrentGeoCachingByGeoDisabledReturnsDefault) { … }

TEST_F(IpProtectionProxyListManagerImplTest,
       CurrentGeoCachingByGeoEnabledReturnsGeoOfProxyList) { … }

// If the geo caching feature is disabled, setting the geo should have no effect
// and should continue returning the default geo.
TEST_F(IpProtectionProxyListManagerImplTest,
       RefreshProxyListForGeoChangeCachingByGeoDisabledNoRefresh) { … }

TEST_F(IpProtectionProxyListManagerImplTest,
       RefreshProxyListForGeoChangeCachingByGeoEnabledGeoChanged) { … }

// If `RefreshProxyListForGeoChange` is called multiple times, the refresh is
// only requested once within the default interval.
TEST_F(IpProtectionProxyListManagerImplTest,
       RefreshProxyListForGeoChangeCachingByGeoEnabledOnlyObservesGeo) { … }

// If a proxy list refresh returns the same geo as the current geo, no callbacks
// to `GeoObserved` are made.
TEST_F(IpProtectionProxyListManagerImplTest,
       CachingByGeoNoGeoObservedWhenNewGeoMatchesCurrent) { … }

}  // namespace
}  // namespace network