| 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 48 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 59 private: | 59 private: |
| 60 DISALLOW_COPY_AND_ASSIGN(MockDelegate); | 60 DISALLOW_COPY_AND_ASSIGN(MockDelegate); |
| 61 }; | 61 }; |
| 62 | 62 |
| 63 class MockQuicClientSessionBase : public QuicClientSessionBase { | 63 class MockQuicClientSessionBase : public QuicClientSessionBase { |
| 64 public: | 64 public: |
| 65 explicit MockQuicClientSessionBase(QuicConnection* connection, | 65 explicit MockQuicClientSessionBase(QuicConnection* connection, |
| 66 QuicClientPushPromiseIndex* index); | 66 QuicClientPushPromiseIndex* index); |
| 67 ~MockQuicClientSessionBase() override; | 67 ~MockQuicClientSessionBase() override; |
| 68 | 68 |
| 69 QuicCryptoStream* GetCryptoStream() override { return crypto_stream_.get(); } | 69 const QuicCryptoStream* GetCryptoStream() const override { |
| 70 return crypto_stream_.get(); |
| 71 } |
| 72 |
| 73 QuicCryptoStream* GetMutableCryptoStream() override { |
| 74 return crypto_stream_.get(); |
| 75 } |
| 70 | 76 |
| 71 // From QuicSession. | 77 // From QuicSession. |
| 72 MOCK_METHOD3(OnConnectionClosed, | 78 MOCK_METHOD3(OnConnectionClosed, |
| 73 void(QuicErrorCode error, | 79 void(QuicErrorCode error, |
| 74 const std::string& error_details, | 80 const std::string& error_details, |
| 75 ConnectionCloseSource source)); | 81 ConnectionCloseSource source)); |
| 76 MOCK_METHOD1(CreateIncomingDynamicStream, QuicSpdyStream*(QuicStreamId id)); | 82 MOCK_METHOD1(CreateIncomingDynamicStream, QuicSpdyStream*(QuicStreamId id)); |
| 77 MOCK_METHOD1(CreateOutgoingDynamicStream, | 83 MOCK_METHOD1(CreateOutgoingDynamicStream, |
| 78 QuicChromiumClientStream*(SpdyPriority priority)); | 84 QuicChromiumClientStream*(SpdyPriority priority)); |
| 79 MOCK_METHOD6( | 85 MOCK_METHOD6( |
| 80 WritevData, | 86 WritevData, |
| 81 QuicConsumedData(QuicStream* stream, | 87 QuicConsumedData(QuicStream* stream, |
| 82 QuicStreamId id, | 88 QuicStreamId id, |
| 83 QuicIOVector data, | 89 QuicIOVector data, |
| 84 QuicStreamOffset offset, | 90 QuicStreamOffset offset, |
| 85 bool fin, | 91 StreamSendingState fin, |
| 86 QuicReferenceCountedPointer<QuicAckListenerInterface>)); | 92 QuicReferenceCountedPointer<QuicAckListenerInterface>)); |
| 87 MOCK_METHOD3(SendRstStream, | 93 MOCK_METHOD3(SendRstStream, |
| 88 void(QuicStreamId stream_id, | 94 void(QuicStreamId stream_id, |
| 89 QuicRstStreamErrorCode error, | 95 QuicRstStreamErrorCode error, |
| 90 QuicStreamOffset bytes_written)); | 96 QuicStreamOffset bytes_written)); |
| 91 | 97 |
| 92 MOCK_METHOD2(OnStreamHeaders, | 98 MOCK_METHOD2(OnStreamHeaders, |
| 93 void(QuicStreamId stream_id, QuicStringPiece headers_data)); | 99 void(QuicStreamId stream_id, QuicStringPiece headers_data)); |
| 94 MOCK_METHOD2(OnStreamHeadersPriority, | 100 MOCK_METHOD2(OnStreamHeadersPriority, |
| 95 void(QuicStreamId stream_id, SpdyPriority priority)); | 101 void(QuicStreamId stream_id, SpdyPriority priority)); |
| 96 MOCK_METHOD3(OnStreamHeadersComplete, | 102 MOCK_METHOD3(OnStreamHeadersComplete, |
| 97 void(QuicStreamId stream_id, bool fin, size_t frame_len)); | 103 void(QuicStreamId stream_id, bool fin, size_t frame_len)); |
| 98 MOCK_METHOD2(OnPromiseHeaders, | 104 MOCK_METHOD2(OnPromiseHeaders, |
| 99 void(QuicStreamId stream_id, QuicStringPiece headers_data)); | 105 void(QuicStreamId stream_id, QuicStringPiece headers_data)); |
| 100 MOCK_METHOD3(OnPromiseHeadersComplete, | 106 MOCK_METHOD3(OnPromiseHeadersComplete, |
| 101 void(QuicStreamId stream_id, | 107 void(QuicStreamId stream_id, |
| 102 QuicStreamId promised_stream_id, | 108 QuicStreamId promised_stream_id, |
| 103 size_t frame_len)); | 109 size_t frame_len)); |
| 104 MOCK_METHOD0(IsCryptoHandshakeConfirmed, bool()); | 110 MOCK_CONST_METHOD0(IsCryptoHandshakeConfirmed, bool()); |
| 105 // Methods taking non-copyable types like SpdyHeaderBlock by value cannot be | 111 // Methods taking non-copyable types like SpdyHeaderBlock by value cannot be |
| 106 // mocked directly. | 112 // mocked directly. |
| 107 size_t WriteHeaders(QuicStreamId id, | 113 size_t WriteHeaders(QuicStreamId id, |
| 108 SpdyHeaderBlock headers, | 114 SpdyHeaderBlock headers, |
| 109 bool fin, | 115 bool fin, |
| 110 SpdyPriority priority, | 116 SpdyPriority priority, |
| 111 QuicReferenceCountedPointer<QuicAckListenerInterface> | 117 QuicReferenceCountedPointer<QuicAckListenerInterface> |
| 112 ack_listener) override { | 118 ack_listener) override { |
| 113 return WriteHeadersMock(id, headers, fin, priority, | 119 return WriteHeadersMock(id, headers, fin, priority, |
| 114 std::move(ack_listener)); | 120 std::move(ack_listener)); |
| (...skipping 479 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 594 stream->SetDelegate(&delegate_); | 600 stream->SetDelegate(&delegate_); |
| 595 ProcessHeadersFull(headers_); | 601 ProcessHeadersFull(headers_); |
| 596 | 602 |
| 597 // Times(2) because OnClose will be called for stream and stream_. | 603 // Times(2) because OnClose will be called for stream and stream_. |
| 598 EXPECT_CALL(delegate_, OnClose()).Times(2); | 604 EXPECT_CALL(delegate_, OnClose()).Times(2); |
| 599 } | 605 } |
| 600 | 606 |
| 601 } // namespace | 607 } // namespace |
| 602 } // namespace test | 608 } // namespace test |
| 603 } // namespace net | 609 } // namespace net |
| OLD | NEW |