Implment BytesConsumer::tee

third_party/WebKit/Source/modules/fetch/BytesConsumerForDataConsumerHandleTest.cpp

 // Copyright 2016 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "modules/fetch/BytesConsumerForDataConsumerHandle.h"
 
+#include "core/testing/DummyPageHolder.h"
 #include "modules/fetch/BytesConsumer.h"
 #include "modules/fetch/DataConsumerHandleTestUtil.h"
 #include "modules/fetch/DataConsumerHandleUtil.h"
 #include "platform/testing/UnitTestHelpers.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace blink {
 
 namespace {
 
 using Command = DataConsumerHandleTestUtil::Command;
 using Checkpoint = ::testing::StrictMock<::testing::MockFunction<void(int)>>;
 using ReplayingHandle = DataConsumerHandleTestUtil::ReplayingHandle;
 using Result = BytesConsumer::Result;
 using ::testing::InSequence;
 using ::testing::Return;
 
 class BytesConsumerForDataConsumerHandleTest : public ::testing::Test {
+public:
+    Document* document() { return &m_page->document(); }
+
 protected:
+    BytesConsumerForDataConsumerHandleTest()
+        : m_page(DummyPageHolder::create())
+    {
+    }
     ~BytesConsumerForDataConsumerHandleTest()
     {
         ThreadHeap::collectAllGarbage();
     }
+    std::unique_ptr<DummyPageHolder> m_page;
 };
 
 class MockClient : public GarbageCollectedFinalized<MockClient>, public BytesConsumer::Client {
     USING_GARBAGE_COLLECTED_MIXIN(MockClient);
 public:
     static MockClient* create() { return new ::testing::StrictMock<MockClient>(); }
     MOCK_METHOD0(onStateChange, void());
 
 protected:
     MockClient() {}
@@ skipping 42 matching lines @@
         }
 
     private:
         Persistent<MockReaderProxy> m_proxy;
     };
 
     std::unique_ptr<FetchDataConsumerHandle::Reader> obtainFetchDataReader(Client*) override { return wrapUnique(new Reader(m_proxy)); }
     Persistent<MockReaderProxy> m_proxy;
 };
 
