Revert 165858 - Add QuicStream and friends to QUIC code.

net/quic/quic_session_test.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 "net/quic/quic_session.h"
-#include "net/quic/quic_connection.h"
-
-#include <set>
-
-#include "base/hash_tables.h"
-#include "net/quic/test_tools/quic_test_utils.h"
-#include "testing/gmock/include/gmock/gmock.h"
-#include "testing/gtest/include/gtest/gtest.h"
-
-using base::hash_map;
-using std::set;
-using testing::_;
-
-namespace net {
-namespace test {
-namespace {
-
-class TestCryptoStream : public QuicCryptoStream {
- public:
-  explicit TestCryptoStream(QuicSession* session)
-      : QuicCryptoStream(session) {
-  }
-
-  void OnHandshakeMessage(const CryptoHandshakeMessage& message) {
-    set_handshake_complete(true);
-  }
-};
-
-class TestStream : public ReliableQuicStream {
- public:
-  TestStream(QuicStreamId id, QuicSession* session)
-      : ReliableQuicStream(id, session) {
-  }
-
-  virtual uint32 ProcessData(const char* data, uint32 data_len) {
-    return data_len;
-  }
-};
-
-class TestSession : public QuicSession {
- public:
-  TestSession(QuicConnection* connection, bool is_server)
-      : QuicSession(connection, is_server),
-        crypto_stream_(this) {
-  }
-
-  virtual QuicCryptoStream* GetCryptoStream() {
-    return &crypto_stream_;
-  }
-
-  virtual TestStream* CreateOutgoingReliableStream() {
-    TestStream* stream = new TestStream(GetNextStreamId(), this);
-    ActivateStream(stream);
-    return stream;
-  }
-
-  virtual TestStream* CreateIncomingReliableStream(QuicStreamId id) {
-    return new TestStream(id, this);
-  }
-
-  bool IsClosedStream(QuicStreamId id) {
-    return QuicSession::IsClosedStream(id);
-  }
-
-  ReliableQuicStream* GetIncomingReliableStream(QuicStreamId stream_id) {
-    return QuicSession::GetIncomingReliableStream(stream_id);
-  }
-
-  TestCryptoStream crypto_stream_;
-};
-
-class QuicSessionTest : public ::testing::Test {
- protected:
-  QuicSessionTest()
-      : guid_(1),
-        connection_(new MockConnection(guid_, IPEndPoint())),
-        session_(connection_, true) {
-  }
-
-  void CheckClosedStreams() {
-    for (int i = kCryptoStreamId; i < 100; i++) {
-      if (closed_streams_.count(i) == 0) {
-        EXPECT_FALSE(session_.IsClosedStream(i)) << " stream id: " << i;
-      } else {
-        EXPECT_TRUE(session_.IsClosedStream(i)) << " stream id: " << i;
-      }
-    }
-  }
-
-  void CloseStream(QuicStreamId id) {
-    session_.CloseStream(id);
-    closed_streams_.insert(id);
-  }
-
-  QuicGuid guid_;
-  MockConnection* connection_;
-  TestSession session_;
-  QuicConnectionVisitorInterface* visitor_;
-  hash_map<QuicStreamId, ReliableQuicStream*>* streams_;
-  set<QuicStreamId> closed_streams_;
-};
-
-TEST_F(QuicSessionTest, IsHandshakeComplete) {
-  EXPECT_FALSE(session_.IsHandshakeComplete());
-  CryptoHandshakeMessage message;
-  session_.crypto_stream_.OnHandshakeMessage(message);
-  EXPECT_TRUE(session_.IsHandshakeComplete());
-}
-
-TEST_F(QuicSessionTest, IsClosedStreamDefault) {
-  // Ensure that no streams are initially closed.
-  for (int i = kCryptoStreamId; i < 100; i++) {
-    EXPECT_FALSE(session_.IsClosedStream(i));
-  }
-}
-
-TEST_F(QuicSessionTest, IsClosedStreamLocallyCreated) {
-  scoped_ptr<TestStream> stream2(session_.CreateOutgoingReliableStream());
-  scoped_ptr<TestStream> stream4(session_.CreateOutgoingReliableStream());
-
-  CheckClosedStreams();
-  CloseStream(4);
-  CheckClosedStreams();
-  CloseStream(2);
-  CheckClosedStreams();
-}
-
-TEST_F(QuicSessionTest, IsClosedStreamPeerCreated) {
-  scoped_ptr<ReliableQuicStream> stream3(session_.GetIncomingReliableStream(3));
-  scoped_ptr<ReliableQuicStream> stream5(session_.GetIncomingReliableStream(5));
-
-  CheckClosedStreams();
-  CloseStream(3);
-  CheckClosedStreams();
-  CloseStream(5);
-  // Create stream id 9, and implicitly 7
-  scoped_ptr<ReliableQuicStream> stream9(session_.GetIncomingReliableStream(9));
-  CheckClosedStreams();
-  // Close 9, but make sure 7 is still not closed
-  CloseStream(9);
-  CheckClosedStreams();
-}
-
-TEST_F(QuicSessionTest, StreamIdTooLarge) {
-  scoped_ptr<ReliableQuicStream> stream3(session_.GetIncomingReliableStream(3));
-  EXPECT_CALL(*connection_, SendConnectionClose(QUIC_INVALID_STREAM_ID));
-  scoped_ptr<ReliableQuicStream> stream5(
-      session_.GetIncomingReliableStream(105));
-}
-
-}  // namespace
-}  // namespace test
-}  // namespace net