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

   1 // Copyright 2015 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "config.h"
   6 #include "core/fetch/Resource.h"
   7
   8 #include "core/fetch/ResourcePtr.h"
   9 #include "platform/network/ResourceRequest.h"
  10 #include "platform/network/ResourceResponse.h"
  11 #include "platform/testing/URLTestHelpers.h"
  12 #include "public/platform/Platform.h"
  13 #include "wtf/Vector.h"
  14
  15 #include <gtest/gtest.h>
  16
  17 using namespace blink;
  18
  19 namespace blink {
  20
  21 namespace {
  22
  23 class MockPlatform final : public Platform {
  24 public:
  25     MockPlatform() : m_oldPlatform(Platform::current()) { }
  26     ~MockPlatform() override { }
  27
  28     // From blink::Platform:
  29     void cacheMetadata(const WebURL& url, int64, const char*, size_t) override
  30     {
  31         m_cachedURLs.append(url);
  32     }
  33     void cryptographicallyRandomValues(unsigned char* buffer, size_t length) override
  34     {
  35         ASSERT_NOT_REACHED();
  36     }
  37
  38     const Vector<WebURL>& cachedURLs() const
  39     {
  40         return m_cachedURLs;
  41     }
  42
  43     WebThread* currentThread() override
  44     {
  45         return m_oldPlatform->currentThread();
  46     }
  47
  48 private:
  49     Platform* m_oldPlatform; // Not owned.
  50     Vector<WebURL> m_cachedURLs;
  51 };
  52
  53 class AutoInstallMockPlatform {
  54 public:
  55     AutoInstallMockPlatform()
  56     {
  57         m_oldPlatform = Platform::current();
  58         Platform::initialize(&m_mockPlatform);
  59     }
  60     ~AutoInstallMockPlatform()
  61     {
  62         Platform::initialize(m_oldPlatform);
  63     }
  64     MockPlatform* platform() { return &m_mockPlatform; }
  65 private:
  66     MockPlatform m_mockPlatform;
  67     Platform* m_oldPlatform;
  68 };
  69
  70 PassOwnPtr<ResourceResponse> createTestResourceResponse()
  71 {
  72     OwnPtr<ResourceResponse> response = adoptPtr(new ResourceResponse);
  73     response->setURL(URLTestHelpers::toKURL("https://example.com/"));
  74     response->setHTTPStatusCode(200);
  75     return response.release();
  76 }
  77
  78 void createTestResourceAndSetCachedMetadata(const ResourceResponse* response)
  79 {
  80     const char testData[] = "test data";
  81     ResourcePtr<Resource> resource = new Resource(ResourceRequest(response->url()), Resource::Raw);
  82     resource->setResponse(*response);
  83     resource->cacheHandler()->setCachedMetadata(100, testData, sizeof(testData), CachedMetadataHandler::SendToPlatform);
  84     return;
  85 }
  86
  87 } // namespace
  88
  89 TEST(ResourceTest, SetCachedMetadata_SendsMetadataToPlatform)
  90 {
  91     AutoInstallMockPlatform mock;
  92     OwnPtr<ResourceResponse> response(createTestResourceResponse());
  93     createTestResourceAndSetCachedMetadata(response.get());
  94     EXPECT_EQ(1u, mock.platform()->cachedURLs().size());
  95 }
  96
  97 TEST(ResourceTest, SetCachedMetadata_DoesNotSendMetadataToPlatformWhenFetchedViaServiceWorker)
  98 {
  99     AutoInstallMockPlatform mock;
 100     OwnPtr<ResourceResponse> response(createTestResourceResponse());
 101     response->setWasFetchedViaServiceWorker(true);
 102     createTestResourceAndSetCachedMetadata(response.get());
 103     EXPECT_EQ(0u, mock.platform()->cachedURLs().size());
 104 }
 105
 106 } // namespace blink