path: root/net/websockets/websocket_basic_stream_test.cc

// Copyright 2013 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.
//
// Tests for WebSocketBasicStream. Note that we do not attempt to verify that
// frame parsing itself functions correctly, as that is covered by the
// WebSocketFrameParser tests.

#include "net/websockets/websocket_basic_stream.h"

#include "base/basictypes.h"
#include "base/port.h"
#include "net/base/capturing_net_log.h"
#include "net/base/test_completion_callback.h"
#include "net/socket/socket_test_util.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace net {
namespace {

// TODO(ricea): Add tests for
// - Empty frames (data & control)
// - Non-NULL masking key
// - A frame larger than kReadBufferSize;

const char kSampleFrame[] = "\x81\x06Sample";
const size_t kSampleFrameSize = arraysize(kSampleFrame) - 1;
const char kPartialLargeFrame[] =
    "\x81\x7F\x00\x00\x00\x00\x7F\xFF\xFF\xFF"
    "chromiunum ad pasco per loca insanis pullum manducat frumenti";
const size_t kPartialLargeFrameSize = arraysize(kPartialLargeFrame) - 1;
const size_t kLargeFrameHeaderSize = 10;
const size_t kLargeFrameDeclaredPayloadSize = 0x7FFFFFFF;
const char kMultipleFrames[] = "\x81\x01X\x81\x01Y\x81\x01Z";
const size_t kMultipleFramesSize = arraysize(kMultipleFrames) - 1;
// This frame encodes a payload length of 7 in two bytes, which is always
// invalid.
const char kInvalidFrame[] = "\x81\x7E\x00\x07Invalid";
const size_t kInvalidFrameSize = arraysize(kInvalidFrame) - 1;
const char kWriteFrame[] = "\x81\x85\x00\x00\x00\x00Write";
const size_t kWriteFrameSize = arraysize(kWriteFrame) - 1;
const WebSocketMaskingKey kNulMaskingKey = {{'\0', '\0', '\0', '\0'}};

// Generates a ScopedVector<WebSocketFrameChunk> which will have a wire format
// matching kWriteFrame.
ScopedVector<WebSocketFrameChunk> GenerateWriteFrame() {
  scoped_ptr<WebSocketFrameChunk> chunk(new WebSocketFrameChunk);
  const size_t payload_size =
      kWriteFrameSize - (WebSocketFrameHeader::kBaseHeaderSize +
                         WebSocketFrameHeader::kMaskingKeyLength);
  chunk->data = new IOBufferWithSize(payload_size);
  memcpy(chunk->data->data(),
         kWriteFrame + kWriteFrameSize - payload_size,
         payload_size);
  chunk->final_chunk = true;
  scoped_ptr<WebSocketFrameHeader> header(
      new WebSocketFrameHeader(WebSocketFrameHeader::kOpCodeText));
  header->final = true;
  header->masked = true;
  header->payload_length = payload_size;
  chunk->header = header.Pass();
  ScopedVector<WebSocketFrameChunk> chunks;
  chunks.push_back(chunk.release());
  return chunks.Pass();
}

// A masking key generator function which generates the identity mask,
// ie. "\0\0\0\0".
WebSocketMaskingKey GenerateNulMaskingKey() { return kNulMaskingKey; }

// Base class for WebSocketBasicStream test fixtures.
class WebSocketBasicStreamTest : public ::testing::Test {
 protected:
  scoped_ptr<WebSocketBasicStream> stream_;
  CapturingNetLog net_log_;
};

// A fixture for tests which only perform normal socket operations.
class WebSocketBasicStreamSocketTest : public WebSocketBasicStreamTest {
 protected:
  WebSocketBasicStreamSocketTest()
      : histograms_("a"), pool_(1, 1, &histograms_, &factory_) {}

  virtual ~WebSocketBasicStreamSocketTest() {
    // stream_ has a reference to socket_data_ (via MockTCPClientSocket) and so
    // should be destroyed first.
    stream_.reset();
  }

