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 |