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_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 25 matching lines...) Expand all Loading... |
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()), completed_(false) { | 44 TestStreamReader() : buffer_(new net::GrowableIOBuffer()), completed_(false) { |
45 } | 45 } |
46 virtual ~TestStreamReader() {} | 46 ~TestStreamReader() override {} |
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 (true) { | 53 while (true) { |
54 Stream::StreamState state = | 54 Stream::StreamState state = |
55 stream->ReadRawData(buffer.get(), kBufferSize, &bytes_read); | 55 stream->ReadRawData(buffer.get(), kBufferSize, &bytes_read); |
56 switch (state) { | 56 switch (state) { |
(...skipping 12 matching lines...) Expand all Loading... |
69 EXPECT_FALSE(completed_); | 69 EXPECT_FALSE(completed_); |
70 return; | 70 return; |
71 } | 71 } |
72 size_t old_capacity = buffer_->capacity(); | 72 size_t old_capacity = buffer_->capacity(); |
73 buffer_->SetCapacity(old_capacity + bytes_read); | 73 buffer_->SetCapacity(old_capacity + bytes_read); |
74 memcpy(buffer_->StartOfBuffer() + old_capacity, | 74 memcpy(buffer_->StartOfBuffer() + old_capacity, |
75 buffer->data(), bytes_read); | 75 buffer->data(), bytes_read); |
76 } | 76 } |
77 } | 77 } |
78 | 78 |
79 virtual void OnDataAvailable(Stream* stream) override { | 79 void OnDataAvailable(Stream* stream) override { Read(stream); } |
80 Read(stream); | |
81 } | |
82 | 80 |
83 scoped_refptr<net::GrowableIOBuffer> buffer() { return buffer_; } | 81 scoped_refptr<net::GrowableIOBuffer> buffer() { return buffer_; } |
84 | 82 |
85 bool completed() const { | 83 bool completed() const { |
86 return completed_; | 84 return completed_; |
87 } | 85 } |
88 | 86 |
89 private: | 87 private: |
90 scoped_refptr<net::GrowableIOBuffer> buffer_; | 88 scoped_refptr<net::GrowableIOBuffer> buffer_; |
91 bool completed_; | 89 bool completed_; |
92 }; | 90 }; |
93 | 91 |
94 class TestStreamWriter : public StreamWriteObserver { | 92 class TestStreamWriter : public StreamWriteObserver { |
95 public: | 93 public: |
96 TestStreamWriter() {} | 94 TestStreamWriter() {} |
97 virtual ~TestStreamWriter() {} | 95 ~TestStreamWriter() override {} |
98 | 96 |
99 void Write(Stream* stream, | 97 void Write(Stream* stream, |
100 scoped_refptr<net::IOBuffer> buffer, | 98 scoped_refptr<net::IOBuffer> buffer, |
101 size_t buffer_size) { | 99 size_t buffer_size) { |
102 stream->AddData(buffer, buffer_size); | 100 stream->AddData(buffer, buffer_size); |
103 } | 101 } |
104 | 102 |
105 virtual void OnSpaceAvailable(Stream* stream) override { | 103 void OnSpaceAvailable(Stream* stream) override {} |
106 } | |
107 | 104 |
108 virtual void OnClose(Stream* stream) override { | 105 void OnClose(Stream* stream) override {} |
109 } | |
110 }; | 106 }; |
111 | 107 |
112 TEST_F(StreamTest, SetReadObserver) { | 108 TEST_F(StreamTest, SetReadObserver) { |
113 TestStreamReader reader; | 109 TestStreamReader reader; |
114 TestStreamWriter writer; | 110 TestStreamWriter writer; |
115 | 111 |
116 GURL url("blob://stream"); | 112 GURL url("blob://stream"); |
117 scoped_refptr<Stream> stream( | 113 scoped_refptr<Stream> stream( |
118 new Stream(registry_.get(), &writer, url)); | 114 new Stream(registry_.get(), &writer, url)); |
119 EXPECT_TRUE(stream->SetReadObserver(&reader)); | 115 EXPECT_TRUE(stream->SetReadObserver(&reader)); |
(...skipping 182 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
302 | 298 |
303 // Run loop to make |reader| consume the data. | 299 // Run loop to make |reader| consume the data. |
304 base::MessageLoop::current()->RunUntilIdle(); | 300 base::MessageLoop::current()->RunUntilIdle(); |
305 | 301 |
306 writer.Write(stream.get(), buffer, kBufferSize); | 302 writer.Write(stream.get(), buffer, kBufferSize); |
307 | 303 |
308 EXPECT_EQ(stream.get(), registry_->GetStream(url).get()); | 304 EXPECT_EQ(stream.get(), registry_->GetStream(url).get()); |
309 } | 305 } |
310 | 306 |
311 } // namespace content | 307 } // namespace content |
OLD | NEW |