// 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 "services/network/network_qualities_pref_delegate.h" #include #include #include "base/bind.h" #include "base/test/metrics/histogram_tester.h" #include "base/test/task_environment.h" #include "components/prefs/testing_pref_service.h" #include "net/base/network_change_notifier.h" #include "net/nqe/cached_network_quality.h" #include "net/nqe/effective_connection_type.h" #include "net/nqe/network_id.h" #include "net/nqe/network_quality_estimator_test_util.h" #include "testing/gtest/include/gtest/gtest.h" namespace network { namespace { class NetworkQualitiesPrefDelegateTest : public testing::Test { public: NetworkQualitiesPrefDelegateTest() : task_environment_(base::test::TaskEnvironment::MainThreadType::IO) {} NetworkQualitiesPrefDelegateTest(const NetworkQualitiesPrefDelegateTest&) = delete; NetworkQualitiesPrefDelegateTest& operator=( const NetworkQualitiesPrefDelegateTest&) = delete; ~NetworkQualitiesPrefDelegateTest() override = default; private: base::test::TaskEnvironment task_environment_; }; // Verify that prefs are writen and read correctly. TEST_F(NetworkQualitiesPrefDelegateTest, WritingReadingToPrefsEnabled) { TestingPrefServiceSimple pref_service_simple; net::TestNetworkQualityEstimator estimator; NetworkQualitiesPrefDelegate::RegisterPrefs(pref_service_simple.registry()); base::HistogramTester initial_histogram_tester; NetworkQualitiesPrefDelegate pref_delegate(&pref_service_simple, &estimator); // NetworkQualityEstimator must be notified of the read prefs at startup. EXPECT_FALSE( initial_histogram_tester.GetAllSamples("NQE.Prefs.ReadSize").empty()); { base::HistogramTester histogram_tester; estimator.set_effective_connection_type( net::EFFECTIVE_CONNECTION_TYPE_OFFLINE); estimator.set_recent_effective_connection_type( net::EFFECTIVE_CONNECTION_TYPE_OFFLINE); estimator.RunOneRequest(); // Prefs are written only if persistent caching was enabled. EXPECT_FALSE( histogram_tester.GetAllSamples("NQE.Prefs.WriteCount").empty()); histogram_tester.ExpectTotalCount("NQE.Prefs.ReadCount", 0); // NetworkQualityEstimator should not be notified of change in prefs. histogram_tester.ExpectTotalCount("NQE.Prefs.ReadSize", 0); } { base::HistogramTester histogram_tester; estimator.set_effective_connection_type( net::EFFECTIVE_CONNECTION_TYPE_SLOW_2G); estimator.set_recent_effective_connection_type( net::EFFECTIVE_CONNECTION_TYPE_SLOW_2G); estimator.RunOneRequest(); // Prefs are written even if the network id was unavailable. EXPECT_FALSE( histogram_tester.GetAllSamples("NQE.Prefs.WriteCount").empty()); histogram_tester.ExpectTotalCount("NQE.Prefs.ReadCount", 0); // NetworkQualityEstimator should not be notified of change in prefs. histogram_tester.ExpectTotalCount("NQE.Prefs.ReadSize", 0); } // Verify the contents of the prefs by reading them again. std::map read_prefs = pref_delegate.ForceReadPrefsForTesting(); // Number of entries must be between 1 and 2. It's possible that 2 entries // are added if the connection type is unknown to network quality estimator // at the time of startup, and shortly after it receives a notification // about the change in the connection type. EXPECT_LE(1u, read_prefs.size()); EXPECT_GE(2u, read_prefs.size()); // Verify that the cached network quality was written correctly. EXPECT_EQ(net::EFFECTIVE_CONNECTION_TYPE_SLOW_2G, read_prefs.begin()->second.effective_connection_type()); if (net::NetworkChangeNotifier::GetConnectionType() == net::NetworkChangeNotifier::CONNECTION_ETHERNET) { // Verify that the network ID was written correctly. net::nqe::internal::NetworkID ethernet_network_id( net::NetworkChangeNotifier::CONNECTION_ETHERNET, std::string(), INT32_MIN); EXPECT_EQ(ethernet_network_id, read_prefs.begin()->first); } } } // namespace } // namespace network