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> | 7 #include <vector> |
8 | 8 |
9 #include "base/files/file_util.h" | 9 #include "base/files/file_util.h" |
10 #include "base/files/scoped_temp_dir.h" | 10 #include "base/files/scoped_temp_dir.h" |
(...skipping 26 matching lines...) Expand all Loading... |
37 : ping_request_(ping_request), | 37 : ping_request_(ping_request), |
38 ping_response_(ping_response), | 38 ping_response_(ping_response), |
39 deletion_callback_(deletion_callback), | 39 deletion_callback_(deletion_callback), |
40 start_called_(false) { | 40 start_called_(false) { |
41 } | 41 } |
42 | 42 |
43 virtual ~FakeDownloadFeedback() { | 43 virtual ~FakeDownloadFeedback() { |
44 deletion_callback_.Run(); | 44 deletion_callback_.Run(); |
45 } | 45 } |
46 | 46 |
47 virtual void Start(const base::Closure& finish_callback) OVERRIDE { | 47 virtual void Start(const base::Closure& finish_callback) override { |
48 start_called_ = true; | 48 start_called_ = true; |
49 finish_callback_ = finish_callback; | 49 finish_callback_ = finish_callback; |
50 } | 50 } |
51 | 51 |
52 virtual const std::string& GetPingRequestForTesting() const OVERRIDE { | 52 virtual const std::string& GetPingRequestForTesting() const override { |
53 return ping_request_; | 53 return ping_request_; |
54 } | 54 } |
55 | 55 |
56 virtual const std::string& GetPingResponseForTesting() const OVERRIDE { | 56 virtual const std::string& GetPingResponseForTesting() const override { |
57 return ping_response_; | 57 return ping_response_; |
58 } | 58 } |
59 | 59 |
60 base::Closure finish_callback() const { | 60 base::Closure finish_callback() const { |
61 return finish_callback_; | 61 return finish_callback_; |
62 } | 62 } |
63 | 63 |
64 bool start_called() const { | 64 bool start_called() const { |
65 return start_called_; | 65 return start_called_; |
66 } | 66 } |
(...skipping 12 matching lines...) Expand all Loading... |
79 | 79 |
80 class FakeDownloadFeedbackFactory : public DownloadFeedbackFactory { | 80 class FakeDownloadFeedbackFactory : public DownloadFeedbackFactory { |
81 public: | 81 public: |
82 virtual ~FakeDownloadFeedbackFactory() {} | 82 virtual ~FakeDownloadFeedbackFactory() {} |
83 | 83 |
84 virtual DownloadFeedback* CreateDownloadFeedback( | 84 virtual DownloadFeedback* CreateDownloadFeedback( |
85 net::URLRequestContextGetter* request_context_getter, | 85 net::URLRequestContextGetter* request_context_getter, |
86 base::TaskRunner* file_task_runner, | 86 base::TaskRunner* file_task_runner, |
87 const base::FilePath& file_path, | 87 const base::FilePath& file_path, |
88 const std::string& ping_request, | 88 const std::string& ping_request, |
89 const std::string& ping_response) OVERRIDE { | 89 const std::string& ping_response) override { |
90 FakeDownloadFeedback* feedback = new FakeDownloadFeedback( | 90 FakeDownloadFeedback* feedback = new FakeDownloadFeedback( |
91 request_context_getter, | 91 request_context_getter, |
92 file_task_runner, | 92 file_task_runner, |
93 file_path, | 93 file_path, |
94 ping_request, | 94 ping_request, |
95 ping_response, | 95 ping_response, |
96 base::Bind(&FakeDownloadFeedbackFactory::DownloadFeedbackDeleted, | 96 base::Bind(&FakeDownloadFeedbackFactory::DownloadFeedbackDeleted, |
97 base::Unretained(this), | 97 base::Unretained(this), |
98 feedbacks_.size())); | 98 feedbacks_.size())); |
99 feedbacks_.push_back(feedback); | 99 feedbacks_.push_back(feedback); |
(...skipping 32 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
132 public: | 132 public: |
133 DownloadFeedbackServiceTest() | 133 DownloadFeedbackServiceTest() |
134 : file_task_runner_(content::BrowserThread::GetMessageLoopProxyForThread( | 134 : file_task_runner_(content::BrowserThread::GetMessageLoopProxyForThread( |
135 content::BrowserThread::FILE)), | 135 content::BrowserThread::FILE)), |
136 io_task_runner_(content::BrowserThread::GetMessageLoopProxyForThread( | 136 io_task_runner_(content::BrowserThread::GetMessageLoopProxyForThread( |
137 content::BrowserThread::IO)), | 137 content::BrowserThread::IO)), |
138 request_context_getter_( | 138 request_context_getter_( |
139 new net::TestURLRequestContextGetter(io_task_runner_)) { | 139 new net::TestURLRequestContextGetter(io_task_runner_)) { |
140 } | 140 } |
141 | 141 |
142 virtual void SetUp() OVERRIDE { | 142 virtual void SetUp() override { |
143 ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); | 143 ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); |
144 DownloadFeedback::RegisterFactory(&download_feedback_factory_); | 144 DownloadFeedback::RegisterFactory(&download_feedback_factory_); |
145 } | 145 } |
146 | 146 |
147 virtual void TearDown() OVERRIDE { | 147 virtual void TearDown() override { |
148 DownloadFeedback::RegisterFactory(NULL); | 148 DownloadFeedback::RegisterFactory(NULL); |
149 } | 149 } |
150 | 150 |
151 base::FilePath CreateTestFile(int n) const { | 151 base::FilePath CreateTestFile(int n) const { |
152 base::FilePath upload_file_path( | 152 base::FilePath upload_file_path( |
153 temp_dir_.path().AppendASCII("test file " + base::IntToString(n))); | 153 temp_dir_.path().AppendASCII("test file " + base::IntToString(n))); |
154 const std::string upload_file_data = "data"; | 154 const std::string upload_file_data = "data"; |
155 int wrote = base::WriteFile( | 155 int wrote = base::WriteFile( |
156 upload_file_path, upload_file_data.data(), upload_file_data.size()); | 156 upload_file_path, upload_file_data.data(), upload_file_data.size()); |
157 EXPECT_EQ(static_cast<int>(upload_file_data.size()), wrote); | 157 EXPECT_EQ(static_cast<int>(upload_file_data.size()), wrote); |
(...skipping 208 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
366 // was deleted. | 366 // was deleted. |
367 EXPECT_FALSE(base::PathExists(file_path[2])); | 367 EXPECT_FALSE(base::PathExists(file_path[2])); |
368 | 368 |
369 // These files should still exist since the FakeDownloadFeedback does not | 369 // These files should still exist since the FakeDownloadFeedback does not |
370 // delete them. | 370 // delete them. |
371 EXPECT_TRUE(base::PathExists(file_path[0])); | 371 EXPECT_TRUE(base::PathExists(file_path[0])); |
372 EXPECT_TRUE(base::PathExists(file_path[1])); | 372 EXPECT_TRUE(base::PathExists(file_path[1])); |
373 } | 373 } |
374 | 374 |
375 } // namespace safe_browsing | 375 } // namespace safe_browsing |
OLD | NEW |