14
14
# along with this program; if not, write to the Free Software
15
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
21
from bzrlib.tests import TestSkipped
17
22
from bzrlib.tests.tree_implementations import TestCaseWithTree
19
24
class TestAnnotate(TestCaseWithTree):
21
26
def test_annotate(self):
22
27
work_tree = self.make_branch_and_tree('wt')
23
28
tree = self.get_tree_no_parents_abc_content(work_tree)
24
29
tree_revision = getattr(tree, 'get_revision_id', lambda: 'current:')()
25
for revision, line in tree.annotate_iter('a-id'):
26
self.assertEqual('contents of a\n', line)
27
self.assertEqual(tree_revision, revision)
32
for revision, line in tree.annotate_iter('a-id'):
33
self.assertEqual('contents of a\n', line)
34
self.assertEqual(tree_revision, revision)
39
class TestReference(TestCaseWithTree):
41
def skip_if_no_reference(self, tree):
42
if not getattr(tree, 'supports_tree_reference', lambda: False)():
43
raise tests.TestSkipped('Tree references not supported')
45
def create_nested(self):
46
work_tree = self.make_branch_and_tree('wt')
47
work_tree.lock_write()
49
self.skip_if_no_reference(work_tree)
50
subtree = self.make_branch_and_tree('wt/subtree')
51
subtree.set_root_id('sub-root')
52
subtree.commit('foo', rev_id='sub-1')
53
work_tree.add_reference(subtree)
56
tree = self._convert_tree(work_tree)
57
self.skip_if_no_reference(tree)
60
def test_get_reference_revision(self):
61
tree = self.create_nested()
62
path = tree.id2path('sub-root')
63
self.assertEqual('sub-1', tree.get_reference_revision('sub-root', path))
65
def test_iter_references(self):
66
tree = self.create_nested()
68
self.addCleanup(tree.unlock)
69
entry = tree.inventory['sub-root']
70
self.assertEqual([(tree.abspath('subtree'), 'sub-root')],
71
list(tree.iter_references()))
73
def test_get_root_id(self):
74
# trees should return some kind of root id; it can be none
75
tree = self.make_branch_and_tree('tree')
76
root_id = tree.get_root_id()
77
if root_id is not None:
78
self.assertIsInstance(root_id, str)
81
class TestFileIds(TestCaseWithTree):
83
def test_id2path(self):
84
# translate from file-id back to path
85
work_tree = self.make_branch_and_tree('wt')
86
tree = self.get_tree_no_parents_abc_content(work_tree)
89
self.assertEqual(u'a', tree.id2path('a-id'))
90
# other ids give an error- don't return None for this case
91
self.assertRaises(errors.NoSuchId, tree.id2path, 'a')