Send the decoded size when response completed and stop summing in ResourceLoader::DidReceiveData()

chrome/renderer/extensions/extension_localization_peer_unittest.cc

 // Copyright 2013 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 "chrome/renderer/extensions/extension_localization_peer.h"
 
 #include <stdint.h>
 
 #include <map>
 #include <memory>
@@ skipping 58 matching lines @@
                     const content::ResourceResponseInfo& info));
   MOCK_METHOD1(OnReceivedResponse,
                void(const content::ResourceResponseInfo& info));
   MOCK_METHOD2(OnDownloadedData, void(int len, int encoded_data_length));
   void OnReceivedData(
       std::unique_ptr<RequestPeer::ReceivedData> data) override {
     OnReceivedDataInternal(data->payload(), data->length());
   }
   MOCK_METHOD2(OnReceivedDataInternal, void(const char* data, int data_length));
   MOCK_METHOD1(OnTransferSizeUpdated, void(int transfer_size_diff));
-  MOCK_METHOD6(OnCompletedRequest,
+  MOCK_METHOD7(OnCompletedRequest,
                void(int error_code,
                     bool was_ignored_by_handler,
                     bool stale_copy_in_cache,
                     const base::TimeTicks& completion_time,
                     int64_t total_transfer_size,
-                    int64_t encoded_body_size));
+                    int64_t encoded_body_size,
+                    int64_t decoded_body_size));
 
  private:
   DISALLOW_COPY_AND_ASSIGN(MockRequestPeer);
 };
 
 }  // namespace
 
 class ExtensionLocalizationPeerTest : public testing::Test {
  protected:
   void SetUp() override {
@@ skipping 50 matching lines @@
 }
 
 MATCHER_P(IsURLRequestEqual, status, "") { return arg.status() == status; }
 
 TEST_F(ExtensionLocalizationPeerTest, OnCompletedRequestBadURLRequestStatus) {
   SetUpExtensionLocalizationPeer("text/css", GURL(kExtensionUrl_1));
 
   EXPECT_CALL(*original_peer_, OnReceivedResponse(_));
   EXPECT_CALL(*original_peer_,
               OnCompletedRequest(net::ERR_ABORTED, false, false,
-                                 base::TimeTicks(), -1, 0));
+                                 base::TimeTicks(), -1, 0, 0));
 
   filter_peer_->OnCompletedRequest(net::ERR_FAILED, false, false,
-                                   base::TimeTicks(), -1, 0);
+                                   base::TimeTicks(), -1, 0, 0);
 }
 
 TEST_F(ExtensionLocalizationPeerTest, OnCompletedRequestEmptyData) {
   SetUpExtensionLocalizationPeer("text/css", GURL(kExtensionUrl_1));
 
   EXPECT_CALL(*original_peer_, OnReceivedDataInternal(_, _)).Times(0);
   EXPECT_CALL(*sender_, Send(_)).Times(0);
 
   EXPECT_CALL(*original_peer_, OnReceivedResponse(_));
   EXPECT_CALL(*original_peer_, OnCompletedRequest(net::OK, false, false,
-                                                  base::TimeTicks(), -1, 0));
+                                                  base::TimeTicks(), -1, 0, 0));
 
   filter_peer_->OnCompletedRequest(net::OK, false, false, base::TimeTicks(), -1,
-                                   0);
+                                   0, 0);
 }
 
 TEST_F(ExtensionLocalizationPeerTest, OnCompletedRequestNoCatalogs) {
   SetUpExtensionLocalizationPeer("text/css", GURL(kExtensionUrl_1));
 
   SetData("some text");
 
   EXPECT_CALL(*sender_, Send(_));
 
   std::string data = GetData();
   EXPECT_CALL(*original_peer_, OnReceivedResponse(_)).Times(1);
   EXPECT_CALL(*original_peer_,
               OnReceivedDataInternal(StrEq(data.c_str()), data.length()))
       .Times(1);
-  EXPECT_CALL(*original_peer_, OnCompletedRequest(net::OK, false, false,
-                                                  base::TimeTicks(), -1, -1))
+  EXPECT_CALL(
+      *original_peer_,
+      OnCompletedRequest(net::OK, false, false, base::TimeTicks(), -1, -1, -1))
       .Times(1);
 
   filter_peer_->OnCompletedRequest(net::OK, false, false, base::TimeTicks(), -1,
-                                   -1);
+                                   -1, -1);
 
   // Test if Send gets called again (it shouldn't be) when first call returned
   // an empty dictionary.
   SetUpExtensionLocalizationPeer("text/css", GURL(kExtensionUrl_1));
   EXPECT_CALL(*original_peer_, OnReceivedResponse(_)).Times(1);
   EXPECT_CALL(*original_peer_,
               OnReceivedDataInternal(StrEq(data.c_str()), data.length()))
       .Times(1);
