Turn off isLinkPreload() when used

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 16 matching lines @@
  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  */
 
 #include "core/fetch/ResourceFetcher.h"
 
 #include "core/fetch/FetchInitiatorInfo.h"
 #include "core/fetch/FetchInitiatorTypeNames.h"
 #include "core/fetch/FetchRequest.h"
 #include "core/fetch/MemoryCache.h"
+#include "core/fetch/MockResourceClients.h"
 #include "core/fetch/RawResource.h"
 #include "core/fetch/ResourceLoader.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"
 #include "platform/scheduler/test/fake_web_task_runner.h"
 #include "platform/testing/URLTestHelpers.h"
@@ skipping 459 matching lines @@
     response.setHTTPStatusCode(200);
     URLTestHelpers::registerMockedURLLoadWithCustomResponse(url, testImageFilename, WebString::fromUTF8(""), WrappedResourceResponse(response));
 
     ResourceFetcher* fetcher = ResourceFetcher::create(ResourceFetcherTestMockFetchContext::create());
     FetchRequest request(url, FetchInitiatorInfo());
     request.makeSynchronous();
     Resource* resource = fetcher->requestResource(request, TestResourceFactory());
     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());
+
+    KURL url(ParsedURLString, "http://127.0.0.1:8000/foo.html");
+    ResourceResponse response;
+    response.setURL(url);
+    response.setHTTPStatusCode(200);
+    URLTestHelpers::registerMockedURLLoadWithCustomResponse(url, testImageFilename, WebString::fromUTF8(""), WrappedResourceResponse(response));
+
+    FetchRequest fetchRequestOriginal = FetchRequest(url, FetchInitiatorInfo());
+    Resource* resource = fetcher->requestResource(fetchRequestOriginal, TestResourceFactory(Resource::Image));
+    ASSERT_TRUE(resource);
+    Platform::current()->getURLLoaderMockFactory()->serveAsynchronousRequests();
+    fetcher->preloadStarted(resource);
+
+    FetchRequest fetchRequest = FetchRequest(url, FetchInitiatorInfo());
+    Resource* newResource = fetcher->requestResource(fetchRequest, TestResourceFactory(Resource::Image));
+    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());
+
+    KURL url(ParsedURLString, "http://127.0.0.1:8000/foo.html");
+    ResourceResponse response;
+    response.setURL(url);
+    response.setHTTPStatusCode(200);
+    URLTestHelpers::registerMockedURLLoadWithCustomResponse(url, testImageFilename, WebString::fromUTF8(""), WrappedResourceResponse(response));
+
+    // Link preload preload scanner
+    FetchRequest fetchRequestOriginal = FetchRequest(url, FetchInitiatorInfo());
+    fetchRequestOriginal.setLinkPreload(true);
+    Resource* resource = fetcher->requestResource(fetchRequestOriginal, TestResourceFactory(Resource::Image));
+    ASSERT_TRUE(resource);
+    EXPECT_TRUE(resource->isLinkPreload());
+    Platform::current()->getURLLoaderMockFactory()->serveAsynchronousRequests();
+    fetcher->preloadStarted(resource);
+
+    // Image preload scanner
+    FetchRequest fetchRequestPreloadScanner = FetchRequest(url, FetchInitiatorInfo());
+    Resource* imgPreloadScannerResource = fetcher->requestResource(fetchRequestPreloadScanner, TestResourceFactory(Resource::Image));
+    EXPECT_EQ(resource, imgPreloadScannerResource);
+    EXPECT_FALSE(resource->isLinkPreload());
+    fetcher->preloadStarted(resource);
+
+    // Image created by parser
+    FetchRequest fetchRequest = FetchRequest(url, FetchInitiatorInfo());
+    Resource* newResource = fetcher->requestResource(fetchRequest, TestResourceFactory(Resource::Image));
+    newResource->testOnlyWillAddClientOrObserver(Resource::MarkAsReferenced);
+    EXPECT_EQ(resource, newResource);
+    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());
+
+    KURL url(ParsedURLString, "http://127.0.0.1:8000/foo.html");
+    ResourceResponse response;
+    response.setURL(url);
+    response.setHTTPStatusCode(200);
+    URLTestHelpers::registerMockedURLLoadWithCustomResponse(url, testImageFilename, WebString::fromUTF8(""), WrappedResourceResponse(response));
+
+    FetchRequest fetchRequestOriginal = FetchRequest(url, FetchInitiatorInfo());
+    fetchRequestOriginal.setLinkPreload(true);
+    Resource* resource = fetcher->requestResource(fetchRequestOriginal, TestResourceFactory(Resource::Image));
+    ASSERT_TRUE(resource);
+    EXPECT_TRUE(resource->isLinkPreload());
+    Platform::current()->getURLLoaderMockFactory()->serveAsynchronousRequests();
+    fetcher->preloadStarted(resource);
+
+    FetchRequest fetchRequestSecond = FetchRequest(url, FetchInitiatorInfo());
+    fetchRequestSecond.setLinkPreload(true);
+    Resource* secondResource = fetcher2->requestResource(fetchRequestSecond, TestResourceFactory(Resource::Image));
+    ASSERT_TRUE(secondResource);
+    EXPECT_TRUE(secondResource->isLinkPreload());
+    Platform::current()->getURLLoaderMockFactory()->serveAsynchronousRequests();
+    fetcher2->preloadStarted(secondResource);
+
+    // Image preload scanner
+    FetchRequest fetchRequestPreloadScanner = FetchRequest(url, FetchInitiatorInfo());
+    Resource* imgPreloadScannerResource = fetcher->requestResource(fetchRequestPreloadScanner, TestResourceFactory(Resource::Image));
+    EXPECT_EQ(resource, imgPreloadScannerResource);
+    EXPECT_FALSE(resource->isLinkPreload());
+    fetcher->preloadStarted(resource);
+
+    // Image created by parser
+    FetchRequest fetchRequest = FetchRequest(url, FetchInitiatorInfo());
+    Resource* newResource = fetcher->requestResource(fetchRequest, TestResourceFactory(Resource::Image));
+    newResource->testOnlyWillAddClientOrObserver(Resource::MarkAsReferenced);
+    EXPECT_EQ(resource, newResource);
+    EXPECT_FALSE(resource->isLinkPreload());
+
+    // DCL reached on first fetcher
+    EXPECT_TRUE(resource->isPreloaded());
+    fetcher->clearPreloads(ResourceFetcher::ClearSpeculativeMarkupPreloads);
+    Platform::current()->getURLLoaderMockFactory()->unregisterURL(url);
+    EXPECT_TRUE(memoryCache()->contains(resource));
+    EXPECT_TRUE(resource->isPreloaded());
+
+    // DCL reached on second fetcher
+    fetcher2->clearPreloads(ResourceFetcher::ClearSpeculativeMarkupPreloads);
+    EXPECT_TRUE(memoryCache()->contains(resource));
+    EXPECT_FALSE(resource->isPreloaded());
+}
+
 } // namespace blink