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/quic_reliable_client_stream.h" | 5 #include "net/quic/quic_reliable_client_stream.h" |
6 | 6 |
7 #include "net/base/net_errors.h" | 7 #include "net/base/net_errors.h" |
8 #include "net/base/test_completion_callback.h" | 8 #include "net/base/test_completion_callback.h" |
9 #include "net/quic/quic_client_session.h" | 9 #include "net/quic/quic_client_session.h" |
10 #include "net/quic/quic_utils.h" | 10 #include "net/quic/quic_utils.h" |
(...skipping 119 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
130 EXPECT_FALSE(stream_->GetDelegate()); | 130 EXPECT_FALSE(stream_->GetDelegate()); |
131 } | 131 } |
132 | 132 |
133 TEST_P(QuicReliableClientStreamTest, WriteStreamData) { | 133 TEST_P(QuicReliableClientStreamTest, WriteStreamData) { |
134 EXPECT_CALL(delegate_, OnClose(QUIC_NO_ERROR)); | 134 EXPECT_CALL(delegate_, OnClose(QUIC_NO_ERROR)); |
135 | 135 |
136 const char kData1[] = "hello world"; | 136 const char kData1[] = "hello world"; |
137 const size_t kDataLen = arraysize(kData1); | 137 const size_t kDataLen = arraysize(kData1); |
138 | 138 |
139 // All data written. | 139 // All data written. |
140 EXPECT_CALL(session_, WritevData(stream_->id(), _, _, _, _)).WillOnce( | 140 EXPECT_CALL(session_, WritevData(stream_->id(), _, _, _, _, _)).WillOnce( |
141 Return(QuicConsumedData(kDataLen, true))); | 141 Return(QuicConsumedData(kDataLen, true))); |
142 TestCompletionCallback callback; | 142 TestCompletionCallback callback; |
143 EXPECT_EQ(OK, stream_->WriteStreamData(base::StringPiece(kData1, kDataLen), | 143 EXPECT_EQ(OK, stream_->WriteStreamData(base::StringPiece(kData1, kDataLen), |
144 true, callback.callback())); | 144 true, callback.callback())); |
145 } | 145 } |
146 | 146 |
147 TEST_P(QuicReliableClientStreamTest, WriteStreamDataAsync) { | 147 TEST_P(QuicReliableClientStreamTest, WriteStreamDataAsync) { |
148 EXPECT_CALL(delegate_, HasSendHeadersComplete()).Times(AnyNumber()); | 148 EXPECT_CALL(delegate_, HasSendHeadersComplete()).Times(AnyNumber()); |
149 EXPECT_CALL(delegate_, OnClose(QUIC_NO_ERROR)); | 149 EXPECT_CALL(delegate_, OnClose(QUIC_NO_ERROR)); |
150 | 150 |
151 const char kData1[] = "hello world"; | 151 const char kData1[] = "hello world"; |
152 const size_t kDataLen = arraysize(kData1); | 152 const size_t kDataLen = arraysize(kData1); |
153 | 153 |
154 // No data written. | 154 // No data written. |
155 EXPECT_CALL(session_, WritevData(stream_->id(), _, _, _, _)).WillOnce( | 155 EXPECT_CALL(session_, WritevData(stream_->id(), _, _, _, _, _)).WillOnce( |
156 Return(QuicConsumedData(0, false))); | 156 Return(QuicConsumedData(0, false))); |
157 TestCompletionCallback callback; | 157 TestCompletionCallback callback; |
158 EXPECT_EQ(ERR_IO_PENDING, | 158 EXPECT_EQ(ERR_IO_PENDING, |
159 stream_->WriteStreamData(base::StringPiece(kData1, kDataLen), | 159 stream_->WriteStreamData(base::StringPiece(kData1, kDataLen), |
160 true, callback.callback())); | 160 true, callback.callback())); |
161 ASSERT_FALSE(callback.have_result()); | 161 ASSERT_FALSE(callback.have_result()); |
162 | 162 |
163 // All data written. | 163 // All data written. |
164 EXPECT_CALL(session_, WritevData(stream_->id(), _, _, _, _)).WillOnce( | 164 EXPECT_CALL(session_, WritevData(stream_->id(), _, _, _, _, _)).WillOnce( |
165 Return(QuicConsumedData(kDataLen, true))); | 165 Return(QuicConsumedData(kDataLen, true))); |
166 stream_->OnCanWrite(); | 166 stream_->OnCanWrite(); |
167 ASSERT_TRUE(callback.have_result()); | 167 ASSERT_TRUE(callback.have_result()); |
168 EXPECT_EQ(OK, callback.WaitForResult()); | 168 EXPECT_EQ(OK, callback.WaitForResult()); |
169 } | 169 } |
170 | 170 |
171 } // namespace | 171 } // namespace |
172 } // namespace test | 172 } // namespace test |
173 } // namespace net | 173 } // namespace net |
OLD | NEW |