// Copyright 2016 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "components/metrics/data_use_tracker.h"

#include "base/time/time.h"
#include "base/values.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() { DataUseTracker::RegisterPrefs(registry()); }

  TestDataUsePrefService(const TestDataUsePrefService&) = delete;
  TestDataUsePrefService& operator=(const TestDataUsePrefService&) = delete;

  void ClearDataUsePrefs() {
    ClearPref(metrics::prefs::kUserCellDataUse);
    ClearPref(metrics::prefs::kUmaCellDataUse);
  }
};

class FakeDataUseTracker : public DataUseTracker {
 public:
  FakeDataUseTracker(PrefService* local_state) : DataUseTracker(local_state) {}

  FakeDataUseTracker(const FakeDataUseTracker&) = delete;
  FakeDataUseTracker& operator=(const FakeDataUseTracker&) = delete;

  base::Time GetCurrentMeasurementDate() const override {
    base::Time today_for_test;
    EXPECT_TRUE(base::Time::FromUTCString(kTodayStr, &today_for_test));
    return today_for_test;
  }

  std::string GetCurrentMeasurementDateAsString() const override {
    return kTodayStr;
  }
};

// Sets up data usage prefs with mock values so that UMA traffic is above the
// allowed ratio.
void SetPrefTestValuesOverRatio(PrefService* local_state) {
  base::Value::Dict user_pref_dict;
  user_pref_dict.Set(kTodayStr, 2 * 100 * 1024);
  user_pref_dict.Set(kYesterdayStr, 2 * 100 * 1024);
  user_pref_dict.Set(kExpiredDateStr1, 2 * 100 * 1024);
  user_pref_dict.Set(kExpiredDateStr2, 2 * 100 * 1024);
  local_state->SetDict(prefs::kUserCellDataUse, std::move(user_pref_dict));

  base::Value::Dict uma_pref_dict;
  uma_pref_dict.Set(kTodayStr, 50 * 1024);
  uma_pref_dict.Set(kYesterdayStr, 50 * 1024);
  uma_pref_dict.Set(kExpiredDateStr1, 50 * 1024);
  uma_pref_dict.Set(kExpiredDateStr2, 50 * 1024);
  local_state->SetDict(prefs::kUmaCellDataUse, std::move(uma_pref_dict));
}

// Sets up data usage prefs with mock values which can be valid.
void SetPrefTestValuesValidRatio(PrefService* local_state) {
  base::Value::Dict user_pref_dict;
  user_pref_dict.Set(kTodayStr, 100 * 100 * 1024);
  user_pref_dict.Set(kYesterdayStr, 100 * 100 * 1024);
  user_pref_dict.Set(kExpiredDateStr1, 100 * 100 * 1024);
  user_pref_dict.Set(kExpiredDateStr2, 100 * 100 * 1024);
  local_state->SetDict(prefs::kUserCellDataUse, std::move(user_pref_dict));

  // Should be 4% of user traffic
  base::Value::Dict uma_pref_dict;
  uma_pref_dict.Set(kTodayStr, 4 * 100 * 1024);
  uma_pref_dict.Set(kYesterdayStr, 4 * 100 * 1024);
  uma_pref_dict.Set(kExpiredDateStr1, 4 * 100 * 1024);
  uma_pref_dict.Set(kExpiredDateStr2, 4 * 100 * 1024);
  local_state->SetDict(prefs::kUmaCellDataUse, std::move(uma_pref_dict));
}

}  // namespace

