| OLD | NEW |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "chrome/renderer/extensions/extension_localization_peer.h" | 5 #include "chrome/renderer/extensions/extension_localization_peer.h" |
| 6 | 6 |
| 7 #include <stdint.h> | 7 #include <stdint.h> |
| 8 | 8 |
| 9 #include <map> | 9 #include <map> |
| 10 #include <memory> | 10 #include <memory> |
| (...skipping 54 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 65 | 65 |
| 66 MOCK_METHOD2(OnUploadProgress, void(uint64_t position, uint64_t size)); | 66 MOCK_METHOD2(OnUploadProgress, void(uint64_t position, uint64_t size)); |
| 67 MOCK_METHOD2(OnReceivedRedirect, | 67 MOCK_METHOD2(OnReceivedRedirect, |
| 68 bool(const net::RedirectInfo& redirect_info, | 68 bool(const net::RedirectInfo& redirect_info, |
| 69 const content::ResourceResponseInfo& info)); | 69 const content::ResourceResponseInfo& info)); |
| 70 MOCK_METHOD1(OnReceivedResponse, | 70 MOCK_METHOD1(OnReceivedResponse, |
| 71 void(const content::ResourceResponseInfo& info)); | 71 void(const content::ResourceResponseInfo& info)); |
| 72 MOCK_METHOD2(OnDownloadedData, void(int len, int encoded_data_length)); | 72 MOCK_METHOD2(OnDownloadedData, void(int len, int encoded_data_length)); |
| 73 void OnReceivedData( | 73 void OnReceivedData( |
| 74 std::unique_ptr<RequestPeer::ReceivedData> data) override { | 74 std::unique_ptr<RequestPeer::ReceivedData> data) override { |
| 75 OnReceivedDataInternal(data->payload(), data->length(), | 75 OnReceivedDataInternal(data->payload(), data->length()); |
| 76 data->encoded_data_length()); | |
| 77 } | 76 } |
| 78 MOCK_METHOD3(OnReceivedDataInternal, | 77 MOCK_METHOD2(OnReceivedDataInternal, void(const char* data, int data_length)); |
| 79 void(const char* data, | 78 MOCK_METHOD1(OnTransferSizeUpdated, void(int transfer_size_diff)); |
| 80 int data_length, | |
| 81 int encoded_data_length)); | |
| 82 MOCK_METHOD6(OnCompletedRequest, | 79 MOCK_METHOD6(OnCompletedRequest, |
| 83 void(int error_code, | 80 void(int error_code, |
| 84 bool was_ignored_by_handler, | 81 bool was_ignored_by_handler, |
| 85 bool stale_copy_in_cache, | 82 bool stale_copy_in_cache, |
| 86 const base::TimeTicks& completion_time, | 83 const base::TimeTicks& completion_time, |
| 87 int64_t total_transfer_size, | 84 int64_t total_transfer_size, |
| 88 int64_t encoded_body_size)); | 85 int64_t encoded_body_size)); |
| 89 | 86 |
| 90 private: | 87 private: |
| 91 DISALLOW_COPY_AND_ASSIGN(MockRequestPeer); | 88 DISALLOW_COPY_AND_ASSIGN(MockRequestPeer); |
| (...skipping 40 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 132 SetUpExtensionLocalizationPeer("text/css", GURL(kExtensionUrl_1)); | 129 SetUpExtensionLocalizationPeer("text/css", GURL(kExtensionUrl_1)); |
| 133 EXPECT_TRUE(NULL != filter_peer_.get()); | 130 EXPECT_TRUE(NULL != filter_peer_.get()); |
| 134 } | 131 } |
| 135 | 132 |
| 136 TEST_F(ExtensionLocalizationPeerTest, OnReceivedData) { | 133 TEST_F(ExtensionLocalizationPeerTest, OnReceivedData) { |
| 137 SetUpExtensionLocalizationPeer("text/css", GURL(kExtensionUrl_1)); | 134 SetUpExtensionLocalizationPeer("text/css", GURL(kExtensionUrl_1)); |
| 138 EXPECT_TRUE(GetData().empty()); | 135 EXPECT_TRUE(GetData().empty()); |
| 139 | 136 |
| 140 const std::string data_chunk("12345"); | 137 const std::string data_chunk("12345"); |
| 141 filter_peer_->OnReceivedData(base::MakeUnique<content::FixedReceivedData>( | 138 filter_peer_->OnReceivedData(base::MakeUnique<content::FixedReceivedData>( |
| 142 data_chunk.data(), data_chunk.length(), -1)); | 139 data_chunk.data(), data_chunk.length())); |
| 143 | 140 |
| 144 EXPECT_EQ(data_chunk, GetData()); | 141 EXPECT_EQ(data_chunk, GetData()); |
| 145 | 142 |
| 146 filter_peer_->OnReceivedData(base::MakeUnique<content::FixedReceivedData>( | 143 filter_peer_->OnReceivedData(base::MakeUnique<content::FixedReceivedData>( |
| 147 data_chunk.data(), data_chunk.length(), -1)); | 144 data_chunk.data(), data_chunk.length())); |
| 148 EXPECT_EQ(data_chunk + data_chunk, GetData()); | 145 EXPECT_EQ(data_chunk + data_chunk, GetData()); |
| 149 } | 146 } |
| 150 | 147 |
| 151 MATCHER_P(IsURLRequestEqual, status, "") { return arg.status() == status; } | 148 MATCHER_P(IsURLRequestEqual, status, "") { return arg.status() == status; } |
| 152 | 149 |
| 153 TEST_F(ExtensionLocalizationPeerTest, OnCompletedRequestBadURLRequestStatus) { | 150 TEST_F(ExtensionLocalizationPeerTest, OnCompletedRequestBadURLRequestStatus) { |
| 154 SetUpExtensionLocalizationPeer("text/css", GURL(kExtensionUrl_1)); | 151 SetUpExtensionLocalizationPeer("text/css", GURL(kExtensionUrl_1)); |
| 155 | 152 |
| 156 EXPECT_CALL(*original_peer_, OnReceivedResponse(_)); | 153 EXPECT_CALL(*original_peer_, OnReceivedResponse(_)); |
| 157 EXPECT_CALL(*original_peer_, | 154 EXPECT_CALL(*original_peer_, |
| 158 OnCompletedRequest(net::ERR_ABORTED, false, false, | 155 OnCompletedRequest(net::ERR_ABORTED, false, false, |
| 159 base::TimeTicks(), -1, 0)); | 156 base::TimeTicks(), -1, 0)); |
| 160 | 157 |
| 161 filter_peer_->OnCompletedRequest(net::ERR_FAILED, false, false, | 158 filter_peer_->OnCompletedRequest(net::ERR_FAILED, false, false, |
| 162 base::TimeTicks(), -1, 0); | 159 base::TimeTicks(), -1, 0); |
| 163 } | 160 } |
| 164 | 161 |
| 165 TEST_F(ExtensionLocalizationPeerTest, OnCompletedRequestEmptyData) { | 162 TEST_F(ExtensionLocalizationPeerTest, OnCompletedRequestEmptyData) { |
| 166 SetUpExtensionLocalizationPeer("text/css", GURL(kExtensionUrl_1)); | 163 SetUpExtensionLocalizationPeer("text/css", GURL(kExtensionUrl_1)); |
| 167 | 164 |
| 168 EXPECT_CALL(*original_peer_, OnReceivedDataInternal(_, _, _)).Times(0); | 165 EXPECT_CALL(*original_peer_, OnReceivedDataInternal(_, _)).Times(0); |
| 169 EXPECT_CALL(*sender_, Send(_)).Times(0); | 166 EXPECT_CALL(*sender_, Send(_)).Times(0); |
| 170 | 167 |
| 171 EXPECT_CALL(*original_peer_, OnReceivedResponse(_)); | 168 EXPECT_CALL(*original_peer_, OnReceivedResponse(_)); |
| 172 EXPECT_CALL(*original_peer_, OnCompletedRequest(net::OK, false, false, | 169 EXPECT_CALL(*original_peer_, OnCompletedRequest(net::OK, false, false, |
| 173 base::TimeTicks(), -1, 0)); | 170 base::TimeTicks(), -1, 0)); |
| 174 | 171 |
| 175 filter_peer_->OnCompletedRequest(net::OK, false, false, base::TimeTicks(), -1, | 172 filter_peer_->OnCompletedRequest(net::OK, false, false, base::TimeTicks(), -1, |
| 176 0); | 173 0); |
| 177 } | 174 } |
| 178 | 175 |
| 179 TEST_F(ExtensionLocalizationPeerTest, OnCompletedRequestNoCatalogs) { | 176 TEST_F(ExtensionLocalizationPeerTest, OnCompletedRequestNoCatalogs) { |
| 180 SetUpExtensionLocalizationPeer("text/css", GURL(kExtensionUrl_1)); | 177 SetUpExtensionLocalizationPeer("text/css", GURL(kExtensionUrl_1)); |
| 181 | 178 |
| 182 SetData("some text"); | 179 SetData("some text"); |
| 183 | 180 |
| 184 EXPECT_CALL(*sender_, Send(_)); | 181 EXPECT_CALL(*sender_, Send(_)); |
| 185 | 182 |
| 186 std::string data = GetData(); | 183 std::string data = GetData(); |
| 187 EXPECT_CALL(*original_peer_, OnReceivedResponse(_)).Times(1); | 184 EXPECT_CALL(*original_peer_, OnReceivedResponse(_)).Times(1); |
| 188 EXPECT_CALL(*original_peer_, | 185 EXPECT_CALL(*original_peer_, |
| 189 OnReceivedDataInternal(StrEq(data.c_str()), data.length(), -1)) | 186 OnReceivedDataInternal(StrEq(data.c_str()), data.length())) |
| 190 .Times(1); | 187 .Times(1); |
| 191 EXPECT_CALL(*original_peer_, OnCompletedRequest(net::OK, false, false, | 188 EXPECT_CALL(*original_peer_, OnCompletedRequest(net::OK, false, false, |
| 192 base::TimeTicks(), -1, -1)) | 189 base::TimeTicks(), -1, -1)) |
| 193 .Times(1); | 190 .Times(1); |
| 194 | 191 |
| 195 filter_peer_->OnCompletedRequest(net::OK, false, false, base::TimeTicks(), -1, | 192 filter_peer_->OnCompletedRequest(net::OK, false, false, base::TimeTicks(), -1, |
| 196 -1); | 193 -1); |
| 197 | 194 |
| 198 // Test if Send gets called again (it shouldn't be) when first call returned | 195 // Test if Send gets called again (it shouldn't be) when first call returned |
| 199 // an empty dictionary. | 196 // an empty dictionary. |
| 200 SetUpExtensionLocalizationPeer("text/css", GURL(kExtensionUrl_1)); | 197 SetUpExtensionLocalizationPeer("text/css", GURL(kExtensionUrl_1)); |
| 201 EXPECT_CALL(*original_peer_, OnReceivedResponse(_)).Times(1); | 198 EXPECT_CALL(*original_peer_, OnReceivedResponse(_)).Times(1); |
| 202 EXPECT_CALL(*original_peer_, | 199 EXPECT_CALL(*original_peer_, |
| 203 OnReceivedDataInternal(StrEq(data.c_str()), data.length(), -1)) | 200 OnReceivedDataInternal(StrEq(data.c_str()), data.length())) |
| 204 .Times(1); | 201 .Times(1); |
| 205 EXPECT_CALL(*original_peer_, OnCompletedRequest(net::OK, false, false, | 202 EXPECT_CALL(*original_peer_, OnCompletedRequest(net::OK, false, false, |
| 206 base::TimeTicks(), -1, -1)) | 203 base::TimeTicks(), -1, -1)) |
| 207 .Times(1); | 204 .Times(1); |
| 208 SetData("some text"); | 205 SetData("some text"); |
| 209 filter_peer_->OnCompletedRequest(net::OK, false, false, base::TimeTicks(), -1, | 206 filter_peer_->OnCompletedRequest(net::OK, false, false, base::TimeTicks(), -1, |
| 210 -1); | 207 -1); |
| 211 } | 208 } |
| 212 | 209 |
| 213 TEST_F(ExtensionLocalizationPeerTest, OnCompletedRequestWithCatalogs) { | 210 TEST_F(ExtensionLocalizationPeerTest, OnCompletedRequestWithCatalogs) { |
| 214 SetUpExtensionLocalizationPeer("text/css", GURL(kExtensionUrl_2)); | 211 SetUpExtensionLocalizationPeer("text/css", GURL(kExtensionUrl_2)); |
| 215 | 212 |
| 216 extensions::L10nMessagesMap messages; | 213 extensions::L10nMessagesMap messages; |
| 217 messages.insert(std::make_pair("text", "new text")); | 214 messages.insert(std::make_pair("text", "new text")); |
| 218 extensions::ExtensionToL10nMessagesMap& l10n_messages_map = | 215 extensions::ExtensionToL10nMessagesMap& l10n_messages_map = |
| 219 *extensions::GetExtensionToL10nMessagesMap(); | 216 *extensions::GetExtensionToL10nMessagesMap(); |
| 220 l10n_messages_map["some_id2"] = messages; | 217 l10n_messages_map["some_id2"] = messages; |
| 221 | 218 |
| 222 SetData("some __MSG_text__"); | 219 SetData("some __MSG_text__"); |
| 223 | 220 |
| 224 // We already have messages in memory, Send will be skipped. | 221 // We already have messages in memory, Send will be skipped. |
| 225 EXPECT_CALL(*sender_, Send(_)).Times(0); | 222 EXPECT_CALL(*sender_, Send(_)).Times(0); |
| 226 | 223 |
| 227 // __MSG_text__ gets replaced with "new text". | 224 // __MSG_text__ gets replaced with "new text". |
| 228 std::string data("some new text"); | 225 std::string data("some new text"); |
| 229 EXPECT_CALL(*original_peer_, OnReceivedResponse(_)); | 226 EXPECT_CALL(*original_peer_, OnReceivedResponse(_)); |
| 230 EXPECT_CALL(*original_peer_, | 227 EXPECT_CALL(*original_peer_, |
| 231 OnReceivedDataInternal(StrEq(data.c_str()), data.length(), -1)); | 228 OnReceivedDataInternal(StrEq(data.c_str()), data.length())); |
| 232 | 229 |
| 233 EXPECT_CALL(*original_peer_, OnCompletedRequest(net::OK, false, false, | 230 EXPECT_CALL(*original_peer_, OnCompletedRequest(net::OK, false, false, |
| 234 base::TimeTicks(), -1, -1)); | 231 base::TimeTicks(), -1, -1)); |
| 235 | 232 |
| 236 filter_peer_->OnCompletedRequest(net::OK, false, false, base::TimeTicks(), -1, | 233 filter_peer_->OnCompletedRequest(net::OK, false, false, base::TimeTicks(), -1, |
| 237 -1); | 234 -1); |
| 238 } | 235 } |
| 239 | 236 |
| 240 TEST_F(ExtensionLocalizationPeerTest, OnCompletedRequestReplaceMessagesFails) { | 237 TEST_F(ExtensionLocalizationPeerTest, OnCompletedRequestReplaceMessagesFails) { |
| 241 SetUpExtensionLocalizationPeer("text/css", GURL(kExtensionUrl_3)); | 238 SetUpExtensionLocalizationPeer("text/css", GURL(kExtensionUrl_3)); |
| 242 | 239 |
| 243 extensions::L10nMessagesMap messages; | 240 extensions::L10nMessagesMap messages; |
| 244 messages.insert(std::make_pair("text", "new text")); | 241 messages.insert(std::make_pair("text", "new text")); |
| 245 extensions::ExtensionToL10nMessagesMap& l10n_messages_map = | 242 extensions::ExtensionToL10nMessagesMap& l10n_messages_map = |
| 246 *extensions::GetExtensionToL10nMessagesMap(); | 243 *extensions::GetExtensionToL10nMessagesMap(); |
| 247 l10n_messages_map["some_id3"] = messages; | 244 l10n_messages_map["some_id3"] = messages; |
| 248 | 245 |
| 249 std::string message("some __MSG_missing_message__"); | 246 std::string message("some __MSG_missing_message__"); |
| 250 SetData(message); | 247 SetData(message); |
| 251 | 248 |
| 252 // We already have messages in memory, Send will be skipped. | 249 // We already have messages in memory, Send will be skipped. |
| 253 EXPECT_CALL(*sender_, Send(_)).Times(0); | 250 EXPECT_CALL(*sender_, Send(_)).Times(0); |
| 254 | 251 |
| 255 // __MSG_missing_message__ is missing, so message stays the same. | 252 // __MSG_missing_message__ is missing, so message stays the same. |
| 256 EXPECT_CALL(*original_peer_, OnReceivedResponse(_)); | 253 EXPECT_CALL(*original_peer_, OnReceivedResponse(_)); |
| 257 EXPECT_CALL(*original_peer_, OnReceivedDataInternal(StrEq(message.c_str()), | 254 EXPECT_CALL(*original_peer_, |
| 258 message.length(), -1)); | 255 OnReceivedDataInternal(StrEq(message.c_str()), message.length())); |
| 259 | 256 |
| 260 EXPECT_CALL(*original_peer_, OnCompletedRequest(net::OK, false, false, | 257 EXPECT_CALL(*original_peer_, OnCompletedRequest(net::OK, false, false, |
| 261 base::TimeTicks(), -1, 0)); | 258 base::TimeTicks(), -1, 0)); |
| 262 | 259 |
| 263 filter_peer_->OnCompletedRequest(net::OK, false, false, base::TimeTicks(), -1, | 260 filter_peer_->OnCompletedRequest(net::OK, false, false, base::TimeTicks(), -1, |
| 264 0); | 261 0); |
| 265 } | 262 } |
| OLD | NEW |