OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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/tools/quic/quic_simple_server_session.h" | 5 #include "net/tools/quic/quic_simple_server_session.h" |
6 | 6 |
7 #include <algorithm> | 7 #include <algorithm> |
8 #include <memory> | 8 #include <memory> |
9 | 9 |
10 #include "base/macros.h" | 10 #include "base/macros.h" |
(...skipping 61 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
72 QuicStreamId promised_stream_id, | 72 QuicStreamId promised_stream_id, |
73 SpdyHeaderBlock headers) override { | 73 SpdyHeaderBlock headers) override { |
74 return WritePushPromiseMock(original_stream_id, promised_stream_id, | 74 return WritePushPromiseMock(original_stream_id, promised_stream_id, |
75 headers); | 75 headers); |
76 } | 76 } |
77 MOCK_METHOD3(WritePushPromiseMock, | 77 MOCK_METHOD3(WritePushPromiseMock, |
78 size_t(QuicStreamId original_stream_id, | 78 size_t(QuicStreamId original_stream_id, |
79 QuicStreamId promised_stream_id, | 79 QuicStreamId promised_stream_id, |
80 const SpdyHeaderBlock& headers)); | 80 const SpdyHeaderBlock& headers)); |
81 | 81 |
82 size_t WriteHeaders( | 82 size_t WriteHeaders(QuicStreamId stream_id, |
83 QuicStreamId stream_id, | 83 SpdyHeaderBlock headers, |
84 SpdyHeaderBlock headers, | 84 bool fin, |
85 bool fin, | 85 SpdyPriority priority, |
86 SpdyPriority priority, | 86 QuicReferenceCountedPointer<QuicAckListenerInterface> |
87 scoped_refptr<QuicAckListenerInterface> ack_listener) override { | 87 ack_listener) override { |
88 return WriteHeadersMock(stream_id, headers, fin, priority, ack_listener); | 88 return WriteHeadersMock(stream_id, headers, fin, priority, ack_listener); |
89 } | 89 } |
90 MOCK_METHOD5( | 90 MOCK_METHOD5( |
91 WriteHeadersMock, | 91 WriteHeadersMock, |
92 size_t(QuicStreamId stream_id, | 92 size_t(QuicStreamId stream_id, |
93 const SpdyHeaderBlock& headers, | 93 const SpdyHeaderBlock& headers, |
94 bool fin, | 94 bool fin, |
95 SpdyPriority priority, | 95 SpdyPriority priority, |
96 const scoped_refptr<QuicAckListenerInterface>& ack_listener)); | 96 const QuicReferenceCountedPointer<QuicAckListenerInterface>& |
| 97 ack_listener)); |
97 }; | 98 }; |
98 | 99 |
99 class MockQuicCryptoServerStream : public QuicCryptoServerStream { | 100 class MockQuicCryptoServerStream : public QuicCryptoServerStream { |
100 public: | 101 public: |
101 explicit MockQuicCryptoServerStream( | 102 explicit MockQuicCryptoServerStream( |
102 const QuicCryptoServerConfig* crypto_config, | 103 const QuicCryptoServerConfig* crypto_config, |
103 QuicCompressedCertsCache* compressed_certs_cache, | 104 QuicCompressedCertsCache* compressed_certs_cache, |
104 QuicServerSessionBase* session, | 105 QuicServerSessionBase* session, |
105 QuicCryptoServerStream::Helper* helper) | 106 QuicCryptoServerStream::Helper* helper) |
106 : QuicCryptoServerStream(crypto_config, | 107 : QuicCryptoServerStream(crypto_config, |
(...skipping 21 matching lines...) Expand all Loading... |
128 MockAlarmFactory* alarm_factory, | 129 MockAlarmFactory* alarm_factory, |
129 Perspective perspective, | 130 Perspective perspective, |
130 const QuicVersionVector& supported_versions) | 131 const QuicVersionVector& supported_versions) |
131 : MockQuicConnection(helper, | 132 : MockQuicConnection(helper, |
132 alarm_factory, | 133 alarm_factory, |
133 perspective, | 134 perspective, |
134 supported_versions) {} | 135 supported_versions) {} |
135 | 136 |
136 MOCK_METHOD5( | 137 MOCK_METHOD5( |
137 SendStreamData, | 138 SendStreamData, |
138 QuicConsumedData(QuicStreamId id, | 139 QuicConsumedData( |
139 QuicIOVector iov, | 140 QuicStreamId id, |
140 QuicStreamOffset offset, | 141 QuicIOVector iov, |
141 bool fin, | 142 QuicStreamOffset offset, |
142 scoped_refptr<QuicAckListenerInterface> listern)); | 143 bool fin, |
| 144 QuicReferenceCountedPointer<QuicAckListenerInterface> listern)); |
143 }; | 145 }; |
144 | 146 |
145 class QuicSimpleServerSessionPeer { | 147 class QuicSimpleServerSessionPeer { |
146 public: | 148 public: |
147 static void SetCryptoStream(QuicSimpleServerSession* s, | 149 static void SetCryptoStream(QuicSimpleServerSession* s, |
148 QuicCryptoServerStream* crypto_stream) { | 150 QuicCryptoServerStream* crypto_stream) { |
149 s->crypto_stream_.reset(crypto_stream); | 151 s->crypto_stream_.reset(crypto_stream); |
150 s->static_streams()[kCryptoStreamId] = crypto_stream; | 152 s->static_streams()[kCryptoStreamId] = crypto_stream; |
151 } | 153 } |
152 | 154 |
(...skipping 466 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
619 .WillOnce(Return(QuicConsumedData(kStreamFlowControlWindowSize, false))); | 621 .WillOnce(Return(QuicConsumedData(kStreamFlowControlWindowSize, false))); |
620 | 622 |
621 EXPECT_CALL(*connection_, SendBlocked(stream_to_open)); | 623 EXPECT_CALL(*connection_, SendBlocked(stream_to_open)); |
622 QuicRstStreamFrame rst(stream_got_reset, QUIC_STREAM_CANCELLED, 0); | 624 QuicRstStreamFrame rst(stream_got_reset, QUIC_STREAM_CANCELLED, 0); |
623 visitor_->OnRstStream(rst); | 625 visitor_->OnRstStream(rst); |
624 } | 626 } |
625 | 627 |
626 } // namespace | 628 } // namespace |
627 } // namespace test | 629 } // namespace test |
628 } // namespace net | 630 } // namespace net |
OLD | NEW |