Introduce AsyncDocumentSubresourceFilter.

third_party/WebKit/Source/core/loader/FrameFetchContextTest.cpp

 /*
  * Copyright (c) 2015, Google Inc. All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions are
  * met:
  *
  *     * Redistributions of source code must retain the above copyright
  * notice, this list of conditions and the following disclaimer.
  *     * Redistributions in binary form must reproduce the above
@@ skipping 32 matching lines @@
 #include "core/page/Page.h"
 #include "core/testing/DummyPageHolder.h"
 #include "platform/loader/fetch/FetchInitiatorInfo.h"
 #include "platform/loader/fetch/MockResource.h"
 #include "platform/loader/fetch/UniqueIdentifier.h"
 #include "platform/network/ResourceRequest.h"
 #include "platform/weborigin/KURL.h"
 #include "public/platform/WebAddressSpace.h"
 #include "public/platform/WebCachePolicy.h"
 #include "public/platform/WebDocumentSubresourceFilter.h"
+#include "public/platform/WebDocumentSubresourceFilterLoadPolicy.h"
 #include "public/platform/WebInsecureRequestPolicy.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace blink {
 
 class StubFrameLoaderClientWithParent final : public EmptyFrameLoaderClient {
  public:
   static StubFrameLoaderClientWithParent* create(Frame* parent) {
     return new StubFrameLoaderClientWithParent(parent);
@@ skipping 15 matching lines @@
 class MockFrameLoaderClient : public EmptyFrameLoaderClient {
  public:
   MockFrameLoaderClient() : EmptyFrameLoaderClient() {}
   MOCK_METHOD1(didDisplayContentWithCertificateErrors, void(const KURL&));
   MOCK_METHOD2(dispatchDidLoadResourceFromMemoryCache,
                void(const ResourceRequest&, const ResourceResponse&));
 };
 
 class FixedPolicySubresourceFilter : public WebDocumentSubresourceFilter {
  public:
-  FixedPolicySubresourceFilter(LoadPolicy policy, int* filteredLoadCounter)
+  FixedPolicySubresourceFilter(WebDocumentSubresourceFilterLoadPolicy policy,
+                               int* filteredLoadCounter)
       : m_policy(policy), m_filteredLoadCounter(filteredLoadCounter) {}
 
-  LoadPolicy getLoadPolicy(const WebURL& resourceUrl,
-                           WebURLRequest::RequestContext) override {
+  WebDocumentSubresourceFilterLoadPolicy getLoadPolicy(
+      const WebURL& resourceUrl,
+      WebURLRequest::RequestContext) override {
     return m_policy;
   }
 
   void reportDisallowedLoad() override { ++*m_filteredLoadCounter; }
 
  private:
-  const LoadPolicy m_policy;
+  const WebDocumentSubresourceFilterLoadPolicy m_policy;
   int* m_filteredLoadCounter;
 };
 
 class FrameFetchContextTest : public ::testing::Test {
  protected:
   void SetUp() override {
     dummyPageHolder = DummyPageHolder::create(IntSize(500, 500));
     dummyPageHolder->page().setDeviceScaleFactor(1.0);
     document = &dummyPageHolder->document();
     fetchContext =
@@ skipping 41 matching lines @@
     m_filteredLoadCallbackCounter = 0;
   }
 
   void TearDown() override {
     document->loader()->setSubresourceFilter(nullptr);
     FrameFetchContextTest::TearDown();
   }
 
   int getFilteredLoadCallCount() const { return m_filteredLoadCallbackCounter; }
 
-  void setFilterPolicy(WebDocumentSubresourceFilter::LoadPolicy policy) {
+  void setFilterPolicy(WebDocumentSubresourceFilterLoadPolicy policy) {
     document->loader()->setSubresourceFilter(
         WTF::makeUnique<FixedPolicySubresourceFilter>(
             policy, &m_filteredLoadCallbackCounter));
   }
 
   ResourceRequestBlockedReason canRequest() {
     return canRequestInternal(
         FetchContext::SecurityViolationReportingPolicy::Report);
   }
 
@@ skipping 692 matching lines @@
     fetchContext->addAdditionalRequestHeaders(mainRequest, FetchMainResource);
     EXPECT_EQ(test.isExternalExpectation, mainRequest.isExternalRequest());
 
     ResourceRequest subRequest(test.url);
     fetchContext->addAdditionalRequestHeaders(subRequest, FetchSubresource);
     EXPECT_EQ(test.isExternalExpectation, subRequest.isExternalRequest());
   }
 }
 
 TEST_F(FrameFetchContextSubresourceFilterTest, Filter) {
-  setFilterPolicy(WebDocumentSubresourceFilter::Disallow);
+  setFilterPolicy(WebDocumentSubresourceFilterLoadPolicy::Disallow);
 
   EXPECT_EQ(ResourceRequestBlockedReason::SubresourceFilter, canRequest());
   EXPECT_EQ(1, getFilteredLoadCallCount());
 
   EXPECT_EQ(ResourceRequestBlockedReason::SubresourceFilter, canRequest());
   EXPECT_EQ(2, getFilteredLoadCallCount());
 
   EXPECT_EQ(ResourceRequestBlockedReason::SubresourceFilter,
             canRequestPreload());
   EXPECT_EQ(2, getFilteredLoadCallCount());
 
   EXPECT_EQ(ResourceRequestBlockedReason::SubresourceFilter, canRequest());
   EXPECT_EQ(3, getFilteredLoadCallCount());
 }
 
 TEST_F(FrameFetchContextSubresourceFilterTest, Allow) {
-  setFilterPolicy(WebDocumentSubresourceFilter::Allow);
+  setFilterPolicy(WebDocumentSubresourceFilterLoadPolicy::Allow);
 
   EXPECT_EQ(ResourceRequestBlockedReason::None, canRequest());
   EXPECT_EQ(0, getFilteredLoadCallCount());
 
   EXPECT_EQ(ResourceRequestBlockedReason::None, canRequestPreload());
   EXPECT_EQ(0, getFilteredLoadCallCount());
 }
 
 TEST_F(FrameFetchContextSubresourceFilterTest, WouldDisallow) {
-  setFilterPolicy(WebDocumentSubresourceFilter::WouldDisallow);
+  setFilterPolicy(WebDocumentSubresourceFilterLoadPolicy::WouldDisallow);
 
   EXPECT_EQ(ResourceRequestBlockedReason::None, canRequest());
   EXPECT_EQ(0, getFilteredLoadCallCount());
 
   EXPECT_EQ(ResourceRequestBlockedReason::None, canRequestPreload());
   EXPECT_EQ(0, getFilteredLoadCallCount());
 }
 
 }  // namespace blink