Loading: Factor out ResourceFetcherMockFetchContext

third_party/WebKit/Source/core/fetch/ResourceFetcherTest.cpp

 /*
  * Copyright (c) 2013, 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 18 matching lines @@
  */
 
 #include "core/fetch/ResourceFetcher.h"
 
 #include "core/fetch/FetchInitiatorInfo.h"
 #include "core/fetch/FetchInitiatorTypeNames.h"
 #include "core/fetch/FetchRequest.h"
 #include "core/fetch/FontResource.h"
 #include "core/fetch/ImageResource.h"
 #include "core/fetch/MemoryCache.h"
+#include "core/fetch/MockFetchContext.h"
 #include "core/fetch/MockResourceClients.h"
 #include "core/fetch/RawResource.h"
 #include "core/fetch/ResourceLoader.h"
 #include "platform/WebTaskRunner.h"
 #include "platform/exported/WrappedResourceResponse.h"
 #include "platform/heap/Handle.h"
 #include "platform/heap/HeapAllocator.h"
 #include "platform/heap/Member.h"
 #include "platform/network/ResourceRequest.h"
 #include "platform/network/ResourceTimingInfo.h"
@@ skipping 10 matching lines @@
 #include "wtf/Vector.h"
 #include <memory>
 
 namespace blink {
 
 namespace {
 const char testImageFilename[] = "white-1x1.png";
 const int testImageSize = 103;  // size of web/tests/data/white-1x1.png
 }
 
-class ResourceFetcherTestMockFetchContext : public FetchContext {
- public:
-  static ResourceFetcherTestMockFetchContext* create() {
-    return new ResourceFetcherTestMockFetchContext;
-  }
-
-  virtual ~ResourceFetcherTestMockFetchContext() {}
-
-  bool allowImage(bool imagesEnabled, const KURL&) const override {
-    return true;
-  }
-  bool canRequest(Resource::Type,
-                  const ResourceRequest&,
-                  const KURL&,
-                  const ResourceLoaderOptions&,
-                  bool forPreload,
-                  FetchRequest::OriginRestriction) const override {
-    return true;
-  }
-  bool shouldLoadNewResource(Resource::Type) const override { return true; }
-  WebTaskRunner* loadingTaskRunner() const override { return m_runner.get(); }
-
-  void setCachePolicy(CachePolicy policy) { m_policy = policy; }
-  CachePolicy getCachePolicy() const override { return m_policy; }
-  void setLoadComplete(bool complete) { m_complete = complete; }
-  bool isLoadComplete() const override { return m_complete; }
-
-  void addResourceTiming(
-      const ResourceTimingInfo& resourceTimingInfo) override {
-    m_transferSize = resourceTimingInfo.transferSize();
-  }
-  long long getTransferSize() const { return m_transferSize; }
-
- private:
-  ResourceFetcherTestMockFetchContext()
-      : m_policy(CachePolicyVerify),
-        m_runner(wrapUnique(new scheduler::FakeWebTaskRunner)),
-        m_complete(false),
-        m_transferSize(-1) {}
-
-  CachePolicy m_policy;
-  std::unique_ptr<scheduler::FakeWebTaskRunner> m_runner;
-  bool m_complete;
-  long long m_transferSize;
-};
-
 class ResourceFetcherTest : public ::testing::Test {};
 
 TEST_F(ResourceFetcherTest, StartLoadAfterFrameDetach) {
   KURL secureURL(ParsedURLString, "https://secureorigin.test/image.png");
   // Try to request a url. The request should fail, and a resource in an error
   // state should be returned, and no resource should be present in the cache.
   ResourceFetcher* fetcher = ResourceFetcher::create(nullptr);
   ResourceRequest resourceRequest(secureURL);
   resourceRequest.setRequestContext(WebURLRequest::RequestContextInternal);
   FetchRequest fetchRequest =
       FetchRequest(resourceRequest, FetchInitiatorInfo());
   Resource* resource = RawResource::fetch(fetchRequest, fetcher);
   ASSERT_TRUE(resource);
   EXPECT_TRUE(resource->errorOccurred());
   EXPECT_TRUE(resource->resourceError().isAccessCheck());
   EXPECT_FALSE(memoryCache()->resourceForURL(secureURL));
 
   // Start by calling startLoad() directly, rather than via requestResource().
   // This shouldn't crash.
   fetcher->startLoad(RawResource::create(secureURL, Resource::Raw));
 }
 
 TEST_F(ResourceFetcherTest, UseExistingResource) {
-  ResourceFetcher* fetcher =
-      ResourceFetcher::create(ResourceFetcherTestMockFetchContext::create());
+  ResourceFetcher* fetcher = ResourceFetcher::create(
+      MockFetchContext::create(MockFetchContext::kShouldLoadNewResource));
 
   KURL url(ParsedURLString, "http://127.0.0.1:8000/foo.html");
   ResourceResponse response;
   response.setURL(url);
   response.setHTTPStatusCode(200);
   response.setHTTPHeaderField(HTTPNames::Cache_Control, "max-age=3600");
   URLTestHelpers::registerMockedURLLoadWithCustomResponse(
       url, testImageFilename, WebString::fromUTF8(""),
       WrappedResourceResponse(response));
 
@@ skipping 15 matching lines @@
   memoryCache()->add(resource);
   ResourceResponse response;
   response.setURL(url);
   response.setHTTPStatusCode(200);
   response.setHTTPHeaderField(HTTPNames::Cache_Control, "max-age=3600");
   response.setHTTPHeaderField(HTTPNames::Vary, "*");
   resource->responseReceived(response, nullptr);
   resource->finish();
   ASSERT_TRUE(resource->hasVaryHeader());
 
-  ResourceFetcher* fetcher =
-      ResourceFetcher::create(ResourceFetcherTestMockFetchContext::create());
+  ResourceFetcher* fetcher = ResourceFetcher::create(
+      MockFetchContext::create(MockFetchContext::kShouldLoadNewResource));
   ResourceRequest resourceRequest(url);
   resourceRequest.setRequestContext(WebURLRequest::RequestContextInternal);
   FetchRequest fetchRequest =
       FetchRequest(resourceRequest, FetchInitiatorInfo());
   Platform::current()->getURLLoaderMockFactory()->registerURL(
       url, WebURLResponse(), "");
   Resource* newResource = RawResource::fetch(fetchRequest, fetcher);
   EXPECT_NE(resource, newResource);
   newResource->loader()->cancel();
   memoryCache()->remove(newResource);
   Platform::current()->getURLLoaderMockFactory()->unregisterURL(url);
 
   memoryCache()->remove(resource);
 }
 
 TEST_F(ResourceFetcherTest, VaryOnBack) {
-  ResourceFetcherTestMockFetchContext* context =
-      ResourceFetcherTestMockFetchContext::create();
+  MockFetchContext* context =
+      MockFetchContext::create(MockFetchContext::kShouldLoadNewResource);
   context->setCachePolicy(CachePolicyHistoryBuffer);
   ResourceFetcher* fetcher = ResourceFetcher::create(context);
 
   KURL url(ParsedURLString, "http://127.0.0.1:8000/foo.html");
   Resource* resource = RawResource::create(url, Resource::Raw);
   memoryCache()->add(resource);
   ResourceResponse response;
   response.setURL(url);
   response.setHTTPStatusCode(200);
   response.setHTTPHeaderField(HTTPNames::Cache_Control, "max-age=3600");
   response.setHTTPHeaderField(HTTPNames::Vary, "*");
   resource->responseReceived(response, nullptr);
   resource->finish();
   ASSERT_TRUE(resource->hasVaryHeader());
 
   ResourceRequest resourceRequest(url);
   resourceRequest.setRequestContext(WebURLRequest::RequestContextInternal);
   FetchRequest fetchRequest =
       FetchRequest(resourceRequest, FetchInitiatorInfo());
   Resource* newResource = RawResource::fetch(fetchRequest, fetcher);
   EXPECT_EQ(resource, newResource);
 
   memoryCache()->remove(newResource);
 }
 
 TEST_F(ResourceFetcherTest, VaryImage) {
-  ResourceFetcher* fetcher =
-      ResourceFetcher::create(ResourceFetcherTestMockFetchContext::create());
+  ResourceFetcher* fetcher = ResourceFetcher::create(
+      MockFetchContext::create(MockFetchContext::kShouldLoadNewResource));
 
   KURL url(ParsedURLString, "http://127.0.0.1:8000/foo.html");
   ResourceResponse response;
   response.setURL(url);
   response.setHTTPStatusCode(200);
   response.setHTTPHeaderField(HTTPNames::Cache_Control, "max-age=3600");
   response.setHTTPHeaderField(HTTPNames::Vary, "*");
   URLTestHelpers::registerMockedURLLoadWithCustomResponse(
       url, testImageFilename, WebString::fromUTF8(""),
       WrappedResourceResponse(response));
@@ skipping 16 matching lines @@
     : public GarbageCollectedFinalized<RequestSameResourceOnComplete>,
       public RawResourceClient {
   USING_GARBAGE_COLLECTED_MIXIN(RequestSameResourceOnComplete);
 
  public:
   explicit RequestSameResourceOnComplete(Resource* resource)
       : m_resource(resource), m_notifyFinishedCalled(false) {}
 
   void notifyFinished(Resource* resource) override {
     EXPECT_EQ(m_resource, resource);
-    ResourceFetcherTestMockFetchContext* context =
-        ResourceFetcherTestMockFetchContext::create();
+    MockFetchContext* context =
+        MockFetchContext::create(MockFetchContext::kShouldLoadNewResource);
     context->setCachePolicy(CachePolicyRevalidate);
     ResourceFetcher* fetcher2 = ResourceFetcher::create(context);
     FetchRequest fetchRequest2(m_resource->url(), FetchInitiatorInfo());
     Resource* resource2 = ImageResource::fetch(fetchRequest2, fetcher2);
     EXPECT_EQ(m_resource, resource2);
     m_notifyFinishedCalled = true;
   }
   bool notifyFinishedCalled() const { return m_notifyFinishedCalled; }
 
   DEFINE_INLINE_TRACE() {
@@ skipping 11 matching lines @@
 TEST_F(ResourceFetcherTest, RevalidateWhileFinishingLoading) {
   KURL url(ParsedURLString, "http://127.0.0.1:8000/foo.png");
   ResourceResponse response;
   response.setURL(url);
   response.setHTTPStatusCode(200);
   response.setHTTPHeaderField(HTTPNames::Cache_Control, "max-age=3600");
   response.setHTTPHeaderField(HTTPNames::ETag, "1234567890");
   URLTestHelpers::registerMockedURLLoadWithCustomResponse(
       url, testImageFilename, WebString::fromUTF8(""),
       WrappedResourceResponse(response));
-  ResourceFetcher* fetcher1 =
-      ResourceFetcher::create(ResourceFetcherTestMockFetchContext::create());
+  ResourceFetcher* fetcher1 = ResourceFetcher::create(
+      MockFetchContext::create(MockFetchContext::kShouldLoadNewResource));
   ResourceRequest request1(url);
   request1.setHTTPHeaderField(HTTPNames::Cache_Control, "no-cache");
   FetchRequest fetchRequest1 = FetchRequest(request1, FetchInitiatorInfo());
   Resource* resource1 = ImageResource::fetch(fetchRequest1, fetcher1);
   Persistent<RequestSameResourceOnComplete> client =
       new RequestSameResourceOnComplete(resource1);
   resource1->addClient(client);
   Platform::current()->getURLLoaderMockFactory()->serveAsynchronousRequests();
   Platform::current()->getURLLoaderMockFactory()->unregisterURL(url);
   EXPECT_TRUE(client->notifyFinishedCalled());
   resource1->removeClient(client);
   memoryCache()->remove(resource1);
 }
 
 TEST_F(ResourceFetcherTest, RevalidateDeferedResourceFromTwoInitiators) {
   KURL url(ParsedURLString, "http://127.0.0.1:8000/font.woff");
   ResourceResponse response;
   response.setURL(url);
   response.setHTTPStatusCode(200);
   response.setHTTPHeaderField(HTTPNames::ETag, "1234567890");
   Platform::current()->getURLLoaderMockFactory()->registerURL(
       url, WrappedResourceResponse(response), "");
 
-  ResourceFetcherTestMockFetchContext* context =
-      ResourceFetcherTestMockFetchContext::create();
+  MockFetchContext* context =
+      MockFetchContext::create(MockFetchContext::kShouldLoadNewResource);
   ResourceFetcher* fetcher = ResourceFetcher::create(context);
 
   // Fetch to cache a resource.
   ResourceRequest request1(url);
   FetchRequest fetchRequest1 = FetchRequest(request1, FetchInitiatorInfo());
   Resource* resource1 = FontResource::fetch(fetchRequest1, fetcher);
   ASSERT_TRUE(resource1);
   fetcher->startLoad(resource1);
   Platform::current()->getURLLoaderMockFactory()->serveAsynchronousRequests();
   EXPECT_TRUE(resource1->isLoaded());
@@ skipping 27 matching lines @@
   Platform::current()->getURLLoaderMockFactory()->serveAsynchronousRequests();
   EXPECT_TRUE(resource3->isLoaded());
   EXPECT_FALSE(resource3->errorOccurred());
   EXPECT_TRUE(resource2->isLoaded());
   EXPECT_FALSE(resource2->errorOccurred());
 
   memoryCache()->remove(resource1);
 }
 
 TEST_F(ResourceFetcherTest, DontReuseMediaDataUrl) {
-  ResourceFetcher* fetcher =
-      ResourceFetcher::create(ResourceFetcherTestMockFetchContext::create());
+  ResourceFetcher* fetcher = ResourceFetcher::create(
+      MockFetchContext::create(MockFetchContext::kShouldLoadNewResource));
   ResourceRequest request(KURL(ParsedURLString, "data:text/html,foo"));
   request.setRequestContext(WebURLRequest::RequestContextVideo);
   ResourceLoaderOptions options;
   options.dataBufferingPolicy = DoNotBufferData;
   FetchRequest fetchRequest =
       FetchRequest(request, FetchInitiatorTypeNames::internal, options);
   Resource* resource1 = RawResource::fetchMedia(fetchRequest, fetcher);
   Resource* resource2 = RawResource::fetchMedia(fetchRequest, fetcher);
   EXPECT_NE(resource1, resource2);
   memoryCache()->remove(resource2);
@@ skipping 43 matching lines @@
 
 // Regression test for http://crbug.com/594072.
 // This emulates a modal dialog triggering a nested run loop inside
 // ResourceLoader::cancel(). If the ResourceLoader doesn't promptly cancel its
 // WebURLLoader before notifying its clients, a nested run loop  may send a
 // network response, leading to an invalid state transition in ResourceLoader.
 TEST_F(ResourceFetcherTest, ResponseOnCancel) {
   KURL url(ParsedURLString, "http://127.0.0.1:8000/foo.png");
   URLTestHelpers::registerMockedURLLoad(url, testImageFilename, "image/png");
 
-  ResourceFetcher* fetcher =
-      ResourceFetcher::create(ResourceFetcherTestMockFetchContext::create());
+  ResourceFetcher* fetcher = ResourceFetcher::create(
+      MockFetchContext::create(MockFetchContext::kShouldLoadNewResource));
   ResourceRequest resourceRequest(url);
   resourceRequest.setRequestContext(WebURLRequest::RequestContextInternal);
   FetchRequest fetchRequest =
       FetchRequest(resourceRequest, FetchInitiatorInfo());
   Resource* resource = RawResource::fetch(fetchRequest, fetcher);
   Persistent<ServeRequestsOnCompleteClient> client =
       new ServeRequestsOnCompleteClient();
   resource->addClient(client);
   resource->loader()->cancel();
   resource->removeClient(client);
@@ skipping 20 matching lines @@
         redirectURL, redirectResponse, "");
   }
 
   void registerFinalResource(const WebString& url) {
     KURL finalURL(ParsedURLString, url);
     URLTestHelpers::registerMockedURLLoad(finalURL, testImageFilename);
   }
 
   void request(const WebString& url) {
     DCHECK(!m_context);
-    m_context = ResourceFetcherTestMockFetchContext::create();
+    m_context =
+        MockFetchContext::create(MockFetchContext::kShouldLoadNewResource);
     ResourceFetcher* fetcher = ResourceFetcher::create(m_context);
     ResourceRequest resourceRequest(url);
     resourceRequest.setRequestContext(WebURLRequest::RequestContextInternal);
     FetchRequest fetchRequest =
         FetchRequest(resourceRequest, FetchInitiatorInfo());
     RawResource::fetch(fetchRequest, fetcher);
     Platform::current()->getURLLoaderMockFactory()->serveAsynchronousRequests();
   }
 
   void cleanUp() {
     Platform::current()->getURLLoaderMockFactory()->unregisterAllURLs();
     memoryCache()->evictResources();
   }
 
-  ResourceFetcherTestMockFetchContext* context() const { return m_context; }
+  MockFetchContext* context() const { return m_context; }
 
  private:
-  Member<ResourceFetcherTestMockFetchContext> m_context;
+  Member<MockFetchContext> m_context;
 };
 
 TEST_F(ResourceFetcherTest, SameOriginRedirect) {
   const char redirectURL[] = "http://127.0.0.1:8000/redirect.html";
   const char finalURL[] = "http://127.0.0.1:8000/final.html";
   ScopedMockRedirectRequester requester;
   requester.registerRedirect(redirectURL, finalURL);
   requester.registerFinalResource(finalURL);
   requester.request(redirectURL);
 
@@ skipping 24 matching lines @@
   requester.registerFinalResource(finalURL);
   requester.request(redirectURL1);
 
   EXPECT_EQ(testImageSize, requester.context()->getTransferSize());
 }
 
 TEST_F(ResourceFetcherTest, SynchronousRequest) {
   KURL url(ParsedURLString, "http://127.0.0.1:8000/foo.png");
   URLTestHelpers::registerMockedURLLoad(url, testImageFilename, "image/png");
 
-  ResourceFetcher* fetcher =
-      ResourceFetcher::create(ResourceFetcherTestMockFetchContext::create());
+  ResourceFetcher* fetcher = ResourceFetcher::create(
+      MockFetchContext::create(MockFetchContext::kShouldLoadNewResource));
   ResourceRequest resourceRequest(url);
   resourceRequest.setRequestContext(WebURLRequest::RequestContextInternal);
   FetchRequest fetchRequest(resourceRequest, FetchInitiatorInfo());
   fetchRequest.makeSynchronous();
   Resource* resource = RawResource::fetch(fetchRequest, fetcher);
   EXPECT_TRUE(resource->isLoaded());
   EXPECT_EQ(ResourceLoadPriorityHighest,
             resource->resourceRequest().priority());
 
   Platform::current()->getURLLoaderMockFactory()->unregisterURL(url);
   memoryCache()->remove(resource);
 }
 
 TEST_F(ResourceFetcherTest, PreloadImageTwice) {
-  ResourceFetcher* fetcher =
-      ResourceFetcher::create(ResourceFetcherTestMockFetchContext::create());
+  ResourceFetcher* fetcher = ResourceFetcher::create(
+      MockFetchContext::create(MockFetchContext::kShouldLoadNewResource));
 
   KURL url(ParsedURLString, "http://127.0.0.1:8000/foo.png");
   URLTestHelpers::registerMockedURLLoad(url, testImageFilename, "image/png");
 
   FetchRequest fetchRequestOriginal = FetchRequest(url, FetchInitiatorInfo());
   Resource* resource = ImageResource::fetch(fetchRequestOriginal, fetcher);
   ASSERT_TRUE(resource);
   Platform::current()->getURLLoaderMockFactory()->serveAsynchronousRequests();
   fetcher->preloadStarted(resource);
 
   FetchRequest fetchRequest = FetchRequest(url, FetchInitiatorInfo());
   Resource* newResource = ImageResource::fetch(fetchRequest, fetcher);
   EXPECT_EQ(resource, newResource);
   fetcher->preloadStarted(resource);
 
   fetcher->clearPreloads(ResourceFetcher::ClearAllPreloads);
   Platform::current()->getURLLoaderMockFactory()->unregisterURL(url);
   EXPECT_FALSE(memoryCache()->contains(resource));
   EXPECT_FALSE(resource->isPreloaded());
 }
 
 TEST_F(ResourceFetcherTest, LinkPreloadImageAndUse) {
-  ResourceFetcher* fetcher =
-      ResourceFetcher::create(ResourceFetcherTestMockFetchContext::create());
+  ResourceFetcher* fetcher = ResourceFetcher::create(
+      MockFetchContext::create(MockFetchContext::kShouldLoadNewResource));
 
   KURL url(ParsedURLString, "http://127.0.0.1:8000/foo.png");
   URLTestHelpers::registerMockedURLLoad(url, testImageFilename, "image/png");
 
   // Link preload preload scanner
   FetchRequest fetchRequestOriginal = FetchRequest(url, FetchInitiatorInfo());
   fetchRequestOriginal.setLinkPreload(true);
   Resource* resource = ImageResource::fetch(fetchRequestOriginal, fetcher);
   ASSERT_TRUE(resource);
   EXPECT_TRUE(resource->isLinkPreload());
@@ skipping 17 matching lines @@
   EXPECT_FALSE(resource->isLinkPreload());
 
   // DCL reached
   fetcher->clearPreloads(ResourceFetcher::ClearSpeculativeMarkupPreloads);
   Platform::current()->getURLLoaderMockFactory()->unregisterURL(url);
   EXPECT_TRUE(memoryCache()->contains(resource));
   EXPECT_FALSE(resource->isPreloaded());
 }
 
 TEST_F(ResourceFetcherTest, LinkPreloadImageMultipleFetchersAndUse) {
-  ResourceFetcher* fetcher =
-      ResourceFetcher::create(ResourceFetcherTestMockFetchContext::create());
-  ResourceFetcher* fetcher2 =
-      ResourceFetcher::create(ResourceFetcherTestMockFetchContext::create());
+  ResourceFetcher* fetcher = ResourceFetcher::create(
+      MockFetchContext::create(MockFetchContext::kShouldLoadNewResource));
+  ResourceFetcher* fetcher2 = ResourceFetcher::create(
+      MockFetchContext::create(MockFetchContext::kShouldLoadNewResource));
 
   KURL url(ParsedURLString, "http://127.0.0.1:8000/foo.png");
   URLTestHelpers::registerMockedURLLoad(url, testImageFilename, "image/png");
 
   FetchRequest fetchRequestOriginal = FetchRequest(url, FetchInitiatorInfo());
   fetchRequestOriginal.setLinkPreload(true);
   Resource* resource = ImageResource::fetch(fetchRequestOriginal, fetcher);
   ASSERT_TRUE(resource);
   EXPECT_TRUE(resource->isLinkPreload());
   Platform::current()->getURLLoaderMockFactory()->serveAsynchronousRequests();
@@ skipping 65 matching lines @@
   KURL url(ParsedURLString, "http://127.0.0.1:8000/foo.html");
   Resource* resource = RawResource::create(url, Resource::Raw);
   memoryCache()->add(resource);
   ResourceResponse response;
   response.setURL(url);
   response.setHTTPStatusCode(304);
   response.setHTTPHeaderField("etag", "1234567890");
   resource->responseReceived(response, nullptr);
   resource->finish();
 
-  ResourceFetcher* fetcher =
-      ResourceFetcher::create(ResourceFetcherTestMockFetchContext::create());
+  ResourceFetcher* fetcher = ResourceFetcher::create(
+      MockFetchContext::create(MockFetchContext::kShouldLoadNewResource));
   ResourceRequest resourceRequest(url);
   resourceRequest.setRequestContext(WebURLRequest::RequestContextInternal);
   FetchRequest fetchRequest =
       FetchRequest(resourceRequest, FetchInitiatorInfo());
   Platform::current()->getURLLoaderMockFactory()->registerURL(
       url, WebURLResponse(), "");
   Resource* newResource = RawResource::fetch(fetchRequest, fetcher);
   fetcher->stopFetching();
   Platform::current()->getURLLoaderMockFactory()->unregisterURL(url);
 
   EXPECT_NE(resource, newResource);
 }
 
 }  // namespace blink