~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_knit.py

  • Committer: John Arbash Meinel
  • Date: 2008-07-09 21:42:24 UTC
  • mto: This revision was merged to the branch mainline in revision 3543.
  • Revision ID: john@arbash-meinel.com-20080709214224-r75k87r6a01pfc3h
Restore a real weave merge to 'bzr merge --weave'.

To do so efficiently, we only add the simple LCAs to the final weave
object, unless we run into complexities with the merge graph.
This gives the same effective result as adding all the texts,
with the advantage of not having to extract all of them.

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2006-2011 Canonical Ltd
 
1
# Copyright (C) 2005, 2006, 2007 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
12
12
#
13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
16
 
17
17
"""Tests for Knit data structure"""
18
18
 
19
19
from cStringIO import StringIO
 
20
import difflib
20
21
import gzip
 
22
import sha
21
23
import sys
22
24
 
23
25
from bzrlib import (
24
26
    errors,
 
27
    generate_ids,
25
28
    knit,
26
 
    multiparent,
27
 
    osutils,
28
29
    pack,
29
 
    tests,
30
 
    transport,
31
30
    )
32
31
from bzrlib.errors import (
 
32
    RevisionAlreadyPresent,
33
33
    KnitHeaderError,
 
34
    RevisionNotPresent,
34
35
    NoSuchFile,
35
36
    )
36
37
from bzrlib.index import *
37
38
from bzrlib.knit import (
38
39
    AnnotatedKnitContent,
39
40
    KnitContent,
 
41
    KnitSequenceMatcher,
40
42
    KnitVersionedFiles,
41
43
    PlainKnitContent,
42
 
    _VFContentMapGenerator,
 
44
    _DirectPackAccess,
43
45
    _KndxIndex,
44
46
    _KnitGraphIndex,
45
47
    _KnitKeyAccess,
46
48
    make_file_factory,
47
49
    )
48
 
from bzrlib.patiencediff import PatienceSequenceMatcher
49
 
from bzrlib.repofmt import (
50
 
    knitpack_repo,
51
 
    pack_repo,
52
 
    )
 
50
from bzrlib.osutils import split_lines
 
51
from bzrlib.symbol_versioning import one_four
53
52
from bzrlib.tests import (
 
53
    Feature,
54
54
    TestCase,
55
55
    TestCaseWithMemoryTransport,
56
56
    TestCaseWithTransport,
57
 
    TestNotApplicable,
58
 
    )
59
 
from bzrlib.versionedfile import (
60
 
    AbsentContentFactory,
61
 
    ConstantMapper,
62
 
    network_bytes_to_kind_and_offset,
63
 
    RecordingVersionedFilesDecorator,
64
 
    )
65
 
from bzrlib.tests import (
66
 
    features,
67
 
    )
68
 
 
69
 
 
70
 
compiled_knit_feature = features.ModuleAvailableFeature(
71
 
    'bzrlib._knit_load_data_pyx')
 
57
    )
 
58
from bzrlib.transport import get_transport
 
59
from bzrlib.transport.memory import MemoryTransport
 
60
from bzrlib.tuned_gzip import GzipFile
 
61
from bzrlib.versionedfile import ConstantMapper
 
62
 
 
63
 
 
64
class _CompiledKnitFeature(Feature):
 
65
 
 
66
    def _probe(self):
 
67
        try:
 
68
            import bzrlib._knit_load_data_c
 
69
        except ImportError:
 
70
            return False
 
71
        return True
 
72
 
 
73
    def feature_name(self):
 
74
        return 'bzrlib._knit_load_data_c'
 
75
 
 
76
CompiledKnitFeature = _CompiledKnitFeature()
72
77
 
73
78
 
74
79
class KnitContentTestsMixin(object):
103
108
        line_delta = source_content.line_delta(target_content)
104
109
        delta_blocks = list(KnitContent.get_line_delta_blocks(line_delta,
105
110
            source_lines, target_lines))
106
 
        matcher = PatienceSequenceMatcher(None, source_lines, target_lines)
107
 
        matcher_blocks = list(matcher.get_matching_blocks())
 
111
        matcher = KnitSequenceMatcher(None, source_lines, target_lines)
 
112
        matcher_blocks = list(list(matcher.get_matching_blocks()))
108
113
        self.assertEqual(matcher_blocks, delta_blocks)
109
114
 
110
115
    def test_get_line_delta_blocks(self):
260
265
        return queue_call
261
266
 
262
267
 
263
 
class MockReadvFailingTransport(MockTransport):
264
 
    """Fail in the middle of a readv() result.
265
 
 
266
 
    This Transport will successfully yield the first two requested hunks, but
267
 
    raise NoSuchFile for the rest.
268
 
    """
269
 
 
270
 
    def readv(self, relpath, offsets):
271
 
        count = 0
272
 
        for result in MockTransport.readv(self, relpath, offsets):
273
 
            count += 1
274
 
            # we use 2 because the first offset is the pack header, the second
275
 
            # is the first actual content requset
276
 
            if count > 2:
277
 
                raise errors.NoSuchFile(relpath)
278
 
            yield result
279
 
 
280
 
 
281
268
class KnitRecordAccessTestsMixin(object):
282
269
    """Tests for getting and putting knit records."""
283
270
 
286
273
        access = self.get_access()
287
274
        memos = access.add_raw_records([('key', 10)], '1234567890')
288
275
        self.assertEqual(['1234567890'], list(access.get_raw_records(memos)))
289
 
 
 
276
 
290
277
    def test_add_several_raw_records(self):
291
278
        """add_raw_records with many records and read some back."""
292
279
        access = self.get_access()
312
299
        mapper = ConstantMapper("foo")
313
300
        access = _KnitKeyAccess(self.get_transport(), mapper)
314
301
        return access
315
 
 
316
 
 
317
 
class _TestException(Exception):
318
 
    """Just an exception for local tests to use."""
319
 
 
 
302
    
320
303
 
321
304
class TestPackKnitAccess(TestCaseWithMemoryTransport, KnitRecordAccessTestsMixin):
322
305
    """Tests for the pack based access."""
330
313
            transport.append_bytes(packname, bytes)
331
314
        writer = pack.ContainerWriter(write_data)
332
315
        writer.begin()
333
 
        access = pack_repo._DirectPackAccess({})
 
316
        access = _DirectPackAccess({})
334
317
        access.set_writer(writer, index, (transport, packname))
335
318
        return access, writer
336
319
 
337
 
    def make_pack_file(self):
338
 
        """Create a pack file with 2 records."""
339
 
        access, writer = self._get_access(packname='packname', index='foo')
340
 
        memos = []
341
 
        memos.extend(access.add_raw_records([('key1', 10)], '1234567890'))
342
 
        memos.extend(access.add_raw_records([('key2', 5)], '12345'))
343
 
        writer.end()
344
 
        return memos
345
 
 
346
 
    def test_pack_collection_pack_retries(self):
347
 
        """An explicit pack of a pack collection succeeds even when a
348
 
        concurrent pack happens.
349
 
        """
350
 
        builder = self.make_branch_builder('.')
351
 
        builder.start_series()
352
 
        builder.build_snapshot('rev-1', None, [
353
 
            ('add', ('', 'root-id', 'directory', None)),
354
 
            ('add', ('file', 'file-id', 'file', 'content\nrev 1\n')),
355
 
            ])
356
 
        builder.build_snapshot('rev-2', ['rev-1'], [
357
 
            ('modify', ('file-id', 'content\nrev 2\n')),
358
 
            ])
359
 
        builder.build_snapshot('rev-3', ['rev-2'], [
360
 
            ('modify', ('file-id', 'content\nrev 3\n')),
361
 
            ])
362
 
        self.addCleanup(builder.finish_series)
363
 
        b = builder.get_branch()
364
 
        self.addCleanup(b.lock_write().unlock)
365
 
        repo = b.repository
366
 
        collection = repo._pack_collection
367
 
        # Concurrently repack the repo.
368
 
        reopened_repo = repo.bzrdir.open_repository()
369
 
        reopened_repo.pack()
370
 
        # Pack the new pack.
371
 
        collection.pack()
372
 
 
373
 
    def make_vf_for_retrying(self):
374
 
        """Create 3 packs and a reload function.
375
 
 
376
 
        Originally, 2 pack files will have the data, but one will be missing.
377
 
        And then the third will be used in place of the first two if reload()
378
 
        is called.
379
 
 
380
 
        :return: (versioned_file, reload_counter)
381
 
            versioned_file  a KnitVersionedFiles using the packs for access
382
 
        """
383
 
        builder = self.make_branch_builder('.', format="1.9")
384
 
        builder.start_series()
385
 
        builder.build_snapshot('rev-1', None, [
386
 
            ('add', ('', 'root-id', 'directory', None)),
387
 
            ('add', ('file', 'file-id', 'file', 'content\nrev 1\n')),
388
 
            ])
389
 
        builder.build_snapshot('rev-2', ['rev-1'], [
390
 
            ('modify', ('file-id', 'content\nrev 2\n')),
391
 
            ])
392
 
        builder.build_snapshot('rev-3', ['rev-2'], [
393
 
            ('modify', ('file-id', 'content\nrev 3\n')),
394
 
            ])
395
 
        builder.finish_series()
396
 
        b = builder.get_branch()
397
 
        b.lock_write()
398
 
        self.addCleanup(b.unlock)
399
 
        # Pack these three revisions into another pack file, but don't remove
400
 
        # the originals
401
 
        repo = b.repository
402
 
        collection = repo._pack_collection
403
 
        collection.ensure_loaded()
404
 
        orig_packs = collection.packs
405
 
        packer = knitpack_repo.KnitPacker(collection, orig_packs, '.testpack')
406
 
        new_pack = packer.pack()
407
 
        # forget about the new pack
408
 
        collection.reset()
409
 
        repo.refresh_data()
410
 
        vf = repo.revisions
411
 
        # Set up a reload() function that switches to using the new pack file
412
 
        new_index = new_pack.revision_index
413
 
        access_tuple = new_pack.access_tuple()
414
 
        reload_counter = [0, 0, 0]
415
 
        def reload():
416
 
            reload_counter[0] += 1
417
 
            if reload_counter[1] > 0:
418
 
                # We already reloaded, nothing more to do
419
 
                reload_counter[2] += 1
420
 
                return False
421
 
            reload_counter[1] += 1
422
 
            vf._index._graph_index._indices[:] = [new_index]
423
 
            vf._access._indices.clear()
424
 
            vf._access._indices[new_index] = access_tuple
425
 
            return True
426
 
        # Delete one of the pack files so the data will need to be reloaded. We
427
 
        # will delete the file with 'rev-2' in it
428
 
        trans, name = orig_packs[1].access_tuple()
429
 
        trans.delete(name)
430
 
        # We don't have the index trigger reloading because we want to test
431
 
        # that we reload when the .pack disappears
432
 
        vf._access._reload_func = reload
433
 
        return vf, reload_counter
434
 
 
435
 
    def make_reload_func(self, return_val=True):
436
 
        reload_called = [0]
437
 
        def reload():
438
 
            reload_called[0] += 1
439
 
            return return_val
440
 
        return reload_called, reload
441
 
 
442
 
    def make_retry_exception(self):
443
 
        # We raise a real exception so that sys.exc_info() is properly
444
 
        # populated
445
 
        try:
446
 
            raise _TestException('foobar')
447
 
        except _TestException, e:
448
 
            retry_exc = errors.RetryWithNewPacks(None, reload_occurred=False,
449
 
                                                 exc_info=sys.exc_info())
450
 
        # GZ 2010-08-10: Cycle with exc_info affects 3 tests
451
 
        return retry_exc
