1
# Copyright (C) 2007 Canonical Ltd
2
# Authors: Robert Collins <robert.collins@canonical.com>
4
# This program is free software; you can redistribute it and/or modify
5
# it under the terms of the GNU General Public License as published by
6
# the Free Software Foundation; either version 2 of the License, or
7
# (at your option) any later version.
9
# This program is distributed in the hope that it will be useful,
10
# but WITHOUT ANY WARRANTY; without even the implied warranty of
11
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12
# GNU General Public License for more details.
14
# You should have received a copy of the GNU General Public License
15
# along with this program; if not, write to the Free Software
16
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19
from bzrlib.tests import TestNotApplicable
20
from bzrlib.transform import TreeTransform
21
from bzrlib.tests.per_workingtree import TestCaseWithWorkingTree
24
class TestNestedSupport(TestCaseWithWorkingTree):
26
def make_branch_and_tree(self, path):
27
tree = TestCaseWithWorkingTree.make_branch_and_tree(self, path)
28
if not tree.supports_tree_reference():
29
raise TestNotApplicable('Tree references not supported')
32
def test_set_get_tree_reference(self):
33
"""This tests that setting a tree reference is persistent."""
34
tree = self.make_branch_and_tree('.')
35
transform = TreeTransform(tree)
36
trans_id = transform.new_directory('reference', transform.root,
38
transform.set_tree_reference('subtree-revision', trans_id)
40
tree = tree.bzrdir.open_workingtree()
42
self.addCleanup(tree.unlock)
43
self.assertEqual('subtree-revision',
44
tree.inventory['subtree-id'].reference_revision)
46
def test_extract_while_locked(self):
47
tree = self.make_branch_and_tree('.')
49
self.addCleanup(tree.unlock)
50
self.build_tree(['subtree/'])
51
tree.add(['subtree'], ['subtree-id'])
52
subtree = tree.extract('subtree-id')
54
def prepare_with_subtree(self):
55
tree = self.make_branch_and_tree('.')
57
self.addCleanup(tree.unlock)
58
subtree = self.make_branch_and_tree('subtree')
59
tree.add(['subtree'], ['subtree-id'])
62
def test_kind_does_not_autodetect_subtree(self):
63
tree = self.prepare_with_subtree()
64
self.assertEqual('directory', tree.kind('subtree-id'))
66
def test_comparison_data_does_not_autodetect_subtree(self):
67
tree = self.prepare_with_subtree()
68
entry = tree.iter_entries_by_dir(['subtree-id']).next()[1]
69
self.assertEqual('directory',
70
tree._comparison_data(entry, 'subtree')[0])