| 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 61 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 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()); | 76 data->encoded_data_length()); |
| 77 } | 77 } |
| 78 MOCK_METHOD3(OnReceivedDataInternal, | 78 MOCK_METHOD3(OnReceivedDataInternal, |
| 79 void(const char* data, | 79 void(const char* data, |
| 80 int data_length, | 80 int data_length, |
| 81 int encoded_data_length)); | 81 int encoded_data_length)); |
| 82 MOCK_METHOD6(OnCompletedRequest, void( | 82 MOCK_METHOD5(OnCompletedRequest, |
| 83 int error_code, | 83 void(int error_code, |
| 84 bool was_ignored_by_handler, | 84 bool was_ignored_by_handler, |
| 85 bool stale_copy_in_cache, | 85 bool stale_copy_in_cache, |
| 86 const std::string& security_info, | 86 const base::TimeTicks& completion_time, |
| 87 const base::TimeTicks& completion_time, | 87 int64_t total_transfer_size)); |
| 88 int64_t total_transfer_size)); | |
| 89 | 88 |
| 90 private: | 89 private: |
| 91 DISALLOW_COPY_AND_ASSIGN(MockRequestPeer); | 90 DISALLOW_COPY_AND_ASSIGN(MockRequestPeer); |
| 92 }; | 91 }; |
| 93 | 92 |
| 94 } // namespace | 93 } // namespace |
| 95 | 94 |
| 96 class ExtensionLocalizationPeerTest : public testing::Test { | 95 class ExtensionLocalizationPeerTest : public testing::Test { |
| 97 protected: | 96 protected: |
| 98 void SetUp() override { | 97 void SetUp() override { |
| (...skipping 49 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 148 EXPECT_EQ(data_chunk + data_chunk, GetData()); | 147 EXPECT_EQ(data_chunk + data_chunk, GetData()); |
| 149 } | 148 } |
| 150 | 149 |
| 151 MATCHER_P(IsURLRequestEqual, status, "") { return arg.status() == status; } | 150 MATCHER_P(IsURLRequestEqual, status, "") { return arg.status() == status; } |
| 152 | 151 |
| 153 TEST_F(ExtensionLocalizationPeerTest, OnCompletedRequestBadURLRequestStatus) { | 152 TEST_F(ExtensionLocalizationPeerTest, OnCompletedRequestBadURLRequestStatus) { |
| 154 SetUpExtensionLocalizationPeer("text/css", GURL(kExtensionUrl_1)); | 153 SetUpExtensionLocalizationPeer("text/css", GURL(kExtensionUrl_1)); |
| 155 | 154 |
| 156 EXPECT_CALL(*original_peer_, OnReceivedResponse(_)); | 155 EXPECT_CALL(*original_peer_, OnReceivedResponse(_)); |
| 157 EXPECT_CALL(*original_peer_, | 156 EXPECT_CALL(*original_peer_, |
| 158 OnCompletedRequest(net::ERR_ABORTED, false, false, "", | 157 OnCompletedRequest(net::ERR_ABORTED, false, false, |
| 159 base::TimeTicks(), -1)); | 158 base::TimeTicks(), -1)); |
| 160 | 159 |
| 161 filter_peer_->OnCompletedRequest(net::ERR_FAILED, false, false, std::string(), | 160 filter_peer_->OnCompletedRequest(net::ERR_FAILED, false, false, |
| 162 base::TimeTicks(), -1); | 161 base::TimeTicks(), -1); |
| 163 } | 162 } |
| 164 | 163 |
| 165 TEST_F(ExtensionLocalizationPeerTest, OnCompletedRequestEmptyData) { | 164 TEST_F(ExtensionLocalizationPeerTest, OnCompletedRequestEmptyData) { |
| 166 SetUpExtensionLocalizationPeer("text/css", GURL(kExtensionUrl_1)); | 165 SetUpExtensionLocalizationPeer("text/css", GURL(kExtensionUrl_1)); |
| 167 | 166 |
| 168 EXPECT_CALL(*original_peer_, OnReceivedDataInternal(_, _, _)).Times(0); | 167 EXPECT_CALL(*original_peer_, OnReceivedDataInternal(_, _, _)).Times(0); |
| 169 EXPECT_CALL(*sender_, Send(_)).Times(0); | 168 EXPECT_CALL(*sender_, Send(_)).Times(0); |
| 170 | 169 |
| 171 EXPECT_CALL(*original_peer_, OnReceivedResponse(_)); | 170 EXPECT_CALL(*original_peer_, OnReceivedResponse(_)); |
| 172 EXPECT_CALL(*original_peer_, OnCompletedRequest(net::OK, false, false, "", | 171 EXPECT_CALL(*original_peer_, |
| 173 base::TimeTicks(), -1)); | 172 OnCompletedRequest(net::OK, false, false, base::TimeTicks(), -1)); |
| 174 | 173 |
| 175 filter_peer_->OnCompletedRequest(net::OK, false, false, std::string(), | 174 filter_peer_->OnCompletedRequest(net::OK, false, false, base::TimeTicks(), |
| 176 base::TimeTicks(), -1); | 175 -1); |
| 177 } | 176 } |
| 178 | 177 |
| 179 TEST_F(ExtensionLocalizationPeerTest, OnCompletedRequestNoCatalogs) { | 178 TEST_F(ExtensionLocalizationPeerTest, OnCompletedRequestNoCatalogs) { |
| 180 SetUpExtensionLocalizationPeer("text/css", GURL(kExtensionUrl_1)); | 179 SetUpExtensionLocalizationPeer("text/css", GURL(kExtensionUrl_1)); |
| 181 | 180 |
| 182 SetData("some text"); | 181 SetData("some text"); |
| 183 | 182 |
| 184 EXPECT_CALL(*sender_, Send(_)); | 183 EXPECT_CALL(*sender_, Send(_)); |
| 185 | 184 |
| 186 std::string data = GetData(); | 185 std::string data = GetData(); |
| 187 EXPECT_CALL(*original_peer_, OnReceivedResponse(_)).Times(1); | 186 EXPECT_CALL(*original_peer_, OnReceivedResponse(_)).Times(1); |
| 188 EXPECT_CALL(*original_peer_, | 187 EXPECT_CALL(*original_peer_, |
| 189 OnReceivedDataInternal(StrEq(data.c_str()), data.length(), -1)) | 188 OnReceivedDataInternal(StrEq(data.c_str()), data.length(), -1)) |
| 190 .Times(1); | 189 .Times(1); |
| 191 EXPECT_CALL(*original_peer_, OnCompletedRequest(net::OK, false, false, "", | 190 EXPECT_CALL(*original_peer_, |
| 192 base::TimeTicks(), -1)) | 191 OnCompletedRequest(net::OK, false, false, base::TimeTicks(), -1)) |
| 193 .Times(1); | 192 .Times(1); |
| 194 | 193 |
| 195 filter_peer_->OnCompletedRequest(net::OK, false, false, std::string(), | 194 filter_peer_->OnCompletedRequest(net::OK, false, false, base::TimeTicks(), |
| 196 base::TimeTicks(), -1); | 195 -1); |
| 197 | 196 |
| 198 // Test if Send gets called again (it shouldn't be) when first call returned | 197 // Test if Send gets called again (it shouldn't be) when first call returned |
| 199 // an empty dictionary. | 198 // an empty dictionary. |
| 200 SetUpExtensionLocalizationPeer("text/css", GURL(kExtensionUrl_1)); | 199 SetUpExtensionLocalizationPeer("text/css", GURL(kExtensionUrl_1)); |
| 201 EXPECT_CALL(*original_peer_, OnReceivedResponse(_)).Times(1); | 200 EXPECT_CALL(*original_peer_, OnReceivedResponse(_)).Times(1); |
| 202 EXPECT_CALL(*original_peer_, | 201 EXPECT_CALL(*original_peer_, |
| 203 OnReceivedDataInternal(StrEq(data.c_str()), data.length(), -1)) | 202 OnReceivedDataInternal(StrEq(data.c_str()), data.length(), -1)) |
| 204 .Times(1); | 203 .Times(1); |
| 205 EXPECT_CALL(*original_peer_, OnCompletedRequest(net::OK, false, false, "", | 204 EXPECT_CALL(*original_peer_, |
| 206 base::TimeTicks(), -1)) | 205 OnCompletedRequest(net::OK, false, false, base::TimeTicks(), -1)) |
| 207 .Times(1); | 206 .Times(1); |
| 208 SetData("some text"); | 207 SetData("some text"); |
| 209 filter_peer_->OnCompletedRequest(net::OK, false, false, std::string(), | 208 filter_peer_->OnCompletedRequest(net::OK, false, false, base::TimeTicks(), |
| 210 base::TimeTicks(), -1); | 209 -1); |
| 211 } | 210 } |
| 212 | 211 |
| 213 TEST_F(ExtensionLocalizationPeerTest, OnCompletedRequestWithCatalogs) { | 212 TEST_F(ExtensionLocalizationPeerTest, OnCompletedRequestWithCatalogs) { |
| 214 SetUpExtensionLocalizationPeer("text/css", GURL(kExtensionUrl_2)); | 213 SetUpExtensionLocalizationPeer("text/css", GURL(kExtensionUrl_2)); |
| 215 | 214 |
| 216 extensions::L10nMessagesMap messages; | 215 extensions::L10nMessagesMap messages; |
| 217 messages.insert(std::make_pair("text", "new text")); | 216 messages.insert(std::make_pair("text", "new text")); |
| 218 extensions::ExtensionToL10nMessagesMap& l10n_messages_map = | 217 extensions::ExtensionToL10nMessagesMap& l10n_messages_map = |
| 219 *extensions::GetExtensionToL10nMessagesMap(); | 218 *extensions::GetExtensionToL10nMessagesMap(); |
| 220 l10n_messages_map["some_id2"] = messages; | 219 l10n_messages_map["some_id2"] = messages; |
| 221 | 220 |
| 222 SetData("some __MSG_text__"); | 221 SetData("some __MSG_text__"); |
| 223 | 222 |
| 224 // We already have messages in memory, Send will be skipped. | 223 // We already have messages in memory, Send will be skipped. |
| 225 EXPECT_CALL(*sender_, Send(_)).Times(0); | 224 EXPECT_CALL(*sender_, Send(_)).Times(0); |
| 226 | 225 |
| 227 // __MSG_text__ gets replaced with "new text". | 226 // __MSG_text__ gets replaced with "new text". |
| 228 std::string data("some new text"); | 227 std::string data("some new text"); |
| 229 EXPECT_CALL(*original_peer_, OnReceivedResponse(_)); | 228 EXPECT_CALL(*original_peer_, OnReceivedResponse(_)); |
| 230 EXPECT_CALL(*original_peer_, | 229 EXPECT_CALL(*original_peer_, |
| 231 OnReceivedDataInternal(StrEq(data.c_str()), data.length(), -1)); | 230 OnReceivedDataInternal(StrEq(data.c_str()), data.length(), -1)); |
| 232 | 231 |
| 233 EXPECT_CALL(*original_peer_, OnCompletedRequest(net::OK, false, false, "", | 232 EXPECT_CALL(*original_peer_, |
| 234 base::TimeTicks(), -1)); | 233 OnCompletedRequest(net::OK, false, false, base::TimeTicks(), -1)); |
| 235 | 234 |
| 236 filter_peer_->OnCompletedRequest(net::OK, false, false, std::string(), | 235 filter_peer_->OnCompletedRequest(net::OK, false, false, base::TimeTicks(), |
| 237 base::TimeTicks(), -1); | 236 -1); |
| 238 } | 237 } |
| 239 | 238 |
| 240 TEST_F(ExtensionLocalizationPeerTest, OnCompletedRequestReplaceMessagesFails) { | 239 TEST_F(ExtensionLocalizationPeerTest, OnCompletedRequestReplaceMessagesFails) { |
| 241 SetUpExtensionLocalizationPeer("text/css", GURL(kExtensionUrl_3)); | 240 SetUpExtensionLocalizationPeer("text/css", GURL(kExtensionUrl_3)); |
| 242 | 241 |
| 243 extensions::L10nMessagesMap messages; | 242 extensions::L10nMessagesMap messages; |
| 244 messages.insert(std::make_pair("text", "new text")); | 243 messages.insert(std::make_pair("text", "new text")); |
| 245 extensions::ExtensionToL10nMessagesMap& l10n_messages_map = | 244 extensions::ExtensionToL10nMessagesMap& l10n_messages_map = |
| 246 *extensions::GetExtensionToL10nMessagesMap(); | 245 *extensions::GetExtensionToL10nMessagesMap(); |
| 247 l10n_messages_map["some_id3"] = messages; | 246 l10n_messages_map["some_id3"] = messages; |
| 248 | 247 |
| 249 std::string message("some __MSG_missing_message__"); | 248 std::string message("some __MSG_missing_message__"); |
| 250 SetData(message); | 249 SetData(message); |
| 251 | 250 |
| 252 // We already have messages in memory, Send will be skipped. | 251 // We already have messages in memory, Send will be skipped. |
| 253 EXPECT_CALL(*sender_, Send(_)).Times(0); | 252 EXPECT_CALL(*sender_, Send(_)).Times(0); |
| 254 | 253 |
| 255 // __MSG_missing_message__ is missing, so message stays the same. | 254 // __MSG_missing_message__ is missing, so message stays the same. |
| 256 EXPECT_CALL(*original_peer_, OnReceivedResponse(_)); | 255 EXPECT_CALL(*original_peer_, OnReceivedResponse(_)); |
| 257 EXPECT_CALL(*original_peer_, OnReceivedDataInternal(StrEq(message.c_str()), | 256 EXPECT_CALL(*original_peer_, OnReceivedDataInternal(StrEq(message.c_str()), |
| 258 message.length(), -1)); | 257 message.length(), -1)); |
| 259 | 258 |
| 260 EXPECT_CALL(*original_peer_, OnCompletedRequest(net::OK, false, false, "", | 259 EXPECT_CALL(*original_peer_, |
| 261 base::TimeTicks(), -1)); | 260 OnCompletedRequest(net::OK, false, false, base::TimeTicks(), -1)); |
| 262 | 261 |
| 263 filter_peer_->OnCompletedRequest(net::OK, false, false, std::string(), | 262 filter_peer_->OnCompletedRequest(net::OK, false, false, base::TimeTicks(), |
| 264 base::TimeTicks(), -1); | 263 -1); |
| 265 } | 264 } |
| OLD | NEW |