452
 
 
453
320
    def test_read_from_several_packs(self):
454
321
        access, writer = self._get_access()
455
322
        memos = []
462
329
        memos.extend(access.add_raw_records([('key', 5)], 'alpha'))
463
330
        writer.end()
464
331
        transport = self.get_transport()
465
 
        access = pack_repo._DirectPackAccess({"FOO":(transport, 'packfile'),
 
332
        access = _DirectPackAccess({"FOO":(transport, 'packfile'),
466
333
            "FOOBAR":(transport, 'pack2'),
467
334
            "BAZ":(transport, 'pack3')})
468
335
        self.assertEqual(['1234567890', '12345', 'alpha'],
478
345
 
479
346
    def test_set_writer(self):
480
347
        """The writer should be settable post construction."""
481
 
        access = pack_repo._DirectPackAccess({})
 
348
        access = _DirectPackAccess({})
482
349
        transport = self.get_transport()
483
350
        packname = 'packfile'
484
351
        index = 'foo'
491
358
        writer.end()
492
359
        self.assertEqual(['1234567890'], list(access.get_raw_records(memos)))
493
360
 
494
 
    def test_missing_index_raises_retry(self):
495
 
        memos = self.make_pack_file()
496
 
        transport = self.get_transport()
497
 
        reload_called, reload_func = self.make_reload_func()
498
 
        # Note that the index key has changed from 'foo' to 'bar'
499
 
        access = pack_repo._DirectPackAccess({'bar':(transport, 'packname')},
500
 
                                   reload_func=reload_func)
501
 
        e = self.assertListRaises(errors.RetryWithNewPacks,
502
 
                                  access.get_raw_records, memos)
503
 
        # Because a key was passed in which does not match our index list, we
504
 
        # assume that the listing was already reloaded
505
 
        self.assertTrue(e.reload_occurred)
506
 
        self.assertIsInstance(e.exc_info, tuple)
507
 
        self.assertIs(e.exc_info[0], KeyError)
508
 
        self.assertIsInstance(e.exc_info[1], KeyError)
509
 
 
510
 
    def test_missing_index_raises_key_error_with_no_reload(self):
511
 
        memos = self.make_pack_file()
512
 
        transport = self.get_transport()
513
 
        # Note that the index key has changed from 'foo' to 'bar'
514
 
        access = pack_repo._DirectPackAccess({'bar':(transport, 'packname')})
515
 
        e = self.assertListRaises(KeyError, access.get_raw_records, memos)
516
 
 
517
 
    def test_missing_file_raises_retry(self):
518
 
        memos = self.make_pack_file()
519
 
        transport = self.get_transport()
520
 
        reload_called, reload_func = self.make_reload_func()
521
 
        # Note that the 'filename' has been changed to 'different-packname'
522
 
        access = pack_repo._DirectPackAccess(
523
 
            {'foo':(transport, 'different-packname')},
524
 
            reload_func=reload_func)
525
 
        e = self.assertListRaises(errors.RetryWithNewPacks,
526
 
                                  access.get_raw_records, memos)
527
 
        # The file has gone missing, so we assume we need to reload
528
 
        self.assertFalse(e.reload_occurred)
529
 
        self.assertIsInstance(e.exc_info, tuple)
530
 
        self.assertIs(e.exc_info[0], errors.NoSuchFile)
531
 
        self.assertIsInstance(e.exc_info[1], errors.NoSuchFile)
532
 
        self.assertEqual('different-packname', e.exc_info[1].path)
533
 
 
534
 
    def test_missing_file_raises_no_such_file_with_no_reload(self):
535
 
        memos = self.make_pack_file()
536
 
        transport = self.get_transport()
537
 
        # Note that the 'filename' has been changed to 'different-packname'
538
 
        access = pack_repo._DirectPackAccess(
539
 
            {'foo': (transport, 'different-packname')})
540
 
        e = self.assertListRaises(errors.NoSuchFile,
541
 
                                  access.get_raw_records, memos)
542
 
 
543
 
    def test_failing_readv_raises_retry(self):
544
 
        memos = self.make_pack_file()
545
 
        transport = self.get_transport()
546
 
        failing_transport = MockReadvFailingTransport(
547
 
                                [transport.get_bytes('packname')])
548
 
        reload_called, reload_func = self.make_reload_func()
549
 
        access = pack_repo._DirectPackAccess(
550
 
            {'foo': (failing_transport, 'packname')},
551
 
            reload_func=reload_func)
552
 
        # Asking for a single record will not trigger the Mock failure
553
 
        self.assertEqual(['1234567890'],
554
 
            list(access.get_raw_records(memos[:1])))
555
 
        self.assertEqual(['12345'],
556
 
            list(access.get_raw_records(memos[1:2])))
557
 
        # A multiple offset readv() will fail mid-way through
558
 
        e = self.assertListRaises(errors.RetryWithNewPacks,
559
 
                                  access.get_raw_records, memos)
560
 
        # The file has gone missing, so we assume we need to reload
561
 
        self.assertFalse(e.reload_occurred)
562
 
        self.assertIsInstance(e.exc_info, tuple)
563
 
        self.assertIs(e.exc_info[0], errors.NoSuchFile)
564
 
        self.assertIsInstance(e.exc_info[1], errors.NoSuchFile)
565
 
        self.assertEqual('packname', e.exc_info[1].path)
566
 
 
567
 
    def test_failing_readv_raises_no_such_file_with_no_reload(self):
568
 
        memos = self.make_pack_file()
569
 
        transport = self.get_transport()
570
 
        failing_transport = MockReadvFailingTransport(
571
 
                                [transport.get_bytes('packname')])
572
 
        reload_called, reload_func = self.make_reload_func()
573
 
        access = pack_repo._DirectPackAccess(
574
 
            {'foo':(failing_transport, 'packname')})
575
 
        # Asking for a single record will not trigger the Mock failure
576
 
        self.assertEqual(['1234567890'],
577
 
            list(access.get_raw_records(memos[:1])))
578
 
        self.assertEqual(['12345'],
579
 
            list(access.get_raw_records(memos[1:2])))
580
 
        # A multiple offset readv() will fail mid-way through
581
 
        e = self.assertListRaises(errors.NoSuchFile,
582
 
                                  access.get_raw_records, memos)
583
 
 
584
 
    def test_reload_or_raise_no_reload(self):
585
 
        access = pack_repo._DirectPackAccess({}, reload_func=None)
586
 
        retry_exc = self.make_retry_exception()
587
 
        # Without a reload_func, we will just re-raise the original exception
588
 
        self.assertRaises(_TestException, access.reload_or_raise, retry_exc)
589
 
 
590
 
    def test_reload_or_raise_reload_changed(self):
591
 
        reload_called, reload_func = self.make_reload_func(return_val=True)
592
 
        access = pack_repo._DirectPackAccess({}, reload_func=reload_func)
593
 
        retry_exc = self.make_retry_exception()
594
 
        access.reload_or_raise(retry_exc)
595
 
        self.assertEqual([1], reload_called)
596
 
        retry_exc.reload_occurred=True
597
 
        access.reload_or_raise(retry_exc)
598
 
        self.assertEqual([2], reload_called)
599
 
 
600
 
    def test_reload_or_raise_reload_no_change(self):
601
 
        reload_called, reload_func = self.make_reload_func(return_val=False)
602
 
        access = pack_repo._DirectPackAccess({}, reload_func=reload_func)
603
 
        retry_exc = self.make_retry_exception()
604
 
        # If reload_occurred is False, then we consider it an error to have
605
 
        # reload_func() return False (no changes).
606
 
        self.assertRaises(_TestException, access.reload_or_raise, retry_exc)
607
 
        self.assertEqual([1], reload_called)
608
 
        retry_exc.reload_occurred=True
609
 
        # If reload_occurred is True, then we assume nothing changed because
610
 
        # it had changed earlier, but didn't change again
611
 
        access.reload_or_raise(retry_exc)
612
 
        self.assertEqual([2], reload_called)
613
 
 
614
 
    def test_annotate_retries(self):
615
 
        vf, reload_counter = self.make_vf_for_retrying()
616
 
        # It is a little bit bogus to annotate the Revision VF, but it works,
617
 
        # as we have ancestry stored there
618
 
        key = ('rev-3',)
619
 
        reload_lines = vf.annotate(key)
620
 
        self.assertEqual([1, 1, 0], reload_counter)
621
 
        plain_lines = vf.annotate(key)
622
 
        self.assertEqual([1, 1, 0], reload_counter) # No extra reloading
623
 
        if reload_lines != plain_lines:
624
 
            self.fail('Annotation was not identical with reloading.')
625
 
        # Now delete the packs-in-use, which should trigger another reload, but
626
 
        # this time we just raise an exception because we can't recover
627
 
        for trans, name in vf._access._indices.itervalues():
628
 
            trans.delete(name)
629
 
        self.assertRaises(errors.NoSuchFile, vf.annotate, key)
630
 
        self.assertEqual([2, 1, 1], reload_counter)
631
 
 
632
 
    def test__get_record_map_retries(self):
633
 
        vf, reload_counter = self.make_vf_for_retrying()
634
 
        keys = [('rev-1',), ('rev-2',), ('rev-3',)]
635
 
        records = vf._get_record_map(keys)
636
 
        self.assertEqual(keys, sorted(records.keys()))
637
 
        self.assertEqual([1, 1, 0], reload_counter)
638
 
        # Now delete the packs-in-use, which should trigger another reload, but
639
 
        # this time we just raise an exception because we can't recover
640
 
        for trans, name in vf._access._indices.itervalues():
641
 
            trans.delete(name)
642
 
        self.assertRaises(errors.NoSuchFile, vf._get_record_map, keys)
643
 
        self.assertEqual([2, 1, 1], reload_counter)
644
 
 
645
 
    def test_get_record_stream_retries(self):
646
 
        vf, reload_counter = self.make_vf_for_retrying()
647
 
        keys = [('rev-1',), ('rev-2',), ('rev-3',)]
648
 
        record_stream = vf.get_record_stream(keys, 'topological', False)
649
 
        record = record_stream.next()
650
 
        self.assertEqual(('rev-1',), record.key)
651
 
        self.assertEqual([0, 0, 0], reload_counter)
652
 
        record = record_stream.next()
653
 
        self.assertEqual(('rev-2',), record.key)
654
 
        self.assertEqual([1, 1, 0], reload_counter)
655
 
        record = record_stream.next()
656
 
        self.assertEqual(('rev-3',), record.key)
657
 
        self.assertEqual([1, 1, 0], reload_counter)
658
 
        # Now delete all pack files, and see that we raise the right error
659
 
        for trans, name in vf._access._indices.itervalues():
660
 
            trans.delete(name)
661
 
        self.assertListRaises(errors.NoSuchFile,
662
 
            vf.get_record_stream, keys, 'topological', False)
663
 
 
664
 
    def test_iter_lines_added_or_present_in_keys_retries(self):
665
 
        vf, reload_counter = self.make_vf_for_retrying()
666
 
        keys = [('rev-1',), ('rev-2',), ('rev-3',)]
667
 
        # Unfortunately, iter_lines_added_or_present_in_keys iterates the
668
 
        # result in random order (determined by the iteration order from a
669
 
        # set()), so we don't have any solid way to trigger whether data is
670
 
        # read before or after. However we tried to delete the middle node to
671
 
        # exercise the code well.
672
 
        # What we care about is that all lines are always yielded, but not
673
 
        # duplicated
674
 
        count = 0
675
 
        reload_lines = sorted(vf.iter_lines_added_or_present_in_keys(keys))
676
 
        self.assertEqual([1, 1, 0], reload_counter)
677
 
        # Now do it again, to make sure the result is equivalent
678
 
        plain_lines = sorted(vf.iter_lines_added_or_present_in_keys(keys))
679
 
        self.assertEqual([1, 1, 0], reload_counter) # No extra reloading
680
 
        self.assertEqual(plain_lines, reload_lines)
681
 
        self.assertEqual(21, len(plain_lines))
682
 
        # Now delete all pack files, and see that we raise the right error
683
 
        for trans, name in vf._access._indices.itervalues():
684
 
            trans.delete(name)
685
 
        self.assertListRaises(errors.NoSuchFile,
686
 
            vf.iter_lines_added_or_present_in_keys, keys)
687
 
        self.assertEqual([2, 1, 1], reload_counter)
688
 
 
689
 
    def test_get_record_stream_yields_disk_sorted_order(self):
690
 
        # if we get 'unordered' pick a semi-optimal order for reading. The
691
 
        # order should be grouped by pack file, and then by position in file
692
 
        repo = self.make_repository('test', format='pack-0.92')
693
 
        repo.lock_write()
694
 
        self.addCleanup(repo.unlock)
695
 
        repo.start_write_group()
696
 
        vf = repo.texts
697
 
        vf.add_lines(('f-id', 'rev-5'), [('f-id', 'rev-4')], ['lines\n'])
698
 
        vf.add_lines(('f-id', 'rev-1'), [], ['lines\n'])
699
 
        vf.add_lines(('f-id', 'rev-2'), [('f-id', 'rev-1')], ['lines\n'])
700
 
        repo.commit_write_group()
701
 
        # We inserted them as rev-5, rev-1, rev-2, we should get them back in
702
 
        # the same order
703
 
        stream = vf.get_record_stream([('f-id', 'rev-1'), ('f-id', 'rev-5'),
704
 
                                       ('f-id', 'rev-2')], 'unordered', False)
705
 
        keys = [r.key for r in stream]
706
 
        self.assertEqual([('f-id', 'rev-5'), ('f-id', 'rev-1'),
707
 
                          ('f-id', 'rev-2')], keys)
708
 
        repo.start_write_group()
709
 
        vf.add_lines(('f-id', 'rev-4'), [('f-id', 'rev-3')], ['lines\n'])
710
 
        vf.add_lines(('f-id', 'rev-3'), [('f-id', 'rev-2')], ['lines\n'])
711
 
        vf.add_lines(('f-id', 'rev-6'), [('f-id', 'rev-5')], ['lines\n'])
712
 
        repo.commit_write_group()
713
 
        # Request in random order, to make sure the output order isn't based on
714
 
        # the request
715
 
        request_keys = set(('f-id', 'rev-%d' % i) for i in range(1, 7))
716
 
        stream = vf.get_record_stream(request_keys, 'unordered', False)
717
 
        keys = [r.key for r in stream]
718
 
        # We want to get the keys back in disk order, but it doesn't matter
719
 
        # which pack we read from first. So this can come back in 2 orders
720
 
        alt1 = [('f-id', 'rev-%d' % i) for i in [4, 3, 6, 5, 1, 2]]
721
 
        alt2 = [('f-id', 'rev-%d' % i) for i in [5, 1, 2, 4, 3, 6]]
722
 
        if keys != alt1 and keys != alt2:
723
 
            self.fail('Returned key order did not match either expected order.'
724
 
                      ' expected %s or %s, not %s'
725
 
                      % (alt1, alt2, keys))
726
 
 
727
361
 
728
362
class LowLevelKnitDataTests(TestCase):
729
363
 
734
368
        gz_file.close()
735
369
        return sio.getvalue()
736
370
 
737
 
    def make_multiple_records(self):
738
 
        """Create the content for multiple records."""
739
 
        sha1sum = osutils.sha_string('foo\nbar\n')
740
 
        total_txt = []
741
 
        gz_txt = self.create_gz_content('version rev-id-1 2 %s\n'
742
 
                                        'foo\n'
743
 
                                        'bar\n'
744
 
                                        'end rev-id-1\n'
745
 
                                        % (sha1sum,))
746
 
        record_1 = (0, len(gz_txt), sha1sum)
747
 
        total_txt.append(gz_txt)
748
 
        sha1sum = osutils.sha_string('baz\n')
749
 
        gz_txt = self.create_gz_content('version rev-id-2 1 %s\n'
750
 
                                        'baz\n'
751
 
                                        'end rev-id-2\n'
752
 
                                        % (sha1sum,))
753
 
        record_2 = (record_1[1], len(gz_txt), sha1sum)
754
 
        total_txt.append(gz_txt)
755
 
        return total_txt, record_1, record_2
756
 
 
757
371
    def test_valid_knit_data(self):
758
 
        sha1sum = osutils.sha_string('foo\nbar\n')
 
372
        sha1sum = sha.new('foo\nbar\n').hexdigest()
759
373
        gz_txt = self.create_gz_content('version rev-id-1 2 %s\n'
760
374
                                        'foo\n'
761
375
                                        'bar\n'
773
387
        raw_contents = list(knit._read_records_iter_raw(records))
774
388
        self.assertEqual([(('rev-id-1',), gz_txt, sha1sum)], raw_contents)
775
389
 
776
 
    def test_multiple_records_valid(self):
777
 
        total_txt, record_1, record_2 = self.make_multiple_records()
778
 
        transport = MockTransport([''.join(total_txt)])
779
 
        access = _KnitKeyAccess(transport, ConstantMapper('filename'))
780
 
        knit = KnitVersionedFiles(None, access)
781
 
        records = [(('rev-id-1',), (('rev-id-1',), record_1[0], record_1[1])),
782
 
                   (('rev-id-2',), (('rev-id-2',), record_2[0], record_2[1]))]
783
 
 
784
 
        contents = list(knit._read_records_iter(records))
785
 
        self.assertEqual([(('rev-id-1',), ['foo\n', 'bar\n'], record_1[2]),
786
 
                          (('rev-id-2',), ['baz\n'], record_2[2])],
787
 
                         contents)
788
 
 
789
 
        raw_contents = list(knit._read_records_iter_raw(records))
790
 
        self.assertEqual([(('rev-id-1',), total_txt[0], record_1[2]),
791
 
                          (('rev-id-2',), total_txt[1], record_2[2])],
792
 
                         raw_contents)
793
 
 
794
390
    def test_not_enough_lines(self):
795
 
        sha1sum = osutils.sha_string('foo\n')
 
391
        sha1sum = sha.new('foo\n').hexdigest()
796
392
        # record says 2 lines data says 1
797
393
        gz_txt = self.create_gz_content('version rev-id-1 2 %s\n'
798
394
                                        'foo\n'
810
406
        self.assertEqual([(('rev-id-1',),  gz_txt, sha1sum)], raw_contents)
811
407
 
812
408
    def test_too_many_lines(self):
813
 
        sha1sum = osutils.sha_string('foo\nbar\n')
 
409
        sha1sum = sha.new('foo\nbar\n').hexdigest()
814
410
        # record says 1 lines data says 2
815
411
        gz_txt = self.create_gz_content('version rev-id-1 1 %s\n'
816
412
                                        'foo\n'
829
425
        self.assertEqual([(('rev-id-1',), gz_txt, sha1sum)], raw_contents)
830
426
 
831
427
    def test_mismatched_version_id(self):
832
 
        sha1sum = osutils.sha_string('foo\nbar\n')
 
428
        sha1sum = sha.new('foo\nbar\n').hexdigest()
833
429
        gz_txt = self.create_gz_content('version rev-id-1 2 %s\n'
834
430
                                        'foo\n'
835
431
                                        'bar\n'
848
444
            knit._read_records_iter_raw(records))
849
445
 
850
446
    def test_uncompressed_data(self):
851
 
        sha1sum = osutils.sha_string('foo\nbar\n')
 
447
        sha1sum = sha.new('foo\nbar\n').hexdigest()
852
448
        txt = ('version rev-id-1 2 %s\n'
853
449
               'foo\n'
854
450
               'bar\n'
868
464
            knit._read_records_iter_raw(records))
869
465
 
870
466
    def test_corrupted_data(self):
871
 
        sha1sum = osutils.sha_string('foo\nbar\n')
 
467
        sha1sum = sha.new('foo\nbar\n').hexdigest()
872
468
        gz_txt = self.create_gz_content('version rev-id-1 2 %s\n'
873
469
                                        'foo\n'
874
470
                                        'bar\n'
891
487
 
892
488
    def get_knit_index(self, transport, name, mode):
893
489
        mapper = ConstantMapper(name)
 
490
        orig = knit._load_data
 
491
        def reset():
 
492
            knit._load_data = orig
 
493
        self.addCleanup(reset)
894
494
        from bzrlib._knit_load_data_py import _load_data_py
895
 
        self.overrideAttr(knit, '_load_data', _load_data_py)
 
495
        knit._load_data = _load_data_py
896
496
        allow_writes = lambda: 'w' in mode
897
497
        return _KndxIndex(transport, mapper, lambda:None, allow_writes, lambda:True)
898
498
 
1114
714
            call[1][1].getvalue())
1115
715
        self.assertEqual({'create_parent_dir': True}, call[2])
1116
716
 
1117
 
    def assertTotalBuildSize(self, size, keys, positions):
1118
 
        self.assertEqual(size,
1119
 
                         knit._get_total_build_size(None, keys, positions))
1120
 
 
1121
 
    def test__get_total_build_size(self):
1122
 
        positions = {
1123
 
            ('a',): (('fulltext', False), (('a',), 0, 100), None),
1124
 
            ('b',): (('line-delta', False), (('b',), 100, 21), ('a',)),
1125
 
            ('c',): (('line-delta', False), (('c',), 121, 35), ('b',)),
1126
 
            ('d',): (('line-delta', False), (('d',), 156, 12), ('b',)),
1127
 
            }
1128
 
        self.assertTotalBuildSize(100, [('a',)], positions)
1129
 
        self.assertTotalBuildSize(121, [('b',)], positions)
1130
 
        # c needs both a & b
1131
 
        self.assertTotalBuildSize(156, [('c',)], positions)
1132
 
        # we shouldn't count 'b' twice
1133
 
        self.assertTotalBuildSize(156, [('b',), ('c',)], positions)
1134
 
        self.assertTotalBuildSize(133, [('d',)], positions)
1135
 
        self.assertTotalBuildSize(168, [('c',), ('d',)], positions)
1136
 
 
1137
717
    def test_get_position(self):
1138
718
        transport = MockTransport([
1139
719
            _KndxIndex.HEADER,
1196
776
            self.assertRaises(errors.KnitCorrupt, index.keys)
1197
777
        except TypeError, e:
1198
778
            if (str(e) == ('exceptions must be strings, classes, or instances,'
1199
 
                           ' not exceptions.IndexError')):
 
779
                           ' not exceptions.IndexError')
 
780
                and sys.version_info[0:2] >= (2,5)):
1200
781
                self.knownFailure('Pyrex <0.9.5 fails with TypeError when'
1201
782
                                  ' raising new style exceptions with python'
1202
783
                                  ' >=2.5')
1215
796
            self.assertRaises(errors.KnitCorrupt, index.keys)
1216
797
        except TypeError, e:
1217
798
            if (str(e) == ('exceptions must be strings, classes, or instances,'
1218
 
                           ' not exceptions.ValueError')):
 
799
                           ' not exceptions.ValueError')
 
800
                and sys.version_info[0:2] >= (2,5)):
1219
801
                self.knownFailure('Pyrex <0.9.5 fails with TypeError when'
1220
802
                                  ' raising new style exceptions with python'
1221
803
                                  ' >=2.5')
1234
816
            self.assertRaises(errors.KnitCorrupt, index.keys)
1235
817
        except TypeError, e:
1236
818
            if (str(e) == ('exceptions must be strings, classes, or instances,'
1237
 
                           ' not exceptions.ValueError')):
 
819
                           ' not exceptions.ValueError')
 
820
                and sys.version_info[0:2] >= (2,5)):
1238
821
                self.knownFailure('Pyrex <0.9.5 fails with TypeError when'
1239
822
                                  ' raising new style exceptions with python'
1240
823
                                  ' >=2.5')
1251
834
            self.assertRaises(errors.KnitCorrupt, index.keys)
1252
835
        except TypeError, e:
1253
836
            if (str(e) == ('exceptions must be strings, classes, or instances,'
1254
 
                           ' not exceptions.ValueError')):
 
837
                           ' not exceptions.ValueError')
 
838
                and sys.version_info[0:2] >= (2,5)):
