OLD | NEW |
1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "net/android/traffic_stats.h" | 5 #include "net/android/traffic_stats.h" |
6 | 6 |
7 #include "base/run_loop.h" | 7 #include "base/run_loop.h" |
8 #include "net/test/embedded_test_server/embedded_test_server.h" | 8 #include "net/test/embedded_test_server/embedded_test_server.h" |
9 #include "net/url_request/url_request_test_util.h" | 9 #include "net/url_request/url_request_test_util.h" |
10 #include "testing/gtest/include/gtest/gtest.h" | 10 #include "testing/gtest/include/gtest/gtest.h" |
11 #include "url/gurl.h" | 11 #include "url/gurl.h" |
12 | 12 |
13 namespace net { | 13 namespace net { |
14 | 14 |
15 namespace { | 15 namespace { |
16 | 16 |
17 TEST(TrafficStatsAndroidTest, BasicsTest) { | 17 TEST(TrafficStatsAndroidTest, BasicsTest) { |
18 test_server::EmbeddedTestServer embedded_test_server; | 18 EmbeddedTestServer embedded_test_server; |
19 embedded_test_server.ServeFilesFromDirectory( | 19 embedded_test_server.ServeFilesFromDirectory( |
20 base::FilePath(FILE_PATH_LITERAL("net/data/url_request_unittest"))); | 20 base::FilePath(FILE_PATH_LITERAL("net/data/url_request_unittest"))); |
21 ASSERT_TRUE(embedded_test_server.InitializeAndWaitUntilReady()); | 21 ASSERT_TRUE(embedded_test_server.Start()); |
22 | 22 |
23 int64_t tx_bytes_before_request = -1; | 23 int64_t tx_bytes_before_request = -1; |
24 int64_t rx_bytes_before_request = -1; | 24 int64_t rx_bytes_before_request = -1; |
25 EXPECT_TRUE( | 25 EXPECT_TRUE( |
26 android::traffic_stats::GetTotalTxBytes(&tx_bytes_before_request)); | 26 android::traffic_stats::GetTotalTxBytes(&tx_bytes_before_request)); |
27 EXPECT_GE(tx_bytes_before_request, 0); | 27 EXPECT_GE(tx_bytes_before_request, 0); |
28 EXPECT_TRUE( | 28 EXPECT_TRUE( |
29 android::traffic_stats::GetTotalRxBytes(&rx_bytes_before_request)); | 29 android::traffic_stats::GetTotalRxBytes(&rx_bytes_before_request)); |
30 EXPECT_GE(rx_bytes_before_request, 0); | 30 EXPECT_GE(rx_bytes_before_request, 0); |
31 | 31 |
32 TestDelegate test_delegate; | 32 TestDelegate test_delegate; |
33 TestURLRequestContext context(false); | 33 TestURLRequestContext context(false); |
34 | 34 |
35 scoped_ptr<URLRequest> request( | 35 scoped_ptr<URLRequest> request( |
36 context.CreateRequest(embedded_test_server.GetURL("/echo.html"), | 36 context.CreateRequest(embedded_test_server.GetURL("/echo.html"), |
37 DEFAULT_PRIORITY, &test_delegate)); | 37 DEFAULT_PRIORITY, &test_delegate)); |
38 request->Start(); | 38 request->Start(); |
39 base::RunLoop().Run(); | 39 base::RunLoop().Run(); |
40 | 40 |
41 // Bytes should increase because of the network traffic. | 41 // Bytes should increase because of the network traffic. |
42 int64_t tx_bytes_after_request = -1; | 42 int64_t tx_bytes_after_request = -1; |
43 int64_t rx_bytes_after_request = -1; | 43 int64_t rx_bytes_after_request = -1; |
44 EXPECT_TRUE(android::traffic_stats::GetTotalTxBytes(&tx_bytes_after_request)); | 44 EXPECT_TRUE(android::traffic_stats::GetTotalTxBytes(&tx_bytes_after_request)); |
45 DCHECK_GT(tx_bytes_after_request, tx_bytes_before_request); | 45 DCHECK_GT(tx_bytes_after_request, tx_bytes_before_request); |
46 EXPECT_TRUE(android::traffic_stats::GetTotalRxBytes(&rx_bytes_after_request)); | 46 EXPECT_TRUE(android::traffic_stats::GetTotalRxBytes(&rx_bytes_after_request)); |
47 DCHECK_GT(rx_bytes_after_request, rx_bytes_before_request); | 47 DCHECK_GT(rx_bytes_after_request, rx_bytes_before_request); |
48 } | 48 } |
49 | 49 |
50 TEST(TrafficStatsAndroidTest, UIDBasicsTest) { | 50 TEST(TrafficStatsAndroidTest, UIDBasicsTest) { |
51 test_server::EmbeddedTestServer embedded_test_server; | 51 EmbeddedTestServer embedded_test_server; |
52 embedded_test_server.ServeFilesFromDirectory( | 52 embedded_test_server.ServeFilesFromDirectory( |
53 base::FilePath(FILE_PATH_LITERAL("net/data/url_request_unittest"))); | 53 base::FilePath(FILE_PATH_LITERAL("net/data/url_request_unittest"))); |
54 ASSERT_TRUE(embedded_test_server.InitializeAndWaitUntilReady()); | 54 ASSERT_TRUE(embedded_test_server.Start()); |
55 | 55 |
56 int64_t tx_bytes_before_request = -1; | 56 int64_t tx_bytes_before_request = -1; |
57 int64_t rx_bytes_before_request = -1; | 57 int64_t rx_bytes_before_request = -1; |
58 EXPECT_TRUE( | 58 EXPECT_TRUE( |
59 android::traffic_stats::GetCurrentUidTxBytes(&tx_bytes_before_request)); | 59 android::traffic_stats::GetCurrentUidTxBytes(&tx_bytes_before_request)); |
60 EXPECT_GE(tx_bytes_before_request, 0); | 60 EXPECT_GE(tx_bytes_before_request, 0); |
61 EXPECT_TRUE( | 61 EXPECT_TRUE( |
62 android::traffic_stats::GetCurrentUidRxBytes(&rx_bytes_before_request)); | 62 android::traffic_stats::GetCurrentUidRxBytes(&rx_bytes_before_request)); |
63 EXPECT_GE(rx_bytes_before_request, 0); | 63 EXPECT_GE(rx_bytes_before_request, 0); |
64 | 64 |
(...skipping 13 matching lines...) Expand all Loading... |
78 android::traffic_stats::GetCurrentUidTxBytes(&tx_bytes_after_request)); | 78 android::traffic_stats::GetCurrentUidTxBytes(&tx_bytes_after_request)); |
79 DCHECK_GT(tx_bytes_after_request, tx_bytes_before_request); | 79 DCHECK_GT(tx_bytes_after_request, tx_bytes_before_request); |
80 EXPECT_TRUE( | 80 EXPECT_TRUE( |
81 android::traffic_stats::GetCurrentUidRxBytes(&rx_bytes_after_request)); | 81 android::traffic_stats::GetCurrentUidRxBytes(&rx_bytes_after_request)); |
82 DCHECK_GT(rx_bytes_after_request, rx_bytes_before_request); | 82 DCHECK_GT(rx_bytes_after_request, rx_bytes_before_request); |
83 } | 83 } |
84 | 84 |
85 } // namespace | 85 } // namespace |
86 | 86 |
87 } // namespace net | 87 } // namespace net |
OLD | NEW |