transferSize implementation

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 126 matching lines @@
       url, WebURLResponse(), "");
   Resource* newResource = RawResource::fetch(fetchRequest, fetcher);
   EXPECT_NE(resource, newResource);
   newResource->loader()->cancel();
   memoryCache()->remove(newResource);
   Platform::current()->getURLLoaderMockFactory()->unregisterURL(url);
 
   memoryCache()->remove(resource);
 }
 
+TEST_F(ResourceFetcherTest, MainResourceTimingInfo) {
+  KURL url(ParsedURLString, "http://127.0.0.1:8000/foo.html");
+  ResourceResponse response;
+  response.setURL(url);
+  response.setHTTPStatusCode(200);
+  long long encodedDataLength = 123;
+  response.setEncodedDataLength(encodedDataLength);
+
+  ResourceFetcher* fetcher = ResourceFetcher::create(
+      MockFetchContext::create(MockFetchContext::kShouldLoadNewResource));
+  ResourceRequest resourceRequest(url);
+  resourceRequest.setFrameType(WebURLRequest::FrameTypeNested);
+  resourceRequest.setRequestContext(WebURLRequest::RequestContextForm);
+  FetchRequest fetchRequest =
+      FetchRequest(resourceRequest, FetchInitiatorInfo());
+  Platform::current()->getURLLoaderMockFactory()->registerURL(
+      url, WebURLResponse(), "");
+  Resource* resource =
+      RawResource::fetchMainResource(fetchRequest, fetcher, SubstituteData());
+  resource->responseReceived(response, nullptr);
+  EXPECT_EQ(resource->getType(), Resource::MainResource);
+  ResourceTimingInfo* mainResourceTimingInfo =
+      fetcher->getMainResourceTimingInfo();
+  ASSERT_TRUE(mainResourceTimingInfo);
+  fetcher->didFinishLoading(resource, 0, ResourceFetcher::DidFinishLoading);
+  EXPECT_EQ(mainResourceTimingInfo->transferSize(), encodedDataLength);
+
+  // When an iframe is loaded.
+  Resource* iframeResource =
+      RawResource::create(ResourceRequest(), Resource::MainResource);
+  iframeResource->setIdentifier(resource->identifier() + 1);
+  ResourceResponse iframeResponse;
+  long long iframeEncodedDataLength = 123;
+  iframeResponse.setEncodedDataLength(iframeEncodedDataLength);
+  iframeResource->responseReceived(iframeResponse, nullptr);
+  fetcher->didFinishLoading(iframeResource, 0,
+                            ResourceFetcher::DidFinishLoading);
+  EXPECT_EQ(mainResourceTimingInfo->transferSize(), encodedDataLength);
+
+  // When there are redirects.
+  KURL redirectURL(ParsedURLString, "http://127.0.0.1:8000/redirect.html");
+  ResourceResponse redirectResponse;
+  redirectResponse.setURL(redirectURL);
+  redirectResponse.setHTTPStatusCode(200);
+  long long redirectEncodedDataLength = 123;
+  redirectResponse.setEncodedDataLength(redirectEncodedDataLength);
+  ResourceRequest redirectResourceRequest(url);
+  fetcher->willFollowRedirect(resource, redirectResourceRequest,
+                              redirectResponse);
+  EXPECT_EQ(mainResourceTimingInfo->transferSize(),
+            encodedDataLength + redirectEncodedDataLength);
+  Platform::current()->getURLLoaderMockFactory()->unregisterURL(url);
+}
+
 TEST_F(ResourceFetcherTest, VaryOnBack) {
   MockFetchContext* context =
       MockFetchContext::create(MockFetchContext::kShouldLoadNewResource);
   context->setCachePolicy(CachePolicyHistoryBuffer);
   ResourceFetcher* fetcher = ResourceFetcher::create(context);
 
   KURL url(ParsedURLString, "http://127.0.0.1:8000/foo.html");
   Resource* resource = RawResource::create(url, Resource::Raw);
   memoryCache()->add(resource);
   ResourceResponse response;
@@ skipping 478 matching lines @@
   FetchRequest fetchRequest2 = FetchRequest(url, FetchInitiatorInfo());
   Resource* newResource2 = ImageResource::fetch(fetchRequest2, fetcher2);
   Persistent<MockResourceClient> client2 = new MockResourceClient(newResource2);
   EXPECT_EQ(resource, newResource2);
   EXPECT_FALSE(fetcher2->isFetching());
   Platform::current()->getURLLoaderMockFactory()->serveAsynchronousRequests();
   Platform::current()->getURLLoaderMockFactory()->unregisterURL(url);
 }
 
 }  // namespace blink