| Index: webrtc/p2p/quic/reliablequicstream_unittest.cc
|
| diff --git a/webrtc/p2p/quic/reliablequicstream_unittest.cc b/webrtc/p2p/quic/reliablequicstream_unittest.cc
|
| index aeb3e1af87ce1f4049100f9205bd42d61c448eab..b8875894576a0f1836cdc84ca8c6df57bacca6c9 100644
|
| --- a/webrtc/p2p/quic/reliablequicstream_unittest.cc
|
| +++ b/webrtc/p2p/quic/reliablequicstream_unittest.cc
|
| @@ -163,6 +163,8 @@ class ReliableQuicStreamTest : public ::testing::Test,
|
| stream_->SignalDataReceived.connect(
|
| this, &ReliableQuicStreamTest::OnDataReceived);
|
| stream_->SignalClosed.connect(this, &ReliableQuicStreamTest::OnClosed);
|
| + stream_->SignalQueuedBytesWritten.connect(
|
| + this, &ReliableQuicStreamTest::OnQueuedBytesWritten);
|
|
|
| session_->register_write_blocked_stream(stream_->id(), kDefaultPriority);
|
| }
|
| @@ -172,7 +174,11 @@ class ReliableQuicStreamTest : public ::testing::Test,
|
| read_buffer_.append(data, length);
|
| }
|
|
|
| - void OnClosed(QuicStreamId id, QuicErrorCode err) { closed_ = true; }
|
| + void OnClosed(QuicStreamId id, int err) { closed_ = true; }
|
| +
|
| + void OnQueuedBytesWritten(QuicStreamId id, uint64_t queued_bytes_written) {
|
| + queued_bytes_written_ = queued_bytes_written;
|
| + }
|
|
|
| protected:
|
| rtc::scoped_ptr<ReliableQuicStream> stream_;
|
| @@ -184,6 +190,8 @@ class ReliableQuicStreamTest : public ::testing::Test,
|
| std::string read_buffer_;
|
| // Whether the ReliableQuicStream is closed.
|
| bool closed_ = false;
|
| + // Bytes written by OnCanWrite().
|
| + uint64_t queued_bytes_written_;
|
| };
|
|
|
| // Write an entire string.
|
| @@ -213,6 +221,7 @@ TEST_F(ReliableQuicStreamTest, BufferData) {
|
|
|
| session_->set_writable(true);
|
| stream_->OnCanWrite();
|
| + EXPECT_EQ(7ul, queued_bytes_written_);
|
|
|
| EXPECT_FALSE(stream_->HasBufferedData());
|
| EXPECT_EQ("Foo bar", write_buffer_);
|
|
|