/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/tree_implementations/test_tree.py

  • Committer: Martin Pool
  • Date: 2007-08-28 02:02:09 UTC
  • mfrom: (2749 +trunk)
  • mto: This revision was merged to the branch mainline in revision 2758.
  • Revision ID: mbp@sourcefrog.net-20070828020209-gbhb0onl14e1fjty
merge trunk

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/tree_implementations/test_tree.py
119
119
            self.assertRaises(errors.NoSuchId, tree.id2path, 'a')
120
120
        finally:
121
121
            tree.unlock()
 
122
 
 
123
 
 
124
class TestExtractFilesBytes(TestCaseWithTree):
 
125
 
 
126
    def test_iter_files_bytes(self):
 
127
        work_tree = self.make_branch_and_tree('wt')
 
128
        self.build_tree_contents([('wt/foo', 'foo'),
 
129
                                  ('wt/bar', 'bar'),
 
130
                                  ('wt/baz', 'baz')])
 
131
        work_tree.add(['foo', 'bar', 'baz'], ['foo-id', 'bar-id', 'baz-id'])
 
132
        tree = self._convert_tree(work_tree)
 
133
        tree.lock_read()
 
134
        self.addCleanup(tree.unlock)
 
135
        extracted = dict((i, ''.join(b)) for i, b in
 
136
                         tree.iter_files_bytes([('foo-id', 'id1'),
 
137
                                                ('bar-id', 'id2'),
 
138
                                                ('baz-id', 'id3')]))
 
139
        self.assertEqual('foo', extracted['id1'])
 
140
        self.assertEqual('bar', extracted['id2'])
 
141
        self.assertEqual('baz', extracted['id3'])
 
142
        self.assertRaises(errors.NoSuchId, lambda: list(
 
143
                          tree.iter_files_bytes(
 
144
                          [('qux-id', 'file1-notpresent')])))