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 |