1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
|
// Copyright 2014 The Chromium Authors
// 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/functional/bind.h"
#include "base/strings/string_number_conversions.h"
#include "base/threading/thread.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace net::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 net::test
|
