OLD | NEW |
| (Empty) |
1 // Copyright (c) 2014 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #include "net/base/network_activity_monitor.h" | |
6 | |
7 #include <vector> | |
8 | |
9 #include "base/bind.h" | |
10 #include "base/port.h" | |
11 #include "base/stl_util.h" | |
12 #include "base/strings/string_number_conversions.h" | |
13 #include "base/synchronization/lock.h" | |
14 #include "base/threading/thread.h" | |
15 #include "testing/gtest/include/gtest/gtest.h" | |
16 | |
17 namespace net { | |
18 | |
19 namespace test { | |
20 | |
21 class NetworkActivityMonitorPeer { | |
22 public: | |
23 static void ResetMonitor() { | |
24 NetworkActivityMonitor* monitor = NetworkActivityMonitor::GetInstance(); | |
25 base::AutoLock lock(monitor->lock_); | |
26 monitor->bytes_sent_ = 0; | |
27 monitor->bytes_received_ = 0; | |
28 monitor->last_received_ticks_ = base::TimeTicks(); | |
29 monitor->last_sent_ticks_ = base::TimeTicks(); | |
30 } | |
31 }; | |
32 | |
33 | |
34 class NetworkActivityMontiorTest : public testing::Test { | |
35 public: | |
36 NetworkActivityMontiorTest() { | |
37 NetworkActivityMonitorPeer::ResetMonitor(); | |
38 } | |
39 }; | |
40 | |
41 TEST_F(NetworkActivityMontiorTest, GetInstance) { | |
42 NetworkActivityMonitor* monitor = NetworkActivityMonitor::GetInstance(); | |
43 EXPECT_TRUE(monitor != NULL); | |
44 EXPECT_TRUE(monitor == NetworkActivityMonitor::GetInstance()); | |
45 } | |
46 | |
47 TEST_F(NetworkActivityMontiorTest, BytesReceived) { | |
48 NetworkActivityMonitor* monitor = NetworkActivityMonitor::GetInstance(); | |
49 | |
50 EXPECT_EQ(0u, monitor->GetBytesReceived()); | |
51 | |
52 base::TimeTicks start = base::TimeTicks::Now(); | |
53 uint64_t bytes = 12345; | |
54 monitor->IncrementBytesReceived(bytes); | |
55 EXPECT_EQ(bytes, monitor->GetBytesReceived()); | |
56 base::TimeDelta delta = monitor->GetTimeSinceLastReceived(); | |
57 EXPECT_LE(base::TimeDelta(), delta); | |
58 EXPECT_GE(base::TimeTicks::Now() - start, delta); | |
59 } | |
60 | |
61 TEST_F(NetworkActivityMontiorTest, BytesSent) { | |
62 NetworkActivityMonitor* monitor = NetworkActivityMonitor::GetInstance(); | |
63 | |
64 EXPECT_EQ(0u, monitor->GetBytesSent()); | |
65 | |
66 base::TimeTicks start = base::TimeTicks::Now(); | |
67 uint64_t bytes = 12345; | |
68 monitor->IncrementBytesSent(bytes); | |
69 EXPECT_EQ(bytes, monitor->GetBytesSent()); | |
70 base::TimeDelta delta = monitor->GetTimeSinceLastSent(); | |
71 EXPECT_LE(base::TimeDelta(), delta); | |
72 EXPECT_GE(base::TimeTicks::Now() - start, delta); | |
73 } | |
74 | |
75 namespace { | |
76 | |
77 void VerifyBytesReceivedIsMultipleOf(uint64_t bytes) { | |
78 EXPECT_EQ(0u, | |
79 NetworkActivityMonitor::GetInstance()->GetBytesReceived() % bytes); | |
80 } | |
81 | |
82 void VerifyBytesSentIsMultipleOf(uint64_t bytes) { | |
83 EXPECT_EQ(0u, NetworkActivityMonitor::GetInstance()->GetBytesSent() % bytes); | |
84 } | |
85 | |
86 void IncrementBytesReceived(uint64_t bytes) { | |
87 NetworkActivityMonitor::GetInstance()->IncrementBytesReceived(bytes); | |
88 } | |
89 | |
90 void IncrementBytesSent(uint64_t bytes) { | |
91 NetworkActivityMonitor::GetInstance()->IncrementBytesSent(bytes); | |
92 } | |
93 | |
94 } // namespace | |
95 | |
96 TEST_F(NetworkActivityMontiorTest, Threading) { | |
97 std::vector<base::Thread*> threads; | |
98 for (size_t i = 0; i < 3; ++i) { | |
99 threads.push_back(new base::Thread(base::UintToString(i))); | |
100 ASSERT_TRUE(threads.back()->Start()); | |
101 } | |
102 | |
103 size_t num_increments = 157; | |
104 uint64_t bytes_received = GG_UINT64_C(7294954321); | |
105 uint64_t bytes_sent = GG_UINT64_C(91294998765); | |
106 for (size_t i = 0; i < num_increments; ++i) { | |
107 size_t thread_num = i % threads.size(); | |
108 threads[thread_num]->task_runner()->PostTask( | |
109 FROM_HERE, | |
110 base::Bind(&IncrementBytesReceived, bytes_received)); | |
111 threads[thread_num]->task_runner()->PostTask( | |
112 FROM_HERE, | |
113 base::Bind(&IncrementBytesSent, bytes_sent)); | |
114 threads[thread_num]->task_runner()->PostTask( | |
115 FROM_HERE, | |
116 base::Bind(&VerifyBytesSentIsMultipleOf, bytes_sent)); | |
117 threads[thread_num]->task_runner()->PostTask( | |
118 FROM_HERE, | |
119 base::Bind(&VerifyBytesReceivedIsMultipleOf, bytes_received)); | |
120 } | |
121 | |
122 STLDeleteElements(&threads); | |
123 | |
124 NetworkActivityMonitor* monitor = NetworkActivityMonitor::GetInstance(); | |
125 EXPECT_EQ(num_increments * bytes_received, monitor->GetBytesReceived()); | |
126 EXPECT_EQ(num_increments * bytes_sent, monitor->GetBytesSent()); | |
127 } | |
128 | |
129 } // namespace test | |
130 | |
131 } // namespace net | |
OLD | NEW |