| 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 <stddef.h> | 5 #include <stddef.h> |
| 6 | 6 |
| 7 #include "base/message_loop/message_loop.h" | 7 #include "base/message_loop/message_loop.h" |
| 8 #include "base/test/test_simple_task_runner.h" | 8 #include "base/test/test_simple_task_runner.h" |
| 9 #include "content/browser/streams/stream.h" | 9 #include "content/browser/streams/stream.h" |
| 10 #include "content/browser/streams/stream_read_observer.h" | 10 #include "content/browser/streams/stream_read_observer.h" |
| (...skipping 16 matching lines...) Expand all Loading... |
| 27 scoped_refptr<net::IOBuffer> buffer(new net::IOBuffer(buffer_size)); | 27 scoped_refptr<net::IOBuffer> buffer(new net::IOBuffer(buffer_size)); |
| 28 char *bufferp = buffer->data(); | 28 char *bufferp = buffer->data(); |
| 29 for (size_t i = 0; i < buffer_size; i++) | 29 for (size_t i = 0; i < buffer_size; i++) |
| 30 bufferp[i] = (i + producing_seed_key_) % (1 << sizeof(char)); | 30 bufferp[i] = (i + producing_seed_key_) % (1 << sizeof(char)); |
| 31 ++producing_seed_key_; | 31 ++producing_seed_key_; |
| 32 return buffer; | 32 return buffer; |
| 33 } | 33 } |
| 34 | 34 |
| 35 protected: | 35 protected: |
| 36 base::MessageLoop message_loop_; | 36 base::MessageLoop message_loop_; |
| 37 scoped_ptr<StreamRegistry> registry_; | 37 std::unique_ptr<StreamRegistry> registry_; |
| 38 | 38 |
| 39 private: | 39 private: |
| 40 int producing_seed_key_; | 40 int producing_seed_key_; |
| 41 }; | 41 }; |
| 42 | 42 |
| 43 class TestStreamReader : public StreamReadObserver { | 43 class TestStreamReader : public StreamReadObserver { |
| 44 public: | 44 public: |
| 45 TestStreamReader() : buffer_(new net::GrowableIOBuffer()), completed_(false) { | 45 TestStreamReader() : buffer_(new net::GrowableIOBuffer()), completed_(false) { |
| 46 } | 46 } |
| 47 ~TestStreamReader() override {} | 47 ~TestStreamReader() override {} |
| (...skipping 77 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 125 StreamRegistry* registry_; | 125 StreamRegistry* registry_; |
| 126 bool registered_; | 126 bool registered_; |
| 127 Stream* stream_; | 127 Stream* stream_; |
| 128 }; | 128 }; |
| 129 | 129 |
| 130 TEST_F(StreamTest, SetAndRemoveRegisterObserver) { | 130 TEST_F(StreamTest, SetAndRemoveRegisterObserver) { |
| 131 TestStreamWriter writer1; | 131 TestStreamWriter writer1; |
| 132 TestStreamWriter writer2; | 132 TestStreamWriter writer2; |
| 133 GURL url1("blob://stream1"); | 133 GURL url1("blob://stream1"); |
| 134 GURL url2("blob://stream2"); | 134 GURL url2("blob://stream2"); |
| 135 scoped_ptr<TestStreamObserver> observer1( | 135 std::unique_ptr<TestStreamObserver> observer1( |
| 136 new TestStreamObserver(url1, registry_.get())); | 136 new TestStreamObserver(url1, registry_.get())); |
| 137 scoped_ptr<TestStreamObserver> observer2( | 137 std::unique_ptr<TestStreamObserver> observer2( |
| 138 new TestStreamObserver(url2, registry_.get())); | 138 new TestStreamObserver(url2, registry_.get())); |
| 139 scoped_refptr<Stream> stream1(new Stream(registry_.get(), &writer1, url1)); | 139 scoped_refptr<Stream> stream1(new Stream(registry_.get(), &writer1, url1)); |
| 140 EXPECT_TRUE(observer1->registered()); | 140 EXPECT_TRUE(observer1->registered()); |
| 141 EXPECT_EQ(observer1->stream(), stream1.get()); | 141 EXPECT_EQ(observer1->stream(), stream1.get()); |
| 142 EXPECT_FALSE(observer2->registered()); | 142 EXPECT_FALSE(observer2->registered()); |
| 143 | 143 |
| 144 observer2.reset(); | 144 observer2.reset(); |
| 145 scoped_refptr<Stream> stream2(new Stream(registry_.get(), &writer2, url2)); | 145 scoped_refptr<Stream> stream2(new Stream(registry_.get(), &writer2, url2)); |
| 146 } | 146 } |
| 147 | 147 |
| (...skipping 227 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 375 TEST_F(StreamTest, AbortPendingStream) { | 375 TEST_F(StreamTest, AbortPendingStream) { |
| 376 TestStreamWriter writer; | 376 TestStreamWriter writer; |
| 377 | 377 |
| 378 GURL url("blob://stream"); | 378 GURL url("blob://stream"); |
| 379 registry_->AbortPendingStream(url); | 379 registry_->AbortPendingStream(url); |
| 380 scoped_refptr<Stream> stream1(new Stream(registry_.get(), &writer, url)); | 380 scoped_refptr<Stream> stream1(new Stream(registry_.get(), &writer, url)); |
| 381 ASSERT_EQ(nullptr, registry_->GetStream(url).get()); | 381 ASSERT_EQ(nullptr, registry_->GetStream(url).get()); |
| 382 } | 382 } |
| 383 | 383 |
| 384 } // namespace content | 384 } // namespace content |
| OLD | NEW |