  scoped_ptr<ClientSocketHandle> MakeTransportSocket(MockRead reads[],
                                                     size_t reads_count,
                                                     MockWrite writes[],
                                                     size_t writes_count) {
    socket_data_.reset(
        new StaticSocketDataProvider(reads, reads_count, writes, writes_count));
    socket_data_->set_connect_data(MockConnect(SYNCHRONOUS, OK));
    factory_.AddSocketDataProvider(socket_data_.get());

    scoped_ptr<ClientSocketHandle> transport_socket(new ClientSocketHandle);
    scoped_refptr<MockTransportSocketParams> params;
    transport_socket->Init("a",
                           params,
                           MEDIUM,
                           CompletionCallback(),
                           &pool_,
                           bound_net_log_.bound());
    return transport_socket.Pass();
  }

  void SetHttpReadBuffer(const char* data, size_t size) {
    http_read_buffer_ = new GrowableIOBuffer;
    http_read_buffer_->SetCapacity(size);
    memcpy(http_read_buffer_->data(), data, size);
    http_read_buffer_->set_offset(size);
  }

  void CreateStream(MockRead reads[],
                    size_t reads_count,
                    MockWrite writes[],
                    size_t writes_count) {
    stream_ = WebSocketBasicStream::CreateWebSocketBasicStreamForTesting(
        MakeTransportSocket(reads, reads_count, writes, writes_count),
        http_read_buffer_,
        sub_protocol_,
        extensions_,
        &GenerateNulMaskingKey);
  }

  template <size_t N>
  void CreateReadOnly(MockRead (&reads)[N]) {
    CreateStream(reads, N, NULL, 0);
  }

  template <size_t N>
  void CreateWriteOnly(MockWrite (&writes)[N]) {
    CreateStream(NULL, 0, writes, N);
  }

  void CreateNullStream() { CreateStream(NULL, 0, NULL, 0); }