1255
839
                self.knownFailure('Pyrex <0.9.5 fails with TypeError when'
1256
840
                                  ' raising new style exceptions with python'
1257
841
                                  ' >=2.5')
1268
852
            self.assertRaises(errors.KnitCorrupt, index.keys)
1269
853
        except TypeError, e:
1270
854
            if (str(e) == ('exceptions must be strings, classes, or instances,'
1271
 
                           ' not exceptions.ValueError')):
 
855
                           ' not exceptions.ValueError')
 
856
                and sys.version_info[0:2] >= (2,5)):
1272
857
                self.knownFailure('Pyrex <0.9.5 fails with TypeError when'
1273
858
                                  ' raising new style exceptions with python'
1274
859
                                  ' >=2.5')
1275
860
            else:
1276
861
                raise
1277
862
 
1278
 
    def test_scan_unvalidated_index_not_implemented(self):
1279
 
        transport = MockTransport()
1280
 
        index = self.get_knit_index(transport, 'filename', 'r')
1281
 
        self.assertRaises(
1282
 
            NotImplementedError, index.scan_unvalidated_index,
1283
 
            'dummy graph_index')
1284
 
        self.assertRaises(
1285
 
            NotImplementedError, index.get_missing_compression_parents)
1286
 
 
1287
863
    def test_short_line(self):
