~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_inv.py

  • Committer: John Arbash Meinel
  • Date: 2007-04-12 21:33:07 UTC
  • mfrom: (2413.4.1 api-doc-builders)
  • mto: This revision was merged to the branch mainline in revision 2566.
  • Revision ID: john@arbash-meinel.com-20070412213307-kuh07cnzaud12wx1
[merge] api-doc-builder and remove the pydoctor build code for now.

Show diffs side-by-side

added added

removed removed

Lines of Context:
14
14
# along with this program; if not, write to the Free Software
15
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
16
 
17
 
from cStringIO import StringIO
18
 
import os
19
 
import time
20
 
 
21
17
from bzrlib import errors, inventory, osutils
22
 
from bzrlib.branch import Branch
23
 
from bzrlib.diff import internal_diff
24
18
from bzrlib.inventory import (Inventory, ROOT_ID, InventoryFile,
25
 
    InventoryDirectory, InventoryEntry)
26
 
from bzrlib.osutils import (has_symlinks, rename, pathjoin, is_inside_any, 
 
19
    InventoryDirectory, InventoryEntry, TreeReference)
 
20
from bzrlib.osutils import (pathjoin, is_inside_any, 
27
21
    is_inside_or_parent_of_any)
28
 
from bzrlib.tests import TestCase, TestCaseWithTransport
29
 
from bzrlib.transform import TreeTransform
30
 
from bzrlib.uncommit import uncommit
 
22
from bzrlib.tests import TestCase
31
23
 
32
24
 
33
25
class TestInventory(TestCase):
184
176
            ], [(path, ie.file_id) for path, ie in inv.iter_entries_by_dir(
185
177
                specific_file_ids=('bye-id',))])
186
178
 
187
 
    def test_version(self):
188
 
        """Inventory remembers the text's version."""
 
179
    def test_add_recursive(self):
 
180
        parent = InventoryDirectory('src-id', 'src', ROOT_ID)
 
181
        child = InventoryFile('hello-id', 'hello.c', 'src-id')
 
182
        parent.children[child.file_id] = child
189
183
        inv = Inventory()
190
 
        ie = inv.add_path('foo.txt', 'file')
191
 
        ## XXX
 
184
        inv.add(parent)
 
185
        self.assertEqual('src/hello.c', inv.id2path('hello-id'))
192
186
 
193
187
 
194
188
class TestInventoryEntry(TestCase):
280
274
            osutils.normalized_filename = orig_normalized_filename
281
275
 
282
276
 
283
 
class TestEntryDiffing(TestCaseWithTransport):
284
 
 
285
 
    def setUp(self):
286
 
        super(TestEntryDiffing, self).setUp()
287
 
        self.wt = self.make_branch_and_tree('.')
288
 
        self.branch = self.wt.branch
289
 
        print >> open('file', 'wb'), 'foo'
290
 
        print >> open('binfile', 'wb'), 'foo'
291
 
        self.wt.add(['file'], ['fileid'])
292
 
        self.wt.add(['binfile'], ['binfileid'])
293
 
        if has_symlinks():
294
 
            os.symlink('target1', 'symlink')
295
 
            self.wt.add(['symlink'], ['linkid'])
296
 
        self.wt.commit('message_1', rev_id = '1')
297
 
        print >> open('file', 'wb'), 'bar'
298
 
        print >> open('binfile', 'wb'), 'x' * 1023 + '\x00'
299
 
        if has_symlinks():
300
 
            os.unlink('symlink')
301
 
            os.symlink('target2', 'symlink')
302
 
        self.tree_1 = self.branch.repository.revision_tree('1')
303
 
        self.inv_1 = self.branch.repository.get_inventory('1')
304
 
        self.file_1 = self.inv_1['fileid']
305
 
        self.file_1b = self.inv_1['binfileid']
306
 
        self.tree_2 = self.wt
307
 
        self.inv_2 = self.tree_2.read_working_inventory()
308
 
        self.file_2 = self.inv_2['fileid']
309
 
        self.file_2b = self.inv_2['binfileid']
310
 
        if has_symlinks():
311
 
            self.link_1 = self.inv_1['linkid']
312
 
            self.link_2 = self.inv_2['linkid']
313
 
 
314
 
    def test_file_diff_deleted(self):
315
 
        output = StringIO()
316
 
        self.file_1.diff(internal_diff, 
317
 
                          "old_label", self.tree_1,
318
 
                          "/dev/null", None, None,
319
 
                          output)