  scoped_ptr<SocketDataProvider> socket_data_;
  MockClientSocketFactory factory_;
  ClientSocketPoolHistograms histograms_;
  MockTransportClientSocketPool pool_;
  CapturingBoundNetLog(bound_net_log_);
  ScopedVector<WebSocketFrameChunk> frame_chunks_;
  TestCompletionCallback cb_;
  scoped_refptr<GrowableIOBuffer> http_read_buffer_;
  std::string sub_protocol_;
  std::string extensions_;
};

TEST_F(WebSocketBasicStreamSocketTest, ConstructionWorks) {
  CreateNullStream();
}

TEST_F(WebSocketBasicStreamSocketTest, SyncReadWorks) {
  MockRead reads[] = {MockRead(SYNCHRONOUS, kSampleFrame, kSampleFrameSize)};
  CreateReadOnly(reads);
  int result = stream_->ReadFrames(&frame_chunks_, cb_.callback());
  EXPECT_EQ(OK, result);
  ASSERT_EQ(1U, frame_chunks_.size());
  ASSERT_TRUE(frame_chunks_[0]->header);
  EXPECT_EQ(GG_UINT64_C(6), frame_chunks_[0]->header->payload_length);
  EXPECT_TRUE(frame_chunks_[0]->header->final);
  EXPECT_TRUE(frame_chunks_[0]->final_chunk);
}

TEST_F(WebSocketBasicStreamSocketTest, AsyncReadWorks) {
  MockRead reads[] = {MockRead(ASYNC, kSampleFrame, kSampleFrameSize)};
  CreateReadOnly(reads);
  int result = stream_->ReadFrames(&frame_chunks_, cb_.callback());
  ASSERT_EQ(ERR_IO_PENDING, result);
  EXPECT_EQ(OK, cb_.WaitForResult());
  ASSERT_EQ(1U, frame_chunks_.size());
  ASSERT_TRUE(frame_chunks_[0]->header);
  EXPECT_EQ(GG_UINT64_C(6), frame_chunks_[0]->header->payload_length);
  // Don't repeat all the tests from SyncReadWorks; just enough to be sure the
  // frame was really read.
}

// ReadFrames will not return a frame whose header has not been wholly received.
TEST_F(WebSocketBasicStreamSocketTest, HeaderFragmentedSync) {
  MockRead reads[] = {
      MockRead(SYNCHRONOUS, kSampleFrame, 1),
      MockRead(SYNCHRONOUS, kSampleFrame + 1, kSampleFrameSize - 1)};
  CreateReadOnly(reads);
  int result = stream_->ReadFrames(&frame_chunks_, cb_.callback());
  ASSERT_EQ(OK, result);
  ASSERT_EQ(1U, frame_chunks_.size());
  ASSERT_TRUE(frame_chunks_[0]->header);
  EXPECT_EQ(GG_UINT64_C(6), frame_chunks_[0]->header->payload_length);
}

// The same behaviour applies to asynchronous reads.
TEST_F(WebSocketBasicStreamSocketTest, HeaderFragmentedAsync) {
  MockRead reads[] = {MockRead(ASYNC, kSampleFrame, 1),
                      MockRead(ASYNC, kSampleFrame + 1, kSampleFrameSize - 1)};
  CreateReadOnly(reads);
  int result = stream_->ReadFrames(&frame_chunks_, cb_.callback());
  ASSERT_EQ(ERR_IO_PENDING, result);
  EXPECT_EQ(OK, cb_.WaitForResult());
  ASSERT_EQ(1U, frame_chunks_.size());
  ASSERT_TRUE(frame_chunks_[0]->header);
  EXPECT_EQ(GG_UINT64_C(6), frame_chunks_[0]->header->payload_length);
}

// If it receives an incomplete header in a synchronous call, then has to wait
// for the rest of the frame, ReadFrames will return ERR_IO_PENDING.
TEST_F(WebSocketBasicStreamSocketTest, HeaderFragmentedSyncAsync) {
  MockRead reads[] = {MockRead(SYNCHRONOUS, kSampleFrame, 1),
                      MockRead(ASYNC, kSampleFrame + 1, kSampleFrameSize - 1)};
  CreateReadOnly(reads);
  int result = stream_->ReadFrames(&frame_chunks_, cb_.callback());
  ASSERT_EQ(ERR_IO_PENDING, result);
  EXPECT_EQ(OK, cb_.WaitForResult());
  ASSERT_EQ(1U, frame_chunks_.size());
  ASSERT_TRUE(frame_chunks_[0]->header);
  EXPECT_EQ(GG_UINT64_C(6), frame_chunks_[0]->header->payload_length);
}

// An extended header should also return ERR_IO_PENDING if it is not completely
// received.
TEST_F(WebSocketBasicStreamSocketTest, FragmentedLargeHeader) {
  MockRead reads[] = {
      MockRead(SYNCHRONOUS, kPartialLargeFrame, kLargeFrameHeaderSize - 1),
      MockRead(SYNCHRONOUS, ERR_IO_PENDING)};
  CreateReadOnly(reads);
  EXPECT_EQ(ERR_IO_PENDING,
            stream_->ReadFrames(&frame_chunks_, cb_.callback()));
}

// A frame that does not arrive in a single read should arrive in chunks.
TEST_F(WebSocketBasicStreamSocketTest, LargeFrameFirstChunk) {
  MockRead reads[] = {
      MockRead(SYNCHRONOUS, kPartialLargeFrame, kPartialLargeFrameSize)};
  CreateReadOnly(reads);
  EXPECT_EQ(OK, stream_->ReadFrames(&frame_chunks_, cb_.callback()));
  ASSERT_EQ(1U, frame_chunks_.size());
  ASSERT_TRUE(frame_chunks_[0]->header);
  EXPECT_EQ(kLargeFrameDeclaredPayloadSize,
            frame_chunks_[0]->header->payload_length);
  EXPECT_TRUE(frame_chunks_[0]->header->final);
  EXPECT_FALSE(frame_chunks_[0]->final_chunk);
  EXPECT_EQ(kPartialLargeFrameSize - kLargeFrameHeaderSize,
            static_cast<size_t>(frame_chunks_[0]->data->size()));
}

// If only the header arrives, we should get a zero-byte chunk.
TEST_F(WebSocketBasicStreamSocketTest, HeaderOnlyChunk) {
  MockRead reads[] = {
      MockRead(SYNCHRONOUS, kPartialLargeFrame, kLargeFrameHeaderSize)};
  CreateReadOnly(reads);
  EXPECT_EQ(OK, stream_->ReadFrames(&frame_chunks_, cb_.callback()));
  ASSERT_EQ(1U, frame_chunks_.size());
  EXPECT_FALSE(frame_chunks_[0]->final_chunk);
  EXPECT_TRUE(frame_chunks_[0]->data.get() == NULL);
}

// The second and subsequent chunks of a frame have no header.
TEST_F(WebSocketBasicStreamSocketTest, LargeFrameTwoChunks) {
  static const size_t kChunkSize = 16;
  MockRead reads[] = {
      MockRead(ASYNC, kPartialLargeFrame, kChunkSize),
      MockRead(ASYNC, kPartialLargeFrame + kChunkSize, kChunkSize)};
  CreateReadOnly(reads);
  TestCompletionCallback cb[2];

  ASSERT_EQ(ERR_IO_PENDING,
            stream_->ReadFrames(&frame_chunks_, cb[0].callback()));
  EXPECT_EQ(OK, cb[0].WaitForResult());
  ASSERT_EQ(1U, frame_chunks_.size());
  ASSERT_TRUE(frame_chunks_[0]->header);

  frame_chunks_.clear();
  ASSERT_EQ(ERR_IO_PENDING,
            stream_->ReadFrames(&frame_chunks_, cb[1].callback()));
  EXPECT_EQ(OK, cb[1].WaitForResult());
  ASSERT_EQ(1U, frame_chunks_.size());
  ASSERT_FALSE(frame_chunks_[0]->header);
}

// Only the final chunk of a frame has final_chunk set.
TEST_F(WebSocketBasicStreamSocketTest, OnlyFinalChunkIsFinal) {
  static const size_t kFirstChunkSize = 4;
  MockRead reads[] = {MockRead(ASYNC, kSampleFrame, kFirstChunkSize),
                      MockRead(ASYNC,
                               kSampleFrame + kFirstChunkSize,
                               kSampleFrameSize - kFirstChunkSize)};
  CreateReadOnly(reads);
  TestCompletionCallback cb[2];

  ASSERT_EQ(ERR_IO_PENDING,
            stream_->ReadFrames(&frame_chunks_, cb[0].callback()));
  EXPECT_EQ(OK, cb[0].WaitForResult());
  ASSERT_EQ(1U, frame_chunks_.size());
  ASSERT_FALSE(frame_chunks_[0]->final_chunk);

  frame_chunks_.clear();
  ASSERT_EQ(ERR_IO_PENDING,
            stream_->ReadFrames(&frame_chunks_, cb[1].callback()));
  EXPECT_EQ(OK, cb[1].WaitForResult());
  ASSERT_EQ(1U, frame_chunks_.size());
  ASSERT_TRUE(frame_chunks_[0]->final_chunk);
}

// Multiple frames that arrive together should be parsed correctly.
TEST_F(WebSocketBasicStreamSocketTest, ThreeFramesTogether) {
  MockRead reads[] = {
      MockRead(SYNCHRONOUS, kMultipleFrames, kMultipleFramesSize)};
  CreateReadOnly(reads);

  ASSERT_EQ(OK, stream_->ReadFrames(&frame_chunks_, cb_.callback()));
  ASSERT_EQ(3U, frame_chunks_.size());
  EXPECT_TRUE(frame_chunks_[0]->final_chunk);
  EXPECT_TRUE(frame_chunks_[1]->final_chunk);
  EXPECT_TRUE(frame_chunks_[2]->final_chunk);
}

// ERR_CONNECTION_CLOSED must be returned on close.
TEST_F(WebSocketBasicStreamSocketTest, SyncClose) {
  MockRead reads[] = {MockRead(SYNCHRONOUS, "", 0)};
  CreateReadOnly(reads);

  EXPECT_EQ(ERR_CONNECTION_CLOSED,
            stream_->ReadFrames(&frame_chunks_, cb_.callback()));
}

TEST_F(WebSocketBasicStreamSocketTest, AsyncClose) {
  MockRead reads[] = {MockRead(ASYNC, "", 0)};
  CreateReadOnly(reads);

  ASSERT_EQ(ERR_IO_PENDING,
            stream_->ReadFrames(&frame_chunks_, cb_.callback()));
  EXPECT_EQ(ERR_CONNECTION_CLOSED, cb_.WaitForResult());
}

// The result should be the same if the socket returns
// ERR_CONNECTION_CLOSED. This is not expected to happen on an established
// connection; a Read of size 0 is the expected behaviour. The key point of this
// test is to confirm that ReadFrames() behaviour is identical in both cases.
TEST_F(WebSocketBasicStreamSocketTest, SyncCloseWithErr) {
  MockRead reads[] = {MockRead(SYNCHRONOUS, ERR_CONNECTION_CLOSED)};
  CreateReadOnly(reads);

  EXPECT_EQ(ERR_CONNECTION_CLOSED,
            stream_->ReadFrames(&frame_chunks_, cb_.callback()));
}

TEST_F(WebSocketBasicStreamSocketTest, AsyncCloseWithErr) {
  MockRead reads[] = {MockRead(ASYNC, ERR_CONNECTION_CLOSED)};
  CreateReadOnly(reads);

  ASSERT_EQ(ERR_IO_PENDING,
            stream_->ReadFrames(&frame_chunks_, cb_.callback()));
  EXPECT_EQ(ERR_CONNECTION_CLOSED, cb_.WaitForResult());
}

TEST_F(WebSocketBasicStreamSocketTest, SyncErrorsPassedThrough) {
  // ERR_INSUFFICIENT_RESOURCES here represents an arbitrary error that
  // WebSocketBasicStream gives no special handling to.
  MockRead reads[] = {MockRead(SYNCHRONOUS, ERR_INSUFFICIENT_RESOURCES)};
  CreateReadOnly(reads);

  EXPECT_EQ(ERR_INSUFFICIENT_RESOURCES,
            stream_->ReadFrames(&frame_chunks_, cb_.callback()));
}

TEST_F(WebSocketBasicStreamSocketTest, AsyncErrorsPassedThrough) {
  MockRead reads[] = {MockRead(ASYNC, ERR_INSUFFICIENT_RESOURCES)};
  CreateReadOnly(reads);

  ASSERT_EQ(ERR_IO_PENDING,
            stream_->ReadFrames(&frame_chunks_, cb_.callback()));
  EXPECT_EQ(ERR_INSUFFICIENT_RESOURCES, cb_.WaitForResult());
}

// If we get a frame followed by a close, we should receive them separately.
TEST_F(WebSocketBasicStreamSocketTest, CloseAfterFrame) {
  MockRead reads[] = {MockRead(SYNCHRONOUS, kSampleFrame, kSampleFrameSize),
                      MockRead(SYNCHRONOUS, "", 0)};
  CreateReadOnly(reads);

  EXPECT_EQ(OK, stream_->ReadFrames(&frame_chunks_, cb_.callback()));
  EXPECT_EQ(1U, frame_chunks_.size());
  frame_chunks_.clear();
  EXPECT_EQ(ERR_CONNECTION_CLOSED,
            stream_->ReadFrames(&frame_chunks_, cb_.callback()));
}

// Synchronous close after an async frame header is handled by a different code
// path.
TEST_F(WebSocketBasicStreamSocketTest, AsyncCloseAfterIncompleteHeader) {
  MockRead reads[] = {MockRead(ASYNC, kSampleFrame, 1U),
                      MockRead(SYNCHRONOUS, "", 0)};
  CreateReadOnly(reads);

  ASSERT_EQ(ERR_IO_PENDING,
            stream_->ReadFrames(&frame_chunks_, cb_.callback()));
  ASSERT_EQ(ERR_CONNECTION_CLOSED, cb_.WaitForResult());
}

// When Stream::Read returns ERR_CONNECTION_CLOSED we get the same result via a
// slightly different code path.
TEST_F(WebSocketBasicStreamSocketTest, AsyncErrCloseAfterIncompleteHeader) {
  MockRead reads[] = {MockRead(ASYNC, kSampleFrame, 1U),
                      MockRead(SYNCHRONOUS, ERR_CONNECTION_CLOSED)};
  CreateReadOnly(reads);

  ASSERT_EQ(ERR_IO_PENDING,
            stream_->ReadFrames(&frame_chunks_, cb_.callback()));
  ASSERT_EQ(ERR_CONNECTION_CLOSED, cb_.WaitForResult());
}

// If there was a frame read at the same time as the response headers (and the
// handshake succeeded), then we should parse it.
TEST_F(WebSocketBasicStreamSocketTest, HttpReadBufferIsUsed) {
  SetHttpReadBuffer(kSampleFrame, kSampleFrameSize);
  CreateNullStream();

  EXPECT_EQ(OK, stream_->ReadFrames(&frame_chunks_, cb_.callback()));
  ASSERT_EQ(1U, frame_chunks_.size());
  ASSERT_TRUE(frame_chunks_[0]->data);
  EXPECT_EQ(6, frame_chunks_[0]->data->size());
}

// Check that a frame whose header partially arrived at the end of the response
// headers works correctly.
TEST_F(WebSocketBasicStreamSocketTest, PartialFrameHeaderInHttpResponse) {
  SetHttpReadBuffer(kSampleFrame, 1);
  MockRead reads[] = {MockRead(ASYNC, kSampleFrame + 1, kSampleFrameSize - 1)};
  CreateReadOnly(reads);

  ASSERT_EQ(ERR_IO_PENDING,
            stream_->ReadFrames(&frame_chunks_, cb_.callback()));
  EXPECT_EQ(OK, cb_.WaitForResult());
  ASSERT_EQ(1U, frame_chunks_.size());
  ASSERT_TRUE(frame_chunks_[0]->data);
  EXPECT_EQ(6, frame_chunks_[0]->data->size());
  ASSERT_TRUE(frame_chunks_[0]->header);
  EXPECT_EQ(WebSocketFrameHeader::kOpCodeText,
            frame_chunks_[0]->header->opcode);
}

// Check that an invalid frame results in an error.
TEST_F(WebSocketBasicStreamSocketTest, SyncInvalidFrame) {
  MockRead reads[] = {MockRead(SYNCHRONOUS, kInvalidFrame, kInvalidFrameSize)};
  CreateReadOnly(reads);

  EXPECT_EQ(ERR_WS_PROTOCOL_ERROR,
            stream_->ReadFrames(&frame_chunks_, cb_.callback()));
}

TEST_F(WebSocketBasicStreamSocketTest, AsyncInvalidFrame) {
  MockRead reads[] = {MockRead(ASYNC, kInvalidFrame, kInvalidFrameSize)};
  CreateReadOnly(reads);

  ASSERT_EQ(ERR_IO_PENDING,
            stream_->ReadFrames(&frame_chunks_, cb_.callback()));
  EXPECT_EQ(ERR_WS_PROTOCOL_ERROR, cb_.WaitForResult());
}

// Check that writing a frame all at once works.
TEST_F(WebSocketBasicStreamSocketTest, WriteAtOnce) {
  MockWrite writes[] = {MockWrite(SYNCHRONOUS, kWriteFrame, kWriteFrameSize)};
  CreateWriteOnly(writes);
  frame_chunks_ = GenerateWriteFrame();

  EXPECT_EQ(OK, stream_->WriteFrames(&frame_chunks_, cb_.callback()));
}

// Check that completely async writing works.
TEST_F(WebSocketBasicStreamSocketTest, AsyncWriteAtOnce) {
  MockWrite writes[] = {MockWrite(ASYNC, kWriteFrame, kWriteFrameSize)};
  CreateWriteOnly(writes);
  frame_chunks_ = GenerateWriteFrame();

  ASSERT_EQ(ERR_IO_PENDING,
            stream_->WriteFrames(&frame_chunks_, cb_.callback()));
  EXPECT_EQ(OK, cb_.WaitForResult());
}

// Check that writing a frame to an extremely full kernel buffer (so that it
// ends up being sent in bits) works. The WriteFrames() callback should not be
// called until all parts have been written.
TEST_F(WebSocketBasicStreamSocketTest, WriteInBits) {
  MockWrite writes[] = {MockWrite(SYNCHRONOUS, kWriteFrame, 4),
                        MockWrite(ASYNC, kWriteFrame + 4, 4),
                        MockWrite(ASYNC, kWriteFrame + 8, kWriteFrameSize - 8)};
  CreateWriteOnly(writes);
  frame_chunks_ = GenerateWriteFrame();

  ASSERT_EQ(ERR_IO_PENDING,
            stream_->WriteFrames(&frame_chunks_, cb_.callback()));
  EXPECT_EQ(OK, cb_.WaitForResult());
}

TEST_F(WebSocketBasicStreamSocketTest, GetExtensionsWorks) {
  extensions_ = "inflate-uuencode";
  CreateNullStream();

  EXPECT_EQ("inflate-uuencode", stream_->GetExtensions());
}

TEST_F(WebSocketBasicStreamSocketTest, GetSubProtocolWorks) {
  sub_protocol_ = "cyberchat";
  CreateNullStream();

  EXPECT_EQ("cyberchat", stream_->GetSubProtocol());
}

}  // namespace
}  // namespace net