| 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/two_phase_uploader.h" | 5 #include "chrome/browser/safe_browsing/two_phase_uploader.h" |
| 6 | 6 |
| 7 #include "base/files/file_path.h" | 7 #include "base/files/file_path.h" |
| 8 #include "base/message_loop.h" | 8 #include "base/message_loop.h" |
| 9 #include "chrome/browser/safe_browsing/local_two_phase_testserver.h" | 9 #include "chrome/browser/safe_browsing/local_two_phase_testserver.h" |
| 10 #include "content/public/test/test_browser_thread.h" | 10 #include "content/public/test/test_browser_thread_bundle.h" |
| 11 #include "content/public/test/test_utils.h" | 11 #include "content/public/test/test_utils.h" |
| 12 #include "net/base/net_errors.h" | 12 #include "net/base/net_errors.h" |
| 13 #include "net/url_request/url_fetcher.h" | 13 #include "net/url_request/url_fetcher.h" |
| 14 #include "net/url_request/url_request_test_util.h" | 14 #include "net/url_request/url_request_test_util.h" |
| 15 #include "testing/gtest/include/gtest/gtest.h" | 15 #include "testing/gtest/include/gtest/gtest.h" |
| 16 | 16 |
| 17 using content::BrowserThread; | 17 using content::BrowserThread; |
| 18 using content::MessageLoopRunner; | 18 using content::MessageLoopRunner; |
| 19 | 19 |
| 20 namespace { | 20 namespace { |
| (...skipping 31 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 52 file_path = file_path.Append(FILE_PATH_LITERAL("url_request_unittest")); | 52 file_path = file_path.Append(FILE_PATH_LITERAL("url_request_unittest")); |
| 53 file_path = file_path.Append(FILE_PATH_LITERAL("BullRunSpeech.txt")); | 53 file_path = file_path.Append(FILE_PATH_LITERAL("BullRunSpeech.txt")); |
| 54 return file_path; | 54 return file_path; |
| 55 } | 55 } |
| 56 | 56 |
| 57 } // namespace | 57 } // namespace |
| 58 | 58 |
| 59 class TwoPhaseUploaderTest : public testing::Test { | 59 class TwoPhaseUploaderTest : public testing::Test { |
| 60 public: | 60 public: |
| 61 TwoPhaseUploaderTest() | 61 TwoPhaseUploaderTest() |
| 62 : db_thread_(BrowserThread::DB), | 62 : thread_bundle_(content::TestBrowserThreadBundle::IO_MAINLOOP), |
| 63 io_thread_(BrowserThread::IO, &message_loop_), | |
| 64 url_request_context_getter_(new net::TestURLRequestContextGetter( | 63 url_request_context_getter_(new net::TestURLRequestContextGetter( |
| 65 BrowserThread::GetMessageLoopProxyForThread(BrowserThread::IO))) { | 64 BrowserThread::GetMessageLoopProxyForThread(BrowserThread::IO))) { |
| 66 } | 65 } |
| 67 | 66 |
| 68 virtual void SetUp() { | |
| 69 db_thread_.Start(); | |
| 70 } | |
| 71 | |
| 72 protected: | 67 protected: |
| 73 base::MessageLoopForIO message_loop_; | 68 content::TestBrowserThreadBundle thread_bundle_; |
| 74 content::TestBrowserThread db_thread_; | |
| 75 content::TestBrowserThread io_thread_; | |
| 76 | 69 |
| 77 scoped_refptr<net::TestURLRequestContextGetter> url_request_context_getter_; | 70 scoped_refptr<net::TestURLRequestContextGetter> url_request_context_getter_; |
| 78 }; | 71 }; |
| 79 | 72 |
| 80 TEST_F(TwoPhaseUploaderTest, UploadFile) { | 73 TEST_F(TwoPhaseUploaderTest, UploadFile) { |
| 81 scoped_refptr<MessageLoopRunner> runner = new MessageLoopRunner; | 74 scoped_refptr<MessageLoopRunner> runner = new MessageLoopRunner; |
| 82 LocalTwoPhaseTestServer test_server; | 75 LocalTwoPhaseTestServer test_server; |
| 83 ASSERT_TRUE(test_server.Start()); | 76 ASSERT_TRUE(test_server.Start()); |
| 84 Delegate delegate; | 77 Delegate delegate; |
| 85 scoped_ptr<TwoPhaseUploader> uploader(TwoPhaseUploader::Create( | 78 scoped_ptr<TwoPhaseUploader> uploader(TwoPhaseUploader::Create( |
| (...skipping 101 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 187 base::Bind(&Delegate::ProgressCallback, base::Unretained(&delegate)), | 180 base::Bind(&Delegate::ProgressCallback, base::Unretained(&delegate)), |
| 188 base::Bind( | 181 base::Bind( |
| 189 &Delegate::FinishCallback, base::Unretained(&delegate), runner))); | 182 &Delegate::FinishCallback, base::Unretained(&delegate), runner))); |
| 190 uploader->Start(); | 183 uploader->Start(); |
| 191 runner->Run(); | 184 runner->Run(); |
| 192 EXPECT_EQ(TwoPhaseUploader::UPLOAD_FILE, delegate.state_); | 185 EXPECT_EQ(TwoPhaseUploader::UPLOAD_FILE, delegate.state_); |
| 193 EXPECT_EQ(net::ERR_EMPTY_RESPONSE, delegate.net_error_); | 186 EXPECT_EQ(net::ERR_EMPTY_RESPONSE, delegate.net_error_); |
| 194 EXPECT_EQ(net::URLFetcher::RESPONSE_CODE_INVALID, delegate.response_code_); | 187 EXPECT_EQ(net::URLFetcher::RESPONSE_CODE_INVALID, delegate.response_code_); |
| 195 EXPECT_EQ("", delegate.response_); | 188 EXPECT_EQ("", delegate.response_); |
| 196 } | 189 } |
| OLD | NEW |