1288
864
        transport = MockTransport([
1289
865
            _KndxIndex.HEADER,
1318
894
 
1319
895
class LowLevelKnitIndexTests_c(LowLevelKnitIndexTests):
1320
896
 
1321
 
    _test_needs_features = [compiled_knit_feature]
 
897
    _test_needs_features = [CompiledKnitFeature]
1322
898
 
1323
899
    def get_knit_index(self, transport, name, mode):
1324
900
        mapper = ConstantMapper(name)
1325
 
        from bzrlib._knit_load_data_pyx import _load_data_c
1326
 
        self.overrideAttr(knit, '_load_data', _load_data_c)
 
901
        orig = knit._load_data
 
902
        def reset():
 
903
            knit._load_data = orig
 
904
        self.addCleanup(reset)
 
905
        from bzrlib._knit_load_data_c import _load_data_c
 
906
        knit._load_data = _load_data_c
1327
907
        allow_writes = lambda: mode == 'w'
1328
 
        return _KndxIndex(transport, mapper, lambda:None,
1329
 
                          allow_writes, lambda:True)
1330
 
 
1331
 
 
1332
 
class Test_KnitAnnotator(TestCaseWithMemoryTransport):
1333
 
 
1334
 
    def make_annotator(self):
1335
 
        factory = knit.make_pack_factory(True, True, 1)
1336
 
        vf = factory(self.get_transport())
1337
 
        return knit._KnitAnnotator(vf)
1338
 
 
1339
 
    def test__expand_fulltext(self):
1340
 
        ann = self.make_annotator()
1341
 
        rev_key = ('rev-id',)
1342
 
        ann._num_compression_children[rev_key] = 1
1343
 
        res = ann._expand_record(rev_key, (('parent-id',),), None,
1344
 
                           ['line1\n', 'line2\n'], ('fulltext', True))
1345
 
        # The content object and text lines should be cached appropriately
1346
 
        self.assertEqual(['line1\n', 'line2'], res)
1347
 
        content_obj = ann._content_objects[rev_key]
1348
 
        self.assertEqual(['line1\n', 'line2\n'], content_obj._lines)
1349
 
        self.assertEqual(res, content_obj.text())
1350
 
        self.assertEqual(res, ann._text_cache[rev_key])
1351
 
 
1352
 
    def test__expand_delta_comp_parent_not_available(self):
1353
 
        # Parent isn't available yet, so we return nothing, but queue up this
1354
 
        # node for later processing
1355
 
        ann = self.make_annotator()
1356
 
        rev_key = ('rev-id',)
1357
 
        parent_key = ('parent-id',)
1358
 
        record = ['0,1,1\n', 'new-line\n']
1359
 
        details = ('line-delta', False)
1360
 
        res = ann._expand_record(rev_key, (parent_key,), parent_key,
1361
 
                                 record, details)
1362
 
        self.assertEqual(None, res)
1363
 
        self.assertTrue(parent_key in ann._pending_deltas)
1364
 
        pending = ann._pending_deltas[parent_key]
1365
 
        self.assertEqual(1, len(pending))
1366
 
        self.assertEqual((rev_key, (parent_key,), record, details), pending[0])
1367
 
 
1368
 
    def test__expand_record_tracks_num_children(self):
1369
 
        ann = self.make_annotator()
1370
 
        rev_key = ('rev-id',)
1371
 
        rev2_key = ('rev2-id',)
1372
 
        parent_key = ('parent-id',)
1373
 
        record = ['0,1,1\n', 'new-line\n']
1374
 
        details = ('line-delta', False)
1375
 
        ann._num_compression_children[parent_key] = 2
1376
 
        ann._expand_record(parent_key, (), None, ['line1\n', 'line2\n'],
1377
 
                           ('fulltext', False))
1378
 
        res = ann._expand_record(rev_key, (parent_key,), parent_key,
1379
 
                                 record, details)
1380
 
        self.assertEqual({parent_key: 1}, ann._num_compression_children)
1381
 
        # Expanding the second child should remove the content object, and the
1382
 
        # num_compression_children entry
1383
 
        res = ann._expand_record(rev2_key, (parent_key,), parent_key,
1384
 
                                 record, details)
1385
 
        self.assertFalse(parent_key in ann._content_objects)
1386
 
        self.assertEqual({}, ann._num_compression_children)
1387
 
        # We should not cache the content_objects for rev2 and rev, because
1388
 
        # they do not have compression children of their own.
1389
 
        self.assertEqual({}, ann._content_objects)
1390
 
 
1391
 
    def test__expand_delta_records_blocks(self):
1392
 
        ann = self.make_annotator()
1393
 
        rev_key = ('rev-id',)
1394
 
        parent_key = ('parent-id',)
1395
 
        record = ['0,1,1\n', 'new-line\n']
1396
 
        details = ('line-delta', True)
1397
 
        ann._num_compression_children[parent_key] = 2
1398
 
        ann._expand_record(parent_key, (), None,
1399
 
                           ['line1\n', 'line2\n', 'line3\n'],
1400
 
                           ('fulltext', False))
1401
 
        ann._expand_record(rev_key, (parent_key,), parent_key, record, details)
1402
 
        self.assertEqual({(rev_key, parent_key): [(1, 1, 1), (3, 3, 0)]},
1403
 
                         ann._matching_blocks)
1404
 
        rev2_key = ('rev2-id',)
1405
 
        record = ['0,1,1\n', 'new-line\n']
1406
 
        details = ('line-delta', False)
1407
 
        ann._expand_record(rev2_key, (parent_key,), parent_key, record, details)
1408
 
        self.assertEqual([(1, 1, 2), (3, 3, 0)],
1409
 
                         ann._matching_blocks[(rev2_key, parent_key)])
1410
 
 
1411
 
    def test__get_parent_ann_uses_matching_blocks(self):
1412
 
        ann = self.make_annotator()
1413
 
        rev_key = ('rev-id',)
1414
 
        parent_key = ('parent-id',)
1415
 
        parent_ann = [(parent_key,)]*3
1416
 
        block_key = (rev_key, parent_key)
1417
 
        ann._annotations_cache[parent_key] = parent_ann
1418
 
        ann._matching_blocks[block_key] = [(0, 1, 1), (3, 3, 0)]
1419
 
        # We should not try to access any parent_lines content, because we know
1420
 
        # we already have the matching blocks
1421
 
        par_ann, blocks = ann._get_parent_annotations_and_matches(rev_key,
1422
 
                                        ['1\n', '2\n', '3\n'], parent_key)
1423
 
        self.assertEqual(parent_ann, par_ann)
1424
 
        self.assertEqual([(0, 1, 1), (3, 3, 0)], blocks)
1425
 
        self.assertEqual({}, ann._matching_blocks)
1426
 
 
1427
 
    def test__process_pending(self):
1428
 
        ann = self.make_annotator()
1429
 
        rev_key = ('rev-id',)
1430
 
        p1_key = ('p1-id',)
1431
 
        p2_key = ('p2-id',)
1432
 
        record = ['0,1,1\n', 'new-line\n']
1433
 
        details = ('line-delta', False)
1434
 
        p1_record = ['line1\n', 'line2\n']
1435
 
        ann._num_compression_children[p1_key] = 1
1436
 
        res = ann._expand_record(rev_key, (p1_key,p2_key), p1_key,
1437
 
                                 record, details)
1438
 
        self.assertEqual(None, res)
1439
 
        # self.assertTrue(p1_key in ann._pending_deltas)
1440
 
        self.assertEqual({}, ann._pending_annotation)
1441
 
        # Now insert p1, and we should be able to expand the delta
1442
 
        res = ann._expand_record(p1_key, (), None, p1_record,
1443
 
                                 ('fulltext', False))
1444
 
        self.assertEqual(p1_record, res)
1445
 
        ann._annotations_cache[p1_key] = [(p1_key,)]*2
1446
 
        res = ann._process_pending(p1_key)
1447
 
        self.assertEqual([], res)
1448
 
        self.assertFalse(p1_key in ann._pending_deltas)
1449
 
        self.assertTrue(p2_key in ann._pending_annotation)
1450
 
        self.assertEqual({p2_key: [(rev_key, (p1_key, p2_key))]},
1451
 
                         ann._pending_annotation)
1452
 
        # Now fill in parent 2, and pending annotation should be satisfied
1453
 
        res = ann._expand_record(p2_key, (), None, [], ('fulltext', False))
1454
 
        ann._annotations_cache[p2_key] = []
1455
 
        res = ann._process_pending(p2_key)
1456
 
        self.assertEqual([rev_key], res)
1457
 
        self.assertEqual({}, ann._pending_annotation)
1458
 
        self.assertEqual({}, ann._pending_deltas)
1459
 
 
1460
 
    def test_record_delta_removes_basis(self):
1461
 
        ann = self.make_annotator()
1462
 
        ann._expand_record(('parent-id',), (), None,
1463
 
                           ['line1\n', 'line2\n'], ('fulltext', False))
1464
 
        ann._num_compression_children['parent-id'] = 2
1465
 
 
1466
 
    def test_annotate_special_text(self):
1467
 
        ann = self.make_annotator()
1468
 
        vf = ann._vf
1469
 
        rev1_key = ('rev-1',)
1470
 
        rev2_key = ('rev-2',)
1471
 
        rev3_key = ('rev-3',)
1472
 
        spec_key = ('special:',)
1473
 
        vf.add_lines(rev1_key, [], ['initial content\n'])
1474
 
        vf.add_lines(rev2_key, [rev1_key], ['initial content\n',
1475
 
                                            'common content\n',
1476
 
                                            'content in 2\n'])
1477
 
        vf.add_lines(rev3_key, [rev1_key], ['initial content\n',
1478
 
                                            'common content\n',
1479
 
                                            'content in 3\n'])
1480
 
        spec_text = ('initial content\n'
1481
 
                     'common content\n'
1482
 
                     'content in 2\n'
1483
 
                     'content in 3\n')
1484
 
        ann.add_special_text(spec_key, [rev2_key, rev3_key], spec_text)
1485
 
        anns, lines = ann.annotate(spec_key)
1486
 
        self.assertEqual([(rev1_key,),
1487
 
                          (rev2_key, rev3_key),
1488
 
                          (rev2_key,),
1489
 
                          (rev3_key,),
1490
 
                         ], anns)
1491
 
        self.assertEqualDiff(spec_text, ''.join(lines))
 
908
        return _KndxIndex(transport, mapper, lambda:None, allow_writes, lambda:True)
1492
909
 
1493
910
 
1494
911
class KnitTests(TestCaseWithTransport):
1499
916
        return make_file_factory(annotate, mapper)(self.get_transport())
1500
917
 
1501
918
 
1502
 
class TestBadShaError(KnitTests):
1503
 
    """Tests for handling of sha errors."""
1504
 
 
1505
 
    def test_sha_exception_has_text(self):
1506
 
        # having the failed text included in the error allows for recovery.
1507
 
        source = self.make_test_knit()
1508
 
        target = self.make_test_knit(name="target")
1509
 
        if not source._max_delta_chain:
1510
 
            raise TestNotApplicable(
1511
 
                "cannot get delta-caused sha failures without deltas.")
1512
 
        # create a basis
1513
 
        basis = ('basis',)
1514
 
        broken = ('broken',)
1515
 
        source.add_lines(basis, (), ['foo\n'])
1516
 
        source.add_lines(broken, (basis,), ['foo\n', 'bar\n'])
1517
 
        # Seed target with a bad basis text
1518
 
        target.add_lines(basis, (), ['gam\n'])
1519
 
        target.insert_record_stream(
1520
 
            source.get_record_stream([broken], 'unordered', False))
1521
 
        err = self.assertRaises(errors.KnitCorrupt,
1522
 
            target.get_record_stream([broken], 'unordered', True
1523
 
            ).next().get_bytes_as, 'chunked')
1524
 
        self.assertEqual(['gam\n', 'bar\n'], err.content)
1525
 
        # Test for formatting with live data
1526
 
        self.assertStartsWith(str(err), "Knit ")
1527
 
 
1528
 
 
1529
919
class TestKnitIndex(KnitTests):
1530
920
 
1531
921
    def test_add_versions_dictionary_compresses(self):
1603
993
        # could leave an empty .kndx file, which bzr would later claim was a
1604
994
        # corrupted file since the header was not present. In reality, the file
1605
995
        # just wasn't created, so it should be ignored.
1606
 
        t = transport.get_transport_from_path('.')
 
996
        t = get_transport('.')
1607
997
        t.put_bytes('test.kndx', '')
1608
998
 
1609
999
        knit = self.make_test_knit()
1610
1000
 
1611
1001
    def test_knit_index_checks_header(self):
1612
 
        t = transport.get_transport_from_path('.')
 
1002
        t = get_transport('.')
1613
1003
        t.put_bytes('test.kndx', '# not really a knit header\n\n')
1614
1004
        k = self.make_test_knit()
1615
1005
        self.assertRaises(KnitHeaderError, k.keys)
1731
1121
            [('parent',)])])
