OLD | NEW |
1 #!/usr/bin/python | 1 #!/usr/bin/python |
2 # Copyright (c) 2009 The Chromium Authors. All rights reserved. | 2 # Copyright (c) 2009 The Chromium Authors. All rights reserved. |
3 # Use of this source code is governed by a BSD-style license that can be | 3 # Use of this source code is governed by a BSD-style license that can be |
4 # found in the LICENSE file. | 4 # found in the LICENSE file. |
5 | 5 |
6 """Unit tests for gcl.py.""" | 6 """Unit tests for gcl.py.""" |
7 | 7 |
8 import unittest | 8 import unittest |
9 | 9 |
10 # Local imports | 10 # Local imports |
(...skipping 166 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
177 self.mox.StubOutWithMock(gcl, 'GetRepositoryRoot') | 177 self.mox.StubOutWithMock(gcl, 'GetRepositoryRoot') |
178 self.mox.StubOutWithMock(gcl, 'SendToRietveld') | 178 self.mox.StubOutWithMock(gcl, 'SendToRietveld') |
179 self.mox.StubOutWithMock(gcl, 'TryChange') | 179 self.mox.StubOutWithMock(gcl, 'TryChange') |
180 | 180 |
181 def testNew(self): | 181 def testNew(self): |
182 change_info = gcl.ChangeInfo('naame', 1, 0, 'deescription', | 182 change_info = gcl.ChangeInfo('naame', 1, 0, 'deescription', |
183 ['aa', 'bb']) | 183 ['aa', 'bb']) |
184 self.mox.StubOutWithMock(change_info, 'Save') | 184 self.mox.StubOutWithMock(change_info, 'Save') |
185 args = ['--foo=bar'] | 185 args = ['--foo=bar'] |
186 change_info.Save() | 186 change_info.Save() |
187 gcl.DoPresubmitChecks(change_info, committing=False).AndReturn(True) | 187 gcl.DoPresubmitChecks(change_info, False, True).AndReturn(True) |
188 gcl.GetCodeReviewSetting('CODE_REVIEW_SERVER').AndReturn('my_server') | 188 gcl.GetCodeReviewSetting('CODE_REVIEW_SERVER').AndReturn('my_server') |
189 gcl.os.getcwd().AndReturn('somewhere') | 189 gcl.os.getcwd().AndReturn('somewhere') |
190 gcl.os.chdir(gcl.GetRepositoryRoot().AndReturn(None)) | 190 gcl.os.chdir(gcl.GetRepositoryRoot().AndReturn(None)) |
191 gcl.GenerateDiff(change_info.FileList()) | 191 gcl.GenerateDiff(change_info.FileList()) |
192 gcl.upload.RealMain(['upload.py', '-y', '--server=my_server', '--foo=bar', | 192 gcl.upload.RealMain(['upload.py', '-y', '--server=my_server', '--foo=bar', |
193 "--message=''", '--issue=1'], change_info.patch).AndReturn(("1", | 193 "--message=''", '--issue=1'], change_info.patch).AndReturn(("1", |
194 "2")) | 194 "2")) |
195 gcl.SendToRietveld("/lint/issue%s_%s" % ('1', '2'), timeout=0.5) | 195 gcl.SendToRietveld("/lint/issue%s_%s" % ('1', '2'), timeout=0.5) |
196 gcl.GetCodeReviewSetting('TRY_ON_UPLOAD').AndReturn('True') | 196 gcl.GetCodeReviewSetting('TRY_ON_UPLOAD').AndReturn('True') |
197 gcl.TryChange(change_info, [], swallow_exception=True) | 197 gcl.TryChange(change_info, [], swallow_exception=True) |
198 gcl.os.chdir('somewhere') | 198 gcl.os.chdir('somewhere') |
199 self.mox.ReplayAll() | 199 self.mox.ReplayAll() |
200 gcl.UploadCL(change_info, args) | 200 gcl.UploadCL(change_info, args) |
201 | 201 |
202 def testServerOverride(self): | 202 def testServerOverride(self): |
203 change_info = gcl.ChangeInfo('naame', 0, 0, 'deescription', | 203 change_info = gcl.ChangeInfo('naame', 0, 0, 'deescription', |
204 ['aa', 'bb']) | 204 ['aa', 'bb']) |
205 change_info.Save = self.mox.CreateMockAnything() | 205 change_info.Save = self.mox.CreateMockAnything() |
206 args = ['--server=a'] | 206 args = ['--server=a'] |
207 change_info.Save() | 207 change_info.Save() |
208 gcl.DoPresubmitChecks(change_info, committing=False).AndReturn(True) | 208 gcl.DoPresubmitChecks(change_info, False, True).AndReturn(True) |
209 gcl.GetCodeReviewSetting('CODE_REVIEW_SERVER').AndReturn('my_server') | 209 gcl.GetCodeReviewSetting('CODE_REVIEW_SERVER').AndReturn('my_server') |
210 gcl.tempfile.mkstemp(text=True).AndReturn((42, 'descfile')) | 210 gcl.tempfile.mkstemp(text=True).AndReturn((42, 'descfile')) |
211 gcl.os.write(42, change_info.description) | 211 gcl.os.write(42, change_info.description) |
212 gcl.os.close(42) | 212 gcl.os.close(42) |
213 gcl.GetCodeReviewSetting('CC_LIST') | 213 gcl.GetCodeReviewSetting('CC_LIST') |
214 gcl.os.getcwd().AndReturn('somewhere') | 214 gcl.os.getcwd().AndReturn('somewhere') |
215 gcl.os.chdir(gcl.GetRepositoryRoot().AndReturn(None)) | 215 gcl.os.chdir(gcl.GetRepositoryRoot().AndReturn(None)) |
216 gcl.GenerateDiff(change_info.FileList()) | 216 gcl.GenerateDiff(change_info.FileList()) |
217 gcl.upload.RealMain(['upload.py', '-y', '--server=my_server', '--server=a', | 217 gcl.upload.RealMain(['upload.py', '-y', '--server=my_server', '--server=a', |
218 "--description_file=descfile", | 218 "--description_file=descfile", |
219 "--message=deescription"], change_info.patch).AndReturn(("1", "2")) | 219 "--message=deescription"], change_info.patch).AndReturn(("1", "2")) |
220 gcl.os.remove('descfile') | 220 gcl.os.remove('descfile') |
221 gcl.SendToRietveld("/lint/issue%s_%s" % ('1', '2'), timeout=0.5) | 221 gcl.SendToRietveld("/lint/issue%s_%s" % ('1', '2'), timeout=0.5) |
222 gcl.os.chdir('somewhere') | 222 gcl.os.chdir('somewhere') |
223 self.mox.ReplayAll() | 223 self.mox.ReplayAll() |
224 gcl.UploadCL(change_info, args) | 224 gcl.UploadCL(change_info, args) |
225 | 225 |
226 def testNoTry(self): | 226 def testNoTry(self): |
227 change_info = gcl.ChangeInfo('naame', 0, 0, 'deescription', | 227 change_info = gcl.ChangeInfo('naame', 0, 0, 'deescription', |
228 ['aa', 'bb']) | 228 ['aa', 'bb']) |
229 change_info.Save = self.mox.CreateMockAnything() | 229 change_info.Save = self.mox.CreateMockAnything() |
230 args = ['--no-try'] | 230 args = ['--no-try'] |
231 change_info.Save() | 231 change_info.Save() |
232 gcl.DoPresubmitChecks(change_info, committing=False).AndReturn(True) | 232 gcl.DoPresubmitChecks(change_info, False, True).AndReturn(True) |
233 gcl.GetCodeReviewSetting('CODE_REVIEW_SERVER').AndReturn('my_server') | 233 gcl.GetCodeReviewSetting('CODE_REVIEW_SERVER').AndReturn('my_server') |
234 gcl.tempfile.mkstemp(text=True).AndReturn((42, 'descfile')) | 234 gcl.tempfile.mkstemp(text=True).AndReturn((42, 'descfile')) |
235 gcl.os.write(42, change_info.description) | 235 gcl.os.write(42, change_info.description) |
236 gcl.os.close(42) | 236 gcl.os.close(42) |
237 gcl.GetCodeReviewSetting('CC_LIST') | 237 gcl.GetCodeReviewSetting('CC_LIST') |
238 gcl.os.getcwd().AndReturn('somewhere') | 238 gcl.os.getcwd().AndReturn('somewhere') |
239 gcl.os.chdir(gcl.GetRepositoryRoot().AndReturn(None)) | 239 gcl.os.chdir(gcl.GetRepositoryRoot().AndReturn(None)) |
240 gcl.GenerateDiff(change_info.FileList()) | 240 gcl.GenerateDiff(change_info.FileList()) |
241 gcl.upload.RealMain(['upload.py', '-y', '--server=my_server', | 241 gcl.upload.RealMain(['upload.py', '-y', '--server=my_server', |
242 "--description_file=descfile", | 242 "--description_file=descfile", |
243 "--message=deescription"], change_info.patch).AndReturn(("1", "2")) | 243 "--message=deescription"], change_info.patch).AndReturn(("1", "2")) |
244 gcl.os.remove('descfile') | 244 gcl.os.remove('descfile') |
245 gcl.SendToRietveld("/lint/issue%s_%s" % ('1', '2'), timeout=0.5) | 245 gcl.SendToRietveld("/lint/issue%s_%s" % ('1', '2'), timeout=0.5) |
246 gcl.os.chdir('somewhere') | 246 gcl.os.chdir('somewhere') |
247 self.mox.ReplayAll() | 247 self.mox.ReplayAll() |
248 gcl.UploadCL(change_info, args) | 248 gcl.UploadCL(change_info, args) |
249 | 249 |
250 def testNormal(self): | 250 def testNormal(self): |
251 change_info = gcl.ChangeInfo('naame', 0, 0, 'deescription', | 251 change_info = gcl.ChangeInfo('naame', 0, 0, 'deescription', |
252 ['aa', 'bb']) | 252 ['aa', 'bb']) |
253 self.mox.StubOutWithMock(change_info, 'Save') | 253 self.mox.StubOutWithMock(change_info, 'Save') |
254 args = [] | 254 args = [] |
255 change_info.Save() | 255 change_info.Save() |
256 gcl.DoPresubmitChecks(change_info, committing=False).AndReturn(True) | 256 gcl.DoPresubmitChecks(change_info, False, True).AndReturn(True) |
257 gcl.GetCodeReviewSetting('CODE_REVIEW_SERVER').AndReturn('my_server') | 257 gcl.GetCodeReviewSetting('CODE_REVIEW_SERVER').AndReturn('my_server') |
258 gcl.tempfile.mkstemp(text=True).AndReturn((42, 'descfile')) | 258 gcl.tempfile.mkstemp(text=True).AndReturn((42, 'descfile')) |
259 gcl.os.write(42, change_info.description) | 259 gcl.os.write(42, change_info.description) |
260 gcl.os.close(42) | 260 gcl.os.close(42) |
261 gcl.GetCodeReviewSetting('CC_LIST') | 261 gcl.GetCodeReviewSetting('CC_LIST') |
262 gcl.os.getcwd().AndReturn('somewhere') | 262 gcl.os.getcwd().AndReturn('somewhere') |
263 gcl.os.chdir(gcl.GetRepositoryRoot().AndReturn(None)) | 263 gcl.os.chdir(gcl.GetRepositoryRoot().AndReturn(None)) |
264 gcl.GenerateDiff(change_info.FileList()) | 264 gcl.GenerateDiff(change_info.FileList()) |
265 gcl.upload.RealMain(['upload.py', '-y', '--server=my_server', | 265 gcl.upload.RealMain(['upload.py', '-y', '--server=my_server', |
266 "--description_file=descfile", | 266 "--description_file=descfile", |
267 "--message=deescription"], change_info.patch).AndReturn(("1", "2")) | 267 "--message=deescription"], change_info.patch).AndReturn(("1", "2")) |
268 gcl.os.remove('descfile') | 268 gcl.os.remove('descfile') |
269 gcl.SendToRietveld("/lint/issue%s_%s" % ('1', '2'), timeout=0.5) | 269 gcl.SendToRietveld("/lint/issue%s_%s" % ('1', '2'), timeout=0.5) |
270 gcl.GetCodeReviewSetting('TRY_ON_UPLOAD').AndReturn('True') | 270 gcl.GetCodeReviewSetting('TRY_ON_UPLOAD').AndReturn('True') |
271 gcl.TryChange(change_info, [], swallow_exception=True) | 271 gcl.TryChange(change_info, [], swallow_exception=True) |
272 gcl.os.chdir('somewhere') | 272 gcl.os.chdir('somewhere') |
273 self.mox.ReplayAll() | 273 self.mox.ReplayAll() |
274 | 274 |
275 gcl.UploadCL(change_info, args) | 275 gcl.UploadCL(change_info, args) |
276 self.assertEquals(change_info.issue, 1) | 276 self.assertEquals(change_info.issue, 1) |
277 self.assertEquals(change_info.patchset, 2) | 277 self.assertEquals(change_info.patchset, 2) |
278 | 278 |
279 | 279 |
280 if __name__ == '__main__': | 280 if __name__ == '__main__': |
281 unittest.main() | 281 unittest.main() |
OLD | NEW |