| OLD | NEW |
| 1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 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 "modules/fetch/BytesConsumerForDataConsumerHandle.h" | 5 #include "modules/fetch/BytesConsumerForDataConsumerHandle.h" |
| 6 | 6 |
| 7 #include "core/testing/DummyPageHolder.h" | 7 #include "core/testing/DummyPageHolder.h" |
| 8 #include "modules/fetch/BytesConsumer.h" | 8 #include "modules/fetch/BytesConsumer.h" |
| 9 #include "modules/fetch/DataConsumerHandleTestUtil.h" | 9 #include "modules/fetch/DataConsumerHandleTestUtil.h" |
| 10 #include "platform/testing/UnitTestHelpers.h" | 10 #include "platform/testing/UnitTestHelpers.h" |
| (...skipping 68 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 79 Result endRead(size_t readSize) override { | 79 Result endRead(size_t readSize) override { |
| 80 return m_proxy->endRead(readSize); | 80 return m_proxy->endRead(readSize); |
| 81 } | 81 } |
| 82 | 82 |
| 83 private: | 83 private: |
| 84 Persistent<MockReaderProxy> m_proxy; | 84 Persistent<MockReaderProxy> m_proxy; |
| 85 }; | 85 }; |
| 86 | 86 |
| 87 std::unique_ptr<WebDataConsumerHandle::Reader> obtainReader( | 87 std::unique_ptr<WebDataConsumerHandle::Reader> obtainReader( |
| 88 Client*) override { | 88 Client*) override { |
| 89 return makeUnique<Reader>(m_proxy); | 89 return WTF::makeUnique<Reader>(m_proxy); |
| 90 } | 90 } |
| 91 Persistent<MockReaderProxy> m_proxy; | 91 Persistent<MockReaderProxy> m_proxy; |
| 92 }; | 92 }; |
| 93 | 93 |
| 94 TEST_F(BytesConsumerForDataConsumerHandleTest, Create) { | 94 TEST_F(BytesConsumerForDataConsumerHandleTest, Create) { |
| 95 std::unique_ptr<ReplayingHandle> handle = ReplayingHandle::create(); | 95 std::unique_ptr<ReplayingHandle> handle = ReplayingHandle::create(); |
| 96 handle->add(Command(Command::Data, "hello")); | 96 handle->add(Command(Command::Data, "hello")); |
| 97 handle->add(Command(Command::Done)); | 97 handle->add(Command(Command::Done)); |
| 98 Persistent<BytesConsumer> consumer = | 98 Persistent<BytesConsumer> consumer = |
| 99 new BytesConsumerForDataConsumerHandle(document(), std::move(handle)); | 99 new BytesConsumerForDataConsumerHandle(document(), std::move(handle)); |
| (...skipping 150 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 250 consumer->cancel(); | 250 consumer->cancel(); |
| 251 const char* buffer = nullptr; | 251 const char* buffer = nullptr; |
| 252 size_t available = 0; | 252 size_t available = 0; |
| 253 ASSERT_EQ(Result::Done, consumer->beginRead(&buffer, &available)); | 253 ASSERT_EQ(Result::Done, consumer->beginRead(&buffer, &available)); |
| 254 } | 254 } |
| 255 | 255 |
| 256 TEST_F(BytesConsumerForDataConsumerHandleTest, drainAsBlobDataHandle) { | 256 TEST_F(BytesConsumerForDataConsumerHandleTest, drainAsBlobDataHandle) { |
| 257 // WebDataConsumerHandle::Reader::drainAsBlobDataHandle should return | 257 // WebDataConsumerHandle::Reader::drainAsBlobDataHandle should return |
| 258 // nullptr from the second time, but we don't care that here. | 258 // nullptr from the second time, but we don't care that here. |
| 259 std::unique_ptr<MockDataConsumerHandle> handle = | 259 std::unique_ptr<MockDataConsumerHandle> handle = |
| 260 wrapUnique(new MockDataConsumerHandle); | 260 WTF::wrapUnique(new MockDataConsumerHandle); |
| 261 Persistent<MockDataConsumerHandle::MockReaderProxy> proxy = handle->proxy(); | 261 Persistent<MockDataConsumerHandle::MockReaderProxy> proxy = handle->proxy(); |
| 262 Persistent<BytesConsumer> consumer = | 262 Persistent<BytesConsumer> consumer = |
| 263 new BytesConsumerForDataConsumerHandle(document(), std::move(handle)); | 263 new BytesConsumerForDataConsumerHandle(document(), std::move(handle)); |
| 264 consumer->setClient(MockClient::create()); | 264 consumer->setClient(MockClient::create()); |
| 265 | 265 |
| 266 Checkpoint checkpoint; | 266 Checkpoint checkpoint; |
| 267 InSequence s; | 267 InSequence s; |
| 268 | 268 |
| 269 EXPECT_FALSE(consumer->drainAsBlobDataHandle( | 269 EXPECT_FALSE(consumer->drainAsBlobDataHandle( |
| 270 BytesConsumer::BlobSizePolicy::DisallowBlobWithInvalidSize)); | 270 BytesConsumer::BlobSizePolicy::DisallowBlobWithInvalidSize)); |
| 271 EXPECT_FALSE(consumer->drainAsBlobDataHandle( | 271 EXPECT_FALSE(consumer->drainAsBlobDataHandle( |
| 272 BytesConsumer::BlobSizePolicy::AllowBlobWithInvalidSize)); | 272 BytesConsumer::BlobSizePolicy::AllowBlobWithInvalidSize)); |
| 273 EXPECT_EQ(BytesConsumer::PublicState::ReadableOrWaiting, | 273 EXPECT_EQ(BytesConsumer::PublicState::ReadableOrWaiting, |
| 274 consumer->getPublicState()); | 274 consumer->getPublicState()); |
| 275 } | 275 } |
| 276 | 276 |
| 277 TEST_F(BytesConsumerForDataConsumerHandleTest, drainAsFormData) { | 277 TEST_F(BytesConsumerForDataConsumerHandleTest, drainAsFormData) { |
| 278 std::unique_ptr<MockDataConsumerHandle> handle = | 278 std::unique_ptr<MockDataConsumerHandle> handle = |
| 279 wrapUnique(new MockDataConsumerHandle); | 279 WTF::wrapUnique(new MockDataConsumerHandle); |
| 280 Persistent<MockDataConsumerHandle::MockReaderProxy> proxy = handle->proxy(); | 280 Persistent<MockDataConsumerHandle::MockReaderProxy> proxy = handle->proxy(); |
| 281 Persistent<BytesConsumer> consumer = | 281 Persistent<BytesConsumer> consumer = |
| 282 new BytesConsumerForDataConsumerHandle(document(), std::move(handle)); | 282 new BytesConsumerForDataConsumerHandle(document(), std::move(handle)); |
| 283 consumer->setClient(MockClient::create()); | 283 consumer->setClient(MockClient::create()); |
| 284 | 284 |
| 285 Checkpoint checkpoint; | 285 Checkpoint checkpoint; |
| 286 InSequence s; | 286 InSequence s; |
| 287 | 287 |
| 288 EXPECT_FALSE(consumer->drainAsFormData()); | 288 EXPECT_FALSE(consumer->drainAsFormData()); |
| 289 EXPECT_EQ(BytesConsumer::PublicState::ReadableOrWaiting, | 289 EXPECT_EQ(BytesConsumer::PublicState::ReadableOrWaiting, |
| 290 consumer->getPublicState()); | 290 consumer->getPublicState()); |
| 291 } | 291 } |
| 292 | 292 |
| 293 } // namespace | 293 } // namespace |
| 294 | 294 |
| 295 } // namespace blink | 295 } // namespace blink |
| OLD | NEW |