Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(1509)

Side by Side Diff: chrome/browser/safe_browsing/two_phase_uploader_unittest.cc

Issue 15984016: Call scoped_refptr<T>::get() rather than relying on implicit "operator T*" (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src
Patch Set: Rebased Created 7 years, 6 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch | Annotate | Revision Log
OLDNEW
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
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 }
OLDNEW
« no previous file with comments | « chrome/browser/safe_browsing/sandboxed_zip_analyzer.cc ('k') | chrome/browser/safe_json_parser.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698