← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/inventory_implementations/basics.py

Merge bzr.dev 4187, and revert the change to fix refcycle issues.

I apparently didn't run the smart fetch tests. Which show that we access inv+chk pages
as a fulltext, and then insert the stream, which expects to get the block as a compressed
block. :(.
Need to rethink how to do it, possibly with weakrefs.


This also brings in CommitBuilder.record_iter_changes() and the updates to btree_index
and backing indices.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/inventory_implementations/basics.py
42
42
    def make_inventory(self, root_id):
43
43
        return self.inventory_class(root_id=root_id)
44
44
 
 
45
    def prepare_inv_with_nested_dirs(self):
 
46
        inv = self.make_inventory('tree-root')
 
47
        for args in [('src', 'directory', 'src-id'),
 
48
                     ('doc', 'directory', 'doc-id'),
 
49
                     ('src/hello.c', 'file', 'hello-id'),
 
50
                     ('src/bye.c', 'file', 'bye-id'),
 
51
                     ('zz', 'file', 'zz-id'),
 
52
                     ('src/sub/', 'directory', 'sub-id'),
 
53
                     ('src/zz.c', 'file', 'zzc-id'),
 
54
                     ('src/sub/a', 'file', 'a-id'),
 
55
                     ('Makefile', 'file', 'makefile-id')]:
 
56
            inv.add_path(*args)
 
57
        return inv
 
58
 
45
59
 
46
60
class TestInventoryUpdates(TestInventory):
47
61
 
226
240
            ], sorted([ie.file_id for ie in inv.iter_just_entries()]))
227
241
 
228
242
    def test_iter_entries_by_dir(self):
229
 
        inv = self.make_inventory('tree-root')
230
 
        for args in [('src', 'directory', 'src-id'),
231
 
                     ('doc', 'directory', 'doc-id'),
232
 
                     ('src/hello.c', 'file', 'hello-id'),
233
 
                     ('src/bye.c', 'file', 'bye-id'),
234
 
                     ('zz', 'file', 'zz-id'),
235
 
                     ('src/sub/', 'directory', 'sub-id'),
236
 
                     ('src/zz.c', 'file', 'zzc-id'),
237
 
                     ('src/sub/a', 'file', 'a-id'),
238
 
                     ('Makefile', 'file', 'makefile-id')]:
239
 
            inv.add_path(*args)
 
243
        inv = self. prepare_inv_with_nested_dirs()
240
244
        self.assertEqual([
241
245
            ('', 'tree-root'),
242
246
            ('Makefile', 'makefile-id'),
300
304
            ('src/bye.c', 'bye-id'),
301
305
            ], [(path, ie.file_id) for path, ie in inv.iter_entries_by_dir(
302
306
                specific_file_ids=('bye-id',), yield_parents=True)])
 
307
 
 
308
 
 
309
class TestInventoryFiltering(TestInventory):
 
310
 
 
311
    def test_inv_filter_empty(self):
 
312
        inv = self.prepare_inv_with_nested_dirs()
 
313
        new_inv = inv.filter([])
 
314
        self.assertEqual([
 
315
            ('', 'tree-root'),
 
316
            ], [(path, ie.file_id) for path, ie in new_inv.iter_entries()])
 
317
    
 
318
    def test_inv_filter_files(self):
 
319
        inv = self.prepare_inv_with_nested_dirs()
 
320
        new_inv = inv.filter(['zz-id', 'hello-id', 'a-id'])
 
321
        self.assertEqual([
 
322
            ('', 'tree-root'),
 
323
            ('src', 'src-id'),
 
324
            ('src/hello.c', 'hello-id'),
 
325
            ('src/sub', 'sub-id'),
 
326
            ('src/sub/a', 'a-id'),
 
327
            ('zz', 'zz-id'),
 
328
            ], [(path, ie.file_id) for path, ie in new_inv.iter_entries()])
 
329
    
 
330
    def test_inv_filter_dirs(self):
 
331
        inv = self.prepare_inv_with_nested_dirs()
 
332
        new_inv = inv.filter(['doc-id', 'sub-id'])
 
333
        self.assertEqual([
 
334
            ('', 'tree-root'),
 
335
            ('doc', 'doc-id'),
 
336
            ('src', 'src-id'),
 
337
            ('src/sub', 'sub-id'),
 
338
            ('src/sub/a', 'a-id'),
 
339
            ], [(path, ie.file_id) for path, ie in new_inv.iter_entries()])
 
340
 
 
341
    def test_inv_filter_files_and_dirs(self):
 
342
        inv = self.prepare_inv_with_nested_dirs()
 
343
        new_inv = inv.filter(['makefile-id', 'src-id'])
 
344
        self.assertEqual([
 
345
            ('', 'tree-root'),
 
346
            ('Makefile', 'makefile-id'),
 
347
            ('src', 'src-id'),
 
348
            ('src/bye.c', 'bye-id'),
 
349
            ('src/hello.c', 'hello-id'),
 
350
            ('src/sub', 'sub-id'),
 
351
            ('src/sub/a', 'a-id'),
 
352
            ('src/zz.c', 'zzc-id'),
 
353
            ], [(path, ie.file_id) for path, ie in new_inv.iter_entries()])