1 files changed, 134 insertions, 0 deletions
diff --git a/chromium/net/quic/quic_client_session_test.cc b/chromium/net/quic/quic_client_session_test.cc
new file mode 100644
index 00000000000..09e7d210116
--- /dev/null
+++ b/chromium/net/quic/quic_client_session_test.cc
@@ -0,0 +1,134 @@
+// 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_client_session.h"
+
+#include <vector>
+
+#include "net/base/capturing_net_log.h"
+#include "net/base/test_completion_callback.h"
+#include "net/quic/crypto/aes_128_gcm_12_encrypter.h"
+#include "net/quic/crypto/crypto_protocol.h"
+#include "net/quic/crypto/quic_decrypter.h"
+#include "net/quic/crypto/quic_encrypter.h"
+#include "net/quic/test_tools/crypto_test_utils.h"
+#include "net/quic/test_tools/quic_client_session_peer.h"
+#include "net/quic/test_tools/quic_test_utils.h"
+#include "net/udp/datagram_client_socket.h"
+
+using testing::_;
+
+namespace net {
+namespace test {
+namespace {
+
+const char kServerHostname[] = "www.example.com";
+
+class QuicClientSessionTest : public ::testing::Test {
+ protected:
+  QuicClientSessionTest()
+      : guid_(1),
+        connection_(new PacketSavingConnection(guid_, IPEndPoint(), false)),
+        session_(connection_, scoped_ptr<DatagramClientSocket>(), NULL,
+                 NULL, kServerHostname, DefaultQuicConfig(), &crypto_config_,
+                 &net_log_) {
+    session_.config()->SetDefaults();
+    crypto_config_.SetDefaults();
+  }
+
+  void CompleteCryptoHandshake() {
+    ASSERT_EQ(ERR_IO_PENDING,
+              session_.CryptoConnect(callback_.callback()));
+    CryptoTestUtils::HandshakeWithFakeServer(
+        connection_, session_.GetCryptoStream());
+    ASSERT_EQ(OK, callback_.WaitForResult());
+  }
+
+  QuicGuid guid_;
+  PacketSavingConnection* connection_;
+  CapturingNetLog net_log_;
+  QuicClientSession session_;
+  MockClock clock_;
+  MockRandom random_;
+  QuicConnectionVisitorInterface* visitor_;
+  TestCompletionCallback callback_;
+  QuicCryptoClientConfig crypto_config_;
+};
+
+TEST_F(QuicClientSessionTest, CryptoConnect) {
+  if (!Aes128Gcm12Encrypter::IsSupported()) {
+    LOG(INFO) << "AES GCM not supported. Test skipped.";
+    return;
+  }
+
+  CompleteCryptoHandshake();
+}
+
+TEST_F(QuicClientSessionTest, MaxNumStreams) {
+  if (!Aes128Gcm12Encrypter::IsSupported()) {
+    LOG(INFO) << "AES GCM not supported. Test skipped.";
+    return;
+  }
+
+  CompleteCryptoHandshake();
+
+  std::vector<QuicReliableClientStream*> streams;
+  for (size_t i = 0; i < kDefaultMaxStreamsPerConnection; i++) {
+    QuicReliableClientStream* stream = session_.CreateOutgoingReliableStream();
+    EXPECT_TRUE(stream);
+    streams.push_back(stream);
+  }
+  EXPECT_FALSE(session_.CreateOutgoingReliableStream());
+
+  // Close a stream and ensure I can now open a new one.
+  session_.CloseStream(streams[0]->id());
+  EXPECT_TRUE(session_.CreateOutgoingReliableStream());
+}
+
+TEST_F(QuicClientSessionTest, MaxNumStreamsViaRequest) {
+  if (!Aes128Gcm12Encrypter::IsSupported()) {
+    LOG(INFO) << "AES GCM not supported. Test skipped.";
+    return;
+  }
+
+  CompleteCryptoHandshake();
+
+  std::vector<QuicReliableClientStream*> streams;
+  for (size_t i = 0; i < kDefaultMaxStreamsPerConnection; i++) {
+    QuicReliableClientStream* stream = session_.CreateOutgoingReliableStream();
+    EXPECT_TRUE(stream);
+    streams.push_back(stream);
+  }
+
+  QuicReliableClientStream* stream;
+  QuicClientSession::StreamRequest stream_request;
+  TestCompletionCallback callback;
+  ASSERT_EQ(ERR_IO_PENDING,
+            stream_request.StartRequest(session_.GetWeakPtr(), &stream,
+                                        callback.callback()));
+
+  // Close a stream and ensure I can now open a new one.
+  session_.CloseStream(streams[0]->id());
+  ASSERT_TRUE(callback.have_result());
+  EXPECT_EQ(OK, callback.WaitForResult());
+  EXPECT_TRUE(stream != NULL);
+}
+
+TEST_F(QuicClientSessionTest, GoAwayReceived) {
+  if (!Aes128Gcm12Encrypter::IsSupported()) {
+    LOG(INFO) << "AES GCM not supported. Test skipped.";
+    return;
+  }
+
+  CompleteCryptoHandshake();
+
+  // After receiving a GoAway, I should no longer be able to create outgoing
+  // streams.
+  session_.OnGoAway(QuicGoAwayFrame(QUIC_PEER_GOING_AWAY, 1u, "Going away."));
+  EXPECT_EQ(NULL, session_.CreateOutgoingReliableStream());
+}
+
+}  // namespace
+}  // namespace test
+}  // namespace net