Collect stats to investigate the viability of UDP

chrome/browser/net/network_stats_unittest.cc

Index: chrome/browser/net/network_stats_unittest.cc
===================================================================
--- chrome/browser/net/network_stats_unittest.cc	(revision 0)
+++ chrome/browser/net/network_stats_unittest.cc	(revision 0)
@@ -0,0 +1,93 @@
+// Copyright (c) 2011 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 "base/basictypes.h"
+#include "base/message_loop.h"
+#include "chrome/browser/net/network_stats.h"
+#include "net/base/test_completion_callback.h"
+#include "net/test/test_server.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "testing/platform_test.h"
+
+namespace chrome_browser_net {
+
+class NetworkStatsTest : public PlatformTest {
+ public:
+  NetworkStatsTest() {}
+ protected:
+  virtual void TearDown() {
+    // Flush the message loop to make Purify happy.
+    message_loop_.RunAllPending();
+  }
+  MessageLoopForIO message_loop_;
+};
+
+class NetworkStatsTestUDP : public NetworkStatsTest {
+ public:
+  NetworkStatsTestUDP()
+      : test_server_(net::TestServer::TYPE_UDP_ECHO,
+                     FilePath(FILE_PATH_LITERAL("net/data"))) {
+  }
+
+ protected:
+  void RunUDPEchoTest(int bytes) {
+    TestCompletionCallback cb;
+
+    scoped_refptr<UDPStatsClient> udp_stats_client(new UDPStatsClient());
+    EXPECT_TRUE(udp_stats_client->Start(test_server_.host_port_pair().host(),
+                                        test_server_.host_port_pair().port(),
+                                        bytes,
+                                        &cb));
+    int rv = cb.WaitForResult();
+    EXPECT_EQ(0, rv);
+    EXPECT_EQ(0, udp_stats_client->error_count());
+  }
+
+  net::TestServer test_server_;
+};
+
+class NetworkStatsTestTCP : public NetworkStatsTest {
+ public:
+  NetworkStatsTestTCP()
+      : test_server_(net::TestServer::TYPE_TCP_ECHO,
+                     FilePath(FILE_PATH_LITERAL("net/data"))) {
+  }
+
+ protected:
+  void RunTCPEchoTest(int bytes) {
+    TestCompletionCallback cb;
+
+    scoped_refptr<TCPStatsClient> tcp_stats_client(new TCPStatsClient());
+    EXPECT_TRUE(tcp_stats_client->Start(test_server_.host_port_pair(),
+                                        bytes,
+                                        &cb));
+    int rv = cb.WaitForResult();
+    EXPECT_EQ(0, rv);
+    EXPECT_EQ(0, tcp_stats_client->error_count());
+  }
+
+  net::TestServer test_server_;
+};
+
+TEST_F(NetworkStatsTestUDP, UDPEcho_50B_Of_Data) {
+  ASSERT_TRUE(test_server_.Start());
+  RunUDPEchoTest(50);
+}
+
+TEST_F(NetworkStatsTestUDP, UDPEcho_1K_Of_Data) {
+  ASSERT_TRUE(test_server_.Start());
+  RunUDPEchoTest(1024);
+}
+
+TEST_F(NetworkStatsTestTCP, TCPEcho_50B_Of_Data) {
+  ASSERT_TRUE(test_server_.Start());
+  RunTCPEchoTest(50);
+}
+
+TEST_F(NetworkStatsTestTCP, TCPEcho_1K_Of_Data) {
+  ASSERT_TRUE(test_server_.Start());
+  RunTCPEchoTest(1024);
+}
+
+}  // namespace chrome_browser_net