| 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 75 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 86 virtual void OnClose(Stream* stream) OVERRIDE { | 86 virtual void OnClose(Stream* stream) OVERRIDE { |
| 87 } | 87 } |
| 88 }; | 88 }; |
| 89 | 89 |
| 90 TEST_F(StreamTest, SetReadObserver) { | 90 TEST_F(StreamTest, SetReadObserver) { |
| 91 TestStreamReader reader; | 91 TestStreamReader reader; |
| 92 TestStreamWriter writer; | 92 TestStreamWriter writer; |
| 93 | 93 |
| 94 GURL url("blob://stream"); | 94 GURL url("blob://stream"); |
| 95 scoped_refptr<Stream> stream( | 95 scoped_refptr<Stream> stream( |
| 96 new Stream(registry_.get(), &writer, GURL(), url)); | 96 new Stream(registry_.get(), &writer, url)); |
| 97 EXPECT_TRUE(stream->SetReadObserver(&reader)); | 97 EXPECT_TRUE(stream->SetReadObserver(&reader)); |
| 98 } | 98 } |
| 99 | 99 |
| 100 TEST_F(StreamTest, SetReadObserver_SecondFails) { | 100 TEST_F(StreamTest, SetReadObserver_SecondFails) { |
| 101 TestStreamReader reader1; | 101 TestStreamReader reader1; |
| 102 TestStreamReader reader2; | 102 TestStreamReader reader2; |
| 103 TestStreamWriter writer; | 103 TestStreamWriter writer; |
| 104 | 104 |
| 105 GURL url("blob://stream"); | 105 GURL url("blob://stream"); |
| 106 scoped_refptr<Stream> stream( | 106 scoped_refptr<Stream> stream( |
| 107 new Stream(registry_.get(), &writer, GURL(), url)); | 107 new Stream(registry_.get(), &writer, url)); |
| 108 EXPECT_TRUE(stream->SetReadObserver(&reader1)); | 108 EXPECT_TRUE(stream->SetReadObserver(&reader1)); |
| 109 EXPECT_FALSE(stream->SetReadObserver(&reader2)); | 109 EXPECT_FALSE(stream->SetReadObserver(&reader2)); |
| 110 } | 110 } |
| 111 | 111 |
| 112 TEST_F(StreamTest, SetReadObserver_TwoReaders) { | 112 TEST_F(StreamTest, SetReadObserver_TwoReaders) { |
| 113 TestStreamReader reader1; | 113 TestStreamReader reader1; |
| 114 TestStreamReader reader2; | 114 TestStreamReader reader2; |
| 115 TestStreamWriter writer; | 115 TestStreamWriter writer; |
| 116 | 116 |
| 117 GURL url("blob://stream"); | 117 GURL url("blob://stream"); |
| 118 scoped_refptr<Stream> stream( | 118 scoped_refptr<Stream> stream( |
| 119 new Stream(registry_.get(), &writer, GURL(), url)); | 119 new Stream(registry_.get(), &writer, url)); |
| 120 EXPECT_TRUE(stream->SetReadObserver(&reader1)); | 120 EXPECT_TRUE(stream->SetReadObserver(&reader1)); |
| 121 | 121 |
| 122 // Once the first read observer is removed, a new one can be added. | 122 // Once the first read observer is removed, a new one can be added. |
| 123 stream->RemoveReadObserver(&reader1); | 123 stream->RemoveReadObserver(&reader1); |
| 124 EXPECT_TRUE(stream->SetReadObserver(&reader2)); | 124 EXPECT_TRUE(stream->SetReadObserver(&reader2)); |
| 125 } | 125 } |
| 126 | 126 |
| 127 TEST_F(StreamTest, Stream) { | 127 TEST_F(StreamTest, Stream) { |
| 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, 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.get(), buffer, kBufferSize); | 138 writer.Write(stream.get(), buffer, kBufferSize); |
| 139 stream->Finalize(); | 139 stream->Finalize(); |
| 140 reader.Read(stream.get()); | 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, 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, 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.get()); | 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, 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, 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.get()); | 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, 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.get()); | 207 ASSERT_FALSE(stream2.get()); |
| 208 } | 208 } |
| 209 | 209 |
| 210 } // namespace content | 210 } // namespace content |
| OLD | NEW |