Implement UMA log throttling for cellular connections

components/metrics/metrics_data_use_measurements_unittest.cc

+#include "components/metrics/metrics_data_use_measurements.h"
+
+#include "base/strings/stringprintf.h"
+#include "components/metrics/metrics_pref_names.h"
+#include "components/prefs/pref_registry_simple.h"
+#include "components/prefs/testing_pref_service.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace metrics {
+
+namespace {
+const char kTodayStr[] = "2016-03-16";
+const char kYesterdayStr[] = "2016-03-15";
+const char kExpiredDateStr1[] = "2016-03-09";
+const char kExpiredDateStr2[] = "2016-03-01";
+
+class TestDataUsePrefService : public TestingPrefServiceSimple {
+ public:
+  TestDataUsePrefService() {
+    registry()->RegisterDictionaryPref(metrics::prefs::kUserCellDataUse);
+    registry()->RegisterDictionaryPref(metrics::prefs::kUmaCellDataUse);
+  }
+
+  void ClearDataUsePrefs() {
+    ClearPref(metrics::prefs::kUserCellDataUse);
+    ClearPref(metrics::prefs::kUmaCellDataUse);
+  }
+};
+
+void SetPrefTestValuesOverRatio(PrefService* local_state) {
+  base::DictionaryValue user_pref_dict;
+  user_pref_dict.SetInteger(kTodayStr, 2*100);
+  user_pref_dict.SetInteger(kYesterdayStr, 2*100);
+  user_pref_dict.SetInteger(kExpiredDateStr1, 2*100);
+  user_pref_dict.SetInteger(kExpiredDateStr2, 2*100);
+  local_state->Set(prefs::kUserCellDataUse, user_pref_dict);
+
+  base::DictionaryValue uma_pref_dict;
+  uma_pref_dict.SetInteger(kTodayStr, 50);
+  uma_pref_dict.SetInteger(kYesterdayStr, 50);
+  uma_pref_dict.SetInteger(kExpiredDateStr1, 50);
+  uma_pref_dict.SetInteger(kExpiredDateStr2, 50);
+  local_state->Set(prefs::kUmaCellDataUse, uma_pref_dict);
+}
+
+void SetPrefTestValuesValidRatio(PrefService* local_state) {
+  base::DictionaryValue user_pref_dict;
+  user_pref_dict.SetInteger(kTodayStr, 100*100);
+  user_pref_dict.SetInteger(kYesterdayStr, 100*100);
+  user_pref_dict.SetInteger(kExpiredDateStr1, 100*100);
+  user_pref_dict.SetInteger(kExpiredDateStr2, 100*100);
+  local_state->Set(prefs::kUserCellDataUse, user_pref_dict);
+
+  // Should be 4% of user traffic
+  base::DictionaryValue uma_pref_dict;
+  uma_pref_dict.SetInteger(kTodayStr, 4*100);
+  uma_pref_dict.SetInteger(kYesterdayStr, 4*100);
+  uma_pref_dict.SetInteger(kExpiredDateStr1, 4*100);
+  uma_pref_dict.SetInteger(kExpiredDateStr2, 4*100);
+  local_state->Set(prefs::kUmaCellDataUse, uma_pref_dict);
+}
+
+}  // namespace
+
+TEST(MetricsDataUseMeasurementsTest, CheckUpdateUsagePref) {
+  TestDataUsePrefService local_state;
+  MetricsDataUseMeasurements::Initialize(&local_state);
+  local_state.ClearDataUsePrefs();
+  MetricsDataUseMeasurements::GetInstance()->SetMeasurementDateForTesting(kTodayStr);
+
+  int user_pref_value = 0;
+  int uma_pref_value = 0;
+
+  MetricsDataUseMeasurements::GetInstance()->UpdateMetricsUsagePrefsOnUIThread("", 2*100);
+  local_state.GetDictionary(prefs::kUserCellDataUse)->GetInteger(kTodayStr, &user_pref_value);
+  EXPECT_EQ(2*100, user_pref_value);
+  local_state.GetDictionary(prefs::kUmaCellDataUse)->GetInteger(kTodayStr, &uma_pref_value);
+  EXPECT_EQ(0, uma_pref_value);
+
+  MetricsDataUseMeasurements::GetInstance()->UpdateMetricsUsagePrefsOnUIThread("UMA", 100);
+  local_state.GetDictionary(prefs::kUserCellDataUse)->GetInteger(kTodayStr, &user_pref_value);
+  EXPECT_EQ(3*100, user_pref_value);
+  local_state.GetDictionary(prefs::kUmaCellDataUse)->GetInteger(kTodayStr, &uma_pref_value);
+  EXPECT_EQ(100, uma_pref_value);
+}
+
+TEST(MetricsDataUseMeasurementsTest, CheckRemoveExpiredEntries) {
+  TestDataUsePrefService local_state;
+  MetricsDataUseMeasurements::Initialize(&local_state);
+  local_state.ClearDataUsePrefs();
+  MetricsDataUseMeasurements::GetInstance()->SetMeasurementDateForTesting(kTodayStr);
+  SetPrefTestValuesOverRatio(&local_state);
+  MetricsDataUseMeasurements::GetInstance()->RemoveExpiredEntries();
+
+  int user_pref_value = 0;
+  int uma_pref_value = 0;
+
+  local_state.GetDictionary(prefs::kUserCellDataUse)->GetInteger(kExpiredDateStr1, &user_pref_value);
+  EXPECT_EQ(0, user_pref_value);
+  local_state.GetDictionary(prefs::kUmaCellDataUse)->GetInteger(kExpiredDateStr1, &uma_pref_value);
+  EXPECT_EQ(0, uma_pref_value);
+
+  local_state.GetDictionary(prefs::kUserCellDataUse)->GetInteger(kExpiredDateStr2, &user_pref_value);
+  EXPECT_EQ(0, user_pref_value);
+  local_state.GetDictionary(prefs::kUmaCellDataUse)->GetInteger(kExpiredDateStr2, &uma_pref_value);
+  EXPECT_EQ(0, uma_pref_value);
+
+  local_state.GetDictionary(prefs::kUserCellDataUse)->GetInteger(kTodayStr, &user_pref_value);
+  EXPECT_EQ(2*100, user_pref_value);
+  local_state.GetDictionary(prefs::kUmaCellDataUse)->GetInteger(kTodayStr, &uma_pref_value);
+  EXPECT_EQ(50, uma_pref_value);
+
+  local_state.GetDictionary(prefs::kUserCellDataUse)->GetInteger(kYesterdayStr, &user_pref_value);
+  EXPECT_EQ(2*100, user_pref_value);
+  local_state.GetDictionary(prefs::kUmaCellDataUse)->GetInteger(kYesterdayStr, &uma_pref_value);
+  EXPECT_EQ(50, uma_pref_value);
+}
+
+TEST(MetricsDataUseMeasurementsTest, CheckTotalDataUse) {
+  TestDataUsePrefService local_state;
+  MetricsDataUseMeasurements::Initialize(&local_state);
+  local_state.ClearDataUsePrefs();
+  MetricsDataUseMeasurements::GetInstance()->SetMeasurementDateForTesting(kTodayStr);
+  SetPrefTestValuesOverRatio(&local_state);
+
+  int user_data_use = MetricsDataUseMeasurements::GetInstance()->TotalDataUse(prefs::kUserCellDataUse);
+  EXPECT_EQ(8*100, user_data_use);
+  int uma_data_use = MetricsDataUseMeasurements::GetInstance()->TotalDataUse(prefs::kUmaCellDataUse);
+  EXPECT_EQ(4*50, uma_data_use);
+}
+
+TEST(MetricsDataUseMeasurementsTest, CheckCanUploadUMALog) {
+  TestDataUsePrefService local_state;
+  MetricsDataUseMeasurements::Initialize(&local_state);
+  MetricsDataUseMeasurements::GetInstance()->SetMeasurementDateForTesting(kTodayStr);
+  local_state.ClearDataUsePrefs();
+  SetPrefTestValuesOverRatio(&local_state);
+
+  bool can_upload = MetricsDataUseMeasurements::GetInstance()->CanUploadUMALog(50);
+  EXPECT_TRUE(can_upload);
+  can_upload = MetricsDataUseMeasurements::GetInstance()->CanUploadUMALog(100);
+  EXPECT_TRUE(can_upload);
+  can_upload = MetricsDataUseMeasurements::GetInstance()->CanUploadUMALog(150);
+  EXPECT_FALSE(can_upload);
+
+  local_state.ClearDataUsePrefs();
+  SetPrefTestValuesValidRatio(&local_state);
+  can_upload = MetricsDataUseMeasurements::GetInstance()->CanUploadUMALog(100);
+  EXPECT_TRUE(can_upload);
+  can_upload = MetricsDataUseMeasurements::GetInstance()->CanUploadUMALog(200); // this is about 0.49%
+  EXPECT_TRUE(can_upload);
+  can_upload = MetricsDataUseMeasurements::GetInstance()->CanUploadUMALog(300);
+  EXPECT_FALSE(can_upload);
+}
+
+}// namespace metrics