Introduce Stream::flush [2/2 chromium]

content/browser/streams/stream_unittest.cc

Index: content/browser/streams/stream_unittest.cc
diff --git a/content/browser/streams/stream_unittest.cc b/content/browser/streams/stream_unittest.cc
index e0346e5a76e9525f8376c9dc433a3cacebd0599d..d601aabc08fa15c2ee167c48a93a98a012f2f1c0 100644
--- a/content/browser/streams/stream_unittest.cc
+++ b/content/browser/streams/stream_unittest.cc
@@ -302,4 +302,32 @@ TEST_F(StreamTest, UnderMemoryUsageLimit) {
   EXPECT_EQ(stream.get(), registry_->GetStream(url).get());
 }
 
+TEST_F(StreamTest, Flush) {
+  TestStreamWriter writer;
+  TestStreamReader reader;
+
+  GURL url("blob://stream");
+  scoped_refptr<Stream> stream(new Stream(registry_.get(), &writer, url));
+  EXPECT_TRUE(stream->SetReadObserver(&reader));
+
+  // If the written data size is smaller than ByteStreamWriter's (total size /
+  // kFractionBufferBeforeSending), StreamReadObserver::OnDataAvailable is not
+  // called.
+  const int kBufferSize = 1;
+  scoped_refptr<net::IOBuffer> buffer(NewIOBuffer(kBufferSize));
+  writer.Write(stream.get(), buffer, kBufferSize);
+
+  // Run loop to make |reader| consume the data.
+  base::MessageLoop::current()->RunUntilIdle();
+  EXPECT_EQ(0, reader.buffer()->capacity());
+
+  stream->Flush();
+
+  // Run loop to make |reader| consume the data.
+  base::MessageLoop::current()->RunUntilIdle();
+  EXPECT_EQ(kBufferSize, reader.buffer()->capacity());
+
+  EXPECT_EQ(stream.get(), registry_->GetStream(url).get());
+}
+
 }  // namespace content