Introduce Stream::flush [2/2 chromium]

content/browser/streams/stream_unittest.cc

 // Copyright (c) 2013 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "base/message_loop/message_loop.h"
 #include "base/test/test_simple_task_runner.h"
 #include "content/browser/streams/stream.h"
 #include "content/browser/streams/stream_read_observer.h"
 #include "content/browser/streams/stream_registry.h"
 #include "content/browser/streams/stream_write_observer.h"
@@ skipping 284 matching lines @@
   writer.Write(stream.get(), buffer, kBufferSize);
 
   // Run loop to make |reader| consume the data.
   base::MessageLoop::current()->RunUntilIdle();
 
   writer.Write(stream.get(), buffer, kBufferSize);
 
   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

[tyoshino (SeeGerritForStatus), 2014/11/26 05:59:49]

period

[horo, 2014/11/26 06:06:29]

Done.

+  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