-  EXPECT_CALL(*original_peer_, OnCompletedRequest(net::OK, false, false,
-                                                  base::TimeTicks(), -1, -1))
+  EXPECT_CALL(
+      *original_peer_,
+      OnCompletedRequest(net::OK, false, false, base::TimeTicks(), -1, -1, -1))
       .Times(1);
   SetData("some text");
   filter_peer_->OnCompletedRequest(net::OK, false, false, base::TimeTicks(), -1,
-                                   -1);
+                                   -1, -1);
 }
 
 TEST_F(ExtensionLocalizationPeerTest, OnCompletedRequestWithCatalogs) {
   SetUpExtensionLocalizationPeer("text/css", GURL(kExtensionUrl_2));
 
   extensions::L10nMessagesMap messages;
   messages.insert(std::make_pair("text", "new text"));
   extensions::ExtensionToL10nMessagesMap& l10n_messages_map =
       *extensions::GetExtensionToL10nMessagesMap();
   l10n_messages_map["some_id2"] = messages;
 
   SetData("some __MSG_text__");
 
   // We already have messages in memory, Send will be skipped.
   EXPECT_CALL(*sender_, Send(_)).Times(0);
 
   // __MSG_text__ gets replaced with "new text".
   std::string data("some new text");
   EXPECT_CALL(*original_peer_, OnReceivedResponse(_));
   EXPECT_CALL(*original_peer_,
               OnReceivedDataInternal(StrEq(data.c_str()), data.length()));
 
-  EXPECT_CALL(*original_peer_, OnCompletedRequest(net::OK, false, false,
-                                                  base::TimeTicks(), -1, -1));
+  EXPECT_CALL(
+      *original_peer_,
+      OnCompletedRequest(net::OK, false, false, base::TimeTicks(), -1, -1, -1));
 
   filter_peer_->OnCompletedRequest(net::OK, false, false, base::TimeTicks(), -1,
-                                   -1);
+                                   -1, -1);
 }
 
 TEST_F(ExtensionLocalizationPeerTest, OnCompletedRequestReplaceMessagesFails) {
   SetUpExtensionLocalizationPeer("text/css", GURL(kExtensionUrl_3));
 
   extensions::L10nMessagesMap messages;
   messages.insert(std::make_pair("text", "new text"));
   extensions::ExtensionToL10nMessagesMap& l10n_messages_map =
       *extensions::GetExtensionToL10nMessagesMap();
   l10n_messages_map["some_id3"] = messages;
 
   std::string message("some __MSG_missing_message__");
   SetData(message);
 
   // We already have messages in memory, Send will be skipped.
   EXPECT_CALL(*sender_, Send(_)).Times(0);
 
   // __MSG_missing_message__ is missing, so message stays the same.
   EXPECT_CALL(*original_peer_, OnReceivedResponse(_));
   EXPECT_CALL(*original_peer_,
               OnReceivedDataInternal(StrEq(message.c_str()), message.length()));
 
   EXPECT_CALL(*original_peer_, OnCompletedRequest(net::OK, false, false,
-                                                  base::TimeTicks(), -1, 0));
+                                                  base::TimeTicks(), -1, 0, 0));
 
   filter_peer_->OnCompletedRequest(net::OK, false, false, base::TimeTicks(), -1,
-                                   0);
+                                   0, 0);
 }