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( |
(...skipping 14 matching lines...) Expand all Loading... |
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 |