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/browser/safe_browsing/download_feedback_service.h" | 5 #include "chrome/browser/safe_browsing/download_feedback_service.h" |
6 | 6 |
| 7 #include <stddef.h> |
| 8 #include <stdint.h> |
| 9 |
7 #include <vector> | 10 #include <vector> |
8 | 11 |
9 #include "base/files/file_util.h" | 12 #include "base/files/file_util.h" |
10 #include "base/files/scoped_temp_dir.h" | 13 #include "base/files/scoped_temp_dir.h" |
11 #include "base/run_loop.h" | 14 #include "base/run_loop.h" |
12 #include "base/strings/string_number_conversions.h" | 15 #include "base/strings/string_number_conversions.h" |
13 #include "chrome/browser/safe_browsing/download_feedback.h" | 16 #include "chrome/browser/safe_browsing/download_feedback.h" |
14 #include "content/public/browser/browser_thread.h" | 17 #include "content/public/browser/browser_thread.h" |
15 #include "content/public/test/mock_download_item.h" | 18 #include "content/public/test/mock_download_item.h" |
16 #include "content/public/test/test_browser_thread_bundle.h" | 19 #include "content/public/test/test_browser_thread_bundle.h" |
(...skipping 91 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
108 | 111 |
109 size_t num_feedbacks() const { | 112 size_t num_feedbacks() const { |
110 return feedbacks_.size(); | 113 return feedbacks_.size(); |
111 } | 114 } |
112 | 115 |
113 private: | 116 private: |
114 std::vector<FakeDownloadFeedback*> feedbacks_; | 117 std::vector<FakeDownloadFeedback*> feedbacks_; |
115 }; | 118 }; |
116 | 119 |
117 bool WillStorePings(DownloadProtectionService::DownloadCheckResult result, | 120 bool WillStorePings(DownloadProtectionService::DownloadCheckResult result, |
118 int64 size) { | 121 int64_t size) { |
119 content::MockDownloadItem item; | 122 content::MockDownloadItem item; |
120 EXPECT_CALL(item, GetReceivedBytes()).WillRepeatedly(Return(size)); | 123 EXPECT_CALL(item, GetReceivedBytes()).WillRepeatedly(Return(size)); |
121 | 124 |
122 EXPECT_FALSE(DownloadFeedbackService::IsEnabledForDownload(item)); | 125 EXPECT_FALSE(DownloadFeedbackService::IsEnabledForDownload(item)); |
123 DownloadFeedbackService::MaybeStorePingsForDownload(result, &item, "a", "b"); | 126 DownloadFeedbackService::MaybeStorePingsForDownload(result, &item, "a", "b"); |
124 return DownloadFeedbackService::IsEnabledForDownload(item); | 127 return DownloadFeedbackService::IsEnabledForDownload(item); |
125 } | 128 } |
126 | 129 |
127 } // namespace | 130 } // namespace |
128 | 131 |
(...skipping 36 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
165 protected: | 168 protected: |
166 base::ScopedTempDir temp_dir_; | 169 base::ScopedTempDir temp_dir_; |
167 content::TestBrowserThreadBundle thread_bundle_; | 170 content::TestBrowserThreadBundle thread_bundle_; |
168 scoped_refptr<base::SingleThreadTaskRunner> file_task_runner_; | 171 scoped_refptr<base::SingleThreadTaskRunner> file_task_runner_; |
169 scoped_refptr<base::SingleThreadTaskRunner> io_task_runner_; | 172 scoped_refptr<base::SingleThreadTaskRunner> io_task_runner_; |
170 scoped_refptr<net::TestURLRequestContextGetter> request_context_getter_; | 173 scoped_refptr<net::TestURLRequestContextGetter> request_context_getter_; |
171 FakeDownloadFeedbackFactory download_feedback_factory_; | 174 FakeDownloadFeedbackFactory download_feedback_factory_; |
172 }; | 175 }; |
173 | 176 |
174 TEST_F(DownloadFeedbackServiceTest, MaybeStorePingsForDownload) { | 177 TEST_F(DownloadFeedbackServiceTest, MaybeStorePingsForDownload) { |
175 const int64 ok_size = DownloadFeedback::kMaxUploadSize; | 178 const int64_t ok_size = DownloadFeedback::kMaxUploadSize; |
176 const int64 bad_size = DownloadFeedback::kMaxUploadSize + 1; | 179 const int64_t bad_size = DownloadFeedback::kMaxUploadSize + 1; |
177 | 180 |
178 EXPECT_FALSE(WillStorePings(DownloadProtectionService::SAFE, ok_size)); | 181 EXPECT_FALSE(WillStorePings(DownloadProtectionService::SAFE, ok_size)); |
179 EXPECT_FALSE(WillStorePings(DownloadProtectionService::DANGEROUS, ok_size)); | 182 EXPECT_FALSE(WillStorePings(DownloadProtectionService::DANGEROUS, ok_size)); |
180 EXPECT_TRUE(WillStorePings(DownloadProtectionService::UNCOMMON, ok_size)); | 183 EXPECT_TRUE(WillStorePings(DownloadProtectionService::UNCOMMON, ok_size)); |
181 EXPECT_TRUE( | 184 EXPECT_TRUE( |
182 WillStorePings(DownloadProtectionService::DANGEROUS_HOST, ok_size)); | 185 WillStorePings(DownloadProtectionService::DANGEROUS_HOST, ok_size)); |
183 EXPECT_TRUE( | 186 EXPECT_TRUE( |
184 WillStorePings(DownloadProtectionService::POTENTIALLY_UNWANTED, ok_size)); | 187 WillStorePings(DownloadProtectionService::POTENTIALLY_UNWANTED, ok_size)); |
185 | 188 |
186 EXPECT_FALSE(WillStorePings(DownloadProtectionService::SAFE, bad_size)); | 189 EXPECT_FALSE(WillStorePings(DownloadProtectionService::SAFE, bad_size)); |
(...skipping 179 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
366 // was deleted. | 369 // was deleted. |
367 EXPECT_FALSE(base::PathExists(file_path[2])); | 370 EXPECT_FALSE(base::PathExists(file_path[2])); |
368 | 371 |
369 // These files should still exist since the FakeDownloadFeedback does not | 372 // These files should still exist since the FakeDownloadFeedback does not |
370 // delete them. | 373 // delete them. |
371 EXPECT_TRUE(base::PathExists(file_path[0])); | 374 EXPECT_TRUE(base::PathExists(file_path[0])); |
372 EXPECT_TRUE(base::PathExists(file_path[1])); | 375 EXPECT_TRUE(base::PathExists(file_path[1])); |
373 } | 376 } |
374 | 377 |
375 } // namespace safe_browsing | 378 } // namespace safe_browsing |
OLD | NEW |