| 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_bundle.h" | 10 #include "content/public/test/test_browser_thread_bundle.h" |
| (...skipping 59 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 70 scoped_refptr<net::TestURLRequestContextGetter> url_request_context_getter_; | 70 scoped_refptr<net::TestURLRequestContextGetter> url_request_context_getter_; |
| 71 }; | 71 }; |
| 72 | 72 |
| 73 TEST_F(TwoPhaseUploaderTest, UploadFile) { | 73 TEST_F(TwoPhaseUploaderTest, UploadFile) { |
| 74 scoped_refptr<MessageLoopRunner> runner = new MessageLoopRunner; | 74 scoped_refptr<MessageLoopRunner> runner = new MessageLoopRunner; |
| 75 LocalTwoPhaseTestServer test_server; | 75 LocalTwoPhaseTestServer test_server; |
| 76 ASSERT_TRUE(test_server.Start()); | 76 ASSERT_TRUE(test_server.Start()); |
| 77 Delegate delegate; | 77 Delegate delegate; |
| 78 scoped_ptr<TwoPhaseUploader> uploader(TwoPhaseUploader::Create( | 78 scoped_ptr<TwoPhaseUploader> uploader(TwoPhaseUploader::Create( |
| 79 url_request_context_getter_.get(), | 79 url_request_context_getter_.get(), |
| 80 BrowserThread::GetMessageLoopProxyForThread(BrowserThread::DB), | 80 BrowserThread::GetMessageLoopProxyForThread(BrowserThread::DB).get(), |
| 81 test_server.GetURL("start"), | 81 test_server.GetURL("start"), |
| 82 "metadata", | 82 "metadata", |
| 83 GetTestFilePath(), | 83 GetTestFilePath(), |
| 84 base::Bind(&Delegate::ProgressCallback, base::Unretained(&delegate)), | 84 base::Bind(&Delegate::ProgressCallback, base::Unretained(&delegate)), |
| 85 base::Bind( | 85 base::Bind( |
| 86 &Delegate::FinishCallback, base::Unretained(&delegate), runner))); | 86 &Delegate::FinishCallback, base::Unretained(&delegate), runner))); |
| 87 uploader->Start(); | 87 uploader->Start(); |
| 88 runner->Run(); | 88 runner->Run(); |
| 89 EXPECT_EQ(TwoPhaseUploader::STATE_SUCCESS, delegate.state_); | 89 EXPECT_EQ(TwoPhaseUploader::STATE_SUCCESS, delegate.state_); |
| 90 EXPECT_EQ(net::OK, delegate.net_error_); | 90 EXPECT_EQ(net::OK, delegate.net_error_); |
| 91 EXPECT_EQ(200, delegate.response_code_); | 91 EXPECT_EQ(200, delegate.response_code_); |
| 92 EXPECT_EQ( | 92 EXPECT_EQ( |
| 93 "/start\n" // path of start request | 93 "/start\n" // path of start request |
| 94 "4c24b2612e94e2ae622e54397663f2b7bf0a2e17\n" // sha1sum of "metadata" | 94 "4c24b2612e94e2ae622e54397663f2b7bf0a2e17\n" // sha1sum of "metadata" |
| 95 "944857cc626f2cafe232521986b4c6d3f9993c97\n", // sha1sum of test file | 95 "944857cc626f2cafe232521986b4c6d3f9993c97\n", // sha1sum of test file |
| 96 delegate.response_); | 96 delegate.response_); |
| 97 } | 97 } |
| 98 | 98 |
| 99 TEST_F(TwoPhaseUploaderTest, BadPhaseOneResponse) { | 99 TEST_F(TwoPhaseUploaderTest, BadPhaseOneResponse) { |
| 100 scoped_refptr<MessageLoopRunner> runner = new MessageLoopRunner; | 100 scoped_refptr<MessageLoopRunner> runner = new MessageLoopRunner; |
| 101 LocalTwoPhaseTestServer test_server; | 101 LocalTwoPhaseTestServer test_server; |
| 102 ASSERT_TRUE(test_server.Start()); | 102 ASSERT_TRUE(test_server.Start()); |
| 103 Delegate delegate; | 103 Delegate delegate; |
| 104 scoped_ptr<TwoPhaseUploader> uploader(TwoPhaseUploader::Create( | 104 scoped_ptr<TwoPhaseUploader> uploader(TwoPhaseUploader::Create( |
| 105 url_request_context_getter_.get(), | 105 url_request_context_getter_.get(), |
| 106 BrowserThread::GetMessageLoopProxyForThread(BrowserThread::DB), | 106 BrowserThread::GetMessageLoopProxyForThread(BrowserThread::DB).get(), |
| 107 test_server.GetURL("start?p1code=500"), | 107 test_server.GetURL("start?p1code=500"), |
| 108 "metadata", | 108 "metadata", |
| 109 GetTestFilePath(), | 109 GetTestFilePath(), |
| 110 base::Bind(&Delegate::ProgressCallback, base::Unretained(&delegate)), | 110 base::Bind(&Delegate::ProgressCallback, base::Unretained(&delegate)), |
| 111 base::Bind( | 111 base::Bind( |
| 112 &Delegate::FinishCallback, base::Unretained(&delegate), runner))); | 112 &Delegate::FinishCallback, base::Unretained(&delegate), runner))); |
| 113 uploader->Start(); | 113 uploader->Start(); |
| 114 runner->Run(); | 114 runner->Run(); |
| 115 EXPECT_EQ(TwoPhaseUploader::UPLOAD_METADATA, delegate.state_); | 115 EXPECT_EQ(TwoPhaseUploader::UPLOAD_METADATA, delegate.state_); |
| 116 EXPECT_EQ(net::OK, delegate.net_error_); | 116 EXPECT_EQ(net::OK, delegate.net_error_); |
| 117 EXPECT_EQ(500, delegate.response_code_); | 117 EXPECT_EQ(500, delegate.response_code_); |
| 118 EXPECT_EQ("", delegate.response_); | 118 EXPECT_EQ("", delegate.response_); |
| 119 } | 119 } |
| 120 | 120 |
| 121 TEST_F(TwoPhaseUploaderTest, BadPhaseTwoResponse) { | 121 TEST_F(TwoPhaseUploaderTest, BadPhaseTwoResponse) { |
| 122 scoped_refptr<MessageLoopRunner> runner = new MessageLoopRunner; | 122 scoped_refptr<MessageLoopRunner> runner = new MessageLoopRunner; |
| 123 LocalTwoPhaseTestServer test_server; | 123 LocalTwoPhaseTestServer test_server; |
| 124 ASSERT_TRUE(test_server.Start()); | 124 ASSERT_TRUE(test_server.Start()); |
| 125 Delegate delegate; | 125 Delegate delegate; |
| 126 scoped_ptr<TwoPhaseUploader> uploader(TwoPhaseUploader::Create( | 126 scoped_ptr<TwoPhaseUploader> uploader(TwoPhaseUploader::Create( |
| 127 url_request_context_getter_.get(), | 127 url_request_context_getter_.get(), |
| 128 BrowserThread::GetMessageLoopProxyForThread(BrowserThread::DB), | 128 BrowserThread::GetMessageLoopProxyForThread(BrowserThread::DB).get(), |
| 129 test_server.GetURL("start?p2code=500"), | 129 test_server.GetURL("start?p2code=500"), |
| 130 "metadata", | 130 "metadata", |
| 131 GetTestFilePath(), | 131 GetTestFilePath(), |
| 132 base::Bind(&Delegate::ProgressCallback, base::Unretained(&delegate)), | 132 base::Bind(&Delegate::ProgressCallback, base::Unretained(&delegate)), |
| 133 base::Bind( | 133 base::Bind( |
| 134 &Delegate::FinishCallback, base::Unretained(&delegate), runner))); | 134 &Delegate::FinishCallback, base::Unretained(&delegate), runner))); |
| 135 uploader->Start(); | 135 uploader->Start(); |
| 136 runner->Run(); | 136 runner->Run(); |
| 137 EXPECT_EQ(TwoPhaseUploader::UPLOAD_FILE, delegate.state_); | 137 EXPECT_EQ(TwoPhaseUploader::UPLOAD_FILE, delegate.state_); |
| 138 EXPECT_EQ(net::OK, delegate.net_error_); | 138 EXPECT_EQ(net::OK, delegate.net_error_); |
| 139 EXPECT_EQ(500, delegate.response_code_); | 139 EXPECT_EQ(500, delegate.response_code_); |
| 140 EXPECT_EQ( | 140 EXPECT_EQ( |
| 141 "/start\n" // path of start request | 141 "/start\n" // path of start request |
| 142 "4c24b2612e94e2ae622e54397663f2b7bf0a2e17\n" // sha1sum of "metadata" | 142 "4c24b2612e94e2ae622e54397663f2b7bf0a2e17\n" // sha1sum of "metadata" |
| 143 "944857cc626f2cafe232521986b4c6d3f9993c97\n", // sha1sum of test file | 143 "944857cc626f2cafe232521986b4c6d3f9993c97\n", // sha1sum of test file |
| 144 delegate.response_); | 144 delegate.response_); |
| 145 } | 145 } |
| 146 | 146 |
| 147 TEST_F(TwoPhaseUploaderTest, PhaseOneConnectionClosed) { | 147 TEST_F(TwoPhaseUploaderTest, PhaseOneConnectionClosed) { |
| 148 scoped_refptr<MessageLoopRunner> runner = new MessageLoopRunner; | 148 scoped_refptr<MessageLoopRunner> runner = new MessageLoopRunner; |
| 149 LocalTwoPhaseTestServer test_server; | 149 LocalTwoPhaseTestServer test_server; |
| 150 ASSERT_TRUE(test_server.Start()); | 150 ASSERT_TRUE(test_server.Start()); |
| 151 Delegate delegate; | 151 Delegate delegate; |
| 152 scoped_ptr<TwoPhaseUploader> uploader(TwoPhaseUploader::Create( | 152 scoped_ptr<TwoPhaseUploader> uploader(TwoPhaseUploader::Create( |
| 153 url_request_context_getter_.get(), | 153 url_request_context_getter_.get(), |
| 154 BrowserThread::GetMessageLoopProxyForThread(BrowserThread::DB), | 154 BrowserThread::GetMessageLoopProxyForThread(BrowserThread::DB).get(), |
| 155 test_server.GetURL("start?p1close=1"), | 155 test_server.GetURL("start?p1close=1"), |
| 156 "metadata", | 156 "metadata", |
| 157 GetTestFilePath(), | 157 GetTestFilePath(), |
| 158 base::Bind(&Delegate::ProgressCallback, base::Unretained(&delegate)), | 158 base::Bind(&Delegate::ProgressCallback, base::Unretained(&delegate)), |
| 159 base::Bind( | 159 base::Bind( |
| 160 &Delegate::FinishCallback, base::Unretained(&delegate), runner))); | 160 &Delegate::FinishCallback, base::Unretained(&delegate), runner))); |
| 161 uploader->Start(); | 161 uploader->Start(); |
| 162 runner->Run(); | 162 runner->Run(); |
| 163 EXPECT_EQ(TwoPhaseUploader::UPLOAD_METADATA, delegate.state_); | 163 EXPECT_EQ(TwoPhaseUploader::UPLOAD_METADATA, delegate.state_); |
| 164 EXPECT_EQ(net::ERR_EMPTY_RESPONSE, delegate.net_error_); | 164 EXPECT_EQ(net::ERR_EMPTY_RESPONSE, delegate.net_error_); |
| 165 EXPECT_EQ(net::URLFetcher::RESPONSE_CODE_INVALID, delegate.response_code_); | 165 EXPECT_EQ(net::URLFetcher::RESPONSE_CODE_INVALID, delegate.response_code_); |
| 166 EXPECT_EQ("", delegate.response_); | 166 EXPECT_EQ("", delegate.response_); |
| 167 } | 167 } |
| 168 | 168 |
| 169 TEST_F(TwoPhaseUploaderTest, PhaseTwoConnectionClosed) { | 169 TEST_F(TwoPhaseUploaderTest, PhaseTwoConnectionClosed) { |
| 170 scoped_refptr<MessageLoopRunner> runner = new MessageLoopRunner; | 170 scoped_refptr<MessageLoopRunner> runner = new MessageLoopRunner; |
| 171 LocalTwoPhaseTestServer test_server; | 171 LocalTwoPhaseTestServer test_server; |
| 172 ASSERT_TRUE(test_server.Start()); | 172 ASSERT_TRUE(test_server.Start()); |
| 173 Delegate delegate; | 173 Delegate delegate; |
| 174 scoped_ptr<TwoPhaseUploader> uploader(TwoPhaseUploader::Create( | 174 scoped_ptr<TwoPhaseUploader> uploader(TwoPhaseUploader::Create( |
| 175 url_request_context_getter_.get(), | 175 url_request_context_getter_.get(), |
| 176 BrowserThread::GetMessageLoopProxyForThread(BrowserThread::DB), | 176 BrowserThread::GetMessageLoopProxyForThread(BrowserThread::DB).get(), |
| 177 test_server.GetURL("start?p2close=1"), | 177 test_server.GetURL("start?p2close=1"), |
| 178 "metadata", | 178 "metadata", |
| 179 GetTestFilePath(), | 179 GetTestFilePath(), |
| 180 base::Bind(&Delegate::ProgressCallback, base::Unretained(&delegate)), | 180 base::Bind(&Delegate::ProgressCallback, base::Unretained(&delegate)), |
| 181 base::Bind( | 181 base::Bind( |
| 182 &Delegate::FinishCallback, base::Unretained(&delegate), runner))); | 182 &Delegate::FinishCallback, base::Unretained(&delegate), runner))); |
| 183 uploader->Start(); | 183 uploader->Start(); |
| 184 runner->Run(); | 184 runner->Run(); |
| 185 EXPECT_EQ(TwoPhaseUploader::UPLOAD_FILE, delegate.state_); | 185 EXPECT_EQ(TwoPhaseUploader::UPLOAD_FILE, delegate.state_); |
| 186 EXPECT_EQ(net::ERR_EMPTY_RESPONSE, delegate.net_error_); | 186 EXPECT_EQ(net::ERR_EMPTY_RESPONSE, delegate.net_error_); |
| 187 EXPECT_EQ(net::URLFetcher::RESPONSE_CODE_INVALID, delegate.response_code_); | 187 EXPECT_EQ(net::URLFetcher::RESPONSE_CODE_INVALID, delegate.response_code_); |
| 188 EXPECT_EQ("", delegate.response_); | 188 EXPECT_EQ("", delegate.response_); |
| 189 } | 189 } |
| OLD | NEW |