channel_socket_adapter_unittest.cc revision 5f1c94371a64b3196d4be9466099bb892df9b88e 
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
16using net::IOBuffer;
17
18using testing::_;
19using testing::Return;
20
21namespace jingle_glue {
22
23namespace {
24const int kBufferSize = 4096;
25const char kTestData[] = "data";
26const int kTestDataSize = 4;
27const int kTestError = -32123;
28}  // namespace
29
30class 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_METHOD4(SendPacket, int(const char* data,
38                               size_t len,
39                               const rtc::PacketOptions& options,
40                               int flags));
41  MOCK_METHOD2(SetOption, int(rtc::Socket::Option opt, int value));
42  MOCK_METHOD0(GetError, int());
43  MOCK_CONST_METHOD0(GetIceRole, cricket::IceRole());
44  MOCK_METHOD1(GetStats, bool(cricket::ConnectionInfos* infos));
45  MOCK_CONST_METHOD0(IsDtlsActive, bool());
46  MOCK_CONST_METHOD1(GetSslRole, bool(rtc::SSLRole* role));
47  MOCK_METHOD1(SetSrtpCiphers, bool(const std::vector<std::string>& ciphers));
48  MOCK_METHOD1(GetSrtpCipher, bool(std::string* cipher));
49  MOCK_CONST_METHOD1(GetLocalIdentity, bool(rtc::SSLIdentity** identity));
50  MOCK_CONST_METHOD1(GetRemoteCertificate,
51                     bool(rtc::SSLCertificate** cert));
52  MOCK_METHOD6(ExportKeyingMaterial, bool(const std::string& label,
53                                          const uint8* context,
54                                          size_t context_len,
55                                          bool use_context,
56                                          uint8* result,
57                                          size_t result_len));
58};
59
60class TransportChannelSocketAdapterTest : public testing::Test {
61 public:
62  TransportChannelSocketAdapterTest()
63      : callback_(base::Bind(&TransportChannelSocketAdapterTest::Callback,
64                             base::Unretained(this))),
65        callback_result_(0) {
66  }
67
68 protected:
69  virtual void SetUp() {
70    target_.reset(new TransportChannelSocketAdapter(&channel_));
71  }
72
73  void Callback(int result) {
74    callback_result_ = result;
75  }
76
77  MockTransportChannel channel_;
78  scoped_ptr<TransportChannelSocketAdapter> target_;
79  net::CompletionCallback callback_;
80  int callback_result_;
81  base::MessageLoopForIO message_loop_;
82};
83
84// Verify that Read() returns net::ERR_IO_PENDING.
85TEST_F(TransportChannelSocketAdapterTest, Read) {
86  scoped_refptr<IOBuffer> buffer(new IOBuffer(kBufferSize));
87
88  int result = target_->Read(buffer.get(), kBufferSize, callback_);
89  ASSERT_EQ(net::ERR_IO_PENDING, result);
90
91  channel_.SignalReadPacket(&channel_, kTestData, kTestDataSize,
92                            rtc::CreatePacketTime(0), 0);
93  EXPECT_EQ(kTestDataSize, callback_result_);
94}
95
96// Verify that Read() after Close() returns error.
97TEST_F(TransportChannelSocketAdapterTest, ReadClose) {
98  scoped_refptr<IOBuffer> buffer(new IOBuffer(kBufferSize));
99
100  int result = target_->Read(buffer.get(), kBufferSize, callback_);
101  ASSERT_EQ(net::ERR_IO_PENDING, result);
102
103  target_->Close(kTestError);
104  EXPECT_EQ(kTestError, callback_result_);
105
106  // All Read() calls after Close() should return the error.
107  EXPECT_EQ(kTestError, target_->Read(buffer.get(), kBufferSize, callback_));
108}
109
110// Verify that Write sends the packet and returns correct result.
111TEST_F(TransportChannelSocketAdapterTest, Write) {
112  scoped_refptr<IOBuffer> buffer(new IOBuffer(kTestDataSize));
113
114  EXPECT_CALL(channel_, SendPacket(buffer->data(), kTestDataSize, _, 0))
115      .WillOnce(Return(kTestDataSize));
116
117  int result = target_->Write(buffer.get(), kTestDataSize, callback_);
118  EXPECT_EQ(kTestDataSize, result);
119}
120
121// Verify that the message is still sent if Write() is called while
122// socket is not open yet. The result is the packet is lost.
123TEST_F(TransportChannelSocketAdapterTest, WritePending) {
124  scoped_refptr<IOBuffer> buffer(new IOBuffer(kTestDataSize));
125
126  EXPECT_CALL(channel_, SendPacket(buffer->data(), kTestDataSize, _, 0))
127      .Times(1)
128      .WillOnce(Return(SOCKET_ERROR));
129
130  EXPECT_CALL(channel_, GetError())
131      .WillOnce(Return(EWOULDBLOCK));
132
133  int result = target_->Write(buffer.get(), kTestDataSize, callback_);
134  ASSERT_EQ(net::OK, result);
135}
136
137}  // namespace jingle_glue
138