chromium/services/network/ip_protection/ip_protection_config_cache_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_config_cache_impl.h"

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

#include "base/strings/stringprintf.h"
#include "base/test/bind.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 "net/base/features.h"
#include "net/base/network_change_notifier.h"
#include "services/network/ip_protection/ip_protection_geo_utils.h"
#include "services/network/ip_protection/ip_protection_proxy_list_manager.h"
#include "services/network/ip_protection/ip_protection_proxy_list_manager_impl.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace network {

namespace {

constexpr char kEmptyTokenCacheHistogram[] = …;

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

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

class MockIpProtectionTokenCacheManager : public IpProtectionTokenCacheManager { … };

class MockIpProtectionProxyListManager : public IpProtectionProxyListManager { … };

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

TEST_F(IpProtectionConfigCacheImplTest,
       AreAuthTokensAvailable_NoProxiesConfigured) { … }

TEST_F(IpProtectionConfigCacheImplTest,
       AuthTokensNotAvailableIfProxyListIsNotAvailable) { … }

// Token cache manager returns available token for proxyA.
TEST_F(IpProtectionConfigCacheImplTest, GetAuthTokenFromManagerForProxyA) { … }

// Token cache manager returns available token for proxyB.
TEST_F(IpProtectionConfigCacheImplTest, GetAuthTokenFromManagerForProxyB) { … }

// If a required token is missing from one of the token caches, the availability
// is set to false.
TEST_F(IpProtectionConfigCacheImplTest,
       AreAuthTokensAvailable_OneTokenCacheIsEmpty) { … }

// GetAuthToken for where proxy list manager's geo is different than the current
// geo of the config cache.
TEST_F(IpProtectionConfigCacheImplTest, GetAuthTokenForOldGeo) { … }

// Proxy list manager returns currently cached proxy hostnames.
TEST_F(IpProtectionConfigCacheImplTest, GetProxyListFromManager) { … }

// When QUIC proxies are enabled, the proxy list has both QUIC and HTTPS
// proxies, and falls back properly when a QUIC proxy fails.
TEST_F(IpProtectionConfigCacheImplTest, GetProxyListFromManagerWithQuic) { … }

// When the network changes, a new proxy list is requested.
TEST_F(IpProtectionConfigCacheImplTest, RefreshProxyListOnNetworkChange) { … }

// Simulates a geo change detected in the IppProxyListManager.
TEST_F(IpProtectionConfigCacheImplTest,
       GeoChangeObservedInIppProxyListManager) { … }

// Current geo for all token cache managers should reflect this empty geo.
TEST_F(IpProtectionConfigCacheImplTest,
       GeoChangeObservedEmptyGeoIdInProxyListManager) { … }

// When token caching by geo is disabled, `GeoObserved` has no impact.
TEST_F(IpProtectionConfigCacheImplTest,
       GeoObservedTokenCachingByGeoDisabledNoImpact) { … }

// Simulates a geo change detected in the IppTokenCacheManager.
TEST_F(IpProtectionConfigCacheImplTest,
       GeoChangeObservedInIppTokenCacheManager) { … }

}  // namespace
}  // namespace network