Multicast socket API

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

 // Copyright (c) 2012 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 "chrome/browser/extensions/api/socket/udp_socket.h"
 
 #include <string>
 
 #include "base/memory/scoped_ptr.h"
 #include "base/message_loop.h"
 #include "chrome/test/base/browser_with_test_window_test.h"
 #include "net/base/io_buffer.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace extensions {
 
 // UDPSocketUnitTest exists solely to make it easier to pass a specific
 // gtest_filter argument during development.
 class UDPSocketUnitTest : public BrowserWithTestWindowTest {
 };
 
 static void OnConnected(int result) {
-  DCHECK(result == 0);
+  EXPECT_EQ(0, result);
 }
 
 static void OnCompleted(int bytes_read,
                         scoped_refptr<net::IOBuffer> io_buffer,
                         const std::string& address,
                         int port) {
   // 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
   // triggering crbug.com/146606.
   socket.Connect("127.0.0.1", 40000, base::Bind(&OnConnected));
   socket.RecvFrom(4096, base::Bind(&OnCompleted));
   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));
+}
+
+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));
+}
+
+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