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 32 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
43 public: | 43 public: |
44 TestStreamReader() : buffer_(new net::GrowableIOBuffer()) { | 44 TestStreamReader() : buffer_(new net::GrowableIOBuffer()) { |
45 } | 45 } |
46 virtual ~TestStreamReader() {} | 46 virtual ~TestStreamReader() {} |
47 | 47 |
48 void Read(Stream* stream) { | 48 void Read(Stream* stream) { |
49 const size_t kBufferSize = 32768; | 49 const size_t kBufferSize = 32768; |
50 scoped_refptr<net::IOBuffer> buffer(new net::IOBuffer(kBufferSize)); | 50 scoped_refptr<net::IOBuffer> buffer(new net::IOBuffer(kBufferSize)); |
51 | 51 |
52 int bytes_read = 0; | 52 int bytes_read = 0; |
53 while (stream->ReadRawData(buffer, kBufferSize, &bytes_read) == | 53 while (stream->ReadRawData(buffer.get(), kBufferSize, &bytes_read) == |
54 Stream::STREAM_HAS_DATA) { | 54 Stream::STREAM_HAS_DATA) { |
55 size_t old_capacity = buffer_->capacity(); | 55 size_t old_capacity = buffer_->capacity(); |
56 buffer_->SetCapacity(old_capacity + bytes_read); | 56 buffer_->SetCapacity(old_capacity + bytes_read); |
57 memcpy(buffer_->StartOfBuffer() + old_capacity, | 57 memcpy(buffer_->StartOfBuffer() + old_capacity, |
58 buffer->data(), bytes_read); | 58 buffer->data(), bytes_read); |
59 } | 59 } |
60 } | 60 } |
61 | 61 |
62 virtual void OnDataAvailable(Stream* stream) OVERRIDE { | 62 virtual void OnDataAvailable(Stream* stream) OVERRIDE { |
63 Read(stream); | 63 Read(stream); |
(...skipping 64 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
128 TestStreamReader reader; | 128 TestStreamReader reader; |
129 TestStreamWriter writer; | 129 TestStreamWriter writer; |
130 | 130 |
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.get(), buffer, kBufferSize); |
139 stream->Finalize(); | 139 stream->Finalize(); |
140 reader.Read(stream); | 140 reader.Read(stream.get()); |
141 base::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"); |
152 scoped_refptr<Stream> stream1( | 152 scoped_refptr<Stream> stream1( |
153 new Stream(registry_.get(), &writer, GURL(), url)); | 153 new Stream(registry_.get(), &writer, GURL(), url)); |
154 | 154 |
155 scoped_refptr<Stream> stream2 = registry_->GetStream(url); | 155 scoped_refptr<Stream> stream2 = registry_->GetStream(url); |
156 ASSERT_EQ(stream1, stream2); | 156 ASSERT_EQ(stream1, stream2); |
157 } | 157 } |
158 | 158 |
159 TEST_F(StreamTest, GetStream_Missing) { | 159 TEST_F(StreamTest, GetStream_Missing) { |
160 TestStreamWriter writer; | 160 TestStreamWriter writer; |
161 | 161 |
162 GURL url1("blob://stream"); | 162 GURL url1("blob://stream"); |
163 scoped_refptr<Stream> stream1( | 163 scoped_refptr<Stream> stream1( |
164 new Stream(registry_.get(), &writer, GURL(), url1)); | 164 new Stream(registry_.get(), &writer, GURL(), url1)); |
165 | 165 |
166 GURL url2("blob://stream2"); | 166 GURL url2("blob://stream2"); |
167 scoped_refptr<Stream> stream2 = registry_->GetStream(url2); | 167 scoped_refptr<Stream> stream2 = registry_->GetStream(url2); |
168 ASSERT_FALSE(stream2); | 168 ASSERT_FALSE(stream2.get()); |
169 } | 169 } |
170 | 170 |
171 TEST_F(StreamTest, CloneStream) { | 171 TEST_F(StreamTest, CloneStream) { |
172 TestStreamWriter writer; | 172 TestStreamWriter writer; |
173 | 173 |
174 GURL url1("blob://stream"); | 174 GURL url1("blob://stream"); |
175 scoped_refptr<Stream> stream1( | 175 scoped_refptr<Stream> stream1( |
176 new Stream(registry_.get(), &writer, GURL(), url1)); | 176 new Stream(registry_.get(), &writer, GURL(), url1)); |
177 | 177 |
178 GURL url2("blob://stream2"); | 178 GURL url2("blob://stream2"); |
179 ASSERT_TRUE(registry_->CloneStream(url2, url1)); | 179 ASSERT_TRUE(registry_->CloneStream(url2, url1)); |
180 scoped_refptr<Stream> stream2 = registry_->GetStream(url2); | 180 scoped_refptr<Stream> stream2 = registry_->GetStream(url2); |
181 ASSERT_EQ(stream1, stream2); | 181 ASSERT_EQ(stream1, stream2); |
182 } | 182 } |
183 | 183 |
184 TEST_F(StreamTest, CloneStream_Missing) { | 184 TEST_F(StreamTest, CloneStream_Missing) { |
185 TestStreamWriter writer; | 185 TestStreamWriter writer; |
186 | 186 |
187 GURL url1("blob://stream"); | 187 GURL url1("blob://stream"); |
188 scoped_refptr<Stream> stream1( | 188 scoped_refptr<Stream> stream1( |
189 new Stream(registry_.get(), &writer, GURL(), url1)); | 189 new Stream(registry_.get(), &writer, GURL(), url1)); |
190 | 190 |
191 GURL url2("blob://stream2"); | 191 GURL url2("blob://stream2"); |
192 GURL url3("blob://stream3"); | 192 GURL url3("blob://stream3"); |
193 ASSERT_FALSE(registry_->CloneStream(url2, url3)); | 193 ASSERT_FALSE(registry_->CloneStream(url2, url3)); |
194 scoped_refptr<Stream> stream2 = registry_->GetStream(url2); | 194 scoped_refptr<Stream> stream2 = registry_->GetStream(url2); |
195 ASSERT_FALSE(stream2); | 195 ASSERT_FALSE(stream2.get()); |
196 } | 196 } |
197 | 197 |
198 TEST_F(StreamTest, UnregisterStream) { | 198 TEST_F(StreamTest, UnregisterStream) { |
199 TestStreamWriter writer; | 199 TestStreamWriter writer; |
200 | 200 |
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.get()); |
208 } | 208 } |
209 | 209 |
210 } // namespace content | 210 } // namespace content |
OLD | NEW |