1732
1122
        # but neither should have added data:
1733
1123
        self.assertEqual([[], [], [], []], self.caught_entries)
1734
 
 
 
1124
        
1735
1125
    def test_add_version_different_dup(self):
1736
1126
        index = self.two_graph_index(deltas=True, catch_adds=True)
1737
1127
        # change options
1738
1128
        self.assertRaises(errors.KnitCorrupt, index.add_records,
1739
 
            [(('tip',), 'line-delta', (None, 0, 100), [('parent',)])])
 
1129
            [(('tip',), 'no-eol,line-delta', (None, 0, 100), [('parent',)])])
 
1130
        self.assertRaises(errors.KnitCorrupt, index.add_records,
 
1131
            [(('tip',), 'line-delta,no-eol', (None, 0, 100), [('parent',)])])
1740
1132
        self.assertRaises(errors.KnitCorrupt, index.add_records,
1741
1133
            [(('tip',), 'fulltext', (None, 0, 100), [('parent',)])])
1742
1134
        # parents
1743
1135
        self.assertRaises(errors.KnitCorrupt, index.add_records,
1744
1136
            [(('tip',), 'fulltext,no-eol', (None, 0, 100), [])])
1745
1137
        self.assertEqual([], self.caught_entries)
1746
 
 
 
1138
        
1747
1139
    def test_add_versions_nodeltas(self):
1748
1140
        index = self.two_graph_index(catch_adds=True)
1749
1141
        index.add_records([
1791
1183
            [('parent',)])])
1792
1184
        # but neither should have added data.
1793
1185
        self.assertEqual([[], [], [], []], self.caught_entries)
1794
 
 
 
1186
        
1795
1187
    def test_add_versions_different_dup(self):
1796
1188
        index = self.two_graph_index(deltas=True, catch_adds=True)
1797
1189
        # change options
1798
1190
        self.assertRaises(errors.KnitCorrupt, index.add_records,
1799
 
            [(('tip',), 'line-delta', (None, 0, 100), [('parent',)])])
 
1191
            [(('tip',), 'no-eol,line-delta', (None, 0, 100), [('parent',)])])
 
1192
        self.assertRaises(errors.KnitCorrupt, index.add_records,
 
1193
            [(('tip',), 'line-delta,no-eol', (None, 0, 100), [('parent',)])])
1800
1194
        self.assertRaises(errors.KnitCorrupt, index.add_records,
1801
1195
            [(('tip',), 'fulltext', (None, 0, 100), [('parent',)])])
1802
1196
        # parents
1805
1199
        # change options in the second record
1806
1200
        self.assertRaises(errors.KnitCorrupt, index.add_records,
1807
1201
            [(('tip',), 'fulltext,no-eol', (None, 0, 100), [('parent',)]),
1808
 
             (('tip',), 'line-delta', (None, 0, 100), [('parent',)])])
 
1202
             (('tip',), 'no-eol,line-delta', (None, 0, 100), [('parent',)])])
1809
1203
        self.assertEqual([], self.caught_entries)
1810
1204
 
1811
 
    def make_g_index_missing_compression_parent(self):
1812
 
        graph_index = self.make_g_index('missing_comp', 2,
1813
 
            [(('tip', ), ' 100 78',
1814
 
              ([('missing-parent', ), ('ghost', )], [('missing-parent', )]))])
1815
 
        return graph_index
1816
 
 
1817
 
    def make_g_index_missing_parent(self):
1818
 
        graph_index = self.make_g_index('missing_parent', 2,
1819
 
            [(('parent', ), ' 100 78', ([], [])),
1820
 
             (('tip', ), ' 100 78',
1821
 
              ([('parent', ), ('missing-parent', )], [('parent', )])),
1822
 
              ])
1823
 
        return graph_index
1824
 
 
1825
 
    def make_g_index_no_external_refs(self):
1826
 
        graph_index = self.make_g_index('no_external_refs', 2,
1827
 
            [(('rev', ), ' 100 78',
1828
 
              ([('parent', ), ('ghost', )], []))])
1829
 
        return graph_index
1830
 
 
1831
 
    def test_add_good_unvalidated_index(self):
1832
 
        unvalidated = self.make_g_index_no_external_refs()
1833
 
        combined = CombinedGraphIndex([unvalidated])
1834
 
        index = _KnitGraphIndex(combined, lambda: True, deltas=True)
1835
 
        index.scan_unvalidated_index(unvalidated)
1836
 
        self.assertEqual(frozenset(), index.get_missing_compression_parents())
1837
 
 
1838
 
    def test_add_missing_compression_parent_unvalidated_index(self):
1839
 
        unvalidated = self.make_g_index_missing_compression_parent()
1840
 
        combined = CombinedGraphIndex([unvalidated])
1841
 
        index = _KnitGraphIndex(combined, lambda: True, deltas=True)
1842
 
        index.scan_unvalidated_index(unvalidated)
1843
 
        # This also checks that its only the compression parent that is
1844
 
        # examined, otherwise 'ghost' would also be reported as a missing
1845
 
        # parent.
1846
 
        self.assertEqual(
1847
 
            frozenset([('missing-parent',)]),
1848
 
            index.get_missing_compression_parents())
1849
 
 
1850
 
    def test_add_missing_noncompression_parent_unvalidated_index(self):
1851
 
        unvalidated = self.make_g_index_missing_parent()
1852
 
        combined = CombinedGraphIndex([unvalidated])
1853
 
        index = _KnitGraphIndex(combined, lambda: True, deltas=True,
1854
 
            track_external_parent_refs=True)
1855
 
        index.scan_unvalidated_index(unvalidated)
1856
 
        self.assertEqual(
1857
 
            frozenset([('missing-parent',)]), index.get_missing_parents())
1858
 
 
1859
 
    def test_track_external_parent_refs(self):
1860
 
        g_index = self.make_g_index('empty', 2, [])
1861
 
        combined = CombinedGraphIndex([g_index])
1862
 
        index = _KnitGraphIndex(combined, lambda: True, deltas=True,
1863
 
            add_callback=self.catch_add, track_external_parent_refs=True)
1864
 
        self.caught_entries = []
1865
 
        index.add_records([
1866
 
            (('new-key',), 'fulltext,no-eol', (None, 50, 60),
1867
 
             [('parent-1',), ('parent-2',)])])
1868
 
        self.assertEqual(
1869
 
            frozenset([('parent-1',), ('parent-2',)]),
1870
 
            index.get_missing_parents())
1871
 
 
1872
 
    def test_add_unvalidated_index_with_present_external_references(self):
1873
 
        index = self.two_graph_index(deltas=True)
1874
 
        # Ugly hack to get at one of the underlying GraphIndex objects that
1875
 
        # two_graph_index built.
1876
 
        unvalidated = index._graph_index._indices[1]
1877
 
        # 'parent' is an external ref of _indices[1] (unvalidated), but is
1878
 
        # present in _indices[0].
1879
 
        index.scan_unvalidated_index(unvalidated)
1880
 
        self.assertEqual(frozenset(), index.get_missing_compression_parents())
1881
 
 
1882
 
    def make_new_missing_parent_g_index(self, name):
1883
 
        missing_parent = name + '-missing-parent'
1884
 
        graph_index = self.make_g_index(name, 2,
1885
 
            [((name + 'tip', ), ' 100 78',
1886
 
              ([(missing_parent, ), ('ghost', )], [(missing_parent, )]))])
1887
 
        return graph_index
1888
 
 
1889
 
    def test_add_mulitiple_unvalidated_indices_with_missing_parents(self):
1890
 
        g_index_1 = self.make_new_missing_parent_g_index('one')
1891
 
        g_index_2 = self.make_new_missing_parent_g_index('two')
1892
 
        combined = CombinedGraphIndex([g_index_1, g_index_2])
1893
 
        index = _KnitGraphIndex(combined, lambda: True, deltas=True)
1894
 
        index.scan_unvalidated_index(g_index_1)
1895
 
        index.scan_unvalidated_index(g_index_2)
1896
 
        self.assertEqual(
1897
 
            frozenset([('one-missing-parent',), ('two-missing-parent',)]),
1898
 
            index.get_missing_compression_parents())
