Index: media/blink/buffered_data_source_unittest.cc |
diff --git a/media/blink/buffered_data_source_unittest.cc b/media/blink/buffered_data_source_unittest.cc |
index aed8453eba0dc1020438c91ed5aa6105839474e3..fb97030506f1a0f554a2eed53b8bff097cb2860b 100644 |
--- a/media/blink/buffered_data_source_unittest.cc |
+++ b/media/blink/buffered_data_source_unittest.cc |
@@ -118,7 +118,8 @@ class BufferedDataSourceTest : public testing::Test { |
: view_(WebView::create(NULL)), |
frame_( |
WebLocalFrame::create(blink::WebTreeScopeType::Document, &client_)), |
- preload_(BufferedDataSource::AUTO) { |
+ preload_(BufferedDataSource::AUTO), |
+ bytes_received_(0) { |
view_->setMainFrame(frame_); |
} |
@@ -197,6 +198,7 @@ class BufferedDataSourceTest : public testing::Test { |
.WillOnce(Invoke(data_source_.get(), |
&MockBufferedDataSource::CreateMockResourceLoader)); |
message_loop_.RunUntilIdle(); |
+ bytes_received_ = 0; |
} |
void Respond(const WebURLResponse& response) { |
@@ -210,6 +212,8 @@ class BufferedDataSourceTest : public testing::Test { |
loader()->didReceiveData(url_loader(), data.get(), size, size); |
message_loop_.RunUntilIdle(); |
+ bytes_received_ += size; |
+ EXPECT_EQ(bytes_received_, data_source_->GetMemoryUsage()); |
} |
void FinishLoading() { |
@@ -299,6 +303,7 @@ class BufferedDataSourceTest : public testing::Test { |
StrictMock<MockBufferedDataSourceHost> host_; |
base::MessageLoop message_loop_; |
+ int bytes_received_; |
private: |
// Used for calling BufferedDataSource::Read(). |
@@ -481,6 +486,7 @@ TEST_F(BufferedDataSourceTest, Http_RetryOnError) { |
DoAll(InvokeWithoutArgs(&run_loop, &base::RunLoop::Quit), |
Invoke(data_source_.get(), |
&MockBufferedDataSource::CreateMockResourceLoader))); |
+ bytes_received_ = 0; |
loader()->didFail(url_loader(), response_generator_->GenerateError()); |
run_loop.Run(); |
Respond(response_generator_->Generate206(kDataSize)); |