Implement ChromiumSocketFactory.

remoting/jingle_glue/chromium_socket_factory_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 "remoting/jingle_glue/chromium_socket_factory.h"
+
+#include "base/message_loop.h"
+#include "base/run_loop.h"
+#include "testing/gmock/include/gmock/gmock.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "third_party/libjingle/source/talk/base/asyncpacketsocket.h"
+#include "third_party/libjingle/source/talk/base/socketaddress.h"
+
+namespace remoting {
+
+class ChromiumSocketFactoryTest : public testing::Test,
+                                  public sigslot::has_slots<> {
+ public:
+  virtual void SetUp() OVERRIDE {
+    socket_factory_.reset(new ChromiumPacketSocketFactory());
+
+    socket_.reset(socket_factory_->CreateUdpSocket(
+        talk_base::SocketAddress("127.0.0.1", 0), 0, 0));
+    ASSERT_TRUE(socket_.get() != NULL);
+    EXPECT_EQ(socket_->GetState(), talk_base::AsyncPacketSocket::STATE_BOUND);
+    socket_->SignalReadPacket.connect(
+        this, &ChromiumSocketFactoryTest::OnPacket);
+  }
+
+  void OnPacket(talk_base::AsyncPacketSocket* socket,
+                const char* data, size_t size,
+                const talk_base::SocketAddress& address) {
+    EXPECT_EQ(socket, socket_.get());
+    last_packet_.assign(data, data + size);
+    last_address_ = address;
+    run_loop_.Quit();
+  }
+
+ protected:
+  MessageLoopForIO message_loop_;
+  base::RunLoop run_loop_;
+
+  scoped_ptr<talk_base::PacketSocketFactory> socket_factory_;
+  scoped_ptr<talk_base::AsyncPacketSocket> socket_;
+
+  std::string last_packet_;
+  talk_base::SocketAddress last_address_;
+};
+
+TEST_F(ChromiumSocketFactoryTest, SendAndReceive) {
+  // UDP packets may be lost, so we have to retry sending it more than once.
+  const int kMaxAttempts = 3;
+  const base::TimeDelta kAttemptPeriod = base::TimeDelta::FromSeconds(1);
+
+  scoped_ptr<talk_base::AsyncPacketSocket> sending_socket;
+  talk_base::SocketAddress address;
+
+  sending_socket.reset(socket_factory_->CreateUdpSocket(
+      talk_base::SocketAddress("127.0.0.1", 0), 0, 0));
+  ASSERT_TRUE(sending_socket.get() != NULL);
+  EXPECT_EQ(sending_socket->GetState(),
+            talk_base::AsyncPacketSocket::STATE_BOUND);
+  address = sending_socket->GetLocalAddress();
+
+  std::string test_packet("TEST PACKET");
+  int attempts = 0;
+  while (last_packet_.empty() && attempts++ < kMaxAttempts) {
+    sending_socket->SendTo(test_packet.data(), test_packet.size(),
+                           socket_->GetLocalAddress());
+    message_loop_.PostDelayedTask(FROM_HERE, run_loop_.QuitClosure(),
+                                  kAttemptPeriod);
+    run_loop_.Run();
+  }
+  EXPECT_EQ(test_packet, last_packet_);
+  EXPECT_EQ(address, last_address_);
+}
+
+TEST_F(ChromiumSocketFactoryTest, SetOptions) {
+  EXPECT_EQ(0, socket_->SetOption(talk_base::Socket::OPT_SNDBUF, 4096));
+  EXPECT_EQ(0, socket_->SetOption(talk_base::Socket::OPT_RCVBUF, 4096));
+}
+
+TEST_F(ChromiumSocketFactoryTest, PortRange) {
+  const int kMinPort = 12400;
+  const int kMaxPort = 12410;
+  socket_.reset(socket_factory_->CreateUdpSocket(
+      talk_base::SocketAddress("127.0.0.1", 0), kMaxPort, kMaxPort));
+  ASSERT_TRUE(socket_.get() != NULL);
+  EXPECT_EQ(socket_->GetState(), talk_base::AsyncPacketSocket::STATE_BOUND);
+  EXPECT_GE(socket_->GetLocalAddress().port(), kMinPort);
+  EXPECT_LE(socket_->GetLocalAddress().port(), kMaxPort);
+}
+
+}  // namespace remoting