1899
 
 
1900
 
    def test_add_mulitiple_unvalidated_indices_with_mutual_dependencies(self):
1901
 
        graph_index_a = self.make_g_index('one', 2,
1902
 
            [(('parent-one', ), ' 100 78', ([('non-compression-parent',)], [])),
1903
 
             (('child-of-two', ), ' 100 78',
1904
 
              ([('parent-two',)], [('parent-two',)]))])
1905
 
        graph_index_b = self.make_g_index('two', 2,
1906
 
            [(('parent-two', ), ' 100 78', ([('non-compression-parent',)], [])),
1907
 
             (('child-of-one', ), ' 100 78',
1908
 
              ([('parent-one',)], [('parent-one',)]))])
1909
 
        combined = CombinedGraphIndex([graph_index_a, graph_index_b])
1910
 
        index = _KnitGraphIndex(combined, lambda: True, deltas=True)
1911
 
        index.scan_unvalidated_index(graph_index_a)
1912
 
        index.scan_unvalidated_index(graph_index_b)
1913
 
        self.assertEqual(
1914
 
            frozenset([]), index.get_missing_compression_parents())
1915
 
 
1916
1205
 
1917
1206
class TestNoParentsGraphIndexKnit(KnitTests):
1918
1207
    """Tests for knits using _KnitGraphIndex with no parents."""
1926
1215
        size = trans.put_file(name, stream)
1927
1216
        return GraphIndex(trans, name, size)
1928
1217
 
1929
 
    def test_add_good_unvalidated_index(self):
1930
 
        unvalidated = self.make_g_index('unvalidated')
1931
 
        combined = CombinedGraphIndex([unvalidated])
1932
 
        index = _KnitGraphIndex(combined, lambda: True, parents=False)
1933
 
        index.scan_unvalidated_index(unvalidated)
1934
 
        self.assertEqual(frozenset(),
1935
 
            index.get_missing_compression_parents())
1936
 
 
1937
1218
    def test_parents_deltas_incompatible(self):
1938
1219
        index = CombinedGraphIndex([])
