Wire NQE Prefs to Profile

net/nqe/network_qualities_prefs_manager_unittest.cc

Index: net/nqe/network_qualities_prefs_manager_unittest.cc
diff --git a/net/nqe/network_qualities_prefs_manager_unittest.cc b/net/nqe/network_qualities_prefs_manager_unittest.cc
index cfd659ba67c4abdfc6a050184266ed94f809920f..72466de84b0e4ca83ad90d37c9dcf91ce0a76ce1 100644
--- a/net/nqe/network_qualities_prefs_manager_unittest.cc
+++ b/net/nqe/network_qualities_prefs_manager_unittest.cc
@@ -9,6 +9,7 @@
 
 #include "base/macros.h"
 #include "base/run_loop.h"
+#include "base/threading/thread_checker.h"
 #include "base/values.h"
 #include "net/base/network_change_notifier.h"
 #include "net/nqe/effective_connection_type.h"
@@ -22,18 +23,49 @@ namespace {
 
 class MockPrefDelegate : public NetworkQualitiesPrefsManager::PrefDelegate {
  public:
-  MockPrefDelegate() : write_count_(0) {}
-  ~MockPrefDelegate() {}
+  MockPrefDelegate()
+      : write_count_(0), read_count_(0), value_(new base::DictionaryValue) {}
+
+  ~MockPrefDelegate() override {
+    DCHECK(thread_checker_.CalledOnValidThread());
+    value_->Clear();
+    EXPECT_EQ(0U, value_->size());
+  }
 
   void SetDictionaryValue(const base::DictionaryValue& value) override {
+    DCHECK(thread_checker_.CalledOnValidThread());
+
     write_count_++;
+    value_.reset(value.DeepCopy());
+    ASSERT_EQ(value.size(), value_->size());
+  }
+
+  const base::DictionaryValue& GetDictionaryValue() override {
+    DCHECK(thread_checker_.CalledOnValidThread());
+
+    read_count_++;
+    return *(value_.get());
   }
 
-  size_t write_count() const { return write_count_; }
+  size_t write_count() const {
+    DCHECK(thread_checker_.CalledOnValidThread());
+    return write_count_;
+  }
+
+  size_t read_count() const {
+    DCHECK(thread_checker_.CalledOnValidThread());
+    return read_count_;
+  }
 
  private:
-  // Number of times prefs were written.
-  mutable size_t write_count_;
+  // Number of times prefs were written and read, respectively..
+  size_t write_count_;
+  size_t read_count_;
+
+  // Current value of the prefs.
+  std::unique_ptr<base::DictionaryValue> value_;
+
+  base::ThreadChecker thread_checker_;
 
   DISALLOW_COPY_AND_ASSIGN(MockPrefDelegate);
 };
@@ -49,6 +81,9 @@ TEST(NetworkQualitiesPrefManager, Write) {
   manager.InitializeOnNetworkThread(&estimator);
   base::RunLoop().RunUntilIdle();
 
+  // Prefs must be read at when NetworkQualitiesPrefsManager is constructed.
+  EXPECT_EQ(1u, prefs_delegate_ptr->read_count());
+
   estimator.SimulateNetworkChange(
       NetworkChangeNotifier::ConnectionType::CONNECTION_UNKNOWN, "test");
   EXPECT_EQ(0u, prefs_delegate_ptr->write_count());
@@ -67,6 +102,130 @@ TEST(NetworkQualitiesPrefManager, Write) {
   base::RunLoop().RunUntilIdle();
   EXPECT_EQ(2u, prefs_delegate_ptr->write_count());
 
+  // Prefs should not be read again.
+  EXPECT_EQ(1u, prefs_delegate_ptr->read_count());
+
+  manager.ShutdownOnPrefThread();
+}
+
+// Verify that the pref is not written if the network ID contains a period.
+TEST(NetworkQualitiesPrefManager, WriteWithPeriodInNetworkID) {
+  std::map<std::string, std::string> variation_params;
+  TestNetworkQualityEstimator estimator(variation_params, nullptr);
+
+  std::unique_ptr<MockPrefDelegate> prefs_delegate(new MockPrefDelegate());
+  MockPrefDelegate* prefs_delegate_ptr = prefs_delegate.get();
+
+  NetworkQualitiesPrefsManager manager(std::move(prefs_delegate));
+  manager.InitializeOnNetworkThread(&estimator);
+  base::RunLoop().RunUntilIdle();
+
+  EXPECT_EQ(1u, prefs_delegate_ptr->read_count());
+
+  estimator.SimulateNetworkChange(
+      NetworkChangeNotifier::ConnectionType::CONNECTION_UNKNOWN, "te.st");
+  EXPECT_EQ(0u, prefs_delegate_ptr->write_count());
+
+  estimator.set_recent_effective_connection_type(EFFECTIVE_CONNECTION_TYPE_2G);
+  // Run a request so that effective connection type is recomputed, and
+  // observers are notified of change in the network quality.
+  estimator.RunOneRequest();
+  base::RunLoop().RunUntilIdle();
+  EXPECT_EQ(0u, prefs_delegate_ptr->write_count());
+
+  manager.ShutdownOnPrefThread();
+}
+
+TEST(NetworkQualitiesPrefManager, WriteAndReadWithMultipleNetworkIDs) {
+  std::map<std::string, std::string> variation_params;
+  TestNetworkQualityEstimator estimator(variation_params, nullptr);
+
+  std::unique_ptr<MockPrefDelegate> prefs_delegate(new MockPrefDelegate());
+
+  NetworkQualitiesPrefsManager manager(std::move(prefs_delegate));
+  manager.InitializeOnNetworkThread(&estimator);
+  base::RunLoop().RunUntilIdle();
+
+  estimator.SimulateNetworkChange(
+      NetworkChangeNotifier::ConnectionType::CONNECTION_UNKNOWN, "test");
+
+  EXPECT_EQ(0u, manager.ForceReadPrefsForTesting().size());
+
+  estimator.set_recent_effective_connection_type(
+      EFFECTIVE_CONNECTION_TYPE_SLOW_2G);
+  // Run a request so that effective connection type is recomputed, and
+  // observers are notified of change in the network quality.
+  estimator.RunOneRequest();
+  base::RunLoop().RunUntilIdle();
+  // Verify that the observer was notified, and the updated network quality was
+  // written to the prefs.
+  EXPECT_EQ(1u, manager.ForceReadPrefsForTesting().size());
+
+  // Chnage the network ID.
+  estimator.SimulateNetworkChange(
+      NetworkChangeNotifier::ConnectionType::CONNECTION_2G, "test");
+  estimator.set_recent_effective_connection_type(EFFECTIVE_CONNECTION_TYPE_2G);
+  estimator.RunOneRequest();
+  base::RunLoop().RunUntilIdle();
+  EXPECT_EQ(2u, manager.ForceReadPrefsForTesting().size());
+
+  estimator.SimulateNetworkChange(
+      NetworkChangeNotifier::ConnectionType::CONNECTION_3G, "test");
+  estimator.set_recent_effective_connection_type(EFFECTIVE_CONNECTION_TYPE_3G);
+  estimator.RunOneRequest();
+  base::RunLoop().RunUntilIdle();
+  EXPECT_EQ(3u, manager.ForceReadPrefsForTesting().size());
+
+  estimator.SimulateNetworkChange(
+      NetworkChangeNotifier::ConnectionType::CONNECTION_4G, "test");
+  estimator.set_recent_effective_connection_type(EFFECTIVE_CONNECTION_TYPE_4G);
+  estimator.RunOneRequest();
+  base::RunLoop().RunUntilIdle();
+  // Size of prefs must not exceed 3.
+  EXPECT_EQ(3u, manager.ForceReadPrefsForTesting().size());
+
+  estimator.SimulateNetworkChange(
+      NetworkChangeNotifier::ConnectionType::CONNECTION_WIFI, "test");
+  estimator.set_recent_effective_connection_type(
+      EFFECTIVE_CONNECTION_TYPE_SLOW_2G);
+  estimator.RunOneRequest();
+  base::RunLoop().RunUntilIdle();
+  std::map<nqe::internal::NetworkID, nqe::internal::CachedNetworkQuality>
+      read_prefs = manager.ForceReadPrefsForTesting();
+  EXPECT_EQ(3u, read_prefs.size());
+
+  // Verify the contents of the prefs.
+  for (std::map<nqe::internal::NetworkID,
+                nqe::internal::CachedNetworkQuality>::const_iterator it =
+           read_prefs.begin();
+       it != read_prefs.end(); ++it) {
+    EXPECT_EQ("test", it->first.id);
+    switch (it->first.type) {
+      case NetworkChangeNotifier::ConnectionType::CONNECTION_UNKNOWN:
+        EXPECT_EQ(EFFECTIVE_CONNECTION_TYPE_SLOW_2G,
+                  it->second.effective_connection_type());
+        break;
+      case NetworkChangeNotifier::ConnectionType::CONNECTION_2G:
+        EXPECT_EQ(EFFECTIVE_CONNECTION_TYPE_2G,
+                  it->second.effective_connection_type());
+        break;
+      case NetworkChangeNotifier::ConnectionType::CONNECTION_3G:
+        EXPECT_EQ(EFFECTIVE_CONNECTION_TYPE_3G,
+                  it->second.effective_connection_type());
+        break;
+      case NetworkChangeNotifier::ConnectionType::CONNECTION_4G:
+        EXPECT_EQ(EFFECTIVE_CONNECTION_TYPE_4G,
+                  it->second.effective_connection_type());
+        break;
+      case NetworkChangeNotifier::ConnectionType::CONNECTION_WIFI:
+        EXPECT_EQ(EFFECTIVE_CONNECTION_TYPE_SLOW_2G,
+                  it->second.effective_connection_type());
+        break;
+      default:
+        NOTREACHED();
+    }
+  }
+
   manager.ShutdownOnPrefThread();
 }