WebFonts cache-aware timeout adaptation

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

 // Copyright 2015 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 "core/fetch/MockResourceClients.h"
 
 #include "core/fetch/ImageResource.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace blink {
@@ skipping 64 matching lines @@
   m_imageNotifyFinishedCount++;
   m_encodedSizeOnImageNotifyFinished = image->encodedSize();
 }
 
 bool MockImageResourceClient::notifyFinishedCalled() const {
   EXPECT_EQ(m_notifyFinishedCalled ? 1 : 0, m_imageNotifyFinishedCount);
 
   return m_notifyFinishedCalled;
 }
 
+MockFontResourceClient::MockFontResourceClient(Resource* resource)
+    : m_resource(resource),
+      m_fontLoadShortLimitExceededCalled(false),
+      m_fontLoadLongLimitExceededCalled(false) {
+  ThreadState::current()->registerPreFinalizer(this);
+  m_resource->addClient(this);
+}
+
+MockFontResourceClient::~MockFontResourceClient() {}
+
+void MockFontResourceClient::fontLoadShortLimitExceeded(FontResource*) {
+  ASSERT_FALSE(m_fontLoadShortLimitExceededCalled);
+  ASSERT_FALSE(m_fontLoadLongLimitExceededCalled);
+  m_fontLoadShortLimitExceededCalled = true;
+}
+
+void MockFontResourceClient::fontLoadLongLimitExceeded(FontResource*) {
+  ASSERT_TRUE(m_fontLoadShortLimitExceededCalled);
+  ASSERT_FALSE(m_fontLoadLongLimitExceededCalled);
+  m_fontLoadLongLimitExceededCalled = true;
+}
+
+void MockFontResourceClient::dispose() {
+  if (m_resource) {
+    m_resource->removeClient(this);
+    m_resource = nullptr;
+  }
+}
+
 }  // namespace blink