/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to breezy/tests/per_repository_reference/test_add_revision.py

  • Committer: Jelmer Vernooij
  • Date: 2018-11-07 22:31:36 UTC
  • mto: This revision was merged to the branch mainline in revision 7306.
  • Revision ID: jelmer@jelmer.uk-20181107223136-2uss9siifwvsq2ha
Add WriteGroup contextmanager.

Show diffs side-by-side

added added

removed removed

Lines of Context:
breezy/tests/per_repository_reference/test_add_revision.py
17
17
"""Tests for add_revision on a repository with external references."""
18
18
 
19
19
from breezy import errors
 
20
from breezy.repository import WriteGroup
20
21
from breezy.tests.per_repository_reference import (
21
22
    TestCaseWithExternalReferenceRepository,
22
23
    )
35
36
        rev = tree.branch.repository.get_revision(revid)
36
37
        base = self.make_repository('base')
37
38
        repo = self.make_referring('referring', base)
38
 
        repo.lock_write()
39
 
        try:
40
 
            repo.start_write_group()
41
 
            try:
42
 
                rev = tree.branch.repository.get_revision(revid)
43
 
                repo.texts.add_lines((inv.root.file_id, revid), [], [])
44
 
                repo.add_revision(revid, rev, inv=inv)
45
 
            except:
46
 
                repo.abort_write_group()
47
 
                raise
48
 
            else:
49
 
                repo.commit_write_group()
50
 
        finally:
51
 
            repo.unlock()
 
39
        with repo.lock_write(), WriteGroup(repo):
 
40
            rev = tree.branch.repository.get_revision(revid)
 
41
            repo.texts.add_lines((inv.root.file_id, revid), [], [])
 
42
            repo.add_revision(revid, rev, inv=inv)
52
43
        rev2 = repo.get_revision(revid)
53
44
        self.assertEqual(rev, rev2)
54
45
        self.assertRaises(errors.NoSuchRevision, base.get_revision, revid)