Migrate data reduction proxy statistics prefs to profile

components/data_reduction_proxy/browser/data_reduction_proxy_prefs_unittest.cc

+// Copyright 2014 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/data_reduction_proxy/browser/data_reduction_proxy_prefs.h"
+
+#include "base/prefs/pref_registry_simple.h"
+#include "base/prefs/pref_service.h"
+#include "base/prefs/scoped_user_pref_update.h"
+#include "base/prefs/testing_pref_service.h"
+#include "base/strings/string_number_conversions.h"
+#include "components/data_reduction_proxy/common/data_reduction_proxy_pref_names.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace data_reduction_proxy {
+
+class DataReductionProxyPrefsTest : public testing::Test {
+ public:
+  virtual void SetUp() override {
+    RegisterPrefs(local_state_prefs_.registry());
+    PrefRegistrySimple* profile_registry = profile_prefs_.registry();
+    RegisterPrefs(profile_registry);
+    profile_registry->RegisterBooleanPref(
+        prefs::kStatisticsPrefsMigrated, false);
+  }
+
+  PrefService* local_state_prefs() {
+    return &local_state_prefs_;
+  }
+
+  PrefService* profile_prefs() {
+    return &profile_prefs_;
+  }
+
+  // Initializes a list with ten string representations of successive int64
+  // values, starting with |starting_value|.
+  void InitializeList(const char* pref_name,
+                      int64 starting_value,
+                      PrefService* pref_service) {
+    ListPrefUpdate list(local_state_prefs(), pref_name);
+    for (int64 i = 0; i < 10L; ++i) {
+      list->Set(i, new base::StringValue(
+          base::Int64ToString(i + starting_value)));
+    }
+  }
+
+  // Verifies that ten string repreentations of successive int64 values starting
+  // with |starting_value| are found in the |ListValue| with the associated
+  // |pref_name|.
+  void VerifyList(const char* pref_name,
+                  int64 starting_value,
+                  PrefService* pref_service) {
+    const base::ListValue* list_value = pref_service->GetList(pref_name);
+    for (int64 i = 0; i < 10L; ++i) {
+      std::string string_value;
+      int64 value;
+      list_value->GetString(i, &string_value);
+      base::StringToInt64(string_value, &value);
+      EXPECT_EQ(i + starting_value, value);
+    }
+  }
+
+ private:
+  void RegisterPrefs(PrefRegistrySimple* registry) {
+    registry->RegisterInt64Pref(prefs::kHttpReceivedContentLength, 0);
+    registry->RegisterInt64Pref(prefs::kHttpOriginalContentLength, 0);
+
+    registry->RegisterListPref(prefs::kDailyHttpOriginalContentLength);
+    registry->RegisterListPref(prefs::kDailyHttpReceivedContentLength);
+    registry->RegisterListPref(
+        prefs::kDailyOriginalContentLengthWithDataReductionProxyEnabled);
+    registry->RegisterListPref(
+        prefs::kDailyContentLengthWithDataReductionProxyEnabled);
+    registry->RegisterListPref(
+        prefs::kDailyContentLengthHttpsWithDataReductionProxyEnabled);
+    registry->RegisterListPref(
+        prefs::kDailyContentLengthShortBypassWithDataReductionProxyEnabled);
+    registry->RegisterListPref(
+        prefs::kDailyContentLengthLongBypassWithDataReductionProxyEnabled);
+    registry->RegisterListPref(
+        prefs::kDailyContentLengthUnknownWithDataReductionProxyEnabled);
+    registry->RegisterListPref(
+        prefs::kDailyOriginalContentLengthViaDataReductionProxy);
+    registry->RegisterListPref(prefs::kDailyContentLengthViaDataReductionProxy);
+    registry->RegisterInt64Pref(
+        prefs::kDailyHttpContentLengthLastUpdateDate, 0L);
+  }
+
+  TestingPrefServiceSimple local_state_prefs_;
+  TestingPrefServiceSimple profile_prefs_;
+};
+
+TEST_F(DataReductionProxyPrefsTest, TestMigration) {
+  local_state_prefs()->SetInt64(prefs::kHttpReceivedContentLength, 123L);
+  local_state_prefs()->SetInt64(prefs::kHttpOriginalContentLength, 234L);
+  local_state_prefs()->SetInt64(
+      prefs::kDailyHttpContentLengthLastUpdateDate, 345L);
+  InitializeList(
+      prefs::kDailyHttpOriginalContentLength, 0L, local_state_prefs());
+  InitializeList(
+      prefs::kDailyHttpReceivedContentLength, 1L, local_state_prefs());
+  InitializeList(
+      prefs::kDailyOriginalContentLengthWithDataReductionProxyEnabled,
+      2L,
+      local_state_prefs());
+  InitializeList(
+      prefs::kDailyContentLengthWithDataReductionProxyEnabled,
+      3L,
+      local_state_prefs());
+  InitializeList(
+      prefs::kDailyContentLengthHttpsWithDataReductionProxyEnabled,
+      4L,
+      local_state_prefs());
+  InitializeList(
+      prefs::kDailyContentLengthShortBypassWithDataReductionProxyEnabled,
+      5L,
+      local_state_prefs());
+  InitializeList(
+      prefs::kDailyContentLengthLongBypassWithDataReductionProxyEnabled,
+      6L,
+      local_state_prefs());
+  InitializeList(
+      prefs::kDailyContentLengthUnknownWithDataReductionProxyEnabled,
+      7L,
+      local_state_prefs());
+  InitializeList(
+      prefs::kDailyOriginalContentLengthViaDataReductionProxy,
+      8L,
+      local_state_prefs());
+  InitializeList(
+      prefs::kDailyContentLengthViaDataReductionProxy,
+      9L,
+      local_state_prefs());
+
+  EXPECT_EQ(0L, profile_prefs()->GetInt64(prefs::kHttpReceivedContentLength));
+  EXPECT_EQ(0L, profile_prefs()->GetInt64(prefs::kHttpOriginalContentLength));
+  EXPECT_EQ(0u, profile_prefs()->GetList(
+      prefs::kDailyHttpOriginalContentLength)->GetSize());
+  EXPECT_EQ(0u, profile_prefs()->GetList(
+      prefs::kDailyHttpReceivedContentLength)->GetSize());
+  EXPECT_EQ(0u, profile_prefs()->GetList(
+      prefs::kDailyOriginalContentLengthWithDataReductionProxyEnabled)->
+          GetSize());
+  EXPECT_EQ(0u, profile_prefs()->GetList(
+      prefs::kDailyContentLengthWithDataReductionProxyEnabled)->GetSize());
+  EXPECT_EQ(0u, profile_prefs()->GetList(
+      prefs::kDailyContentLengthHttpsWithDataReductionProxyEnabled)->GetSize());
+  EXPECT_EQ(0u, profile_prefs()->GetList(
+      prefs::kDailyContentLengthShortBypassWithDataReductionProxyEnabled)->
+          GetSize());
+  EXPECT_EQ(0u, profile_prefs()->GetList(
+      prefs::kDailyContentLengthLongBypassWithDataReductionProxyEnabled)->
+          GetSize());
+  EXPECT_EQ(0u, profile_prefs()->GetList(
+      prefs::kDailyContentLengthUnknownWithDataReductionProxyEnabled)->
+          GetSize());
+  EXPECT_EQ(0u, profile_prefs()->GetList(
+      prefs::kDailyOriginalContentLengthViaDataReductionProxy)->GetSize());
+  EXPECT_EQ(0u, profile_prefs()->GetList(
+      prefs::kDailyContentLengthViaDataReductionProxy)->GetSize());
+  EXPECT_EQ(0L, profile_prefs()->GetInt64(
+      prefs::kDailyHttpContentLengthLastUpdateDate));
+  EXPECT_FALSE(profile_prefs()->GetBoolean(prefs::kStatisticsPrefsMigrated));
+
+  data_reduction_proxy::MigrateStatisticsPrefs(local_state_prefs(),
+                                               profile_prefs());
+
+  EXPECT_EQ(123L, profile_prefs()->GetInt64(prefs::kHttpReceivedContentLength));
+  EXPECT_EQ(234L, profile_prefs()->GetInt64(prefs::kHttpOriginalContentLength));
+  VerifyList(prefs::kDailyHttpOriginalContentLength, 0L, profile_prefs());
+  VerifyList(prefs::kDailyHttpReceivedContentLength, 1L, profile_prefs());
+  VerifyList(
+      prefs::kDailyOriginalContentLengthWithDataReductionProxyEnabled,
+      2L,
+      profile_prefs());
+  VerifyList(
+      prefs::kDailyContentLengthWithDataReductionProxyEnabled,
+      3L,
+      profile_prefs());
+  VerifyList(
+      prefs::kDailyContentLengthHttpsWithDataReductionProxyEnabled,
+      4L,
+      profile_prefs());
+  VerifyList(
+      prefs::kDailyContentLengthShortBypassWithDataReductionProxyEnabled,
+      5L,
+      profile_prefs());
+  VerifyList(
+      prefs::kDailyContentLengthLongBypassWithDataReductionProxyEnabled,
+      6L,
+      profile_prefs());
+  VerifyList(
+      prefs::kDailyContentLengthUnknownWithDataReductionProxyEnabled,
+      7L,
+      profile_prefs());
+  VerifyList(
+      prefs::kDailyOriginalContentLengthViaDataReductionProxy,
+      8L,
+      profile_prefs());
+  VerifyList(
+      prefs::kDailyContentLengthViaDataReductionProxy,
+      9L,
+      profile_prefs());
+  EXPECT_EQ(345L, profile_prefs()->GetInt64(
+      prefs::kDailyHttpContentLengthLastUpdateDate));
+  EXPECT_TRUE(profile_prefs()->GetBoolean(prefs::kStatisticsPrefsMigrated));
+
+  // Migration should only happen once.
+  local_state_prefs()->SetInt64(prefs::kHttpReceivedContentLength, 456L);
+  data_reduction_proxy::MigrateStatisticsPrefs(local_state_prefs(),
+                                               profile_prefs());
+  EXPECT_EQ(123L, profile_prefs()->GetInt64(prefs::kHttpReceivedContentLength));
+}
+
+}  // namespace data_reduction_proxy