Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(84)

Side by Side Diff: net/quic/quic_client_session_test.cc

Issue 114933003: Minor cleanup of QUIC MockConnection and PacketSavingConnection (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src
Patch Set: Created 7 years ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch | Annotate | Revision Log
« no previous file with comments | « no previous file | net/quic/quic_crypto_client_stream_test.cc » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. 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 2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file. 3 // found in the LICENSE file.
4 4
5 #include "net/quic/quic_client_session.h" 5 #include "net/quic/quic_client_session.h"
6 6
7 #include <vector> 7 #include <vector>
8 8
9 #include "base/rand_util.h" 9 #include "base/rand_util.h"
10 #include "net/base/capturing_net_log.h" 10 #include "net/base/capturing_net_log.h"
(...skipping 46 matching lines...) Expand 10 before | Expand all | Expand 10 after
57 // Returns the header from the last packet written. 57 // Returns the header from the last packet written.
58 const QuicPacketHeader& header() { return header_; } 58 const QuicPacketHeader& header() { return header_; }
59 59
60 private: 60 private:
61 QuicPacketHeader header_; 61 QuicPacketHeader header_;
62 }; 62 };
63 63
64 class QuicClientSessionTest : public ::testing::Test { 64 class QuicClientSessionTest : public ::testing::Test {
65 protected: 65 protected:
66 QuicClientSessionTest() 66 QuicClientSessionTest()
67 : guid_(1), 67 : writer_(new TestPacketWriter()),
68 writer_(new TestPacketWriter()), 68 connection_(new PacketSavingConnection(false)),
69 connection_(new PacketSavingConnection(guid_, IPEndPoint(), false)),
70 session_(connection_, GetSocket().Pass(), writer_.Pass(), NULL, NULL, 69 session_(connection_, GetSocket().Pass(), writer_.Pass(), NULL, NULL,
71 kServerHostname, DefaultQuicConfig(), &crypto_config_, 70 kServerHostname, DefaultQuicConfig(), &crypto_config_,
72 &net_log_) { 71 &net_log_) {
73 session_.config()->SetDefaults(); 72 session_.config()->SetDefaults();
74 crypto_config_.SetDefaults(); 73 crypto_config_.SetDefaults();
75 } 74 }
76 75
77 virtual void TearDown() OVERRIDE { 76 virtual void TearDown() OVERRIDE {
78 session_.CloseSessionOnError(ERR_ABORTED); 77 session_.CloseSessionOnError(ERR_ABORTED);
79 } 78 }
80 79
81 scoped_ptr<DatagramClientSocket> GetSocket() { 80 scoped_ptr<DatagramClientSocket> GetSocket() {
82 socket_factory_.AddSocketDataProvider(&socket_data_); 81 socket_factory_.AddSocketDataProvider(&socket_data_);
83 return socket_factory_.CreateDatagramClientSocket( 82 return socket_factory_.CreateDatagramClientSocket(
84 DatagramSocket::DEFAULT_BIND, base::Bind(&base::RandInt), 83 DatagramSocket::DEFAULT_BIND, base::Bind(&base::RandInt),
85 &net_log_, NetLog::Source()); 84 &net_log_, NetLog::Source());
86 } 85 }
87 86
88 void CompleteCryptoHandshake() { 87 void CompleteCryptoHandshake() {
89 ASSERT_EQ(ERR_IO_PENDING, 88 ASSERT_EQ(ERR_IO_PENDING,
90 session_.CryptoConnect(false, callback_.callback())); 89 session_.CryptoConnect(false, callback_.callback()));
91 CryptoTestUtils::HandshakeWithFakeServer( 90 CryptoTestUtils::HandshakeWithFakeServer(
92 connection_, session_.GetCryptoStream()); 91 connection_, session_.GetCryptoStream());
93 ASSERT_EQ(OK, callback_.WaitForResult()); 92 ASSERT_EQ(OK, callback_.WaitForResult());
94 } 93 }
95 94
96 QuicGuid guid_;
97 scoped_ptr<QuicDefaultPacketWriter> writer_; 95 scoped_ptr<QuicDefaultPacketWriter> writer_;
98 PacketSavingConnection* connection_; 96 PacketSavingConnection* connection_;
99 CapturingNetLog net_log_; 97 CapturingNetLog net_log_;
100 MockClientSocketFactory socket_factory_; 98 MockClientSocketFactory socket_factory_;
101 StaticSocketDataProvider socket_data_; 99 StaticSocketDataProvider socket_data_;
102 QuicClientSession session_; 100 QuicClientSession session_;
103 MockClock clock_; 101 MockClock clock_;
104 MockRandom random_; 102 MockRandom random_;
105 QuicConnectionVisitorInterface* visitor_; 103 QuicConnectionVisitorInterface* visitor_;
106 TestCompletionCallback callback_; 104 TestCompletionCallback callback_;
(...skipping 49 matching lines...) Expand 10 before | Expand all | Expand 10 after
156 154
157 // After receiving a GoAway, I should no longer be able to create outgoing 155 // After receiving a GoAway, I should no longer be able to create outgoing
158 // streams. 156 // streams.
159 session_.OnGoAway(QuicGoAwayFrame(QUIC_PEER_GOING_AWAY, 1u, "Going away.")); 157 session_.OnGoAway(QuicGoAwayFrame(QUIC_PEER_GOING_AWAY, 1u, "Going away."));
160 EXPECT_EQ(NULL, session_.CreateOutgoingDataStream()); 158 EXPECT_EQ(NULL, session_.CreateOutgoingDataStream());
161 } 159 }
162 160
163 } // namespace 161 } // namespace
164 } // namespace test 162 } // namespace test
165 } // namespace net 163 } // namespace net
OLDNEW
« no previous file with comments | « no previous file | net/quic/quic_crypto_client_stream_test.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698