Amortize data usage using TrafficStats on Android.

components/data_usage/android/traffic_stats_amortizer_unittest.cc

+// Copyright 2015 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_usage/android/traffic_stats_amortizer.h"
+
+#include <stdint.h>
+
+#include <vector>
+
+#include "components/data_usage/core/data_use.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "url/gurl.h"
+
+namespace data_usage {
+namespace android {
+
+namespace {
+
+class TestTrafficStatsAmortizer : public TrafficStatsAmortizer {
+ public:
+  TestTrafficStatsAmortizer()
+      : next_is_tx_bytes_available_(false),
+        next_tx_bytes_(0),
+        next_is_rx_bytes_available_(false),
+        next_rx_bytes_(0) {}
+
+  ~TestTrafficStatsAmortizer() override {}
+
+  void SetNextTxBytesResult(bool available, int64_t tx_bytes) {
+    next_is_tx_bytes_available_ = available;
+    next_tx_bytes_ = tx_bytes;
+  }
+
+  void SetNextRxBytesResult(bool available, int64_t rx_bytes) {
+    next_is_rx_bytes_available_ = available;
+    next_rx_bytes_ = rx_bytes;
+  }
+
+ protected:
+  bool GetTrafficStatsTxBytes(int64_t* tx_bytes) const override {
+    *tx_bytes = next_tx_bytes_;
+    return next_is_tx_bytes_available_;
+  }
+
+  bool GetTrafficStatsRxBytes(int64_t* rx_bytes) const override {
+    *rx_bytes = next_rx_bytes_;
+    return next_is_rx_bytes_available_;
+  }
+
+ private:
+  bool next_is_tx_bytes_available_;
+  int64_t next_tx_bytes_;
+  bool next_is_rx_bytes_available_;
+  int64_t next_rx_bytes_;
+};
+
+TEST(TrafficStatsAmortizerTest, TrafficStatsUnavailable) {
+  // TODO in this CL: Write tests.
+}
+
+}  // namespace
+
+}  // namespace android
+}  // namespace data_usage