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 |