| OLD | NEW |
| 1 // Copyright (c) 2013 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2013 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "base/message_loop/message_loop.h" | 5 #include "base/message_loop/message_loop.h" |
| 6 #include "base/test/test_simple_task_runner.h" | 6 #include "base/test/test_simple_task_runner.h" |
| 7 #include "content/browser/streams/stream.h" | 7 #include "content/browser/streams/stream.h" |
| 8 #include "content/browser/streams/stream_read_observer.h" | 8 #include "content/browser/streams/stream_read_observer.h" |
| 9 #include "content/browser/streams/stream_register_observer.h" | 9 #include "content/browser/streams/stream_register_observer.h" |
| 10 #include "content/browser/streams/stream_registry.h" | 10 #include "content/browser/streams/stream_registry.h" |
| (...skipping 93 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 104 void OnClose(Stream* stream) override {} | 104 void OnClose(Stream* stream) override {} |
| 105 }; | 105 }; |
| 106 | 106 |
| 107 class TestStreamObserver : public StreamRegisterObserver { | 107 class TestStreamObserver : public StreamRegisterObserver { |
| 108 public: | 108 public: |
| 109 TestStreamObserver(const GURL& url, StreamRegistry* registry) | 109 TestStreamObserver(const GURL& url, StreamRegistry* registry) |
| 110 : url_(url), registry_(registry), registered_(false), stream_(nullptr) { | 110 : url_(url), registry_(registry), registered_(false), stream_(nullptr) { |
| 111 registry->SetRegisterObserver(url, this); | 111 registry->SetRegisterObserver(url, this); |
| 112 } | 112 } |
| 113 ~TestStreamObserver() override { registry_->RemoveRegisterObserver(url_); } | 113 ~TestStreamObserver() override { registry_->RemoveRegisterObserver(url_); } |
| 114 virtual void OnStreamRegistered(Stream* stream) override { | 114 void OnStreamRegistered(Stream* stream) override { |
| 115 registered_ = true; | 115 registered_ = true; |
| 116 stream_ = stream; | 116 stream_ = stream; |
| 117 } | 117 } |
| 118 bool registered() const { return registered_; } | 118 bool registered() const { return registered_; } |
| 119 Stream* stream() const { return stream_; } | 119 Stream* stream() const { return stream_; } |
| 120 | 120 |
| 121 private: | 121 private: |
| 122 const GURL url_; | 122 const GURL url_; |
| 123 StreamRegistry* registry_; | 123 StreamRegistry* registry_; |
| 124 bool registered_; | 124 bool registered_; |
| (...skipping 248 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 373 TEST_F(StreamTest, AbortPendingStream) { | 373 TEST_F(StreamTest, AbortPendingStream) { |
| 374 TestStreamWriter writer; | 374 TestStreamWriter writer; |
| 375 | 375 |
| 376 GURL url("blob://stream"); | 376 GURL url("blob://stream"); |
| 377 registry_->AbortPendingStream(url); | 377 registry_->AbortPendingStream(url); |
| 378 scoped_refptr<Stream> stream1(new Stream(registry_.get(), &writer, url)); | 378 scoped_refptr<Stream> stream1(new Stream(registry_.get(), &writer, url)); |
| 379 ASSERT_EQ(nullptr, registry_->GetStream(url).get()); | 379 ASSERT_EQ(nullptr, registry_->GetStream(url).get()); |
| 380 } | 380 } |
| 381 | 381 |
| 382 } // namespace content | 382 } // namespace content |
| OLD | NEW |