def test_upgrade_branches_packagebranch(self):
"""Test that upgrade_branches can upgrade package branches."""
self.useBzrBranches()
package_branch = self.factory.makePackageBranch()
target, target_tree = self.create_branch_and_tree(
db_branch=package_branch, format='knit')
target.branch_format = BranchFormat.BZR_BRANCH_5
target.repository_format = RepositoryFormat.BZR_KNIT_1
self.assertEqual(
target_tree.branch.repository._format.get_format_string(),
'Bazaar-NG Knit Repository Format 1')
BranchUpgradeJob.create(target, self.factory.makePerson())
transaction.commit()
retcode, stdout, stderr = run_script(
'cronscripts/process-job-source.py', ['IBranchUpgradeJobSource'],
expect_returncode=0)
self.assertEqual('', stdout)
self.assertIn(
'INFO Ran 1 BranchUpgradeJob jobs.\n', stderr)
target_branch = BzrBranch.open(target_tree.branch.base)
self.assertEqual(
target_branch.repository._format.get_format_string(),
'Bazaar repository format 2a (needs bzr 1.16 or later)\n')
def test_upgrade_branches_packagebranch(self):
"""Test that upgrade_branches can upgrade package branches."""
self.useBzrBranches()
package_branch = self.factory.makePackageBranch()
target, target_tree = self.create_branch_and_tree(
db_branch=package_branch, format='knit')
target.branch_format = BranchFormat.BZR_BRANCH_5
target.repository_format = RepositoryFormat.BZR_KNIT_1
self.assertEqual(
target_tree.branch.repository._format.get_format_string(),
'Bazaar-NG Knit Repository Format 1')
BranchUpgradeJob.create(target, self.factory.makePerson())
transaction.commit()
retcode, stdout, stderr = run_script(
'cronscripts/process-job-source.py', ['IBranchUpgradeJobSource'],
expect_returncode=0)
self.assertEqual('', stdout)
self.assertIn('INFO Ran 1 BranchUpgradeJob jobs.\n', stderr)
target_branch = BzrBranch.open(target_tree.branch.base)
self.assertEqual(
target_branch.repository._format.get_format_string(),
'Bazaar repository format 2a (needs bzr 1.16 or later)\n')
def test_BranchJobPruner(self):
# Garbo should remove jobs completed over 30 days ago.
switch_dbuser('testadmin')
store = IMasterStore(Job)
db_branch = self.factory.makeAnyBranch()
db_branch.branch_format = BranchFormat.BZR_BRANCH_5
db_branch.repository_format = RepositoryFormat.BZR_KNIT_1
Store.of(db_branch).flush()
branch_job = BranchUpgradeJob.create(
db_branch, self.factory.makePerson())
branch_job.job.date_finished = THIRTY_DAYS_AGO
self.assertEqual(
store.find(
BranchJob,
BranchJob.branch == db_branch.id).count(),
1)
self.runDaily()
switch_dbuser('testadmin')
self.assertEqual(
store.find(
BranchJob,
BranchJob.branch == db_branch.id).count(),
0)
def test_BranchJobPruner_doesnt_prune_recent_jobs(self):
# Check to make sure the garbo doesn't remove jobs that aren't more
# than thirty days old.
switch_dbuser('testadmin')
store = IMasterStore(Job)
db_branch = self.factory.makeAnyBranch(
branch_format=BranchFormat.BZR_BRANCH_5,
repository_format=RepositoryFormat.BZR_KNIT_1)
branch_job = BranchUpgradeJob.create(
db_branch, self.factory.makePerson())
branch_job.job.date_finished = THIRTY_DAYS_AGO
db_branch2 = self.factory.makeAnyBranch(
branch_format=BranchFormat.BZR_BRANCH_5,
repository_format=RepositoryFormat.BZR_KNIT_1)
BranchUpgradeJob.create(db_branch2, self.factory.makePerson())
self.runDaily()
switch_dbuser('testadmin')
self.assertEqual(store.find(BranchJob).count(), 1)
