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 |