← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_memorytree.py

Various changes to allow non-workingtree specific tests to run entirely
from MemoryTransports:
 * Create MemoryTree and pull up common code for it from WorkingTree to
   a new common base class MutableTree.
 * Add MutableTree.mkdir().
 * Add MutableTree.put_file_bytes_nonatomic().
 * New test helper make_branch_and_memory_tree().
(Robert Collins)

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_memorytree.py
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 MemoryTree class."""
19
19
 
24
24
 
25
25
 
26
26
class TestMemoryTree(TestCaseWithTransport):
27
 
 
 
27
    
28
28
    def test_create_on_branch(self):
29
29
        """Creating a mutable tree on a trivial branch works."""
30
30
        branch = self.make_branch('branch')
32
32
        self.assertEqual(branch.bzrdir, tree.bzrdir)
33
33
        self.assertEqual(branch, tree.branch)
34
34
        self.assertEqual([], tree.get_parent_ids())
35
 
 
 
35
    
36
36
    def test_create_on_branch_with_content(self):
37
37
        """Creating a mutable tree on a non-trivial branch works."""
38
38
        branch = self.make_branch('branch')
52
52
            tree.get_file(tree.path2id('foo')).read())
53
53
        tree.unlock()
54
54
 
55
 
    def test_get_root_id(self):
56
 
        branch = self.make_branch('branch')
57
 
        tree = MemoryTree.create_on_branch(branch)
58
 
        tree.lock_write()
59
 
        try:
60
 
            tree.add([''])
61
 
            self.assertIsNot(None, tree.get_root_id())
62
 
        finally:
63
 
            tree.unlock()
64
 
 
65
 
    def test_lock_tree_write(self):
66
 
        """Check we can lock_tree_write and unlock MemoryTrees."""
67
 
        branch = self.make_branch('branch')
68
 
        tree = MemoryTree.create_on_branch(branch)
69
 
        tree.lock_tree_write()
70
 
        tree.unlock()
71
 
 
72
 
    def test_lock_tree_write_after_read_fails(self):
73
 
        """Check that we error when trying to upgrade a read lock to write."""
74
 
        branch = self.make_branch('branch')
75
 
        tree = MemoryTree.create_on_branch(branch)
76
 
        tree.lock_read()
77
 
        self.assertRaises(errors.ReadOnlyError, tree.lock_tree_write)
78
 
        tree.unlock()
79
 
 
80
55
    def test_lock_write(self):
81
56
        """Check we can lock_write and unlock MemoryTrees."""
82
57
        branch = self.make_branch('branch')
96
71
        branch = self.make_branch('branch')
97
72
        tree = MemoryTree.create_on_branch(branch)
98
73
        tree.lock_write()
99
 
        tree.add(['', 'afile', 'adir'], None,
100
 
                 ['directory', 'file', 'directory'])
 
74
        tree.add(['afile', 'adir'], None, ['file', 'directory'])
101
75
        self.assertEqual('afile', tree.id2path(tree.path2id('afile')))
102
76
        self.assertEqual('adir', tree.id2path(tree.path2id('adir')))
103
77
        self.assertFalse(tree.has_filename('afile'))
108
82
        branch = self.make_branch('branch')
109
83
        tree = MemoryTree.create_on_branch(branch)
110
84
        tree.lock_write()
111
 
        tree.add(['', 'foo'], ids=['root-id', 'foo-id'],
112
 
                  kinds=['directory', 'file'])
 
85
        tree.add(['foo'], ids=['foo-id'], kinds=['file'])
113
86
        tree.put_file_bytes_non_atomic('foo-id', 'barshoom')
114
87
        self.assertEqual('barshoom', tree.get_file('foo-id').read())
115
88
        tree.unlock()
118
91
        branch = self.make_branch('branch')
119
92
        tree = MemoryTree.create_on_branch(branch)
120
93
        tree.lock_write()
121
 
        tree.add(['', 'foo'], ids=['root-id', 'foo-id'],
122
 
                 kinds=['directory', 'file'])
 
94
        tree.add(['foo'], ids=['foo-id'], kinds=['file'])
123
95
        tree.put_file_bytes_non_atomic('foo-id', 'first-content')
124
96
        tree.put_file_bytes_non_atomic('foo-id', 'barshoom')
125
97
        self.assertEqual('barshoom', tree.get_file('foo-id').read())
126
98
        tree.unlock()
127
99
 
128
 
    def test_add_in_subdir(self):
129
 
        branch = self.make_branch('branch')
130
 
        tree = MemoryTree.create_on_branch(branch)
131
 
        tree.lock_write()
132
 
        self.addCleanup(tree.unlock)
133
 
        tree.add([''], ['root-id'], ['directory'])
134
 
        # Unfortunately, the only way to 'mkdir' is to call 'tree.mkdir', but
135
 
        # that *always* adds the directory as well. So if you want to create a
136
 
        # file in a subdirectory, you have to split out the 'mkdir()' calls
137
 
        # from the add and put_file_bytes_non_atomic calls. :(
138
 
        tree.mkdir('adir', 'dir-id')
139
 
        tree.add(['adir/afile'], ['file-id'], ['file'])
140
 
        self.assertEqual('adir/afile', tree.id2path('file-id'))
141
 
        self.assertEqual('adir', tree.id2path('dir-id'))
142
 
        tree.put_file_bytes_non_atomic('file-id', 'barshoom')
143
 
 
144
100
    def test_commit_trivial(self):
145
101
        """Smoke test for commit on a MemoryTree.
