~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_memorytree.py

• browse files at revision 4634.136.1
• compare with another revision
• download diff
• download tarball
• view history from revision 4634.136.1

• reverse the comparison (4634.136.1 to 3537)
• stop comparing with revision 3537

Show diffs side-by-side

added

removed

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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

 

16

# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 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')

96

96

        branch = self.make_branch('branch')

97

97

        tree = MemoryTree.create_on_branch(branch)

98

98

        tree.lock_write()

99

 

        tree.add(['', 'afile', 'adir'], None, 

 

99

        tree.add(['', 'afile', 'adir'], None,

100

100

                 ['directory', 'file', 'directory'])

101

101

        self.assertEqual('afile', tree.id2path(tree.path2id('afile')))

102

102

        self.assertEqual('adir', tree.id2path(tree.path2id('adir')))

108

108

        branch = self.make_branch('branch')

109

109

        tree = MemoryTree.create_on_branch(branch)

110

110

        tree.lock_write()

111

 

        tree.add(['', 'foo'], ids=['root-id', 'foo-id'], 

 

111

        tree.add(['', 'foo'], ids=['root-id', 'foo-id'],

112

112

                  kinds=['directory', 'file'])

113

113

        tree.put_file_bytes_non_atomic('foo-id', 'barshoom')

114

114

        self.assertEqual('barshoom', tree.get_file('foo-id').read())

118

118

        branch = self.make_branch('branch')

119

119

        tree = MemoryTree.create_on_branch(branch)

120

120

        tree.lock_write()

121

 

        tree.add(['', 'foo'], ids=['root-id', 'foo-id'], 

 

121

        tree.add(['', 'foo'], ids=['root-id', 'foo-id'],

122

122

                 kinds=['directory', 'file'])

123

123

        tree.put_file_bytes_non_atomic('foo-id', 'first-content')

124

124

        tree.put_file_bytes_non_atomic('foo-id', 'barshoom')

125

125

        self.assertEqual('barshoom', tree.get_file('foo-id').read())

126

126

        tree.unlock()

127

127

 

 

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

 

128

144

    def test_commit_trivial(self):

129

145

        """Smoke test for commit on a MemoryTree.

130

146

 

134

150

        branch = self.make_branch('branch')

135

151

        tree = MemoryTree.create_on_branch(branch)

136

152

        tree.lock_write()

137

 

        tree.add(['', 'foo'], ids=['root-id', 'foo-id'], 

 

153

        tree.add(['', 'foo'], ids=['root-id', 'foo-id'],

138

154

                 kinds=['directory', 'file'])

139

155

        tree.put_file_bytes_non_atomic('foo-id', 'barshoom')

140

156

        revision_id = tree.commit('message baby')

152

168

        branch = self.make_branch('branch')

153

169

        tree = MemoryTree.create_on_branch(branch)

154

170

        tree.lock_write()

155

 

        tree.add(['', 'foo'], ids=['root-id', 'foo-id'], 

 

171

        tree.add(['', 'foo'], ids=['root-id', 'foo-id'],

156

172

                 kinds=['directory', 'file'])

157

173

        tree.unversion(['foo-id'])

158

174

        self.assertFalse(tree.has_id('foo-id'))

166

182

        rev_id = tree.commit('first post')

167

183

        tree.unlock()

168

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'))

• Older »

Loggerhead is a web-based interface for Breezy
Version: 2.0.1