jingle/glue/channel_socket_adapter_unittest.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "base/memory/ref_counted.h"
   6 #include "base/memory/scoped_ptr.h"
   7 #include "base/message_loop/message_loop.h"
   8 #include "jingle/glue/channel_socket_adapter.h"
   9 #include "net/base/io_buffer.h"
  10 #include "net/base/net_errors.h"
  11 #include "net/socket/socket.h"
  12 #include "testing/gmock/include/gmock/gmock.h"
  13 #include "testing/gtest/include/gtest/gtest.h"
  14 #include "third_party/libjingle/source/talk/p2p/base/transportchannel.h"
  15
  16 using net::IOBuffer;
  17
  18 using testing::_;
  19 using testing::Return;
  20
  21 namespace jingle_glue {
  22
  23 namespace {
  24 const int kBufferSize = 4096;
  25 const char kTestData[] = "data";
  26 const int kTestDataSize = 4;
  27 const int kTestError = -32123;
  28 }  // namespace
  29
  30 class MockTransportChannel : public cricket::TransportChannel {
  31  public:
  32   MockTransportChannel() : cricket::TransportChannel(std::string(), 0) {
  33     set_writable(true);
  34     set_readable(true);
  35   }
  36
  37   MOCK_METHOD3(SendPacket, int(const char* data, size_t len, int flags));
  38   MOCK_METHOD2(SetOption, int(talk_base::Socket::Option opt, int value));
  39   MOCK_METHOD0(GetError, int());
  40   MOCK_CONST_METHOD0(GetIceRole, cricket::IceRole());
  41   MOCK_METHOD1(GetStats, bool(cricket::ConnectionInfos* infos));
  42   MOCK_CONST_METHOD0(IsDtlsActive, bool());
  43   MOCK_CONST_METHOD1(GetSslRole, bool(talk_base::SSLRole* role));
  44   MOCK_METHOD1(SetSrtpCiphers, bool(const std::vector<std::string>& ciphers));
  45   MOCK_METHOD1(GetSrtpCipher, bool(std::string* cipher));
  46   MOCK_METHOD6(ExportKeyingMaterial, bool(const std::string& label,
  47                                           const uint8* context,
  48                                           size_t context_len,
  49                                           bool use_context,
  50                                           uint8* result,
  51                                           size_t result_len));
  52 };
  53
  54 class TransportChannelSocketAdapterTest : public testing::Test {
  55  public:
  56   TransportChannelSocketAdapterTest()
  57       : callback_(base::Bind(&TransportChannelSocketAdapterTest::Callback,
  58                              base::Unretained(this))),
  59         callback_result_(0) {
  60   }
  61
  62  protected:
  63   virtual void SetUp() {
  64     target_.reset(new TransportChannelSocketAdapter(&channel_));
  65   }
  66
  67   void Callback(int result) {
  68     callback_result_ = result;
  69   }
  70
  71   MockTransportChannel channel_;
  72   scoped_ptr<TransportChannelSocketAdapter> target_;
  73   net::CompletionCallback callback_;
  74   int callback_result_;
  75   base::MessageLoopForIO message_loop_;
  76 };
  77
  78 // Verify that Read() returns net::ERR_IO_PENDING.
  79 TEST_F(TransportChannelSocketAdapterTest, Read) {
  80   scoped_refptr<IOBuffer> buffer(new IOBuffer(kBufferSize));
  81
  82   int result = target_->Read(buffer.get(), kBufferSize, callback_);
  83   ASSERT_EQ(net::ERR_IO_PENDING, result);
  84
  85   channel_.SignalReadPacket(&channel_, kTestData, kTestDataSize, 0);
  86   EXPECT_EQ(kTestDataSize, callback_result_);
  87 }
  88
  89 // Verify that Read() after Close() returns error.
  90 TEST_F(TransportChannelSocketAdapterTest, ReadClose) {
  91   scoped_refptr<IOBuffer> buffer(new IOBuffer(kBufferSize));
  92
  93   int result = target_->Read(buffer.get(), kBufferSize, callback_);
  94   ASSERT_EQ(net::ERR_IO_PENDING, result);
  95
  96   target_->Close(kTestError);
  97   EXPECT_EQ(kTestError, callback_result_);
  98
  99   // All Read() calls after Close() should return the error.
 100   EXPECT_EQ(kTestError, target_->Read(buffer.get(), kBufferSize, callback_));
 101 }
 102
 103 // Verify that Write sends the packet and returns correct result.
 104 TEST_F(TransportChannelSocketAdapterTest, Write) {
 105   scoped_refptr<IOBuffer> buffer(new IOBuffer(kTestDataSize));
 106
 107   EXPECT_CALL(channel_, SendPacket(buffer->data(), kTestDataSize, 0))
 108       .WillOnce(Return(kTestDataSize));
 109
 110   int result = target_->Write(buffer.get(), kTestDataSize, callback_);
 111   EXPECT_EQ(kTestDataSize, result);
 112 }
 113
 114 // Verify that the message is still sent if Write() is called while
 115 // socket is not open yet. The result is the packet is lost.
 116 TEST_F(TransportChannelSocketAdapterTest, WritePending) {
 117   scoped_refptr<IOBuffer> buffer(new IOBuffer(kTestDataSize));
 118
 119   EXPECT_CALL(channel_, SendPacket(buffer->data(), kTestDataSize, 0))
 120       .Times(1)
 121       .WillOnce(Return(SOCKET_ERROR));
 122
 123   EXPECT_CALL(channel_, GetError())
 124       .WillOnce(Return(EWOULDBLOCK));
 125
 126   int result = target_->Write(buffer.get(), kTestDataSize, callback_);
 127   ASSERT_EQ(net::OK, result);
 128 }
 129
 130 }  // namespace jingle_glue