320
 
        self.assertEqual(output.getvalue(), "--- old_label\n"
321
 
                                            "+++ /dev/null\n"
322
 
                                            "@@ -1,1 +0,0 @@\n"
323
 
                                            "-foo\n"
324
 
                                            "\n")
325
 
 
326
 
    def test_file_diff_added(self):
327
 
        output = StringIO()
328
 
        self.file_1.diff(internal_diff, 
329
 
                          "new_label", self.tree_1,
330
 
                          "/dev/null", None, None,
331
 
                          output, reverse=True)
332
 
        self.assertEqual(output.getvalue(), "--- /dev/null\n"
333
 
                                            "+++ new_label\n"
334
 
                                            "@@ -0,0 +1,1 @@\n"
335
 
                                            "+foo\n"
336
 
                                            "\n")
337
 
 
338
 
    def test_file_diff_changed(self):
339
 
        output = StringIO()
340
 
        self.file_1.diff(internal_diff, 
341
 
                          "/dev/null", self.tree_1, 
342
 
                          "new_label", self.file_2, self.tree_2,
343
 
                          output)
344
 
        self.assertEqual(output.getvalue(), "--- /dev/null\n"
345
 
                                            "+++ new_label\n"
346
 
                                            "@@ -1,1 +1,1 @@\n"
347
 
                                            "-foo\n"
348
 
                                            "+bar\n"
349
 
                                            "\n")
350
 
        
351
 
    def test_file_diff_binary(self):
352
 
        output = StringIO()
353
 
        self.file_1.diff(internal_diff, 
354
 
                          "/dev/null", self.tree_1, 
355
 
                          "new_label", self.file_2b, self.tree_2,
356
 
                          output)
357
 
        self.assertEqual(output.getvalue(), 
358
 
                         "Binary files /dev/null and new_label differ\n")
359
 
    def test_link_diff_deleted(self):
360
 
        if not has_symlinks():
361
 
            return
362
 
        output = StringIO()
363
 
        self.link_1.diff(internal_diff, 
364
 
                          "old_label", self.tree_1,
365
 
                          "/dev/null", None, None,
366
 
                          output)
367
 
        self.assertEqual(output.getvalue(),
368
 
                         "=== target was 'target1'\n")
369
 
 
370
 
    def test_link_diff_added(self):
371
 
        if not has_symlinks():
372
 
            return
373
 
        output = StringIO()
374
 
        self.link_1.diff(internal_diff, 
375
 
                          "new_label", self.tree_1,
376
 
                          "/dev/null", None, None,
377
 
                          output, reverse=True)
378
 
        self.assertEqual(output.getvalue(),
379
 
                         "=== target is 'target1'\n")
380
 
 
381
 
    def test_link_diff_changed(self):
382
 
        if not has_symlinks():
383
 
            return
384
 
        output = StringIO()
385
 
        self.link_1.diff(internal_diff, 
386
 
                          "/dev/null", self.tree_1, 
387
 
                          "new_label", self.link_2, self.tree_2,
388
 
                          output)
389
 
        self.assertEqual(output.getvalue(),
390
 
                         "=== target changed 'target1' => 'target2'\n")
391
 
 
392
 
 
393
 
class TestSnapshot(TestCaseWithTransport):
394
 
 
395
 
    def setUp(self):
396
 
        # for full testing we'll need a branch
397
 
        # with a subdir to test parent changes.
398
 
        # and a file, link and dir under that.
399
 
        # but right now I only need one attribute
400
 
        # to change, and then test merge patterns
401
 
        # with fake parent entries.
402
 
        super(TestSnapshot, self).setUp()
403
 
        self.wt = self.make_branch_and_tree('.')
404
 
        self.branch = self.wt.branch
405
 
        self.build_tree(['subdir/', 'subdir/file'], line_endings='binary')
406
 
        self.wt.add(['subdir', 'subdir/file'],
407
 
                                       ['dirid', 'fileid'])
408
 
        if has_symlinks():
409
 
            pass
410
 
        self.wt.commit('message_1', rev_id = '1')
411
 
        self.tree_1 = self.branch.repository.revision_tree('1')
412
 
        self.inv_1 = self.branch.repository.get_inventory('1')
413
 
        self.file_1 = self.inv_1['fileid']
414
 
        self.wt.lock_write()
415
 
        self.addCleanup(self.wt.unlock)
416
 
        self.file_active = self.wt.inventory['fileid']
417
 
        self.builder = self.branch.get_commit_builder([], timestamp=time.time(), revision_id='2')
418
 
 
419
 
    def test_snapshot_new_revision(self):
