/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 bzrlib/tests/test_treebuilder.py

  • Committer: Robert Collins
  • Date: 2007-04-19 02:27:44 UTC
  • mto: This revision was merged to the branch mainline in revision 2426.
  • Revision ID: robertc@robertcollins.net-20070419022744-pfdqz42kp1wizh43
``make docs`` now creates a man page at ``man1/bzr.1`` fixing bug 107388.
(Robert Collins)

Show diffs side-by-side

added added

removed removed

Lines of Context:
13
13
#
14
14
# You should have received a copy of the GNU General Public License
15
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
 
16
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
17
17
 
18
18
"""Tests for the TreeBuilder helper class."""
19
19
 
20
 
from breezy import tests
21
 
from breezy.memorytree import MemoryTree
22
 
from breezy.tests import TestCaseWithTransport
23
 
from breezy.treebuilder import (
24
 
    AlreadyBuilding,
25
 
    NotBuilding,
26
 
    TreeBuilder,
27
 
    )
 
20
from bzrlib import errors
 
21
from bzrlib.memorytree import MemoryTree
 
22
from bzrlib.tests import TestCaseWithTransport
 
23
from bzrlib.treebuilder import TreeBuilder
28
24
 
29
25
 
30
26
class FakeTree(object):
41
37
 
42
38
 
43
39
class TestFakeTree(TestCaseWithTransport):
44
 
 
 
40
    
45
41
    def testFakeTree(self):
46
42
        """Check that FakeTree works as required for the TreeBuilder tests."""
47
43
        tree = FakeTree()
52
48
        self.assertEqual(["lock_tree_write", "unlock"], tree._calls)
53
49
 
54
50
 
55
 
class TestTreeBuilderMemoryTree(tests.TestCaseWithMemoryTransport):
56
 
 
 
51
class TestTreeBuilder(TestCaseWithTransport):
 
52
    
57
53
    def test_create(self):
58
 
        TreeBuilder()
 
54
        builder = TreeBuilder()
59
55
 
60
56
    def test_start_tree_locks_write(self):
61
57
        builder = TreeBuilder()
67
63
        builder = TreeBuilder()
68
64
        tree = FakeTree()
69
65
        builder.start_tree(tree)
70
 
        self.assertRaises(AlreadyBuilding, builder.start_tree, tree)
71
 
 
 
66
        self.assertRaises(errors.AlreadyBuilding, builder.start_tree, tree)
 
67
        
72
68
    def test_finish_tree_not_started_errors(self):
73
69
        builder = TreeBuilder()
74
 
        self.assertRaises(NotBuilding, builder.finish_tree)
 
70
        self.assertRaises(errors.NotBuilding, builder.finish_tree)
75
71
 
76
72
    def test_finish_tree_unlocks(self):
77
73
        builder = TreeBuilder()
82
78
 
83
79
    def test_build_tree_not_started_errors(self):
84
80
        builder = TreeBuilder()
85
 
        self.assertRaises(NotBuilding, builder.build, "foo")
 
81
        self.assertRaises(errors.NotBuilding, builder.build, "foo")
86
82
 
87
83
    def test_build_tree(self):
88
84
        """Test building works using a MemoryTree."""
91
87
        builder = TreeBuilder()
92
88
        builder.start_tree(tree)
93
89
        builder.build(['foo', "bar/", "bar/file"])
94
 
        self.assertEqual(
95
 
            b'contents of foo\n',
96
 
            tree.get_file('foo').read())
97
 
        self.assertEqual(
98
 
            b'contents of bar/file\n',
99
 
            tree.get_file('bar/file').read())
 
90
        self.assertEqual('contents of foo\n',
 
91
            tree.get_file(tree.path2id('foo')).read())
 
92
        self.assertEqual('contents of bar/file\n',
 
93
            tree.get_file(tree.path2id('bar/file')).read())
100
94
        builder.finish_tree()