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

  • Committer: Jelmer Vernooij
  • Date: 2017-07-23 22:06:41 UTC
  • mfrom: (6738 trunk)
  • mto: This revision was merged to the branch mainline in revision 6739.
  • Revision ID: jelmer@jelmer.uk-20170723220641-69eczax9bmv8d6kk
Merge trunk, address review comments.

Show diffs side-by-side

added added

removed removed

Lines of Context:
breezy/tests/test_treebuilder.py
17
17
 
18
18
"""Tests for the TreeBuilder helper class."""
19
19
 
20
 
from breezy import tests
 
20
from breezy import errors, tests
21
21
from breezy.memorytree import MemoryTree
22
22
from breezy.tests import TestCaseWithTransport
23
 
from breezy.treebuilder import (
24
 
    AlreadyBuilding,
25
 
    NotBuilding,
26
 
    TreeBuilder,
27
 
    )
 
23
from breezy.treebuilder import TreeBuilder
28
24
 
29
25
 
30
26
class FakeTree(object):
55
51
class TestTreeBuilderMemoryTree(tests.TestCaseWithMemoryTransport):
56
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)
 
66
        self.assertRaises(errors.AlreadyBuilding, builder.start_tree, tree)
71
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()
 
95