| 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 <vector> |
| 8 |
| 7 #include "base/file_util.h" | 9 #include "base/file_util.h" |
| 8 #include "base/files/scoped_temp_dir.h" | 10 #include "base/files/scoped_temp_dir.h" |
| 9 #include "base/run_loop.h" | 11 #include "base/run_loop.h" |
| 10 #include "base/strings/string_number_conversions.h" | 12 #include "base/strings/string_number_conversions.h" |
| 11 #include "chrome/browser/safe_browsing/download_feedback.h" | 13 #include "chrome/browser/safe_browsing/download_feedback.h" |
| 12 #include "content/public/browser/browser_thread.h" | 14 #include "content/public/browser/browser_thread.h" |
| 13 #include "content/public/test/mock_download_item.h" | 15 #include "content/public/test/mock_download_item.h" |
| 14 #include "content/public/test/test_browser_thread_bundle.h" | 16 #include "content/public/test/test_browser_thread_bundle.h" |
| 15 #include "net/url_request/url_request_test_util.h" | 17 #include "net/url_request/url_request_test_util.h" |
| 16 #include "testing/gmock/include/gmock/gmock.h" | 18 #include "testing/gmock/include/gmock/gmock.h" |
| (...skipping 139 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 156 return upload_file_path; | 158 return upload_file_path; |
| 157 } | 159 } |
| 158 | 160 |
| 159 FakeDownloadFeedback* feedback(size_t n) const { | 161 FakeDownloadFeedback* feedback(size_t n) const { |
| 160 return download_feedback_factory_.feedback(n); | 162 return download_feedback_factory_.feedback(n); |
| 161 } | 163 } |
| 162 | 164 |
| 163 size_t num_feedbacks() const { | 165 size_t num_feedbacks() const { |
| 164 return download_feedback_factory_.num_feedbacks(); | 166 return download_feedback_factory_.num_feedbacks(); |
| 165 } | 167 } |
| 168 |
| 166 protected: | 169 protected: |
| 167 base::ScopedTempDir temp_dir_; | 170 base::ScopedTempDir temp_dir_; |
| 168 content::TestBrowserThreadBundle thread_bundle_; | 171 content::TestBrowserThreadBundle thread_bundle_; |
| 169 scoped_refptr<base::SingleThreadTaskRunner> file_task_runner_; | 172 scoped_refptr<base::SingleThreadTaskRunner> file_task_runner_; |
| 170 scoped_refptr<base::SingleThreadTaskRunner> io_task_runner_; | 173 scoped_refptr<base::SingleThreadTaskRunner> io_task_runner_; |
| 171 scoped_refptr<net::TestURLRequestContextGetter> request_context_getter_; | 174 scoped_refptr<net::TestURLRequestContextGetter> request_context_getter_; |
| 172 FakeDownloadFeedbackFactory download_feedback_factory_; | 175 FakeDownloadFeedbackFactory download_feedback_factory_; |
| 173 | |
| 174 }; | 176 }; |
| 175 | 177 |
| 176 TEST_F(DownloadFeedbackServiceTest, MaybeStorePingsForDownload) { | 178 TEST_F(DownloadFeedbackServiceTest, MaybeStorePingsForDownload) { |
| 177 const int64 ok_size = DownloadFeedback::kMaxUploadSize; | 179 const int64 ok_size = DownloadFeedback::kMaxUploadSize; |
| 178 const int64 bad_size = DownloadFeedback::kMaxUploadSize + 1; | 180 const int64 bad_size = DownloadFeedback::kMaxUploadSize + 1; |
| 179 | 181 |
| 180 EXPECT_FALSE(WillStorePings(DownloadProtectionService::SAFE, ok_size)); | 182 EXPECT_FALSE(WillStorePings(DownloadProtectionService::SAFE, ok_size)); |
| 181 EXPECT_FALSE(WillStorePings(DownloadProtectionService::DANGEROUS, ok_size)); | 183 EXPECT_FALSE(WillStorePings(DownloadProtectionService::DANGEROUS, ok_size)); |
| 182 EXPECT_TRUE(WillStorePings(DownloadProtectionService::UNCOMMON, ok_size)); | 184 EXPECT_TRUE(WillStorePings(DownloadProtectionService::UNCOMMON, ok_size)); |
| 183 EXPECT_TRUE( | 185 EXPECT_TRUE( |
| (...skipping 180 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 364 // was deleted. | 366 // was deleted. |
| 365 EXPECT_FALSE(base::PathExists(file_path[2])); | 367 EXPECT_FALSE(base::PathExists(file_path[2])); |
| 366 | 368 |
| 367 // These files should still exist since the FakeDownloadFeedback does not | 369 // These files should still exist since the FakeDownloadFeedback does not |
| 368 // delete them. | 370 // delete them. |
| 369 EXPECT_TRUE(base::PathExists(file_path[0])); | 371 EXPECT_TRUE(base::PathExists(file_path[0])); |
| 370 EXPECT_TRUE(base::PathExists(file_path[1])); | 372 EXPECT_TRUE(base::PathExists(file_path[1])); |
| 371 } | 373 } |
| 372 | 374 |
| 373 } // namespace safe_browsing | 375 } // namespace safe_browsing |
| OLD | NEW |