1939
1220
        self.assertRaises(errors.KnitError, _KnitGraphIndex, lambda:True,
2020
1301
        index.add_records([(('tip',), 'fulltext,no-eol', (None, 0, 1000), [])])
2021
1302
        # but neither should have added data.
2022
1303
        self.assertEqual([[], [], [], []], self.caught_entries)
2023
 
 
 
1304
        
2024
1305
    def test_add_version_different_dup(self):
2025
1306
        index = self.two_graph_index(catch_adds=True)
2026
1307
        # change options
2034
1315
        self.assertRaises(errors.KnitCorrupt, index.add_records,
2035
1316
            [(('tip',), 'fulltext,no-eol', (None, 0, 100), [('parent',)])])
2036
1317
        self.assertEqual([], self.caught_entries)
2037
 
 
 
1318
        
2038
1319
    def test_add_versions(self):
2039
1320
        index = self.two_graph_index(catch_adds=True)
2040
1321
        index.add_records([
2072
1353
        index.add_records([(('tip',), 'fulltext,no-eol', (None, 0, 1000), [])])
2073
1354
        # but neither should have added data.
2074
1355
        self.assertEqual([[], [], [], []], self.caught_entries)
2075
 
 
 
1356
        
2076
1357
    def test_add_versions_different_dup(self):
2077
1358
        index = self.two_graph_index(catch_adds=True)
2078
1359
        # change options
2090
1371
            [(('tip',), 'fulltext,no-eol', (None, 0, 100), []),
2091
1372
             (('tip',), 'no-eol,line-delta', (None, 0, 100), [])])
2092
1373
        self.assertEqual([], self.caught_entries)
2093
 
 
2094
 
 
2095
 
class TestKnitVersionedFiles(KnitTests):
2096
 
 
2097
 
    def assertGroupKeysForIo(self, exp_groups, keys, non_local_keys,
2098
 
                             positions, _min_buffer_size=None):
2099
 
        kvf = self.make_test_knit()
2100
 
        if _min_buffer_size is None:
2101
 
            _min_buffer_size = knit._STREAM_MIN_BUFFER_SIZE
2102
 
        self.assertEqual(exp_groups, kvf._group_keys_for_io(keys,
2103
 
                                        non_local_keys, positions,
2104
 
                                        _min_buffer_size=_min_buffer_size))
2105
 
 
2106
 
    def assertSplitByPrefix(self, expected_map, expected_prefix_order,
2107
 
                            keys):
2108
 
        split, prefix_order = KnitVersionedFiles._split_by_prefix(keys)
2109
 
        self.assertEqual(expected_map, split)
2110
 
        self.assertEqual(expected_prefix_order, prefix_order)
2111
 
 
2112
 
    def test__group_keys_for_io(self):
2113
 
        ft_detail = ('fulltext', False)
2114
 
        ld_detail = ('line-delta', False)
2115
 
        f_a = ('f', 'a')
2116
 
        f_b = ('f', 'b')
2117
 
        f_c = ('f', 'c')
2118
 
        g_a = ('g', 'a')
2119
 
        g_b = ('g', 'b')
2120
 
        g_c = ('g', 'c')
2121
 
        positions = {
2122
 
            f_a: (ft_detail, (f_a, 0, 100), None),
2123
 
            f_b: (ld_detail, (f_b, 100, 21), f_a),
2124
 
            f_c: (ld_detail, (f_c, 180, 15), f_b),
2125
 
            g_a: (ft_detail, (g_a, 121, 35), None),
2126
 
            g_b: (ld_detail, (g_b, 156, 12), g_a),
2127
 
            g_c: (ld_detail, (g_c, 195, 13), g_a),
2128
 
            }
2129
 
        self.assertGroupKeysForIo([([f_a], set())],
2130
 
                                  [f_a], [], positions)
2131
 
        self.assertGroupKeysForIo([([f_a], set([f_a]))],
2132
 
                                  [f_a], [f_a], positions)
2133
 
        self.assertGroupKeysForIo([([f_a, f_b], set([]))],
2134
 
                                  [f_a, f_b], [], positions)
2135
 
        self.assertGroupKeysForIo([([f_a, f_b], set([f_b]))],
2136
 
                                  [f_a, f_b], [f_b], positions)
2137
 
        self.assertGroupKeysForIo([([f_a, f_b, g_a, g_b], set())],
2138
 
                                  [f_a, g_a, f_b, g_b], [], positions)
2139
 
        self.assertGroupKeysForIo([([f_a, f_b, g_a, g_b], set())],
2140
 
                                  [f_a, g_a, f_b, g_b], [], positions,
2141
 
                                  _min_buffer_size=150)
2142
 
        self.assertGroupKeysForIo([([f_a, f_b], set()), ([g_a, g_b], set())],
2143
 
                                  [f_a, g_a, f_b, g_b], [], positions,
2144
 
                                  _min_buffer_size=100)
2145
 
        self.assertGroupKeysForIo([([f_c], set()), ([g_b], set())],
2146
 
                                  [f_c, g_b], [], positions,
2147
 
                                  _min_buffer_size=125)
2148
 
        self.assertGroupKeysForIo([([g_b, f_c], set())],
2149
 
                                  [g_b, f_c], [], positions,
2150
 
                                  _min_buffer_size=125)
2151
 
 
2152
 
    def test__split_by_prefix(self):
2153
 
        self.assertSplitByPrefix({'f': [('f', 'a'), ('f', 'b')],
2154
 
                                  'g': [('g', 'b'), ('g', 'a')],
2155
 
                                 }, ['f', 'g'],
2156
 
                                 [('f', 'a'), ('g', 'b'),
2157
 
                                  ('g', 'a'), ('f', 'b')])
2158
 
 
2159
 
        self.assertSplitByPrefix({'f': [('f', 'a'), ('f', 'b')],
2160
 
                                  'g': [('g', 'b'), ('g', 'a')],
2161
 
                                 }, ['f', 'g'],
2162
 
                                 [('f', 'a'), ('f', 'b'),
2163
 
                                  ('g', 'b'), ('g', 'a')])
2164
 
 
2165
 
        self.assertSplitByPrefix({'f': [('f', 'a'), ('f', 'b')],
2166
 
                                  'g': [('g', 'b'), ('g', 'a')],
2167
 
                                 }, ['f', 'g'],
2168
 
                                 [('f', 'a'), ('f', 'b'),
2169
 
                                  ('g', 'b'), ('g', 'a')])
2170
 
 
2171
 
        self.assertSplitByPrefix({'f': [('f', 'a'), ('f', 'b')],
2172
 
                                  'g': [('g', 'b'), ('g', 'a')],
2173
 
                                  '': [('a',), ('b',)]
2174
 
                                 }, ['f', 'g', ''],
2175
 
                                 [('f', 'a'), ('g', 'b'),
2176
 
                                  ('a',), ('b',),
2177
 
                                  ('g', 'a'), ('f', 'b')])
2178
 
 
2179
 
 
2180
 
class TestStacking(KnitTests):
2181
 
 
2182
 
    def get_basis_and_test_knit(self):
2183
 
        basis = self.make_test_knit(name='basis')
2184
 
        basis = RecordingVersionedFilesDecorator(basis)
2185
 
        test = self.make_test_knit(name='test')
2186
 
        test.add_fallback_versioned_files(basis)
2187
 
        return basis, test
2188
 
 
2189
 
    def test_add_fallback_versioned_files(self):
2190
 
        basis = self.make_test_knit(name='basis')
2191
 
        test = self.make_test_knit(name='test')
2192
 
        # It must not error; other tests test that the fallback is referred to
2193
 
        # when accessing data.
2194
 
        test.add_fallback_versioned_files(basis)
2195
 
 
2196
 
    def test_add_lines(self):
2197
 
        # lines added to the test are not added to the basis
2198
 
        basis, test = self.get_basis_and_test_knit()
2199
 
        key = ('foo',)
2200
 
        key_basis = ('bar',)
2201
 
        key_cross_border = ('quux',)
2202
 
        key_delta = ('zaphod',)
2203
 
        test.add_lines(key, (), ['foo\n'])
2204
 
        self.assertEqual({}, basis.get_parent_map([key]))
2205
 
        # lines added to the test that reference across the stack do a
2206
 
        # fulltext.
2207
 
        basis.add_lines(key_basis, (), ['foo\n'])
2208
 
        basis.calls = []
2209
 
        test.add_lines(key_cross_border, (key_basis,), ['foo\n'])
2210
 
        self.assertEqual('fulltext', test._index.get_method(key_cross_border))
2211
 
        # we don't even need to look at the basis to see that this should be
2212
 
        # stored as a fulltext
2213
 
        self.assertEqual([], basis.calls)
2214
 
        # Subsequent adds do delta.
2215
 
        basis.calls = []
2216
 
        test.add_lines(key_delta, (key_cross_border,), ['foo\n'])
2217
 
        self.assertEqual('line-delta', test._index.get_method(key_delta))
2218
 
        self.assertEqual([], basis.calls)
2219
 
 
2220
 
    def test_annotate(self):
2221
 
        # annotations from the test knit are answered without asking the basis
2222
 
        basis, test = self.get_basis_and_test_knit()
2223
 
        key = ('foo',)
2224
 
        key_basis = ('bar',)
2225
 
        key_missing = ('missing',)
2226
 
        test.add_lines(key, (), ['foo\n'])
2227
 
        details = test.annotate(key)
2228
 
        self.assertEqual([(key, 'foo\n')], details)
2229
 
        self.assertEqual([], basis.calls)
2230
 
        # But texts that are not in the test knit are looked for in the basis
2231
 
        # directly.
2232
 
        basis.add_lines(key_basis, (), ['foo\n', 'bar\n'])
2233
 
        basis.calls = []
2234
 
        details = test.annotate(key_basis)
2235
 
        self.assertEqual([(key_basis, 'foo\n'), (key_basis, 'bar\n')], details)
2236
 
        # Not optimised to date:
2237
 
        # self.assertEqual([("annotate", key_basis)], basis.calls)
2238
 
        self.assertEqual([('get_parent_map', set([key_basis])),
2239
 
            ('get_parent_map', set([key_basis])),
2240
 
            ('get_record_stream', [key_basis], 'topological', True)],
2241
 
            basis.calls)
2242
 
 
2243
 
    def test_check(self):
2244
 
        # At the moment checking a stacked knit does implicitly check the
2245
 
        # fallback files.
2246
 
        basis, test = self.get_basis_and_test_knit()
2247
 
        test.check()
2248
 
 
2249
 
    def test_get_parent_map(self):
2250
 
        # parents in the test knit are answered without asking the basis
2251
 
        basis, test = self.get_basis_and_test_knit()
2252
 
        key = ('foo',)
2253
 
        key_basis = ('bar',)
2254
 
        key_missing = ('missing',)
2255
 
        test.add_lines(key, (), [])
2256
 
        parent_map = test.get_parent_map([key])
2257
 
        self.assertEqual({key: ()}, parent_map)
2258
 
        self.assertEqual([], basis.calls)
2259
 
        # But parents that are not in the test knit are looked for in the basis
2260
 
        basis.add_lines(key_basis, (), [])
2261
 
        basis.calls = []
2262
 
        parent_map = test.get_parent_map([key, key_basis, key_missing])
2263
 
        self.assertEqual({key: (),
2264
 
            key_basis: ()}, parent_map)
2265
 
        self.assertEqual([("get_parent_map", set([key_basis, key_missing]))],
2266
 
            basis.calls)
2267
 
 
2268
 
    def test_get_record_stream_unordered_fulltexts(self):
2269
 
        # records from the test knit are answered without asking the basis:
2270
 
        basis, test = self.get_basis_and_test_knit()
2271
 
        key = ('foo',)
2272
 
        key_basis = ('bar',)
2273
 
        key_missing = ('missing',)
2274
 
        test.add_lines(key, (), ['foo\n'])
2275
 
        records = list(test.get_record_stream([key], 'unordered', True))
2276
 
        self.assertEqual(1, len(records))
2277
 
        self.assertEqual([], basis.calls)
2278
 
        # Missing (from test knit) objects are retrieved from the basis:
2279
 
        basis.add_lines(key_basis, (), ['foo\n', 'bar\n'])
2280
 
        basis.calls = []
2281
 
        records = list(test.get_record_stream([key_basis, key_missing],
2282
 
            'unordered', True))
2283
 
        self.assertEqual(2, len(records))
2284
 
        calls = list(basis.calls)
2285
 
        for record in records:
2286
 
            self.assertSubset([record.key], (key_basis, key_missing))
2287
 
            if record.key == key_missing:
2288
 
                self.assertIsInstance(record, AbsentContentFactory)
2289
 
            else:
2290
 
                reference = list(basis.get_record_stream([key_basis],
2291
 
                    'unordered', True))[0]
2292
 
                self.assertEqual(reference.key, record.key)
2293
 
                self.assertEqual(reference.sha1, record.sha1)
2294
 
                self.assertEqual(reference.storage_kind, record.storage_kind)
2295
 
                self.assertEqual(reference.get_bytes_as(reference.storage_kind),
2296
 
                    record.get_bytes_as(record.storage_kind))
2297
 
                self.assertEqual(reference.get_bytes_as('fulltext'),
2298
 
                    record.get_bytes_as('fulltext'))
2299
 
        # It's not strictly minimal, but it seems reasonable for now for it to
2300
 
        # ask which fallbacks have which parents.
2301
 
        self.assertEqual([
2302
 
            ("get_parent_map", set([key_basis, key_missing])),
2303
 
            ("get_record_stream", [key_basis], 'unordered', True)],
2304
 
            calls)
2305
 
 
2306
 
    def test_get_record_stream_ordered_fulltexts(self):
2307
 
        # ordering is preserved down into the fallback store.
2308
 
        basis, test = self.get_basis_and_test_knit()
2309
 
        key = ('foo',)
2310
 
        key_basis = ('bar',)
2311
 
        key_basis_2 = ('quux',)
2312
 
        key_missing = ('missing',)
2313
 
        test.add_lines(key, (key_basis,), ['foo\n'])
2314
 
        # Missing (from test knit) objects are retrieved from the basis:
2315
 
        basis.add_lines(key_basis, (key_basis_2,), ['foo\n', 'bar\n'])
2316
 
        basis.add_lines(key_basis_2, (), ['quux\n'])
2317
 
        basis.calls = []
2318
 
        # ask for in non-topological order
2319
 
        records = list(test.get_record_stream(
2320
 
            [key, key_basis, key_missing, key_basis_2], 'topological', True))
2321
 
        self.assertEqual(4, len(records))
2322
 
        results = []
2323
 
        for record in records:
2324
 
            self.assertSubset([record.key],
2325
 
                (key_basis, key_missing, key_basis_2, key))
2326
 
            if record.key == key_missing:
2327
 
                self.assertIsInstance(record, AbsentContentFactory)
2328
 
            else:
2329
 
                results.append((record.key, record.sha1, record.storage_kind,
2330
 
                    record.get_bytes_as('fulltext')))
2331
 
        calls = list(basis.calls)
2332
 
        order = [record[0] for record in results]
2333
 
        self.assertEqual([key_basis_2, key_basis, key], order)
2334
 
        for result in results:
2335
 
            if result[0] == key:
2336
 
                source = test
2337
 
            else:
2338
 
                source = basis
2339
 
            record = source.get_record_stream([result[0]], 'unordered',
2340
 
                True).next()
2341
 
            self.assertEqual(record.key, result[0])
2342
 
            self.assertEqual(record.sha1, result[1])
2343
 
            # We used to check that the storage kind matched, but actually it
2344
 
            # depends on whether it was sourced from the basis, or in a single
2345
 
            # group, because asking for full texts returns proxy objects to a
2346
 
            # _ContentMapGenerator object; so checking the kind is unneeded.
2347
 
            self.assertEqual(record.get_bytes_as('fulltext'), result[3])
2348
 
        # It's not strictly minimal, but it seems reasonable for now for it to
2349
 
        # ask which fallbacks have which parents.
2350
 
        self.assertEqual([
2351
 
            ("get_parent_map", set([key_basis, key_basis_2, key_missing])),
2352
 
            # topological is requested from the fallback, because that is what
2353
 
            # was requested at the top level.
2354
 
            ("get_record_stream", [key_basis_2, key_basis], 'topological', True)],
2355
 
            calls)
2356
 
 
2357
 
    def test_get_record_stream_unordered_deltas(self):
2358
 
        # records from the test knit are answered without asking the basis:
2359
 
        basis, test = self.get_basis_and_test_knit()
2360
 
        key = ('foo',)
2361
 
        key_basis = ('bar',)
2362
 
        key_missing = ('missing',)
2363
 
        test.add_lines(key, (), ['foo\n'])
2364
 
        records = list(test.get_record_stream([key], 'unordered', False))
2365
 
        self.assertEqual(1, len(records))
2366
 
        self.assertEqual([], basis.calls)
2367
 
        # Missing (from test knit) objects are retrieved from the basis:
2368
 
        basis.add_lines(key_basis, (), ['foo\n', 'bar\n'])
2369
 
        basis.calls = []
2370
 
        records = list(test.get_record_stream([key_basis, key_missing],
2371
 
            'unordered', False))
2372
 
        self.assertEqual(2, len(records))
2373
 
        calls = list(basis.calls)
2374
 
        for record in records:
2375
 
            self.assertSubset([record.key], (key_basis, key_missing))
2376
 
            if record.key == key_missing:
2377
 
                self.assertIsInstance(record, AbsentContentFactory)
2378
 
            else:
2379
 
                reference = list(basis.get_record_stream([key_basis],
2380
 
                    'unordered', False))[0]
2381
 
                self.assertEqual(reference.key, record.key)
2382
 
                self.assertEqual(reference.sha1, record.sha1)
2383
 
                self.assertEqual(reference.storage_kind, record.storage_kind)
2384
 
                self.assertEqual(reference.get_bytes_as(reference.storage_kind),
2385
 
                    record.get_bytes_as(record.storage_kind))
2386
 
        # It's not strictly minimal, but it seems reasonable for now for it to
2387
 
        # ask which fallbacks have which parents.
2388
 
        self.assertEqual([
2389
 
            ("get_parent_map", set([key_basis, key_missing])),
2390
 
            ("get_record_stream", [key_basis], 'unordered', False)],
2391
 
            calls)
2392
 
 
2393
 
    def test_get_record_stream_ordered_deltas(self):
2394
 
        # ordering is preserved down into the fallback store.
2395
 
        basis, test = self.get_basis_and_test_knit()
2396
 
        key = ('foo',)
2397
 
        key_basis = ('bar',)
2398
 
        key_basis_2 = ('quux',)
2399
 
        key_missing = ('missing',)
2400
 
        test.add_lines(key, (key_basis,), ['foo\n'])
2401
 
        # Missing (from test knit) objects are retrieved from the basis:
2402
 
        basis.add_lines(key_basis, (key_basis_2,), ['foo\n', 'bar\n'])
2403
 
        basis.add_lines(key_basis_2, (), ['quux\n'])
2404
 
        basis.calls = []
2405
 
        # ask for in non-topological order
2406
 
        records = list(test.get_record_stream(
2407
 
            [key, key_basis, key_missing, key_basis_2], 'topological', False))
2408
 
        self.assertEqual(4, len(records))
2409
 
        results = []
2410
 
        for record in records:
2411
 
            self.assertSubset([record.key],
2412
 
                (key_basis, key_missing, key_basis_2, key))
2413
 
            if record.key == key_missing:
2414
 
                self.assertIsInstance(record, AbsentContentFactory)
2415
 
            else:
2416
 
                results.append((record.key, record.sha1, record.storage_kind,
2417
 
                    record.get_bytes_as(record.storage_kind)))
2418
 
        calls = list(basis.calls)
2419
 
        order = [record[0] for record in results]
2420
 
        self.assertEqual([key_basis_2, key_basis, key], order)
2421
 
        for result in results:
2422
 
            if result[0] == key:
2423
 
                source = test
2424
 
            else:
2425
 
                source = basis
2426
 
            record = source.get_record_stream([result[0]], 'unordered',
2427
 
                False).next()
2428
 
            self.assertEqual(record.key, result[0])
2429
 
            self.assertEqual(record.sha1, result[1])
2430
 
            self.assertEqual(record.storage_kind, result[2])
2431
 
            self.assertEqual(record.get_bytes_as(record.storage_kind), result[3])
2432
 
        # It's not strictly minimal, but it seems reasonable for now for it to
2433
 
        # ask which fallbacks have which parents.
2434
 
        self.assertEqual([
2435
 
            ("get_parent_map", set([key_basis, key_basis_2, key_missing])),
2436
 
            ("get_record_stream", [key_basis_2, key_basis], 'topological', False)],
2437
 
            calls)
2438
 
 
2439
 
    def test_get_sha1s(self):
2440
 
        # sha1's in the test knit are answered without asking the basis
2441
 
        basis, test = self.get_basis_and_test_knit()
2442
 
        key = ('foo',)
2443
 
        key_basis = ('bar',)
2444
 
        key_missing = ('missing',)
2445
 
        test.add_lines(key, (), ['foo\n'])
2446
 
        key_sha1sum = osutils.sha_string('foo\n')
2447
 
        sha1s = test.get_sha1s([key])
2448
 
        self.assertEqual({key: key_sha1sum}, sha1s)
2449
 
        self.assertEqual([], basis.calls)
2450
 
        # But texts that are not in the test knit are looked for in the basis
2451
 
        # directly (rather than via text reconstruction) so that remote servers
2452
 
        # etc don't have to answer with full content.
2453
 
        basis.add_lines(key_basis, (), ['foo\n', 'bar\n'])
2454
 
        basis_sha1sum = osutils.sha_string('foo\nbar\n')
2455
 
        basis.calls = []
2456
 
        sha1s = test.get_sha1s([key, key_missing, key_basis])
2457
 
        self.assertEqual({key: key_sha1sum,
2458
 
            key_basis: basis_sha1sum}, sha1s)
2459
 
        self.assertEqual([("get_sha1s", set([key_basis, key_missing]))],
2460
 
            basis.calls)
2461
 
 
2462
 
    def test_insert_record_stream(self):
2463
 
        # records are inserted as normal; insert_record_stream builds on
2464
 
        # add_lines, so a smoke test should be all that's needed:
2465
 
        key = ('foo',)
2466
 
        key_basis = ('bar',)
2467
 
        key_delta = ('zaphod',)
2468
 
        basis, test = self.get_basis_and_test_knit()
2469
 
        source = self.make_test_knit(name='source')
2470
 
        basis.add_lines(key_basis, (), ['foo\n'])
2471
 
        basis.calls = []
2472
 
        source.add_lines(key_basis, (), ['foo\n'])
2473
 
        source.add_lines(key_delta, (key_basis,), ['bar\n'])
2474
 
        stream = source.get_record_stream([key_delta], 'unordered', False)
2475
 
        test.insert_record_stream(stream)
2476
 
        # XXX: this does somewhat too many calls in making sure of whether it
2477
 
        # has to recreate the full text.
2478
 
        self.assertEqual([("get_parent_map", set([key_basis])),
2479
 
             ('get_parent_map', set([key_basis])),
2480
 
             ('get_record_stream', [key_basis], 'unordered', True)],
2481
 
            basis.calls)
2482
 
        self.assertEqual({key_delta:(key_basis,)},
2483
 
            test.get_parent_map([key_delta]))
2484
 
        self.assertEqual('bar\n', test.get_record_stream([key_delta],
2485
 
            'unordered', True).next().get_bytes_as('fulltext'))
2486
 
 
2487
 
    def test_iter_lines_added_or_present_in_keys(self):
2488
 
        # Lines from the basis are returned, and lines for a given key are only
2489
 
        # returned once.
2490
 
        key1 = ('foo1',)
2491
 
        key2 = ('foo2',)
2492
 
        # all sources are asked for keys:
2493
 
        basis, test = self.get_basis_and_test_knit()
2494
 
        basis.add_lines(key1, (), ["foo"])
2495
 
        basis.calls = []
2496
 
        lines = list(test.iter_lines_added_or_present_in_keys([key1]))
2497
 
        self.assertEqual([("foo\n", key1)], lines)
2498
 
        self.assertEqual([("iter_lines_added_or_present_in_keys", set([key1]))],
2499
 
            basis.calls)
2500
 
        # keys in both are not duplicated:
2501
 
        test.add_lines(key2, (), ["bar\n"])
2502
 
        basis.add_lines(key2, (), ["bar\n"])
2503
 
        basis.calls = []
2504
 
        lines = list(test.iter_lines_added_or_present_in_keys([key2]))
2505
 
        self.assertEqual([("bar\n", key2)], lines)
2506
 
        self.assertEqual([], basis.calls)
2507
 
 
2508
 
    def test_keys(self):
2509
 
        key1 = ('foo1',)
2510
 
        key2 = ('foo2',)
2511
 
        # all sources are asked for keys:
2512
 
        basis, test = self.get_basis_and_test_knit()
2513
 
        keys = test.keys()
2514
 
        self.assertEqual(set(), set(keys))
2515
 
        self.assertEqual([("keys",)], basis.calls)
2516
 
        # keys from a basis are returned:
2517
 
        basis.add_lines(key1, (), [])
2518
 
        basis.calls = []
2519
 
        keys = test.keys()
2520
 
        self.assertEqual(set([key1]), set(keys))
2521
 
        self.assertEqual([("keys",)], basis.calls)
2522
 
        # keys in both are not duplicated:
2523
 
        test.add_lines(key2, (), [])
2524
 
        basis.add_lines(key2, (), [])
2525
 
        basis.calls = []
2526
 
        keys = test.keys()
2527
 
        self.assertEqual(2, len(keys))
2528
 
        self.assertEqual(set([key1, key2]), set(keys))
2529
 
        self.assertEqual([("keys",)], basis.calls)
2530
 
 
2531
 
    def test_add_mpdiffs(self):
2532
 
        # records are inserted as normal; add_mpdiff builds on
2533
 
        # add_lines, so a smoke test should be all that's needed:
2534
 
        key = ('foo',)
2535
 
        key_basis = ('bar',)
2536
 
        key_delta = ('zaphod',)
2537
 
        basis, test = self.get_basis_and_test_knit()
2538
 
        source = self.make_test_knit(name='source')
2539
 
        basis.add_lines(key_basis, (), ['foo\n'])
2540
 
        basis.calls = []
2541
 
        source.add_lines(key_basis, (), ['foo\n'])
2542
 
        source.add_lines(key_delta, (key_basis,), ['bar\n'])
2543
 
        diffs = source.make_mpdiffs([key_delta])
2544
 
        test.add_mpdiffs([(key_delta, (key_basis,),
2545
 
            source.get_sha1s([key_delta])[key_delta], diffs[0])])
2546
 
        self.assertEqual([("get_parent_map", set([key_basis])),
2547
 
            ('get_record_stream', [key_basis], 'unordered', True),],
2548
 
            basis.calls)
2549
 
        self.assertEqual({key_delta:(key_basis,)},
2550
 
            test.get_parent_map([key_delta]))
2551
 
        self.assertEqual('bar\n', test.get_record_stream([key_delta],
2552
 
            'unordered', True).next().get_bytes_as('fulltext'))
2553
 
 
2554
 
    def test_make_mpdiffs(self):
2555
 
        # Generating an mpdiff across a stacking boundary should detect parent
2556
 
        # texts regions.
2557
 
        key = ('foo',)
2558
 
        key_left = ('bar',)
2559
 
        key_right = ('zaphod',)
2560
 
        basis, test = self.get_basis_and_test_knit()
2561
 
        basis.add_lines(key_left, (), ['bar\n'])
2562
 
        basis.add_lines(key_right, (), ['zaphod\n'])
2563
 
        basis.calls = []
2564
 
        test.add_lines(key, (key_left, key_right),
2565
 
            ['bar\n', 'foo\n', 'zaphod\n'])
2566
 
        diffs = test.make_mpdiffs([key])
2567
 
        self.assertEqual([
2568
 
            multiparent.MultiParent([multiparent.ParentText(0, 0, 0, 1),
2569
 
                multiparent.NewText(['foo\n']),
2570
 
                multiparent.ParentText(1, 0, 2, 1)])],
2571
 
            diffs)
2572
 
        self.assertEqual(3, len(basis.calls))
2573
 
        self.assertEqual([
2574
 
            ("get_parent_map", set([key_left, key_right])),
2575
 
            ("get_parent_map", set([key_left, key_right])),
2576
 
            ],
2577
 
            basis.calls[:-1])
2578
 
        last_call = basis.calls[-1]
2579
 
        self.assertEqual('get_record_stream', last_call[0])
2580
 
        self.assertEqual(set([key_left, key_right]), set(last_call[1]))
2581
 
        self.assertEqual('topological', last_call[2])
2582
 
        self.assertEqual(True, last_call[3])
2583
 
 
2584
 
 
2585
 
class TestNetworkBehaviour(KnitTests):
2586
 
    """Tests for getting data out of/into knits over the network."""
2587
 
 
2588
 
    def test_include_delta_closure_generates_a_knit_delta_closure(self):
2589
 
        vf = self.make_test_knit(name='test')
2590
 
        # put in three texts, giving ft, delta, delta
2591
 
        vf.add_lines(('base',), (), ['base\n', 'content\n'])
2592
 
        vf.add_lines(('d1',), (('base',),), ['d1\n'])
2593
 
        vf.add_lines(('d2',), (('d1',),), ['d2\n'])
2594
 
        # But heuristics could interfere, so check what happened:
2595
 
        self.assertEqual(['knit-ft-gz', 'knit-delta-gz', 'knit-delta-gz'],
2596
 
            [record.storage_kind for record in
2597
 
             vf.get_record_stream([('base',), ('d1',), ('d2',)],
2598
 
                'topological', False)])
2599
 
        # generate a stream of just the deltas include_delta_closure=True,
2600
 
        # serialise to the network, and check that we get a delta closure on the wire.
2601
 
        stream = vf.get_record_stream([('d1',), ('d2',)], 'topological', True)
2602
 
        netb = [record.get_bytes_as(record.storage_kind) for record in stream]
2603
 
        # The first bytes should be a memo from _ContentMapGenerator, and the
2604
 
        # second bytes should be empty (because its a API proxy not something
2605
 
        # for wire serialisation.
2606
 
        self.assertEqual('', netb[1])
2607
 
        bytes = netb[0]
2608
 
        kind, line_end = network_bytes_to_kind_and_offset(bytes)
2609
 
        self.assertEqual('knit-delta-closure', kind)
2610
 
 
2611
 
 
2612
 
class TestContentMapGenerator(KnitTests):
2613
 
    """Tests for ContentMapGenerator"""
2614
 
 
2615
 
    def test_get_record_stream_gives_records(self):
2616
 
        vf = self.make_test_knit(name='test')
2617
 
        # put in three texts, giving ft, delta, delta
2618
 
        vf.add_lines(('base',), (), ['base\n', 'content\n'])
2619
 
        vf.add_lines(('d1',), (('base',),), ['d1\n'])
2620
 
        vf.add_lines(('d2',), (('d1',),), ['d2\n'])
2621
 
        keys = [('d1',), ('d2',)]
2622
 
        generator = _VFContentMapGenerator(vf, keys,
2623
 
            global_map=vf.get_parent_map(keys))
2624
 
        for record in generator.get_record_stream():
2625
 
            if record.key == ('d1',):
2626
 
                self.assertEqual('d1\n', record.get_bytes_as('fulltext'))
2627
 
            else:
2628
 
                self.assertEqual('d2\n', record.get_bytes_as('fulltext'))
2629
 
 
2630
 
    def test_get_record_stream_kinds_are_raw(self):
2631
 
        vf = self.make_test_knit(name='test')
2632
 
        # put in three texts, giving ft, delta, delta
2633
 
        vf.add_lines(('base',), (), ['base\n', 'content\n'])
2634
 
        vf.add_lines(('d1',), (('base',),), ['d1\n'])
2635
 
        vf.add_lines(('d2',), (('d1',),), ['d2\n'])
2636
 
        keys = [('base',), ('d1',), ('d2',)]
2637
 
        generator = _VFContentMapGenerator(vf, keys,
2638
 
            global_map=vf.get_parent_map(keys))
2639
 
        kinds = {('base',): 'knit-delta-closure',
2640
 
            ('d1',): 'knit-delta-closure-ref',
2641
 
            ('d2',): 'knit-delta-closure-ref',
2642
 
            }
2643
 
        for record in generator.get_record_stream():
2644
 
            self.assertEqual(kinds[record.key], record.storage_kind)