| 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 |