aiden
/
depot_tools
mirror of https://chromium.googlesource.com/chromium/tools/depot_tools
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Use function decorator to simplify main()

Browse Source 
Review URL: http://codereview.chromium.org/2198001

git-svn-id: svn://svn.chromium.org/chrome/trunk/tools/depot_tools@48225 0039d316-1c4b-4281-b951-d872f2087c98
experimental/szager/collated-output
maruel@chromium.org 16 years ago
parent 95f0f4eb86
commit 35fe9ada4e
2 changed files with 158 additions and 136 deletions
Show Stats Download Patch File Download Diff File

251
gcl.py
Unescape Escape View File

@ -27,7 +27,7 @@ __pychecker__ = ''
from scm import SVN from scm import SVN
import gclient_utils import gclient_utils
__version__ = '1.1.3' __version__ = '1.1.4'
CODEREVIEW_SETTINGS = { CODEREVIEW_SETTINGS = {
@ -81,6 +81,7 @@ def CheckHomeForFile(filename):
return full_path return full_path
return None return None
def UnknownFiles(extra_args): def UnknownFiles(extra_args):
"""Runs svn status and returns unknown files. """Runs svn status and returns unknown files.
@ -622,25 +623,26 @@ def ListFiles(show_unknown_files):
return 0 return 0
def CMDopened(argv): def CMDopened(args):
"""Lists modified files in the current directory down.""" """Lists modified files in the current directory down."""
__pychecker__ = 'unusednames=argv' if args:
ErrorExit("Doesn't support arguments")
return ListFiles(False) return ListFiles(False)
def CMDstatus(argv): def CMDstatus(args):
"""Lists modified and unknown files in the current directory down.""" """Lists modified and unknown files in the current directory down."""
__pychecker__ = 'unusednames=argv' if args:
ErrorExit("Doesn't support arguments")
return ListFiles(True) return ListFiles(True)
def CMDhelp(argv=None): def CMDhelp(args):
"""Prints this help or help for the given command.""" """Prints this help or help for the given command."""
if argv and len(argv) > 2: if args and len(args) > 2:
if argv[2] == 'try': if args[2] == 'try':
TryChange(None, ['--help'], swallow_exception=False) return TryChange(None, ['--help'], swallow_exception=False)
return 0 if args[2] == 'upload':
if argv[2] == 'upload':
upload.RealMain(['upload.py', '--help']) upload.RealMain(['upload.py', '--help'])
return 0 return 0
@ -744,12 +746,25 @@ def OptionallyDoPresubmitChecks(change_info, committing, args):
return DoPresubmitChecks(change_info, committing, True) return DoPresubmitChecks(change_info, committing, True)
def CMDupload(change_info, args): def need_change(function):
"""Converts args -> change_info."""
def hook(args):
if not len(args) == 1:
ErrorExit("You need to pass a change list name")
change_info = ChangeInfo.Load(args[0], GetRepositoryRoot(), True, True)
return function(change_info)
return hook
def CMDupload(args):
if not args:
ErrorExit("You need to pass a change list name")
change_info = ChangeInfo.Load(args.pop(0), GetRepositoryRoot(), True, True)
if not change_info.GetFiles(): if not change_info.GetFiles():
print "Nothing to upload, changelist is empty." print "Nothing to upload, changelist is empty."
return return 0
if not OptionallyDoPresubmitChecks(change_info, False, args): if not OptionallyDoPresubmitChecks(change_info, False, args):
return return 1
# Might want to support GetInfoDir()/GetRepositoryRoot() like # Might want to support GetInfoDir()/GetRepositoryRoot() like
# CheckHomeForFile() so the skip of tries can be per tree basis instead # CheckHomeForFile() so the skip of tries can be per tree basis instead
# of user global. # of user global.
@ -856,13 +871,13 @@ def CMDupload(change_info, args):
trychange_args = [] trychange_args = []
if clobber: if clobber:
trychange_args.append('--clobber') trychange_args.append('--clobber')
TryChange(change_info, trychange_args, swallow_exception=True) return TryChange(change_info, trychange_args, swallow_exception=True)
return 0
def CMDpresubmit(change_info, argv): @need_change
def CMDpresubmit(change_info):
"""Runs presubmit checks on the change.""" """Runs presubmit checks on the change."""
__pychecker__ = 'unusednames=argv'
if not change_info.GetFiles(): if not change_info.GetFiles():
print "Nothing to presubmit check, changelist is empty." print "Nothing to presubmit check, changelist is empty."
return 0 return 0
@ -881,7 +896,7 @@ def TryChange(change_info, args, swallow_exception):
import trychange import trychange
except ImportError: except ImportError:
if swallow_exception: if swallow_exception:
return return 1
ErrorExit("You need to install trychange.py to use the try server.") ErrorExit("You need to install trychange.py to use the try server.")
trychange_args = [] trychange_args = []
@ -896,13 +911,14 @@ def TryChange(change_info, args, swallow_exception):
else: else:
trychange_args.extend(args) trychange_args.extend(args)
file_list = None file_list = None
trychange.TryChange(trychange_args, return trychange.TryChange(trychange_args,
file_list=file_list, file_list=file_list,
swallow_exception=swallow_exception, swallow_exception=swallow_exception,
prog='gcl try') prog='gcl try')
def CMDcommit(change_info, args): @need_change
def CMDcommit(change_info):
if not change_info.GetFiles(): if not change_info.GetFiles():
print "Nothing to commit, changelist is empty." print "Nothing to commit, changelist is empty."
return 1 return 1
@ -959,8 +975,16 @@ def CMDcommit(change_info, args):
return 0 return 0
def CMDchange(change_info, args): def CMDchange(args):
"""Creates/edits a changelist.""" """Creates/edits a changelist."""
if len(args) == 0:
# Generate a random changelist name.
changename = GenerateChangeName()
elif args[0] == '--force':
changename = GenerateChangeName()
else:
changename = args[0]
change_info = ChangeInfo.Load(changename, GetRepositoryRoot(), False, True)
silent = FilterFlag(args, "--silent") silent = FilterFlag(args, "--silent")
# Verify the user is running the change command from a read-write checkout. # Verify the user is running the change command from a read-write checkout.
@ -969,9 +993,8 @@ def CMDchange(change_info, args):
ErrorExit("Current checkout is unversioned. Please retry with a versioned " ErrorExit("Current checkout is unversioned. Please retry with a versioned "
"directory.") "directory.")
if (len(args) == 1): if len(args) == 2:
filename = args[0] f = open(args[1], 'rU')
f = open(filename, 'rU')
override_description = f.read() override_description = f.read()
f.close() f.close()
else: else:
@ -1073,13 +1096,13 @@ def CMDchange(change_info, args):
return 0 return 0
def CMDlint(change_info, args): @need_change
def CMDlint(change_info):
"""Runs cpplint.py on all the files in |change_info|""" """Runs cpplint.py on all the files in |change_info|"""
try: try:
import cpplint import cpplint
except ImportError: except ImportError:
ErrorExit("You need to install cpplint.py to lint C++ files.") ErrorExit("You need to install cpplint.py to lint C++ files.")
# Change the current working directory before calling lint so that it # Change the current working directory before calling lint so that it
# shows the correct base. # shows the correct base.
previous_cwd = os.getcwd() previous_cwd = os.getcwd()
@ -1133,9 +1156,10 @@ def DoPresubmitChecks(change_info, committing, may_prompt):
return result return result
def CMDchanges(argv): def CMDchanges(args):
"""Lists all the changelists and their files.""" """Lists all the changelists and their files."""
__pychecker__ = 'unusednames=argv' if args:
ErrorExit("Doesn't support arguments")
for cl in GetCLs(): for cl in GetCLs():
change_info = ChangeInfo.Load(cl, GetRepositoryRoot(), True, True) change_info = ChangeInfo.Load(cl, GetRepositoryRoot(), True, True)
print "\n--- Changelist " + change_info.name + ":" print "\n--- Changelist " + change_info.name + ":"
@ -1144,9 +1168,10 @@ def CMDchanges(argv):
return 0 return 0
def CMDdeleteempties(argv): def CMDdeleteempties(args):
"""Delete all changelists that have no files.""" """Delete all changelists that have no files."""
__pychecker__ = 'unusednames=argv' if args:
ErrorExit("Doesn't support arguments")
print "\n--- Deleting:" print "\n--- Deleting:"
for cl in GetCLs(): for cl in GetCLs():
change_info = ChangeInfo.Load(cl, GetRepositoryRoot(), True, True) change_info = ChangeInfo.Load(cl, GetRepositoryRoot(), True, True)
@ -1156,25 +1181,31 @@ def CMDdeleteempties(argv):
return 0 return 0
def CMDnothave(argv): def CMDnothave(args):
"""Lists files unknown to Subversion.""" """Lists files unknown to Subversion."""
__pychecker__ = 'unusednames=argv' if args:
for filename in UnknownFiles(argv[2:]): ErrorExit("Doesn't support arguments")
for filename in UnknownFiles(args):
print "? " + "".join(filename) print "? " + "".join(filename)
return 0 return 0
def CMDdiff(argv): def CMDdiff(args):
"""Diffs all files in the changelist.""" """Diffs all files in the changelist."""
__pychecker__ = 'unusednames=argv' files = None
files = GetFilesNotInCL() if args:
print GenerateDiff([x[1] for x in files]) change_info = ChangeInfo.Load(args[0], GetRepositoryRoot(), True, True)
return 0 args.pop(0)
files = change_info.GetFileNames()
else:
files = GetFilesNotInCL()
return RunShellWithReturnCode(['svn', 'diff'] + files + args,
print_output=True)[1]
def CMDsettings(argv): def CMDsettings(args):
"""Prints code review settings.""" """Prints code review settings."""
__pychecker__ = 'unusednames=argv' __pychecker__ = 'unusednames=args'
# Force load settings # Force load settings
GetCodeReviewSetting("UNKNOWN"); GetCodeReviewSetting("UNKNOWN");
del CODEREVIEW_SETTINGS['__just_initialized'] del CODEREVIEW_SETTINGS['__just_initialized']
@ -1183,21 +1214,21 @@ def CMDsettings(argv):
return 0 return 0
def CMDdescription(change_info, argv): @need_change
def CMDdescription(change_info):
"""Prints the description of the specified change to stdout.""" """Prints the description of the specified change to stdout."""
__pychecker__ = 'unusednames=argv'
print change_info.description print change_info.description
return 0 return 0
def CMDdelete(change_info, argv): @need_change
def CMDdelete(change_info):
"""Deletes a changelist.""" """Deletes a changelist."""
__pychecker__ = 'unusednames=argv'
change_info.Delete() change_info.Delete()
return 0 return 0
def CMDtry(change_info, argv): def CMDtry(args):
"""Sends the change to the tryserver so a trybot can do a test run on your """Sends the change to the tryserver so a trybot can do a test run on your
code. code.
@ -1205,21 +1236,28 @@ def CMDtry(change_info, argv):
changenames. Use 'gcl help try' for more information!""" changenames. Use 'gcl help try' for more information!"""
# When the change contains no file, send the "changename" positional # When the change contains no file, send the "changename" positional
# argument to trychange.py. # argument to trychange.py.
__pychecker__ = 'unusednames=argv' # When the command is 'try' and --patchset is used, the patch to try
# is on the Rietveld server.
if not args:
ErrorExit("You need to pass a change list name")
if args[0].find(',') != -1:
change_info = LoadChangelistInfoForMultiple(args[0], GetRepositoryRoot(),
True, True)
else:
change_info = ChangeInfo.Load(args[0], GetRepositoryRoot(),
False, True)
if change_info.GetFiles(): if change_info.GetFiles():
args = argv[3:] args = args[1:]
else: else:
change_info = None change_info = None
args = argv[2:] return TryChange(change_info, args, swallow_exception=False)
TryChange(change_info, args, swallow_exception=False)
return 0
def CMDrename(argv): def CMDrename(args):
"""Renames an existing change.""" """Renames an existing change."""
if len(argv) != 4: if len(args) != 2:
ErrorExit("Usage: gcl rename <old-name> <new-name>.") ErrorExit("Usage: gcl rename <old-name> <new-name>.")
src, dst = argv[2:4] src, dst = args
src_file = GetChangelistInfoFile(src) src_file = GetChangelistInfoFile(src)
if not os.path.isfile(src_file): if not os.path.isfile(src_file):
ErrorExit("Change '%s' does not exist." % src) ErrorExit("Change '%s' does not exist." % src)
@ -1231,14 +1269,20 @@ def CMDrename(argv):
return 0 return 0
def main(argv=None): def CMDpassthru(args):
__pychecker__ = 'maxreturns=0' # Everything else that is passed into gcl we redirect to svn, after adding
if argv is None: # the files.
argv = sys.argv # args is guaranteed to be len(args) >= 1
args = ["svn", args[0]]
if len(args) > 1:
root = GetRepositoryRoot()
change_info = ChangeInfo.Load(args[1], root, True, True)
args.extend([os.path.join(root, x) for x in change_info.GetFileNames()])
return RunShellWithReturnCode(args, print_output=True)[1]
if len(argv) == 1:
return CMDhelp()
def main(argv):
__pychecker__ = 'maxreturns=0'
try: try:
# Create the directories where we store information about changelists if it # Create the directories where we store information about changelists if it
# doesn't exist. # doesn't exist.
@ -1252,80 +1296,47 @@ def main(argv=None):
# Will throw an exception if not run in a svn checkout. # Will throw an exception if not run in a svn checkout.
pass pass
if not argv:
argv = ['help']
# Commands that don't require an argument. # Commands that don't require an argument.
command = argv[1] command = argv[0]
if command == "opened": if command == "opened":
return CMDopened(argv) return CMDopened(argv[1:])
if command == "status": if command == "status":
return CMDstatus(argv) return CMDstatus(argv[1:])
if command == "nothave": if command == "nothave":
return CMDnothave(argv) return CMDnothave(argv[1:])
if command == "changes": if command == "changes":
return CMDchanges(argv) return CMDchanges(argv[1:])
if command == "help": if command == "help":
return CMDhelp(argv) return CMDhelp(argv[1:])
if command == "diff" and len(argv) == 2: if command == "diff":
return CMDdiff(argv) return CMDdiff(argv[1:])
if command == "settings": if command == "settings":
return CMDsettings(argv) return CMDsettings(argv[1:])
if command == "deleteempties": if command == "deleteempties":
return CMDdeleteempties(argv) return CMDdeleteempties(argv[1:])
if command == "rename": if command == "rename":
return CMDrename(argv) return CMDrename(argv[1:])
if command == "change": elif command == "change":
if len(argv) == 2: return CMDchange(argv[1:])
# Generate a random changelist name.
changename = GenerateChangeName()
elif argv[2] == '--force':
changename = GenerateChangeName()
# argv[3:] is passed to Change() as |args| later. Change() should receive
# |args| which includes '--force'.
argv.insert(2, changename)
else:
changename = argv[2]
elif len(argv) == 2:
ErrorExit("Need a changelist name.")
else:
changename = argv[2]
# When the command is 'try' and --patchset is used, the patch to try
# is on the Rietveld server. 'change' creates a change so it's fine if the
# change didn't exist. All other commands require an existing change.
fail_on_not_found = command != "try" and command != "change"
if command == "try" and changename.find(',') != -1:
change_info = LoadChangelistInfoForMultiple(changename, GetRepositoryRoot(),
True, True)
else:
change_info = ChangeInfo.Load(changename, GetRepositoryRoot(),
fail_on_not_found, True)
if command == "change":
return CMDchange(change_info, argv[3:])
elif command == "description": elif command == "description":
return CMDdescription(change_info, argv[3:]) return CMDdescription(argv[1:])
elif command == "lint": elif command == "lint":
return CMDlint(change_info, argv[3:]) return CMDlint(argv[1:])
elif command == "upload": elif command == "upload":
return CMDupload(change_info, argv[3:]) return CMDupload(argv[1:])
elif command == "presubmit": elif command == "presubmit":
return CMDpresubmit(change_info, argv[3:]) return CMDpresubmit(argv[1:])
elif command in ("commit", "submit"): elif command in ("commit", "submit"):
return CMDcommit(change_info, argv[3:]) return CMDcommit(argv[1:])
elif command == "delete": elif command == "delete":
return CMDdelete(change_info, argv[3:]) return CMDdelete(argv[1:])
elif command == "try": elif command == "try":
return CMDtry(change_info, argv) return CMDtry(argv[1:])
else: else:
# Everything else that is passed into gcl we redirect to svn, after adding return CMDpassthru(argv)
# the files. This allows commands such as 'gcl diff xxx' to work.
if command == "diff" and not change_info.GetFileNames():
return 0
args =["svn", command]
root = GetRepositoryRoot()
args.extend([os.path.join(root, x) for x in change_info.GetFileNames()])
RunShell(args, True)
return 0
if __name__ == "__main__": if __name__ == "__main__":
sys.exit(main()) sys.exit(main(sys.argv[1:]))

43
tests/gcl_unittest.py
Unescape Escape View File

@ -34,8 +34,8 @@ class GclUnittest(GclTestsBase):
'CODEREVIEW_SETTINGS', 'CODEREVIEW_SETTINGS_FILE', 'CODEREVIEW_SETTINGS', 'CODEREVIEW_SETTINGS_FILE',
'CMDchange', 'CMDchanges', 'CMDcommit', 'CMDdelete', 'CMDdeleteempties', 'CMDchange', 'CMDchanges', 'CMDcommit', 'CMDdelete', 'CMDdeleteempties',
'CMDdescription', 'CMDdiff', 'CMDhelp', 'CMDlint', 'CMDnothave', 'CMDdescription', 'CMDdiff', 'CMDhelp', 'CMDlint', 'CMDnothave',
'CMDopened', 'CMDpresubmit', 'CMDrename', 'CMDsettings', 'CMDstatus', 'CMDopened', 'CMDpassthru', 'CMDpresubmit', 'CMDrename', 'CMDsettings',
'CMDtry', 'CMDupload', 'CMDstatus', 'CMDtry', 'CMDupload',
'ChangeInfo', 'DEFAULT_LINT_IGNORE_REGEX', 'ChangeInfo', 'DEFAULT_LINT_IGNORE_REGEX',
'DEFAULT_LINT_REGEX', 'CheckHomeForFile', 'DEFAULT_LINT_REGEX', 'CheckHomeForFile',
'DoPresubmitChecks', 'ErrorExit', 'FILES_CACHE', 'FilterFlag', 'DoPresubmitChecks', 'ErrorExit', 'FILES_CACHE', 'FilterFlag',
@ -48,9 +48,9 @@ class GclUnittest(GclTestsBase):
'OptionallyDoPresubmitChecks', 'REPOSITORY_ROOT', 'OptionallyDoPresubmitChecks', 'REPOSITORY_ROOT',
'RunShell', 'RunShellWithReturnCode', 'SVN', 'RunShell', 'RunShellWithReturnCode', 'SVN',
'SendToRietveld', 'TryChange', 'UnknownFiles', 'Warn', 'SendToRietveld', 'TryChange', 'UnknownFiles', 'Warn',
'breakpad', 'gclient_utils', 'getpass', 'main', 'os', 'random', 're', 'breakpad', 'gclient_utils', 'getpass', 'main', 'need_change', 'os',
'shutil', 'string', 'subprocess', 'sys', 'tempfile', 'time', 'upload', 'random', 're', 'shutil', 'string', 'subprocess', 'sys', 'tempfile',
'urllib2', 'time', 'upload', 'urllib2',
] ]
# If this test fails, you should add the relevant test. # If this test fails, you should add the relevant test.
self.compareMembers(gcl, members) self.compareMembers(gcl, members)
@ -88,7 +88,7 @@ class GclUnittest(GclTestsBase):
gcl.sys.stdout.write(mox.StrContains('GCL is a wrapper for Subversion')) gcl.sys.stdout.write(mox.StrContains('GCL is a wrapper for Subversion'))
gcl.sys.stdout.write('\n') gcl.sys.stdout.write('\n')
self.mox.ReplayAll() self.mox.ReplayAll()
gcl.CMDhelp() gcl.CMDhelp([])
class ChangeInfoUnittest(GclTestsBase): class ChangeInfoUnittest(GclTestsBase):
@ -185,6 +185,7 @@ class CMDuploadUnittest(GclTestsBase):
self.mox.StubOutWithMock(gcl, 'GetRepositoryRoot') self.mox.StubOutWithMock(gcl, 'GetRepositoryRoot')
self.mox.StubOutWithMock(gcl, 'SendToRietveld') self.mox.StubOutWithMock(gcl, 'SendToRietveld')
self.mox.StubOutWithMock(gcl, 'TryChange') self.mox.StubOutWithMock(gcl, 'TryChange')
self.mox.StubOutWithMock(gcl.ChangeInfo, 'Load')
def testNew(self): def testNew(self):
change_info = self.mox.CreateMock(gcl.ChangeInfo) change_info = self.mox.CreateMock(gcl.ChangeInfo)
@ -195,7 +196,6 @@ class CMDuploadUnittest(GclTestsBase):
change_info.files = [('A', 'aa'), ('M', 'bb')] change_info.files = [('A', 'aa'), ('M', 'bb')]
change_info.patch = None change_info.patch = None
files = [item[1] for item in change_info.files] files = [item[1] for item in change_info.files]
args = ['--foo=bar']
gcl.CheckHomeForFile('.gcl_upload_no_try').AndReturn(None) gcl.CheckHomeForFile('.gcl_upload_no_try').AndReturn(None)
gcl.DoPresubmitChecks(change_info, False, True).AndReturn(True) gcl.DoPresubmitChecks(change_info, False, True).AndReturn(True)
gcl.GetCodeReviewSetting('CODE_REVIEW_SERVER').AndReturn('my_server') gcl.GetCodeReviewSetting('CODE_REVIEW_SERVER').AndReturn('my_server')
@ -205,24 +205,28 @@ class CMDuploadUnittest(GclTestsBase):
gcl.os.chdir('proout') gcl.os.chdir('proout')
change_info.GetFileNames().AndReturn(files) change_info.GetFileNames().AndReturn(files)
gcl.GenerateDiff(files) gcl.GenerateDiff(files)
gcl.upload.RealMain(['upload.py', '-y', '--server=my_server', '--foo=bar', gcl.upload.RealMain(['upload.py', '-y', '--server=my_server',
"--message=''", '--issue=1'], change_info.patch).AndReturn(("1", '-r', 'georges@example.com',
'--message=\'\'', '--issue=1'],
change_info.patch).AndReturn(("1",
"2")) "2"))
gcl.SendToRietveld("/lint/issue%s_%s" % ('1', '2'), timeout=0.5) gcl.SendToRietveld("/lint/issue%s_%s" % ('1', '2'), timeout=0.5)
gcl.GetCodeReviewSetting('TRY_ON_UPLOAD').AndReturn('True') gcl.GetCodeReviewSetting('TRY_ON_UPLOAD').AndReturn('True')
gcl.TryChange(change_info, [], swallow_exception=True) gcl.TryChange(change_info, [], swallow_exception=True)
gcl.os.chdir('somewhere') gcl.os.chdir('somewhere')
change_info.Save() change_info.Save()
gcl.GetRepositoryRoot().AndReturn(self.fake_root_dir)
gcl.ChangeInfo.Load('naame', self.fake_root_dir, True, True
).AndReturn(change_info)
self.mox.ReplayAll() self.mox.ReplayAll()
gcl.CMDupload(change_info, args) gcl.CMDupload(['naame', '-r', 'georges@example.com'])
def testServerOverride(self): def testServerOverride(self):
change_info = gcl.ChangeInfo('naame', 0, 0, 'deescription', change_info = gcl.ChangeInfo('naame', 0, 0, 'deescription',
[('A', 'aa'), ('M', 'bb')], [('A', 'aa'), ('M', 'bb')],
self.fake_root_dir) self.fake_root_dir)
self.mox.StubOutWithMock(change_info, 'Save') self.mox.StubOutWithMock(change_info, 'Save')
args = ['--server=a', '--no_watchlists']
change_info.Save() change_info.Save()
gcl.CheckHomeForFile('.gcl_upload_no_try').AndReturn(None) gcl.CheckHomeForFile('.gcl_upload_no_try').AndReturn(None)
gcl.DoPresubmitChecks(change_info, False, True).AndReturn(True) gcl.DoPresubmitChecks(change_info, False, True).AndReturn(True)
@ -241,16 +245,18 @@ class CMDuploadUnittest(GclTestsBase):
gcl.os.remove('descfile') gcl.os.remove('descfile')
gcl.SendToRietveld("/lint/issue%s_%s" % ('1', '2'), timeout=0.5) gcl.SendToRietveld("/lint/issue%s_%s" % ('1', '2'), timeout=0.5)
gcl.os.chdir('somewhere') gcl.os.chdir('somewhere')
gcl.GetRepositoryRoot().AndReturn(self.fake_root_dir)
gcl.ChangeInfo.Load('naame', self.fake_root_dir, True, True
).AndReturn(change_info)
self.mox.ReplayAll() self.mox.ReplayAll()
gcl.CMDupload(change_info, args) gcl.CMDupload(['naame', '--server=a', '--no_watchlists'])
def testNoTry(self): def testNoTry(self):
change_info = gcl.ChangeInfo('naame', 0, 0, 'deescription', change_info = gcl.ChangeInfo('naame', 0, 0, 'deescription',
[('A', 'aa'), ('M', 'bb')], [('A', 'aa'), ('M', 'bb')],
self.fake_root_dir) self.fake_root_dir)
change_info.Save = self.mox.CreateMockAnything() change_info.Save = self.mox.CreateMockAnything()
args = ['--no-try', '--no_watchlists']
change_info.Save() change_info.Save()
gcl.DoPresubmitChecks(change_info, False, True).AndReturn(True) gcl.DoPresubmitChecks(change_info, False, True).AndReturn(True)
gcl.GetCodeReviewSetting('CODE_REVIEW_SERVER').AndReturn('my_server') gcl.GetCodeReviewSetting('CODE_REVIEW_SERVER').AndReturn('my_server')
@ -268,16 +274,18 @@ class CMDuploadUnittest(GclTestsBase):
gcl.os.remove('descfile') gcl.os.remove('descfile')
gcl.SendToRietveld("/lint/issue%s_%s" % ('1', '2'), timeout=0.5) gcl.SendToRietveld("/lint/issue%s_%s" % ('1', '2'), timeout=0.5)
gcl.os.chdir('somewhere') gcl.os.chdir('somewhere')
gcl.GetRepositoryRoot().AndReturn(self.fake_root_dir)
gcl.ChangeInfo.Load('naame', self.fake_root_dir, True, True
).AndReturn(change_info)
self.mox.ReplayAll() self.mox.ReplayAll()
gcl.CMDupload(change_info, args) gcl.CMDupload(['naame', '--no-try', '--no_watchlists'])
def testNormal(self): def testNormal(self):
change_info = gcl.ChangeInfo('naame', 0, 0, 'deescription', change_info = gcl.ChangeInfo('naame', 0, 0, 'deescription',
[('A', 'aa'), ('M', 'bb')], [('A', 'aa'), ('M', 'bb')],
self.fake_root_dir) self.fake_root_dir)
self.mox.StubOutWithMock(change_info, 'Save') self.mox.StubOutWithMock(change_info, 'Save')
args = ['--no_watchlists']
change_info.Save() change_info.Save()
gcl.CheckHomeForFile('.gcl_upload_no_try').AndReturn(None) gcl.CheckHomeForFile('.gcl_upload_no_try').AndReturn(None)
gcl.DoPresubmitChecks(change_info, False, True).AndReturn(True) gcl.DoPresubmitChecks(change_info, False, True).AndReturn(True)
@ -298,9 +306,12 @@ class CMDuploadUnittest(GclTestsBase):
gcl.GetCodeReviewSetting('TRY_ON_UPLOAD').AndReturn('True') gcl.GetCodeReviewSetting('TRY_ON_UPLOAD').AndReturn('True')
gcl.TryChange(change_info, [], swallow_exception=True) gcl.TryChange(change_info, [], swallow_exception=True)
gcl.os.chdir('somewhere') gcl.os.chdir('somewhere')
gcl.GetRepositoryRoot().AndReturn(self.fake_root_dir)
gcl.ChangeInfo.Load('naame', self.fake_root_dir, True, True
).AndReturn(change_info)
self.mox.ReplayAll() self.mox.ReplayAll()
gcl.CMDupload(change_info, args) gcl.CMDupload(['naame', '--no_watchlists'])
self.assertEquals(change_info.issue, 1) self.assertEquals(change_info.issue, 1)
self.assertEquals(change_info.patchset, 2) self.assertEquals(change_info.patchset, 2)

Write Preview
Loading…
Cancel
Save