Loading: Factor out ResourceFetcherMockFetchContext

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

 /*
  * Copyright (c) 2014, 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 13 matching lines @@
  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  * (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/FetchContext.h"
 #include "core/fetch/ImageResource.h"
 #include "core/fetch/MemoryCache.h"
+#include "core/fetch/MockFetchContext.h"
 #include "core/fetch/RawResource.h"
 #include "core/fetch/Resource.h"
 #include "core/fetch/ResourceFetcher.h"
 #include "platform/network/ResourceRequest.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "wtf/RefPtr.h"
 
 namespace blink {
 
 // An URL for the original request.
 const char kResourceURL[] = "http://resource.com/";
 
 // The origin time of our first request.
 const char kOriginalRequestDateAsString[] = "Thu, 25 May 1977 18:30:00 GMT";
 const double kOriginalRequestDateAsDouble = 233433000.;
 
 const char kOneDayBeforeOriginalRequest[] = "Wed, 24 May 1977 18:30:00 GMT";
 const char kOneDayAfterOriginalRequest[] = "Fri, 26 May 1977 18:30:00 GMT";
 
-class MockFetchContext : public FetchContext {
- public:
-  static MockFetchContext* create() { return new MockFetchContext; }
-
-  ~MockFetchContext() {}
-
-  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;
-  }
-
- private:
-  MockFetchContext() {}
-};
-
 class CachingCorrectnessTest : public ::testing::Test {
  protected:
   static void advanceClock(double seconds) { s_timeElapsed += seconds; }
 
   Resource* resourceFromResourceResponse(ResourceResponse response,
                                          Resource::Type type = Resource::Raw) {
     if (response.url().isNull())
       response.setURL(KURL(ParsedURLString, kResourceURL));
     Resource* resource = nullptr;
     switch (type) {
@@ skipping 49 matching lines @@
 
  private:
   static double returnMockTime() {
     return kOriginalRequestDateAsDouble + s_timeElapsed;
   }
 
   virtual void SetUp() {
     // Save the global memory cache to restore it upon teardown.
     m_globalMemoryCache = replaceMemoryCacheForTesting(MemoryCache::create());
 
-    m_fetcher = ResourceFetcher::create(MockFetchContext::create());
+    m_fetcher = ResourceFetcher::create(
+        MockFetchContext::create(MockFetchContext::kShouldNotLoadNewResource));
 
     s_timeElapsed = 0.0;
     m_originalTimeFunction = setTimeFunctionsForTesting(returnMockTime);
   }
 
   virtual void TearDown() {
     memoryCache()->evictResources();
 
     // Yield the ownership of the global memory cache back.
     replaceMemoryCacheForTesting(m_globalMemoryCache.release());
@@ skipping 446 matching lines @@
   firstResource->finish();
   memoryCache()->add(firstResource);
 
   advanceClock(500.);
 
   Resource* fetched = fetch();
   EXPECT_EQ(firstResource, fetched);
 }
 
 }  // namespace blink