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.h" | 5 #include "base/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_registry.h" | 9 #include "content/browser/streams/stream_registry.h" |
10 #include "content/browser/streams/stream_write_observer.h" | 10 #include "content/browser/streams/stream_write_observer.h" |
(...skipping 14 matching lines...) Expand all Loading... |
25 scoped_refptr<net::IOBuffer> NewIOBuffer(size_t buffer_size) { | 25 scoped_refptr<net::IOBuffer> NewIOBuffer(size_t buffer_size) { |
26 scoped_refptr<net::IOBuffer> buffer(new net::IOBuffer(buffer_size)); | 26 scoped_refptr<net::IOBuffer> buffer(new net::IOBuffer(buffer_size)); |
27 char *bufferp = buffer->data(); | 27 char *bufferp = buffer->data(); |
28 for (size_t i = 0; i < buffer_size; i++) | 28 for (size_t i = 0; i < buffer_size; i++) |
29 bufferp[i] = (i + producing_seed_key_) % (1 << sizeof(char)); | 29 bufferp[i] = (i + producing_seed_key_) % (1 << sizeof(char)); |
30 ++producing_seed_key_; | 30 ++producing_seed_key_; |
31 return buffer; | 31 return buffer; |
32 } | 32 } |
33 | 33 |
34 protected: | 34 protected: |
35 MessageLoop message_loop_; | 35 base::MessageLoop message_loop_; |
36 scoped_ptr<StreamRegistry> registry_; | 36 scoped_ptr<StreamRegistry> registry_; |
37 | 37 |
38 private: | 38 private: |
39 int producing_seed_key_; | 39 int producing_seed_key_; |
40 }; | 40 }; |
41 | 41 |
42 class TestStreamReader : public StreamReadObserver { | 42 class TestStreamReader : public StreamReadObserver { |
43 public: | 43 public: |
44 TestStreamReader() : buffer_(new net::GrowableIOBuffer()) { | 44 TestStreamReader() : buffer_(new net::GrowableIOBuffer()) { |
45 } | 45 } |
(...skipping 85 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
131 GURL url("blob://stream"); | 131 GURL url("blob://stream"); |
132 scoped_refptr<Stream> stream( | 132 scoped_refptr<Stream> stream( |
133 new Stream(registry_.get(), &writer, GURL(), url)); | 133 new Stream(registry_.get(), &writer, GURL(), url)); |
134 EXPECT_TRUE(stream->SetReadObserver(&reader)); | 134 EXPECT_TRUE(stream->SetReadObserver(&reader)); |
135 | 135 |
136 const int kBufferSize = 1000000; | 136 const int kBufferSize = 1000000; |
137 scoped_refptr<net::IOBuffer> buffer(NewIOBuffer(kBufferSize)); | 137 scoped_refptr<net::IOBuffer> buffer(NewIOBuffer(kBufferSize)); |
138 writer.Write(stream, buffer, kBufferSize); | 138 writer.Write(stream, buffer, kBufferSize); |
139 stream->Finalize(); | 139 stream->Finalize(); |
140 reader.Read(stream); | 140 reader.Read(stream); |
141 MessageLoop::current()->RunUntilIdle(); | 141 base::MessageLoop::current()->RunUntilIdle(); |
142 | 142 |
143 ASSERT_EQ(reader.buffer()->capacity(), kBufferSize); | 143 ASSERT_EQ(reader.buffer()->capacity(), kBufferSize); |
144 for (int i = 0; i < kBufferSize; i++) | 144 for (int i = 0; i < kBufferSize; i++) |
145 EXPECT_EQ(buffer->data()[i], reader.buffer()->data()[i]); | 145 EXPECT_EQ(buffer->data()[i], reader.buffer()->data()[i]); |
146 } | 146 } |
147 | 147 |
148 TEST_F(StreamTest, GetStream) { | 148 TEST_F(StreamTest, GetStream) { |
149 TestStreamWriter writer; | 149 TestStreamWriter writer; |
150 | 150 |
151 GURL url("blob://stream"); | 151 GURL url("blob://stream"); |
(...skipping 49 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
201 GURL url("blob://stream"); | 201 GURL url("blob://stream"); |
202 scoped_refptr<Stream> stream1( | 202 scoped_refptr<Stream> stream1( |
203 new Stream(registry_.get(), &writer, GURL(), url)); | 203 new Stream(registry_.get(), &writer, GURL(), url)); |
204 | 204 |
205 registry_->UnregisterStream(url); | 205 registry_->UnregisterStream(url); |
206 scoped_refptr<Stream> stream2 = registry_->GetStream(url); | 206 scoped_refptr<Stream> stream2 = registry_->GetStream(url); |
207 ASSERT_FALSE(stream2); | 207 ASSERT_FALSE(stream2); |
208 } | 208 } |
209 | 209 |
210 } // namespace content | 210 } // namespace content |
OLD | NEW |