TEST(DataUseTrackerTest, CheckUpdateUsagePref) {
  TestDataUsePrefService local_state;
  FakeDataUseTracker data_use_tracker(&local_state);
  local_state.ClearDataUsePrefs();

  data_use_tracker.UpdateMetricsUsagePrefsInternal(2 * 100 * 1024, true, false);
  EXPECT_EQ(2 * 100 * 1024,
            local_state.GetDict(prefs::kUserCellDataUse).FindInt(kTodayStr));
  EXPECT_FALSE(local_state.GetDict(prefs::kUmaCellDataUse).FindInt(kTodayStr));

  data_use_tracker.UpdateMetricsUsagePrefsInternal(100 * 1024, true, true);
  EXPECT_EQ(3 * 100 * 1024,
            local_state.GetDict(prefs::kUserCellDataUse).FindInt(kTodayStr));
  EXPECT_EQ(100 * 1024,
            local_state.GetDict(prefs::kUmaCellDataUse).FindInt(kTodayStr));
}

TEST(DataUseTrackerTest, CheckRemoveExpiredEntries) {
  TestDataUsePrefService local_state;
  FakeDataUseTracker data_use_tracker(&local_state);
  local_state.ClearDataUsePrefs();
  SetPrefTestValuesOverRatio(&local_state);
  data_use_tracker.RemoveExpiredEntries();

  EXPECT_FALSE(
      local_state.GetDict(prefs::kUserCellDataUse).FindInt(kExpiredDateStr1));
  EXPECT_FALSE(
      local_state.GetDict(prefs::kUmaCellDataUse).FindInt(kExpiredDateStr1));

  EXPECT_FALSE(
      local_state.GetDict(prefs::kUserCellDataUse).FindInt(kExpiredDateStr2));
  EXPECT_FALSE(
      local_state.GetDict(prefs::kUmaCellDataUse).FindInt(kExpiredDateStr2));

  EXPECT_EQ(2 * 100 * 1024,
            local_state.GetDict(prefs::kUserCellDataUse).FindInt(kTodayStr));
  EXPECT_EQ(50 * 1024,
            local_state.GetDict(prefs::kUmaCellDataUse).FindInt(kTodayStr));

  EXPECT_EQ(
      2 * 100 * 1024,
      local_state.GetDict(prefs::kUserCellDataUse).FindInt(kYesterdayStr));
  EXPECT_EQ(50 * 1024,
            local_state.GetDict(prefs::kUmaCellDataUse).FindInt(kYesterdayStr));
}

TEST(DataUseTrackerTest, CheckComputeTotalDataUse) {
  TestDataUsePrefService local_state;
  FakeDataUseTracker data_use_tracker(&local_state);
  local_state.ClearDataUsePrefs();
  SetPrefTestValuesOverRatio(&local_state);

  int user_data_use =
      data_use_tracker.ComputeTotalDataUse(prefs::kUserCellDataUse);
  EXPECT_EQ(8 * 100 * 1024, user_data_use);
  int uma_data_use =
      data_use_tracker.ComputeTotalDataUse(prefs::kUmaCellDataUse);
  EXPECT_EQ(4 * 50 * 1024, uma_data_use);
}

TEST(DataUseTrackerTest, CheckShouldUploadLogOnCellular) {
  TestDataUsePrefService local_state;
  FakeDataUseTracker data_use_tracker(&local_state);
  local_state.ClearDataUsePrefs();
  SetPrefTestValuesOverRatio(&local_state);

  bool can_upload = data_use_tracker.ShouldUploadLogOnCellular(50 * 1024);
  EXPECT_TRUE(can_upload);
  can_upload = data_use_tracker.ShouldUploadLogOnCellular(100 * 1024);
  EXPECT_TRUE(can_upload);
  can_upload = data_use_tracker.ShouldUploadLogOnCellular(150 * 1024);
  EXPECT_FALSE(can_upload);

  local_state.ClearDataUsePrefs();
  SetPrefTestValuesValidRatio(&local_state);
  can_upload = data_use_tracker.ShouldUploadLogOnCellular(100 * 1024);
  EXPECT_TRUE(can_upload);
  // this is about 0.49%
  can_upload = data_use_tracker.ShouldUploadLogOnCellular(200 * 1024);
  EXPECT_TRUE(can_upload);
  can_upload = data_use_tracker.ShouldUploadLogOnCellular(300 * 1024);
  EXPECT_FALSE(can_upload);
}

}  // namespace metrics