420
 
        # This tests that a simple commit with no parents makes a new
421
 
        # revision value in the inventory entry
422
 
        self.file_active.snapshot('2', 'subdir/file', {}, self.wt, self.builder)
423
 
        # expected outcome - file_1 has a revision id of '2', and we can get
424
 
        # its text of 'file contents' out of the weave.
425
 
        self.assertEqual(self.file_1.revision, '1')
426
 
        self.assertEqual(self.file_active.revision, '2')
427
 
        # this should be a separate test probably, but lets check it once..
428
 
        lines = self.branch.repository.weave_store.get_weave(
429
 
            'fileid', 
430
 
            self.branch.get_transaction()).get_lines('2')
431
 
        self.assertEqual(lines, ['contents of subdir/file\n'])
432
 
 
433
 
    def test_snapshot_unchanged(self):
434
 
        #This tests that a simple commit does not make a new entry for
435
 
        # an unchanged inventory entry
436
 
        self.file_active.snapshot('2', 'subdir/file', {'1':self.file_1},
437
 
                                  self.wt, self.builder)
438
 
        self.assertEqual(self.file_1.revision, '1')
439
 
        self.assertEqual(self.file_active.revision, '1')
440
 
        vf = self.branch.repository.weave_store.get_weave(
441
 
            'fileid', 
442
 
            self.branch.repository.get_transaction())
443
 
        self.assertRaises(errors.RevisionNotPresent,
444
 
                          vf.get_lines,
445
 
                          '2')
446
 
 
447
 
    def test_snapshot_merge_identical_different_revid(self):
448
 
        # This tests that a commit with two identical parents, one of which has
449
 
        # a different revision id, results in a new revision id in the entry.
450
 
        # 1->other, commit a merge of other against 1, results in 2.
451
 
        other_ie = inventory.InventoryFile('fileid', 'newname', self.file_1.parent_id)
452
 
        other_ie = inventory.InventoryFile('fileid', 'file', self.file_1.parent_id)
453
 
        other_ie.revision = '1'
454
 
        other_ie.text_sha1 = self.file_1.text_sha1
455
 
        other_ie.text_size = self.file_1.text_size
456
 
        self.assertEqual(self.file_1, other_ie)
457
 
        other_ie.revision = 'other'
458
 
        self.assertNotEqual(self.file_1, other_ie)
459
 
        versionfile = self.branch.repository.weave_store.get_weave(
460
 
            'fileid', self.branch.repository.get_transaction())
461
 
        versionfile.clone_text('other', '1', ['1'])
462
 
        self.file_active.snapshot('2', 'subdir/file', 
463
 
                                  {'1':self.file_1, 'other':other_ie},
464
 
                                  self.wt, self.builder)
465
 
        self.assertEqual(self.file_active.revision, '2')
466
 
 
467
 
    def test_snapshot_changed(self):
468
 
        # This tests that a commit with one different parent results in a new
469
 
        # revision id in the entry.
470
 
        self.wt.rename_one('subdir/file', 'subdir/newname')
471
 
        self.file_active = self.wt.inventory['fileid']
472
 
        self.file_active.snapshot('2', 'subdir/newname', {'1':self.file_1}, 
473
 
                                  self.wt, self.builder)
474
 
        # expected outcome - file_1 has a revision id of '2'
475
 
        self.assertEqual(self.file_active.revision, '2')
476
 
 
477
 
 
478
 
class TestPreviousHeads(TestCaseWithTransport):
479
 
 
480
 
    def setUp(self):
481
 
        # we want several inventories, that respectively
482
 
        # give use the following scenarios:
483
 
        # A) fileid not in any inventory (A),
484
 
        # B) fileid present in one inventory (B) and (A,B)
485
 
        # C) fileid present in two inventories, and they
486
 
        #   are not mutual descendents (B, C)
487
 
        # D) fileid present in two inventories and one is
488
 
        #   a descendent of the other. (B, D)
489
 
        super(TestPreviousHeads, self).setUp()
490
 
        self.wt = self.make_branch_and_tree('.')
491
 
        self.branch = self.wt.branch
492
 
        self.build_tree(['file'])
493
 
        self.wt.commit('new branch', allow_pointless=True, rev_id='A')
494
 
        self.inv_A = self.branch.repository.get_inventory('A')
495
 
        self.wt.add(['file'], ['fileid'])
496
 
        self.wt.commit('add file', rev_id='B')
497
 
        self.inv_B = self.branch.repository.get_inventory('B')
498
 
        uncommit(self.branch, tree=self.wt)
