WebFonts cache-aware timeout adaptation

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 26 matching lines @@
 #include "core/fetch/ImageResource.h"
 #include "core/fetch/MemoryCache.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/ResourceError.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"
 #include "platform/testing/weburl_loader_mock.h"
 #include "platform/weborigin/KURL.h"
 #include "public/platform/Platform.h"
 #include "public/platform/WebURLLoaderMockFactory.h"
+#include "public/platform/WebURLLoaderTestDelegate.h"
 #include "public/platform/WebURLResponse.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "wtf/Allocator.h"
 #include "wtf/PtrUtil.h"
 #include "wtf/Vector.h"
 #include <memory>
 
 namespace blink {
 
 namespace {
@@ skipping 644 matching lines @@
       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);
 }
 
+// Wrapper for accessing private members of FontResource.
+class FontResourceTestWrapper {
+  STACK_ALLOCATED();
+
+ public:
+  FontResourceTestWrapper(FontResource* resource) : m_resource(resource) {}
+  FontResource* operator->() const { return m_resource.get(); }

[yhirano, 2016/11/09 07:29:52]

I generally don't like define operators: Are these really needed? There are
other solutions, such as

1. FontResourceTestWrapper doesn't hold a FontResource. Instead, each function
takes a FontResource.
2. Make ResourceFetcher friend of FontResource
3. Expose private functions with the ForTesting suffix.

What do you think?

[Shao-Chuan Lee, 2016/11/09 08:04:31]

Now using 1. and methods are made static. This should be more elegant.

+  operator FontResource*() const { return m_resource.get(); }
+
+  void initLoadLimitState() {
+    m_resource->m_loadLimitState = FontResource::UnderLimit;
+  }
+
+  void callFontLoadShortLimitCallback() {
+    m_resource->fontLoadShortLimitCallback(nullptr);
+  }
+
+  void callFontLoadLongLimitCallback() {
+    m_resource->fontLoadLongLimitCallback(nullptr);
+  }
+
+ private:
+  Member<FontResource> m_resource;
+};
+
+TEST_F(ResourceFetcherTest, CacheAwareFontLoading) {
+  KURL url(ParsedURLString, "http://127.0.0.1:8000/font.woff");
+  ResourceResponse response;
+  response.setURL(url);
+  response.setHTTPStatusCode(200);
+  Platform::current()->getURLLoaderMockFactory()->registerURL(
+      url, WrappedResourceResponse(response), "");
+
+  ResourceFetcher* fetcher =
+      ResourceFetcher::create(ResourceFetcherTestMockFetchContext::create());
+
+  FetchRequest fetchRequest = FetchRequest(url, FetchInitiatorInfo());
+  fetchRequest.setCacheAwareLoadingEnabled(IsCacheAwareLoadingEnabled);
+  FontResourceTestWrapper resource(FontResource::fetch(fetchRequest, fetcher));
+  ASSERT_TRUE(resource);
+
+  Persistent<MockFontResourceClient> client =
+      new MockFontResourceClient(resource);
+  fetcher->startLoad(resource);
+  EXPECT_TRUE(resource->loader()->isCacheAwareLoadingActivated());
+  resource.initLoadLimitState();
+
+  // FontResource callbacks should be blocked during cache-aware loading.
+  resource.callFontLoadShortLimitCallback();
+  EXPECT_FALSE(client->fontLoadShortLimitExceededCalled());
+
+  // Fail first request as disk cache miss.
+  resource->loader()->didFail(ResourceError::cacheMissError(url));
+
+  // Once cache miss error returns, previously blocked callbacks should be
+  // called immediately.
+  EXPECT_FALSE(resource->loader()->isCacheAwareLoadingActivated());
+  EXPECT_TRUE(client->fontLoadShortLimitExceededCalled());
+  EXPECT_FALSE(client->fontLoadLongLimitExceededCalled());
+
+  // FontResource callbacks are not blocked now.
+  resource.callFontLoadLongLimitCallback();
+  EXPECT_TRUE(client->fontLoadLongLimitExceededCalled());
+
+  Platform::current()->getURLLoaderMockFactory()->serveAsynchronousRequests();
+  Platform::current()->getURLLoaderMockFactory()->unregisterURL(url);
+  memoryCache()->remove(resource);
+}
+
 }  // namespace blink