Add a StyleResourceClient in CSSStyleSheetResourceTest to keep Resource alive

third_party/WebKit/Source/core/css/CSSStyleSheetResourceTest.cpp

 // Copyright 2016 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/CSSStyleSheetResource.h"
 
 #include "core/css/CSSCrossfadeValue.h"
 #include "core/css/CSSImageValue.h"
 #include "core/css/CSSPrimitiveValue.h"
 #include "core/css/CSSProperty.h"
 #include "core/css/CSSSelectorList.h"
 #include "core/css/CSSStyleSheet.h"
 #include "core/css/StylePropertySet.h"
 #include "core/css/StyleRule.h"
 #include "core/css/StyleSheetContents.h"
 #include "core/css/parser/CSSParserMode.h"
 #include "core/css/parser/CSSParserSelector.h"
 #include "core/dom/Document.h"
 #include "core/fetch/FetchContext.h"
 #include "core/fetch/FetchInitiatorTypeNames.h"
 #include "core/fetch/FetchRequest.h"
 #include "core/fetch/MemoryCache.h"
 #include "core/fetch/ResourceFetcher.h"
+#include "core/fetch/StyleSheetResourceClient.h"
 #include "core/testing/DummyPageHolder.h"
 #include "platform/heap/Handle.h"
 #include "platform/heap/Heap.h"
 #include "platform/network/ResourceRequest.h"
 #include "platform/testing/URLTestHelpers.h"
 #include "platform/testing/UnitTestHelpers.h"
 #include "platform/weborigin/KURL.h"
 #include "public/platform/Platform.h"
 #include "public/platform/WebURLResponse.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "wtf/RefPtr.h"
 #include "wtf/text/WTFString.h"
 
 namespace blink {
 
 class Document;
 
 namespace {
 
+class MockStyleSheetResourceClient : public StyleSheetResourceClient {
+public:
+    String debugName() const override { return "MockStyleSheetResourceClient"; }
+};
+
 class CSSStyleSheetResourceTest : public ::testing::Test {
 protected:
     CSSStyleSheetResourceTest()
     {
         m_originalMemoryCache = replaceMemoryCacheForTesting(MemoryCache::create());
         m_page = DummyPageHolder::create();
         document()->setURL(KURL(KURL(), "https://localhost/"));
     }
 
     ~CSSStyleSheetResourceTest() override
@@ skipping 12 matching lines @@
     {
         // We need this scope to remove all references.
         KURL imageURL(KURL(), "https://localhost/image");
         KURL cssURL(KURL(), "https://localhost/style.css");
 
         // We need to disable loading because we manually give a response to
         // the image resource.
         document()->fetcher()->setAutoLoadImages(false);
 
         CSSStyleSheetResource* cssResource = CSSStyleSheetResource::createForTest(ResourceRequest(cssURL), "utf-8");
+        MockStyleSheetResourceClient client;
+        cssResource->addClient(&client);
         memoryCache()->add(cssResource);
         cssResource->responseReceived(ResourceResponse(cssURL, "style/css", 0, nullAtom, String()), nullptr);
         cssResource->finish();
 
         StyleSheetContents* contents = StyleSheetContents::create(CSSParserContext(HTMLStandardMode, nullptr));
         CSSStyleSheet* sheet = CSSStyleSheet::create(contents, document());
         EXPECT_TRUE(sheet);
         CSSCrossfadeValue* crossfade = CSSCrossfadeValue::create(
             CSSImageValue::create(String("image"), imageURL),
             CSSImageValue::create(String("image"), imageURL),
@@ skipping 20 matching lines @@
         memoryCache()->update(cssResource, cssResource->size(), cssResource->size(), false);
         memoryCache()->update(imageResource, imageResource->size(), imageResource->size(), false);
         if (!memoryCache()->isInSameLRUListForTest(cssResource, imageResource)) {
             // We assume that the LRU list is determined by |size / accessCount|.
             for (size_t i = 0; i < cssResource->size() + 1; ++i)
                 memoryCache()->update(cssResource, cssResource->size(), cssResource->size(), true);
             for (size_t i = 0; i < imageResource->size() + 1; ++i)
                 memoryCache()->update(imageResource, imageResource->size(), imageResource->size(), true);
         }
         ASSERT_TRUE(memoryCache()->isInSameLRUListForTest(cssResource, imageResource));
+        cssResource->removeClient(&client);
     }
     ThreadHeap::collectAllGarbage();
     // This operation should not lead to crash!
     memoryCache()->pruneAll();
 }
 
 TEST_F(CSSStyleSheetResourceTest, DuplicateResourceNotCached)
 {
     const char url[] = "https://localhost/style.css";
     KURL imageURL(KURL(), url);
@@ skipping 22 matching lines @@
     // The underlying |contents| for the stylesheet resource must have a
     // matching reference status.
     EXPECT_TRUE(memoryCache()->contains(imageResource));
     EXPECT_FALSE(memoryCache()->contains(cssResource));
     EXPECT_FALSE(contents->isReferencedFromResource());
     EXPECT_FALSE(cssResource->restoreParsedStyleSheet(parserContext));
 }
 
 } // namespace
 } // namespace blink