Introduce StreamRegistry::NotifyReaderAbortedBeforeRegistration()

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_register_observer.h"
 #include "content/browser/streams/stream_registry.h"
@@ skipping 89 matching lines @@
   }
 
   void OnSpaceAvailable(Stream* stream) override {}
 
   void OnClose(Stream* stream) override {}
 };
 
 class TestStreamObserver : public StreamRegisterObserver {
  public:
   TestStreamObserver(const GURL& url, StreamRegistry* registry)
-      : url_(url), registry_(registry), registered_(false), stream_(NULL) {
+      : url_(url), registry_(registry), registered_(false), stream_(nullptr) {
     registry->SetRegisterObserver(url, this);
   }
   ~TestStreamObserver() override { registry_->RemoveRegisterObserver(url_); }
   virtual void OnStreamRegistered(Stream* stream) override {
     registered_ = true;
     stream_ = stream;
   }
   bool registered() const { return registered_; }
   Stream* stream() const { return stream_; }
 
@@ skipping 184 matching lines @@
   const int kBufferSize = 1000000;
   scoped_refptr<net::IOBuffer> buffer(NewIOBuffer(kBufferSize));
   writer1.Write(stream1.get(), buffer, kBufferSize);
   // Make transfer happen.
   base::MessageLoop::current()->RunUntilIdle();
 
   writer2.Write(stream2.get(), buffer, kBufferSize);
 
   // Written data (1000000 * 2) exceeded limit (1500000). |stream2| should be
   // unregistered with |registry_|.
-  EXPECT_EQ(NULL, registry_->GetStream(url2).get());
+  EXPECT_EQ(nullptr, registry_->GetStream(url2).get());
 
   writer1.Write(stream1.get(), buffer, kMaxMemoryUsage - kBufferSize);
   // Should be accepted since stream2 is unregistered and the new data is not
   // so big to exceed the limit.
-  EXPECT_FALSE(registry_->GetStream(url1).get() == NULL);
+  EXPECT_FALSE(registry_->GetStream(url1).get() == nullptr);
 }
 
 TEST_F(StreamTest, UnderMemoryUsageLimit) {
   TestStreamWriter writer;
   TestStreamReader reader;
 
   GURL url("blob://stream");
   scoped_refptr<Stream> stream(new Stream(registry_.get(), &writer, url));
   EXPECT_TRUE(stream->SetReadObserver(&reader));
 
@@ skipping 32 matching lines @@
 
   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());
 }
 
+TEST_F(StreamTest, AbortPendingStream) {
+  TestStreamWriter writer;
+
+  GURL url("blob://stream");
+  registry_->AbortPendingStream(url);
+  scoped_refptr<Stream> stream1(new Stream(registry_.get(), &writer, url));
+  ASSERT_EQ(nullptr, registry_->GetStream(url).get());
+}
+
 }  // namespace content