OLD | NEW |
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/chromium/quic_chromium_client_stream.h" | 5 #include "net/quic/chromium/quic_chromium_client_stream.h" |
6 | 6 |
7 #include <string> | 7 #include <string> |
8 | 8 |
9 #include "base/memory/ptr_util.h" | 9 #include "base/memory/ptr_util.h" |
10 #include "base/run_loop.h" | 10 #include "base/run_loop.h" |
(...skipping 61 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
72 QuicCryptoStream* GetCryptoStream() override { return crypto_stream_.get(); } | 72 QuicCryptoStream* GetCryptoStream() override { return crypto_stream_.get(); } |
73 | 73 |
74 // From QuicSession. | 74 // From QuicSession. |
75 MOCK_METHOD3(OnConnectionClosed, | 75 MOCK_METHOD3(OnConnectionClosed, |
76 void(QuicErrorCode error, | 76 void(QuicErrorCode error, |
77 const std::string& error_details, | 77 const std::string& error_details, |
78 ConnectionCloseSource source)); | 78 ConnectionCloseSource source)); |
79 MOCK_METHOD1(CreateIncomingDynamicStream, QuicSpdyStream*(QuicStreamId id)); | 79 MOCK_METHOD1(CreateIncomingDynamicStream, QuicSpdyStream*(QuicStreamId id)); |
80 MOCK_METHOD1(CreateOutgoingDynamicStream, | 80 MOCK_METHOD1(CreateOutgoingDynamicStream, |
81 QuicChromiumClientStream*(SpdyPriority priority)); | 81 QuicChromiumClientStream*(SpdyPriority priority)); |
82 MOCK_METHOD6(WritevData, | 82 MOCK_METHOD6( |
83 QuicConsumedData(QuicStream* stream, | 83 WritevData, |
84 QuicStreamId id, | 84 QuicConsumedData(QuicStream* stream, |
85 QuicIOVector data, | 85 QuicStreamId id, |
86 QuicStreamOffset offset, | 86 QuicIOVector data, |
87 bool fin, | 87 QuicStreamOffset offset, |
88 scoped_refptr<QuicAckListenerInterface>)); | 88 bool fin, |
| 89 QuicReferenceCountedPointer<QuicAckListenerInterface>)); |
89 MOCK_METHOD3(SendRstStream, | 90 MOCK_METHOD3(SendRstStream, |
90 void(QuicStreamId stream_id, | 91 void(QuicStreamId stream_id, |
91 QuicRstStreamErrorCode error, | 92 QuicRstStreamErrorCode error, |
92 QuicStreamOffset bytes_written)); | 93 QuicStreamOffset bytes_written)); |
93 | 94 |
94 MOCK_METHOD2(OnStreamHeaders, | 95 MOCK_METHOD2(OnStreamHeaders, |
95 void(QuicStreamId stream_id, base::StringPiece headers_data)); | 96 void(QuicStreamId stream_id, base::StringPiece headers_data)); |
96 MOCK_METHOD2(OnStreamHeadersPriority, | 97 MOCK_METHOD2(OnStreamHeadersPriority, |
97 void(QuicStreamId stream_id, SpdyPriority priority)); | 98 void(QuicStreamId stream_id, SpdyPriority priority)); |
98 MOCK_METHOD3(OnStreamHeadersComplete, | 99 MOCK_METHOD3(OnStreamHeadersComplete, |
99 void(QuicStreamId stream_id, bool fin, size_t frame_len)); | 100 void(QuicStreamId stream_id, bool fin, size_t frame_len)); |
100 MOCK_METHOD2(OnPromiseHeaders, | 101 MOCK_METHOD2(OnPromiseHeaders, |
101 void(QuicStreamId stream_id, StringPiece headers_data)); | 102 void(QuicStreamId stream_id, StringPiece headers_data)); |
102 MOCK_METHOD3(OnPromiseHeadersComplete, | 103 MOCK_METHOD3(OnPromiseHeadersComplete, |
103 void(QuicStreamId stream_id, | 104 void(QuicStreamId stream_id, |
104 QuicStreamId promised_stream_id, | 105 QuicStreamId promised_stream_id, |
105 size_t frame_len)); | 106 size_t frame_len)); |
106 MOCK_METHOD0(IsCryptoHandshakeConfirmed, bool()); | 107 MOCK_METHOD0(IsCryptoHandshakeConfirmed, bool()); |
107 // Methods taking non-copyable types like SpdyHeaderBlock by value cannot be | 108 // Methods taking non-copyable types like SpdyHeaderBlock by value cannot be |
108 // mocked directly. | 109 // mocked directly. |
109 size_t WriteHeaders( | 110 size_t WriteHeaders(QuicStreamId id, |
110 QuicStreamId id, | 111 SpdyHeaderBlock headers, |
111 SpdyHeaderBlock headers, | 112 bool fin, |
112 bool fin, | 113 SpdyPriority priority, |
113 SpdyPriority priority, | 114 QuicReferenceCountedPointer<QuicAckListenerInterface> |
114 scoped_refptr<QuicAckListenerInterface> ack_notifier_delegate) override { | 115 ack_notifier_delegate) override { |
115 return WriteHeadersMock(id, headers, fin, priority, | 116 return WriteHeadersMock(id, headers, fin, priority, |
116 std::move(ack_notifier_delegate)); | 117 std::move(ack_notifier_delegate)); |
117 } | 118 } |
118 MOCK_METHOD5(WriteHeadersMock, | 119 MOCK_METHOD5( |
119 size_t(QuicStreamId id, | 120 WriteHeadersMock, |
120 const SpdyHeaderBlock& headers, | 121 size_t(QuicStreamId id, |
121 bool fin, | 122 const SpdyHeaderBlock& headers, |
122 SpdyPriority priority, | 123 bool fin, |
123 const scoped_refptr<QuicAckListenerInterface>& | 124 SpdyPriority priority, |
124 ack_notifier_delegate)); | 125 const QuicReferenceCountedPointer<QuicAckListenerInterface>& |
| 126 ack_notifier_delegate)); |
125 MOCK_METHOD1(OnHeadersHeadOfLineBlocking, void(QuicTime::Delta delta)); | 127 MOCK_METHOD1(OnHeadersHeadOfLineBlocking, void(QuicTime::Delta delta)); |
126 | 128 |
127 using QuicSession::ActivateStream; | 129 using QuicSession::ActivateStream; |
128 | 130 |
129 // Returns a QuicConsumedData that indicates all of |data| (and |fin| if set) | 131 // Returns a QuicConsumedData that indicates all of |data| (and |fin| if set) |
130 // has been consumed. | 132 // has been consumed. |
131 static QuicConsumedData ConsumeAllData( | 133 static QuicConsumedData ConsumeAllData( |
132 QuicStreamId id, | 134 QuicStreamId id, |
133 const QuicIOVector& data, | 135 const QuicIOVector& data, |
134 QuicStreamOffset offset, | 136 QuicStreamOffset offset, |
(...skipping 457 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
592 stream->SetDelegate(&delegate_); | 594 stream->SetDelegate(&delegate_); |
593 ProcessHeadersFull(headers_); | 595 ProcessHeadersFull(headers_); |
594 | 596 |
595 // Times(2) because OnClose will be called for stream and stream_. | 597 // Times(2) because OnClose will be called for stream and stream_. |
596 EXPECT_CALL(delegate_, OnClose()).Times(2); | 598 EXPECT_CALL(delegate_, OnClose()).Times(2); |
597 } | 599 } |
598 | 600 |
599 } // namespace | 601 } // namespace |
600 } // namespace test | 602 } // namespace test |
601 } // namespace net | 603 } // namespace net |
OLD | NEW |