-} // namespace
-
 TEST_F(BytesConsumerForDataConsumerHandleTest, Create)
 {
     std::unique_ptr<ReplayingHandle> handle = ReplayingHandle::create();
     handle->add(Command(Command::Data, "hello"));
     handle->add(Command(Command::Done));
-    Persistent<BytesConsumer> consumer = new BytesConsumerForDataConsumerHandle(createFetchDataConsumerHandleFromWebHandle(std::move(handle)));
+    Persistent<BytesConsumer> consumer = new BytesConsumerForDataConsumerHandle(document(), createFetchDataConsumerHandleFromWebHandle(std::move(handle)));
 }
 
 TEST_F(BytesConsumerForDataConsumerHandleTest, BecomeReadable)
 {
     Checkpoint checkpoint;
     Persistent<MockClient> client = MockClient::create();
 
     InSequence s;
     EXPECT_CALL(checkpoint, Call(1));
     EXPECT_CALL(*client, onStateChange());
     EXPECT_CALL(checkpoint, Call(2));
 
     std::unique_ptr<ReplayingHandle> handle = ReplayingHandle::create();
     handle->add(Command(Command::Data, "hello"));
-    Persistent<BytesConsumer> consumer = new BytesConsumerForDataConsumerHandle(createFetchDataConsumerHandleFromWebHandle(std::move(handle)));
+    Persistent<BytesConsumer> consumer = new BytesConsumerForDataConsumerHandle(document(), createFetchDataConsumerHandleFromWebHandle(std::move(handle)));
     consumer->setClient(client);
     EXPECT_EQ(BytesConsumer::PublicState::ReadableOrWaiting, consumer->getPublicState());
 
     checkpoint.Call(1);
     testing::runPendingTasks();
     checkpoint.Call(2);
     EXPECT_EQ(BytesConsumer::PublicState::ReadableOrWaiting, consumer->getPublicState());
 }
 
 TEST_F(BytesConsumerForDataConsumerHandleTest, BecomeClosed)
 {
     Checkpoint checkpoint;
     Persistent<MockClient> client = MockClient::create();
 
     InSequence s;
     EXPECT_CALL(checkpoint, Call(1));
     EXPECT_CALL(*client, onStateChange());
     EXPECT_CALL(checkpoint, Call(2));
 
     std::unique_ptr<ReplayingHandle> handle = ReplayingHandle::create();
     handle->add(Command(Command::Done));
-    Persistent<BytesConsumer> consumer = new BytesConsumerForDataConsumerHandle(createFetchDataConsumerHandleFromWebHandle(std::move(handle)));
+    Persistent<BytesConsumer> consumer = new BytesConsumerForDataConsumerHandle(document(), createFetchDataConsumerHandleFromWebHandle(std::move(handle)));
     consumer->setClient(client);
     EXPECT_EQ(BytesConsumer::PublicState::ReadableOrWaiting, consumer->getPublicState());
 
     checkpoint.Call(1);
     testing::runPendingTasks();
     checkpoint.Call(2);
     EXPECT_EQ(BytesConsumer::PublicState::Closed, consumer->getPublicState());
 }
 
 TEST_F(BytesConsumerForDataConsumerHandleTest, BecomeErrored)
 {
     Checkpoint checkpoint;
     Persistent<MockClient> client = MockClient::create();
 
     InSequence s;
     EXPECT_CALL(checkpoint, Call(1));
     EXPECT_CALL(*client, onStateChange());
     EXPECT_CALL(checkpoint, Call(2));
 
     std::unique_ptr<ReplayingHandle> handle = ReplayingHandle::create();
     handle->add(Command(Command::Error));
-    Persistent<BytesConsumer> consumer = new BytesConsumerForDataConsumerHandle(createFetchDataConsumerHandleFromWebHandle(std::move(handle)));
+    Persistent<BytesConsumer> consumer = new BytesConsumerForDataConsumerHandle(document(), createFetchDataConsumerHandleFromWebHandle(std::move(handle)));
     consumer->setClient(client);
     EXPECT_EQ(BytesConsumer::PublicState::ReadableOrWaiting, consumer->getPublicState());
 
     checkpoint.Call(1);
     testing::runPendingTasks();
     checkpoint.Call(2);
     EXPECT_EQ(BytesConsumer::PublicState::Errored, consumer->getPublicState());
 }
 
 TEST_F(BytesConsumerForDataConsumerHandleTest, ClearClient)
 {
     Checkpoint checkpoint;
     Persistent<MockClient> client = MockClient::create();
 
     InSequence s;
     EXPECT_CALL(checkpoint, Call(1));
     EXPECT_CALL(checkpoint, Call(2));
 
     std::unique_ptr<ReplayingHandle> handle = ReplayingHandle::create();
     handle->add(Command(Command::Error));
-    Persistent<BytesConsumer> consumer = new BytesConsumerForDataConsumerHandle(createFetchDataConsumerHandleFromWebHandle(std::move(handle)));
+    Persistent<BytesConsumer> consumer = new BytesConsumerForDataConsumerHandle(document(), createFetchDataConsumerHandleFromWebHandle(std::move(handle)));
     consumer->setClient(client);
     consumer->clearClient();
 
     checkpoint.Call(1);
     testing::runPendingTasks();
     checkpoint.Call(2);
 }
 
 TEST_F(BytesConsumerForDataConsumerHandleTest, ReadWhenReadable)
 {
     std::unique_ptr<ReplayingHandle> handle = ReplayingHandle::create();
     handle->add(Command(Command::Data, "hello"));
-    Persistent<BytesConsumer> consumer = new BytesConsumerForDataConsumerHandle(createFetchDataConsumerHandleFromWebHandle(std::move(handle)));
+    Persistent<BytesConsumer> consumer = new BytesConsumerForDataConsumerHandle(document(), createFetchDataConsumerHandleFromWebHandle(std::move(handle)));
     consumer->setClient(MockClient::create());
 
     char buffer[16];
     size_t read;
     Result r = consumer->read(buffer, sizeof(buffer), &read);
     ASSERT_EQ(Result::Ok, r);
     EXPECT_EQ("hello", String(buffer, read));
 }
 
 TEST_F(BytesConsumerForDataConsumerHandleTest, ReadWhenWaiting)
 {
     std::unique_ptr<ReplayingHandle> handle = ReplayingHandle::create();
-    Persistent<BytesConsumer> consumer = new BytesConsumerForDataConsumerHandle(createFetchDataConsumerHandleFromWebHandle(std::move(handle)));
+    Persistent<BytesConsumer> consumer = new BytesConsumerForDataConsumerHandle(document(), createFetchDataConsumerHandleFromWebHandle(std::move(handle)));
     consumer->setClient(MockClient::create());
 
     char buffer[16];
     size_t read = 42;
     Result r = consumer->read(buffer, sizeof(buffer), &read);
     ASSERT_EQ(Result::ShouldWait, r);
     EXPECT_EQ(0u, read);
 }
 
 TEST_F(BytesConsumerForDataConsumerHandleTest, ReadWhenClosed)
 {
     std::unique_ptr<ReplayingHandle> handle = ReplayingHandle::create();
     handle->add(Command(Command::Done));
-    Persistent<BytesConsumer> consumer = new BytesConsumerForDataConsumerHandle(createFetchDataConsumerHandleFromWebHandle(std::move(handle)));
+    Persistent<BytesConsumer> consumer = new BytesConsumerForDataConsumerHandle(document(), createFetchDataConsumerHandleFromWebHandle(std::move(handle)));
     consumer->setClient(MockClient::create());
 
     char buffer[16];
     size_t read = 42;
     Result r = consumer->read(buffer, sizeof(buffer), &read);
     ASSERT_EQ(Result::Done, r);
     EXPECT_EQ(0u, read);
 }
 
 TEST_F(BytesConsumerForDataConsumerHandleTest, ReadWhenErrored)
 {
     std::unique_ptr<ReplayingHandle> handle = ReplayingHandle::create();
     handle->add(Command(Command::Error));
-    Persistent<BytesConsumer> consumer = new BytesConsumerForDataConsumerHandle(createFetchDataConsumerHandleFromWebHandle(std::move(handle)));
+    Persistent<BytesConsumer> consumer = new BytesConsumerForDataConsumerHandle(document(), createFetchDataConsumerHandleFromWebHandle(std::move(handle)));
     consumer->setClient(MockClient::create());
 
     char buffer[16];
     size_t read;
     Result r = consumer->read(buffer, sizeof(buffer), &read);
     ASSERT_EQ(Result::Error, r);
     EXPECT_EQ(BytesConsumer::Error("error"), consumer->getError());
 }
 
 TEST_F(BytesConsumerForDataConsumerHandleTest, TwoPhaseReadWhenReadable)
 {
     std::unique_ptr<ReplayingHandle> handle = ReplayingHandle::create();
     handle->add(Command(Command::Data, "hello"));
-    Persistent<BytesConsumer> consumer = new BytesConsumerForDataConsumerHandle(createFetchDataConsumerHandleFromWebHandle(std::move(handle)));
+    Persistent<BytesConsumer> consumer = new BytesConsumerForDataConsumerHandle(document(), createFetchDataConsumerHandleFromWebHandle(std::move(handle)));
     consumer->setClient(MockClient::create());
 
     const char* buffer = nullptr;
     size_t available = 0;
     ASSERT_EQ(Result::Ok, consumer->beginRead(&buffer, &available));
     EXPECT_EQ("hello", String(buffer, available));
 
     ASSERT_EQ(Result::Ok, consumer->endRead(1));
     ASSERT_EQ(Result::Ok, consumer->beginRead(&buffer, &available));
     EXPECT_EQ("ello", String(buffer, available));
 
     ASSERT_EQ(Result::Ok, consumer->endRead(4));
     ASSERT_EQ(Result::ShouldWait, consumer->beginRead(&buffer, &available));
 }
 
 TEST_F(BytesConsumerForDataConsumerHandleTest, TwoPhaseReadWhenWaiting)
 {
     std::unique_ptr<ReplayingHandle> handle = ReplayingHandle::create();
-    Persistent<BytesConsumer> consumer = new BytesConsumerForDataConsumerHandle(createFetchDataConsumerHandleFromWebHandle(std::move(handle)));
+    Persistent<BytesConsumer> consumer = new BytesConsumerForDataConsumerHandle(document(), createFetchDataConsumerHandleFromWebHandle(std::move(handle)));
     consumer->setClient(MockClient::create());
     const char* buffer = nullptr;
     size_t available = 0;
     ASSERT_EQ(Result::ShouldWait, consumer->beginRead(&buffer, &available));
 }
 
 TEST_F(BytesConsumerForDataConsumerHandleTest, TwoPhaseReadWhenClosed)
 {
     std::unique_ptr<ReplayingHandle> handle = ReplayingHandle::create();
     handle->add(Command(Command::Done));
-    Persistent<BytesConsumer> consumer = new BytesConsumerForDataConsumerHandle(createFetchDataConsumerHandleFromWebHandle(std::move(handle)));
+    Persistent<BytesConsumer> consumer = new BytesConsumerForDataConsumerHandle(document(), createFetchDataConsumerHandleFromWebHandle(std::move(handle)));
     consumer->setClient(MockClient::create());
     const char* buffer = nullptr;
     size_t available = 0;
     ASSERT_EQ(Result::Done, consumer->beginRead(&buffer, &available));
 }
 
 TEST_F(BytesConsumerForDataConsumerHandleTest, TwoPhaseReadWhenErrored)
 {
     std::unique_ptr<ReplayingHandle> handle = ReplayingHandle::create();
     handle->add(Command(Command::Error));
-    Persistent<BytesConsumer> consumer = new BytesConsumerForDataConsumerHandle(createFetchDataConsumerHandleFromWebHandle(std::move(handle)));
+    Persistent<BytesConsumer> consumer = new BytesConsumerForDataConsumerHandle(document(), createFetchDataConsumerHandleFromWebHandle(std::move(handle)));
     consumer->setClient(MockClient::create());
     const char* buffer = nullptr;
     size_t available = 0;
     ASSERT_EQ(Result::Error, consumer->beginRead(&buffer, &available));
     EXPECT_EQ(BytesConsumer::Error("error"), consumer->getError());
 }
 
 TEST_F(BytesConsumerForDataConsumerHandleTest, Cancel)
 {
     std::unique_ptr<ReplayingHandle> handle = ReplayingHandle::create();
-    Persistent<BytesConsumer> consumer = new BytesConsumerForDataConsumerHandle(createFetchDataConsumerHandleFromWebHandle(std::move(handle)));
+    Persistent<BytesConsumer> consumer = new BytesConsumerForDataConsumerHandle(document(), createFetchDataConsumerHandleFromWebHandle(std::move(handle)));
     consumer->setClient(MockClient::create());
     consumer->cancel();
     const char* buffer = nullptr;
     size_t available = 0;
     ASSERT_EQ(Result::Done, consumer->beginRead(&buffer, &available));
 }
 
 TEST_F(BytesConsumerForDataConsumerHandleTest, drainAsBlobDataHandle)
 {
     // FetchDataConsumerHandle::Reader::drainAsBlobDataHandle should return
     // nullptr from the second time, but we don't care that here.
     std::unique_ptr<MockDataConsumerHandle> handle = wrapUnique(new MockDataConsumerHandle);
     Persistent<MockDataConsumerHandle::MockReaderProxy> proxy = handle->proxy();
-    Persistent<BytesConsumer> consumer = new BytesConsumerForDataConsumerHandle(std::move(handle));
+    Persistent<BytesConsumer> consumer = new BytesConsumerForDataConsumerHandle(document(), std::move(handle));
     consumer->setClient(MockClient::create());
     RefPtr<BlobDataHandle> blobDataHandle = BlobDataHandle::create();
 
     Checkpoint checkpoint;
     InSequence s;
     EXPECT_CALL(*proxy, drainAsBlobDataHandle(FetchDataConsumerHandle::Reader::DisallowBlobWithInvalidSize)).WillOnce(Return(nullptr));
     EXPECT_CALL(*proxy, drainAsBlobDataHandle(FetchDataConsumerHandle::Reader::AllowBlobWithInvalidSize)).WillOnce(Return(nullptr));
     EXPECT_CALL(*proxy, drainAsBlobDataHandle(FetchDataConsumerHandle::Reader::AllowBlobWithInvalidSize)).WillOnce(Return(blobDataHandle));
 
     EXPECT_FALSE(consumer->drainAsBlobDataHandle(BytesConsumer::BlobSizePolicy::DisallowBlobWithInvalidSize));
     EXPECT_FALSE(consumer->drainAsBlobDataHandle(BytesConsumer::BlobSizePolicy::AllowBlobWithInvalidSize));
     EXPECT_TRUE(blobDataHandle);
     EXPECT_EQ(blobDataHandle, consumer->drainAsBlobDataHandle(BytesConsumer::BlobSizePolicy::AllowBlobWithInvalidSize));
     EXPECT_EQ(BytesConsumer::PublicState::Closed, consumer->getPublicState());
 }
 
 TEST_F(BytesConsumerForDataConsumerHandleTest, drainAsFormData)
 {
     std::unique_ptr<MockDataConsumerHandle> handle = wrapUnique(new MockDataConsumerHandle);
     Persistent<MockDataConsumerHandle::MockReaderProxy> proxy = handle->proxy();
-    Persistent<BytesConsumer> consumer = new BytesConsumerForDataConsumerHandle(std::move(handle));
+    Persistent<BytesConsumer> consumer = new BytesConsumerForDataConsumerHandle(document(), std::move(handle));
     consumer->setClient(MockClient::create());
     RefPtr<EncodedFormData> formData = EncodedFormData::create();
 
     Checkpoint checkpoint;
     InSequence s;
     EXPECT_CALL(*proxy, drainAsFormData()).WillOnce(Return(formData));
 
     EXPECT_EQ(formData, consumer->drainAsFormData());
     EXPECT_EQ(BytesConsumer::PublicState::Closed, consumer->getPublicState());
 }
 
+} // namespace
+
 } // namespace blink