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..31ed7b8896f2040035c0fa9b2bd6e6ab242803cb 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,57 @@ 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)); |
+} |
+ |
+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 |