net/base/network_activity_monitor_unittest.cc

// Copyright (c) 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 "net/base/network_activity_monitor.h"

#include <stdint.h>

#include <memory>
#include <vector>

#include "base/bind.h"
#include "base/strings/string_number_conversions.h"
#include "base/threading/thread.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace net {

namespace test {

class NetworkActivityMontiorTest : public testing::Test {
 public:
  NetworkActivityMontiorTest() {
    activity_monitor::ResetBytesReceivedForTesting();
  }
};

TEST_F(NetworkActivityMontiorTest, BytesReceived) {
  EXPECT_EQ(0u, activity_monitor::GetBytesReceived());

  uint64_t bytes = 12345;
  activity_monitor::IncrementBytesReceived(bytes);
  EXPECT_EQ(bytes, activity_monitor::GetBytesReceived());
}

namespace {

void VerifyBytesReceivedIsMultipleOf(uint64_t bytes) {
  EXPECT_EQ(0u, activity_monitor::GetBytesReceived() % bytes);
}

}  // namespace

TEST_F(NetworkActivityMontiorTest, Threading) {
  std::vector<std::unique_ptr<base::Thread>> threads;
  for (size_t i = 0; i < 3; ++i) {
    threads.push_back(std::make_unique<base::Thread>(base::NumberToString(i)));
    ASSERT_TRUE(threads.back()->Start());
  }

  size_t num_increments = 157;
  uint64_t bytes_received = UINT64_C(7294954321);
  for (size_t i = 0; i < num_increments; ++i) {
    size_t thread_num = i % threads.size();
    threads[thread_num]->task_runner()->PostTask(
        FROM_HERE, base::BindOnce(&activity_monitor::IncrementBytesReceived,
                                  bytes_received));
    threads[thread_num]->task_runner()->PostTask(
        FROM_HERE,
        base::BindOnce(&VerifyBytesReceivedIsMultipleOf, bytes_received));
  }

  threads.clear();

  EXPECT_EQ(num_increments * bytes_received,
            activity_monitor::GetBytesReceived());
}

}  // namespace test

}  // namespace net