146
102
 
150
106
        branch = self.make_branch('branch')
151
107
        tree = MemoryTree.create_on_branch(branch)
152
108
        tree.lock_write()
153
 
        tree.add(['', 'foo'], ids=['root-id', 'foo-id'],
154
 
                 kinds=['directory', 'file'])
 
109
        tree.add(['foo'], ids=['foo-id'], kinds=['file'])
155
110
        tree.put_file_bytes_non_atomic('foo-id', 'barshoom')
156
111
        revision_id = tree.commit('message baby')
157
112
        # the parents list for the tree should have changed.
159
114
        tree.unlock()
160
115
        # and we should have a revision that is accessible outside the tree lock
161
116
        revtree = tree.branch.repository.revision_tree(revision_id)
162
 
        revtree.lock_read()
163
 
        self.addCleanup(revtree.unlock)
164
117
        self.assertEqual('barshoom', revtree.get_file('foo-id').read())
165
 
 
166
 
    def test_unversion(self):
167
 
        """Some test for unversion of a memory tree."""
168
 
        branch = self.make_branch('branch')
169
 
        tree = MemoryTree.create_on_branch(branch)
170
 
        tree.lock_write()
171
 
        tree.add(['', 'foo'], ids=['root-id', 'foo-id'],
172
 
                 kinds=['directory', 'file'])
173
 
        tree.unversion(['foo-id'])
174
 
        self.assertFalse(tree.has_id('foo-id'))
175
 
        tree.unlock()
176
 
 
177
 
    def test_last_revision(self):
178
 
        """There should be a last revision method we can call."""
179
 
        tree = self.make_branch_and_memory_tree('branch')
180
 
        tree.lock_write()
181
 
        tree.add('')
182
 
        rev_id = tree.commit('first post')
183
 
        tree.unlock()
184
 
        self.assertEqual(rev_id, tree.last_revision())
185
 
 
186
 
    def test_rename_file(self):
187
 
        tree = self.make_branch_and_memory_tree('branch')
188
 
        tree.lock_write()
189
 
        self.addCleanup(tree.unlock)
190
 
        tree.add(['', 'foo'], ['root-id', 'foo-id'], ['directory', 'file'])
191
 
        tree.put_file_bytes_non_atomic('foo-id', 'content\n')
192
 
        tree.commit('one', rev_id='rev-one')
193
 
        tree.rename_one('foo', 'bar')
194
 
        self.assertEqual('bar', tree.id2path('foo-id'))
195
 
        self.assertEqual('content\n', tree._file_transport.get_bytes('bar'))
196
 
        self.assertRaises(errors.NoSuchFile,
197
 
                          tree._file_transport.get_bytes, 'foo')
198
 
        tree.commit('two', rev_id='rev-two')
199
 
        self.assertEqual('content\n', tree._file_transport.get_bytes('bar'))
200
 
        self.assertRaises(errors.NoSuchFile,
201
 
                          tree._file_transport.get_bytes, 'foo')
202
 
 
203
 
        rev_tree2 = tree.branch.repository.revision_tree('rev-two')
204
 
        self.assertEqual('bar', rev_tree2.id2path('foo-id'))
205
 
        self.assertEqual('content\n', rev_tree2.get_file_text('foo-id'))
206
 
 
207
 
    def test_rename_file_to_subdir(self):
208
 
        tree = self.make_branch_and_memory_tree('branch')
209
 
        tree.lock_write()
210
 
        self.addCleanup(tree.unlock)
211
 
        tree.add('')
212
 
        tree.mkdir('subdir', 'subdir-id')
213
 
        tree.add('foo', 'foo-id', 'file')
214
 
        tree.put_file_bytes_non_atomic('foo-id', 'content\n')
215
 
        tree.commit('one', rev_id='rev-one')
216
 
 
217
 
        tree.rename_one('foo', 'subdir/bar')
218
 
        self.assertEqual('subdir/bar', tree.id2path('foo-id'))
219
 
        self.assertEqual('content\n',
220
 
                         tree._file_transport.get_bytes('subdir/bar'))
221
 
        tree.commit('two', rev_id='rev-two')
222
 
        rev_tree2 = tree.branch.repository.revision_tree('rev-two')
223
 
        self.assertEqual('subdir/bar', rev_tree2.id2path('foo-id'))