499
 
        self.assertEqual(self.branch.revision_history(), ['A'])
500
 
        self.wt.commit('another add of file', rev_id='C')
501
 
        self.inv_C = self.branch.repository.get_inventory('C')
502
 
        self.wt.add_parent_tree_id('B')
503
 
        self.wt.commit('merge in B', rev_id='D')
504
 
        self.inv_D = self.branch.repository.get_inventory('D')
505
 
        self.file_active = self.wt.inventory['fileid']
506
 
        self.weave = self.branch.repository.weave_store.get_weave('fileid',
507
 
            self.branch.repository.get_transaction())
508
 
        
509
 
    def get_previous_heads(self, inventories):
510
 
        return self.file_active.find_previous_heads(
511
 
            inventories, 
512
 
            self.branch.repository.weave_store,
513
 
            self.branch.repository.get_transaction())
514
 
        
515
 
    def test_fileid_in_no_inventory(self):
516
 
        self.assertEqual({}, self.get_previous_heads([self.inv_A]))
517
 
 
518
 
    def test_fileid_in_one_inventory(self):
519
 
        self.assertEqual({'B':self.inv_B['fileid']},
520
 
                         self.get_previous_heads([self.inv_B]))
521
 
        self.assertEqual({'B':self.inv_B['fileid']},
522
 
                         self.get_previous_heads([self.inv_A, self.inv_B]))
523
 
        self.assertEqual({'B':self.inv_B['fileid']},
524
 
                         self.get_previous_heads([self.inv_B, self.inv_A]))
525
 
 
526
 
    def test_fileid_in_two_inventories_gives_both_entries(self):
527
 
        self.assertEqual({'B':self.inv_B['fileid'],
528
 
                          'C':self.inv_C['fileid']},
529
 
                          self.get_previous_heads([self.inv_B, self.inv_C]))
530
 
        self.assertEqual({'B':self.inv_B['fileid'],
531
 
                          'C':self.inv_C['fileid']},
532
 
                          self.get_previous_heads([self.inv_C, self.inv_B]))
533
 
 
534
 
    def test_fileid_in_two_inventories_already_merged_gives_head(self):
535
 
        self.assertEqual({'D':self.inv_D['fileid']},
536
 
                         self.get_previous_heads([self.inv_B, self.inv_D]))
537
 
        self.assertEqual({'D':self.inv_D['fileid']},
538
 
                         self.get_previous_heads([self.inv_D, self.inv_B]))
539
 
 
540
 
    # TODO: test two inventories with the same file revision 
541
 
 
542
 
 
543
277
class TestDescribeChanges(TestCase):
544
278
 
545
279
    def test_describe_change(self):
598
332
        self.assertEqual(expected_change, change)
599
333
 
600
334
 
601
 
class TestRevert(TestCaseWithTransport):
602
 
 
603
 
    def test_dangling_id(self):
604
 
        wt = self.make_branch_and_tree('b1')
605
 
        self.assertEqual(len(wt.inventory), 1)
606
 
        open('b1/a', 'wb').write('a test\n')
607
 
        wt.add('a')
608
 
        self.assertEqual(len(wt.inventory), 2)
609
 
        os.unlink('b1/a')
610
 
        wt.revert([])
611
 
        self.assertEqual(len(wt.inventory), 1)
612
 
 
613
 
 
614
335
class TestIsRoot(TestCase):
615
336
    """Ensure our root-checking code is accurate."""
616
337
 
625
346
        inv.root = None
626
347
        self.assertFalse(inv.is_root('TREE_ROOT'))
627
348
        self.assertFalse(inv.is_root('booga'))
 
349
 
 
350
 
 
351
class TestTreeReference(TestCase):
 
352
    
 
353
    def test_create(self):
 
354
        inv = Inventory('tree-root-123')
 
355
        inv.add(TreeReference('nested-id', 'nested', parent_id='tree-root-123',
 
356
                              revision='rev', reference_revision='rev2'))
 
357
 
 
358
 
 
359
class TestEncoding(TestCase):
 
360
 
 
361
    def test_error_encoding(self):
 
362
        inv = Inventory('tree-root')
 
363
        inv.add(InventoryFile('a-id', u'\u1234', 'tree-root'))
 
364
        try:
 
365
            inv.add(InventoryFile('b-id', u'\u1234', 'tree-root'))
 
366
        except errors.BzrError, e:
 
367
            self.assertContainsRe(str(e), u'\u1234'.encode('utf-8'))
 
368
        else:
 
369
            self.fail('BzrError not raised')