Multicast socket API

chrome/browser/extensions/api/socket/udp_socket_unittest.cc

Index: chrome/browser/extensions/api/socket/udp_socket_unittest.cc
diff --git a/chrome/browser/extensions/api/socket/udp_socket_unittest.cc b/chrome/browser/extensions/api/socket/udp_socket_unittest.cc
index c625efecdf27b66d73dbb2bde3d4639d2cacfa3c..050b17ab966b79e3373251be7c94a6e4ae8b0500 100644
--- a/chrome/browser/extensions/api/socket/udp_socket_unittest.cc
+++ b/chrome/browser/extensions/api/socket/udp_socket_unittest.cc
@@ -20,7 +20,7 @@ class UDPSocketUnitTest : public BrowserWithTestWindowTest {
 };
 
 static void OnConnected(int result) {
-  DCHECK(result == 0);
+  EXPECT_EQ(0, result);
 }
 
 static void OnCompleted(int bytes_read,
@@ -30,8 +30,29 @@ static void OnCompleted(int bytes_read,
   // Do nothing; don't care.
 }
 
+static const char test_message[] = "$$TESTMESSAGETESTMESSAGETESTMESSAGETEST$$";
+
+static void OnReadCompleted(int count,
+                            scoped_refptr<net::IOBuffer> io_buffer) {
+  size_t expected_size = arraysize(test_message);
+  EXPECT_EQ(static_cast<int>(expected_size), count);
+  EXPECT_EQ(0, strncmp(io_buffer->data(), test_message, expected_size));
+}
+
+static void OnExpectTimeoutMessage(int count,
+                            scoped_refptr<net::IOBuffer> io_buffer) {
+  size_t expected_size = arraysize(test_message);
+  EXPECT_EQ(static_cast<int>(expected_size), count);
+  EXPECT_EQ(0, strncmp(io_buffer->data(), test_message, expected_size));
+}
+
+static void OnSendCompleted(int result) {
+  size_t expected_size = arraysize(test_message);
+  EXPECT_EQ(static_cast<int>(expected_size), result);
+}
+
 TEST(UDPSocketUnitTest, TestUDPSocketRecvFrom) {
-  MessageLoopForIO io_loop;  // for RecvFrom to do its threaded work.
+  MessageLoopForIO io_loop;  // For RecvFrom to do its threaded work.
   UDPSocket socket("abcdefghijklmnopqrst");
 
   // Confirm that we can call two RecvFroms in quick succession without
@@ -41,4 +62,60 @@ TEST(UDPSocketUnitTest, TestUDPSocketRecvFrom) {
   socket.RecvFrom(4096, base::Bind(&OnCompleted));
 }
 
+TEST(UDPSocketUnitTest, TestUDPMulticastJoinGroup) {
+  const char* kGroup = "237.132.100.17";
+  MessageLoopForIO io_loop;  // For Read to do its threaded work.
+  UDPSocket src("abcdefghijklmnopqrst");
+  UDPSocket dest("abcdefghijklmnopqrst");
+
+  EXPECT_EQ(0, dest.Bind("0.0.0.0", 13333));
+  EXPECT_EQ(0, dest.JoinGroup(kGroup));
+  base::hash_set<std::string> groups;
+  dest.GetJoinedGroups(&groups);
+  EXPECT_EQ(static_cast<size_t>(1), groups.size());
+  EXPECT_EQ(kGroup, *groups.begin());
+  dest.Read(1024, base::Bind(&OnReadCompleted));
+
+  EXPECT_NE(0, dest.LeaveGroup("237.132.100.13"));
+  EXPECT_EQ(0, dest.LeaveGroup(kGroup));
+  dest.GetJoinedGroups(&groups);
+  EXPECT_EQ(static_cast<size_t>(0), groups.size());
+}
+
+TEST(UDPSocketUnitTest, TestUDPMulticastTimeToLive) {
+  const char* kGroup = "237.132.100.17";
+  UDPSocket socket("abcdefghijklmnopqrst");
+  EXPECT_NE(0, socket.SetMulticastTimeToLive(-1));  // Negative TTL shall fail.
+  EXPECT_EQ(0, socket.SetMulticastTimeToLive(3));
+  socket.Connect(kGroup, 13333, base::Bind(&OnConnected));
+  EXPECT_NE(0, socket.SetMulticastTimeToLive(-1));
+  EXPECT_EQ(0, socket.SetMulticastTimeToLive(3));
+}
+
+TEST(UDPSocketUnitTest, TestUDPMulticastLoopbackMode) {
+  const char* kGroup = "237.132.100.17";
+  UDPSocket socket("abcdefghijklmnopqrst");
+  EXPECT_EQ(0, socket.SetMulticastLoopbackMode(false));
+  socket.Connect(kGroup, 13333, base::Bind(&OnConnected));
+  EXPECT_EQ(0, socket.SetMulticastLoopbackMode(false));
+}
+
+TEST(UDPSocketUnitTest, TestUDPMulticastRecv) {
+  const char* kGroup = "237.132.100.17";
+  MessageLoopForIO io_loop;  // For Read to do its threaded work.
+  UDPSocket src("abcdefghijklmnopqrst");
+  UDPSocket dest("abcdefghijklmnopqrst");
+
+  EXPECT_EQ(0, dest.Bind("0.0.0.0", 13333));
+  EXPECT_EQ(0, dest.JoinGroup(kGroup));
+  dest.Read(1024, base::Bind(&OnReadCompleted));
+
+  size_t size = arraysize(test_message);
+  scoped_refptr<net::IOBuffer> data = new net::WrappedIOBuffer(test_message);
+  // Setting multicast setting s should work even before connection.
+  EXPECT_EQ(0, src.SetMulticastTimeToLive(0));
+  src.Connect(kGroup, 13333, base::Bind(&OnConnected));
+  src.SendTo(data, size, kGroup, 13333, base::Bind(&OnSendCompleted));
+}
+
 }  // namespace extensions