def main(self):
if len(self.args) < 1:
raise LaunchpadScriptFailure('Please specify a target directory.')
if len(self.args) > 1:
raise LaunchpadScriptFailure('Too many arguments.')
target_dir = self.args[0]
with server(get_rw_server()):
if self.options.finish:
Upgrader.finish_all_upgrades(target_dir, self.logger)
else:
Upgrader.start_all_upgrades(target_dir, self.logger)
def main(self):
if len(self.args) < 1:
raise LaunchpadScriptFailure('Please specify a target directory.')
if len(self.args) > 1:
raise LaunchpadScriptFailure('Too many arguments.')
target_dir = self.args[0]
with server(get_rw_server()):
if self.options.finish:
Upgrader.finish_all_upgrades(target_dir, self.logger)
else:
Upgrader.start_all_upgrades(target_dir, self.logger)
def test_finish_all_upgrades(self):
"""Finish all upgrades behaves as expected."""
upgrader = self.prepare()
branch_changed(upgrader.branch, upgrader.bzr_branch)
upgrader.start_upgrade()
Upgrader.finish_all_upgrades(upgrader.target_dir, upgrader.logger)
upgraded = upgrader.branch.getBzrBranch()
self.assertIs(RepositoryFormat2a,
upgraded.repository._format.__class__)
self.assertEqual(
'foo',
upgraded.repository.get_revision('prepare-commit').message)
