466
397
# Arguably, the deserialise_inventory should detect a mismatch, and
467
398
# raise an error, rather than silently using one revision_id over the
469
self.assertRaises(AssertionError, repo._deserialise_inventory,
470
'test-rev-id', inv_xml)
471
inv = repo._deserialise_inventory('other-rev-id', inv_xml)
400
inv = repo.deserialise_inventory('test-rev-id', inv_xml)
472
401
self.assertEqual('other-rev-id', inv.root.revision)
474
def test_supports_external_lookups(self):
475
repo = self.make_repository('.',
476
format=bzrdir.format_registry.get('knit')())
477
self.assertFalse(repo._format.supports_external_lookups)
404
class KnitRepositoryStreamTests(test_knit.KnitTests):
405
"""Tests for knitrepo._get_stream_as_bytes."""
407
def test_get_stream_as_bytes(self):
409
k1 = self.make_test_knit()
410
k1.add_lines('text-a', [], test_knit.split_lines(test_knit.TEXT_1))
412
# Serialise it, check the output.
413
bytes = knitrepo._get_stream_as_bytes(k1, ['text-a'])
414
data = bencode.bdecode(bytes)
415
format, record = data
416
self.assertEqual('knit-plain', format)
417
self.assertEqual(['text-a', ['fulltext'], []], record[:3])
418
self.assertRecordContentEqual(k1, 'text-a', record[3])
420
def test_get_stream_as_bytes_all(self):
421
"""Get a serialised data stream for all the records in a knit.
423
Much like test_get_stream_all, except for get_stream_as_bytes.
425
k1 = self.make_test_knit()
426
# Insert the same data as BasicKnitTests.test_knit_join, as they seem
427
# to cover a range of cases (no parents, one parent, multiple parents).
429
('text-a', [], test_knit.TEXT_1),
430
('text-b', ['text-a'], test_knit.TEXT_1),
431
('text-c', [], test_knit.TEXT_1),
432
('text-d', ['text-c'], test_knit.TEXT_1),
433
('text-m', ['text-b', 'text-d'], test_knit.TEXT_1),
435
expected_data_list = [
436
# version, options, parents
437
('text-a', ['fulltext'], []),
438
('text-b', ['line-delta'], ['text-a']),
439
('text-c', ['fulltext'], []),
440
('text-d', ['line-delta'], ['text-c']),
441
('text-m', ['line-delta'], ['text-b', 'text-d']),
443
for version_id, parents, lines in test_data:
444
k1.add_lines(version_id, parents, test_knit.split_lines(lines))
446
bytes = knitrepo._get_stream_as_bytes(
447
k1, ['text-a', 'text-b', 'text-c', 'text-d', 'text-m'])
449
data = bencode.bdecode(bytes)
451
self.assertEqual('knit-plain', format)
453
for expected, actual in zip(expected_data_list, data):
454
expected_version = expected[0]
455
expected_options = expected[1]
456
expected_parents = expected[2]
457
version, options, parents, bytes = actual
458
self.assertEqual(expected_version, version)
459
self.assertEqual(expected_options, options)
460
self.assertEqual(expected_parents, parents)
461
self.assertRecordContentEqual(k1, version, bytes)
480
464
class DummyRepository(object):
481
465
"""A dummy repository for testing."""
484
467
_serializer = None
486
469
def supports_rich_root(self):
487
if self._format is not None:
488
return self._format.rich_root_data
492
raise NotImplementedError
494
def get_parent_map(self, revision_ids):
495
raise NotImplementedError
498
473
class InterDummy(repository.InterRepository):
499
474
"""An inter-repository optimised code path for DummyRepository.
650
648
tree = self.make_branch_and_tree('.', format)
651
649
tree.commit("Dull commit", rev_id="dull")
652
650
revision_tree = tree.branch.repository.revision_tree('dull')
653
revision_tree.lock_read()
655
self.assertRaises(errors.NoSuchFile, revision_tree.get_file_lines,
656
revision_tree.inventory.root.file_id)
658
revision_tree.unlock()
651
self.assertRaises(errors.NoSuchFile, revision_tree.get_file_lines,
652
revision_tree.inventory.root.file_id)
659
653
format = bzrdir.BzrDirMetaFormat1()
660
654
format.repository_format = knitrepo.RepositoryFormatKnit3()
661
655
upgrade.Convert('.', format)
662
656
tree = workingtree.WorkingTree.open('.')
663
657
revision_tree = tree.branch.repository.revision_tree('dull')
664
revision_tree.lock_read()
666
revision_tree.get_file_lines(revision_tree.inventory.root.file_id)
668
revision_tree.unlock()
658
revision_tree.get_file_lines(revision_tree.inventory.root.file_id)
669
659
tree.commit("Another dull commit", rev_id='dull2')
670
660
revision_tree = tree.branch.repository.revision_tree('dull2')
671
revision_tree.lock_read()
672
self.addCleanup(revision_tree.unlock)
673
661
self.assertEqual('dull', revision_tree.inventory.root.revision)
675
def test_supports_external_lookups(self):
663
def test_exposed_versioned_files_are_marked_dirty(self):
676
664
format = bzrdir.BzrDirMetaFormat1()
677
665
format.repository_format = knitrepo.RepositoryFormatKnit3()
678
666
repo = self.make_repository('.', format=format)
679
self.assertFalse(repo._format.supports_external_lookups)
682
class Test2a(tests.TestCaseWithMemoryTransport):
684
def test_fetch_combines_groups(self):
685
builder = self.make_branch_builder('source', format='2a')
686
builder.start_series()
687
builder.build_snapshot('1', None, [
688
('add', ('', 'root-id', 'directory', '')),
689
('add', ('file', 'file-id', 'file', 'content\n'))])
690
builder.build_snapshot('2', ['1'], [
691
('modify', ('file-id', 'content-2\n'))])
692
builder.finish_series()
693
source = builder.get_branch()
694
target = self.make_repository('target', format='2a')
695
target.fetch(source.repository)
697
self.addCleanup(target.unlock)
698
details = target.texts._index.get_build_details(
699
[('file-id', '1',), ('file-id', '2',)])
700
file_1_details = details[('file-id', '1')]
701
file_2_details = details[('file-id', '2')]
702
# The index, and what to read off disk, should be the same for both
703
# versions of the file.
704
self.assertEqual(file_1_details[0][:3], file_2_details[0][:3])
706
def test_fetch_combines_groups(self):
707
builder = self.make_branch_builder('source', format='2a')
708
builder.start_series()
709
builder.build_snapshot('1', None, [
710
('add', ('', 'root-id', 'directory', '')),
711
('add', ('file', 'file-id', 'file', 'content\n'))])
712
builder.build_snapshot('2', ['1'], [
713
('modify', ('file-id', 'content-2\n'))])
714
builder.finish_series()
715
source = builder.get_branch()
716
target = self.make_repository('target', format='2a')
717
target.fetch(source.repository)
719
self.addCleanup(target.unlock)
720
details = target.texts._index.get_build_details(
721
[('file-id', '1',), ('file-id', '2',)])
722
file_1_details = details[('file-id', '1')]
723
file_2_details = details[('file-id', '2')]
724
# The index, and what to read off disk, should be the same for both
725
# versions of the file.
726
self.assertEqual(file_1_details[0][:3], file_2_details[0][:3])
728
def test_fetch_combines_groups(self):
729
builder = self.make_branch_builder('source', format='2a')
730
builder.start_series()
731
builder.build_snapshot('1', None, [
732
('add', ('', 'root-id', 'directory', '')),
733
('add', ('file', 'file-id', 'file', 'content\n'))])
734
builder.build_snapshot('2', ['1'], [
735
('modify', ('file-id', 'content-2\n'))])
736
builder.finish_series()
737
source = builder.get_branch()
738
target = self.make_repository('target', format='2a')
739
target.fetch(source.repository)
741
self.addCleanup(target.unlock)
742
details = target.texts._index.get_build_details(
743
[('file-id', '1',), ('file-id', '2',)])
744
file_1_details = details[('file-id', '1')]
745
file_2_details = details[('file-id', '2')]
746
# The index, and what to read off disk, should be the same for both
747
# versions of the file.
748
self.assertEqual(file_1_details[0][:3], file_2_details[0][:3])
750
def test_format_pack_compresses_True(self):
751
repo = self.make_repository('repo', format='2a')
752
self.assertTrue(repo._format.pack_compresses)
754
def test_inventories_use_chk_map_with_parent_base_dict(self):
755
tree = self.make_branch_and_memory_tree('repo', format="2a")
757
tree.add([''], ['TREE_ROOT'])
758
revid = tree.commit("foo")
761
self.addCleanup(tree.unlock)
762
inv = tree.branch.repository.get_inventory(revid)
763
self.assertNotEqual(None, inv.parent_id_basename_to_file_id)
764
inv.parent_id_basename_to_file_id._ensure_root()
765
inv.id_to_entry._ensure_root()
766
self.assertEqual(65536, inv.id_to_entry._root_node.maximum_size)
767
self.assertEqual(65536,
768
inv.parent_id_basename_to_file_id._root_node.maximum_size)
770
def test_autopack_unchanged_chk_nodes(self):
771
# at 20 unchanged commits, chk pages are packed that are split into
772
# two groups such that the new pack being made doesn't have all its
773
# pages in the source packs (though they are in the repository).
774
# Use a memory backed repository, we don't need to hit disk for this
775
tree = self.make_branch_and_memory_tree('tree', format='2a')
777
self.addCleanup(tree.unlock)
778
tree.add([''], ['TREE_ROOT'])
779
for pos in range(20):
780
tree.commit(str(pos))
782
def test_pack_with_hint(self):
783
tree = self.make_branch_and_memory_tree('tree', format='2a')
785
self.addCleanup(tree.unlock)
786
tree.add([''], ['TREE_ROOT'])
787
# 1 commit to leave untouched
789
to_keep = tree.branch.repository._pack_collection.names()
793
all = tree.branch.repository._pack_collection.names()
794
combine = list(set(all) - set(to_keep))
795
self.assertLength(3, all)
796
self.assertLength(2, combine)
797
tree.branch.repository.pack(hint=combine)
798
final = tree.branch.repository._pack_collection.names()
799
self.assertLength(2, final)
800
self.assertFalse(combine[0] in final)
801
self.assertFalse(combine[1] in final)
802
self.assertSubset(to_keep, final)
804
def test_stream_source_to_gc(self):
805
source = self.make_repository('source', format='2a')
806
target = self.make_repository('target', format='2a')
807
stream = source._get_source(target._format)
808
self.assertIsInstance(stream, groupcompress_repo.GroupCHKStreamSource)
810
def test_stream_source_to_non_gc(self):
811
source = self.make_repository('source', format='2a')
812
target = self.make_repository('target', format='rich-root-pack')
813
stream = source._get_source(target._format)
814
# We don't want the child GroupCHKStreamSource
815
self.assertIs(type(stream), repository.StreamSource)
817
def test_get_stream_for_missing_keys_includes_all_chk_refs(self):
818
source_builder = self.make_branch_builder('source',
820
# We have to build a fairly large tree, so that we are sure the chk
821
# pages will have split into multiple pages.
822
entries = [('add', ('', 'a-root-id', 'directory', None))]
823
for i in 'abcdefghijklmnopqrstuvwxyz123456789':
824
for j in 'abcdefghijklmnopqrstuvwxyz123456789':
827
content = 'content for %s\n' % (fname,)
828
entries.append(('add', (fname, fid, 'file', content)))
829
source_builder.start_series()
830
source_builder.build_snapshot('rev-1', None, entries)
831
# Now change a few of them, so we get a few new pages for the second
833
source_builder.build_snapshot('rev-2', ['rev-1'], [
834
('modify', ('aa-id', 'new content for aa-id\n')),
835
('modify', ('cc-id', 'new content for cc-id\n')),
836
('modify', ('zz-id', 'new content for zz-id\n')),
838
source_builder.finish_series()
839
source_branch = source_builder.get_branch()
840
source_branch.lock_read()
841
self.addCleanup(source_branch.unlock)
842
target = self.make_repository('target', format='2a')
843
source = source_branch.repository._get_source(target._format)
844
self.assertIsInstance(source, groupcompress_repo.GroupCHKStreamSource)
846
# On a regular pass, getting the inventories and chk pages for rev-2
847
# would only get the newly created chk pages
848
search = graph.SearchResult(set(['rev-2']), set(['rev-1']), 1,
850
simple_chk_records = []
851
for vf_name, substream in source.get_stream(search):
852
if vf_name == 'chk_bytes':
853
for record in substream:
854
simple_chk_records.append(record.key)
858
# 3 pages, the root (InternalNode), + 2 pages which actually changed
859
self.assertEqual([('sha1:91481f539e802c76542ea5e4c83ad416bf219f73',),
860
('sha1:4ff91971043668583985aec83f4f0ab10a907d3f',),
861
('sha1:81e7324507c5ca132eedaf2d8414ee4bb2226187',),
862
('sha1:b101b7da280596c71a4540e9a1eeba8045985ee0',)],
864
# Now, when we do a similar call using 'get_stream_for_missing_keys'
865
# we should get a much larger set of pages.
866
missing = [('inventories', 'rev-2')]
867
full_chk_records = []
868
for vf_name, substream in source.get_stream_for_missing_keys(missing):
869
if vf_name == 'inventories':
870
for record in substream:
871
self.assertEqual(('rev-2',), record.key)
872
elif vf_name == 'chk_bytes':
873
for record in substream:
874
full_chk_records.append(record.key)
876
self.fail('Should not be getting a stream of %s' % (vf_name,))
877
# We have 257 records now. This is because we have 1 root page, and 256
878
# leaf pages in a complete listing.
879
self.assertEqual(257, len(full_chk_records))
880
self.assertSubset(simple_chk_records, full_chk_records)
882
def test_inconsistency_fatal(self):
883
repo = self.make_repository('repo', format='2a')
884
self.assertTrue(repo.revisions._index._inconsistency_fatal)
885
self.assertFalse(repo.texts._index._inconsistency_fatal)
886
self.assertFalse(repo.inventories._index._inconsistency_fatal)
887
self.assertFalse(repo.signatures._index._inconsistency_fatal)
888
self.assertFalse(repo.chk_bytes._index._inconsistency_fatal)
891
class TestKnitPackStreamSource(tests.TestCaseWithMemoryTransport):
893
def test_source_to_exact_pack_092(self):
894
source = self.make_repository('source', format='pack-0.92')
895
target = self.make_repository('target', format='pack-0.92')
896
stream_source = source._get_source(target._format)
897
self.assertIsInstance(stream_source, pack_repo.KnitPackStreamSource)
899
def test_source_to_exact_pack_rich_root_pack(self):
900
source = self.make_repository('source', format='rich-root-pack')
901
target = self.make_repository('target', format='rich-root-pack')
902
stream_source = source._get_source(target._format)
903
self.assertIsInstance(stream_source, pack_repo.KnitPackStreamSource)
905
def test_source_to_exact_pack_19(self):
906
source = self.make_repository('source', format='1.9')
907
target = self.make_repository('target', format='1.9')
908
stream_source = source._get_source(target._format)
909
self.assertIsInstance(stream_source, pack_repo.KnitPackStreamSource)
911
def test_source_to_exact_pack_19_rich_root(self):
912
source = self.make_repository('source', format='1.9-rich-root')
913
target = self.make_repository('target', format='1.9-rich-root')
914
stream_source = source._get_source(target._format)
915
self.assertIsInstance(stream_source, pack_repo.KnitPackStreamSource)
917
def test_source_to_remote_exact_pack_19(self):
918
trans = self.make_smart_server('target')
920
source = self.make_repository('source', format='1.9')
921
target = self.make_repository('target', format='1.9')
922
target = repository.Repository.open(trans.base)
923
stream_source = source._get_source(target._format)
924
self.assertIsInstance(stream_source, pack_repo.KnitPackStreamSource)
926
def test_stream_source_to_non_exact(self):
927
source = self.make_repository('source', format='pack-0.92')
928
target = self.make_repository('target', format='1.9')
929
stream = source._get_source(target._format)
930
self.assertIs(type(stream), repository.StreamSource)
932
def test_stream_source_to_non_exact_rich_root(self):
933
source = self.make_repository('source', format='1.9')
934
target = self.make_repository('target', format='1.9-rich-root')
935
stream = source._get_source(target._format)
936
self.assertIs(type(stream), repository.StreamSource)
938
def test_source_to_remote_non_exact_pack_19(self):
939
trans = self.make_smart_server('target')
941
source = self.make_repository('source', format='1.9')
942
target = self.make_repository('target', format='1.6')
943
target = repository.Repository.open(trans.base)
944
stream_source = source._get_source(target._format)
945
self.assertIs(type(stream_source), repository.StreamSource)
947
def test_stream_source_to_knit(self):
948
source = self.make_repository('source', format='pack-0.92')
949
target = self.make_repository('target', format='dirstate')
950
stream = source._get_source(target._format)
951
self.assertIs(type(stream), repository.StreamSource)
954
class TestDevelopment6FindParentIdsOfRevisions(TestCaseWithTransport):
955
"""Tests for _find_parent_ids_of_revisions."""
958
super(TestDevelopment6FindParentIdsOfRevisions, self).setUp()
959
self.builder = self.make_branch_builder('source',
960
format='development6-rich-root')
961
self.builder.start_series()
962
self.builder.build_snapshot('initial', None,
963
[('add', ('', 'tree-root', 'directory', None))])
964
self.repo = self.builder.get_branch().repository
965
self.addCleanup(self.builder.finish_series)
967
def assertParentIds(self, expected_result, rev_set):
968
self.assertEqual(sorted(expected_result),
969
sorted(self.repo._find_parent_ids_of_revisions(rev_set)))
971
def test_simple(self):
972
self.builder.build_snapshot('revid1', None, [])
973
self.builder.build_snapshot('revid2', ['revid1'], [])
975
self.assertParentIds(['revid1'], rev_set)
977
def test_not_first_parent(self):
978
self.builder.build_snapshot('revid1', None, [])
979
self.builder.build_snapshot('revid2', ['revid1'], [])
980
self.builder.build_snapshot('revid3', ['revid2'], [])
981
rev_set = ['revid3', 'revid2']
982
self.assertParentIds(['revid1'], rev_set)
984
def test_not_null(self):
985
rev_set = ['initial']
986
self.assertParentIds([], rev_set)
988
def test_not_null_set(self):
989
self.builder.build_snapshot('revid1', None, [])
990
rev_set = [_mod_revision.NULL_REVISION]
991
self.assertParentIds([], rev_set)
993
def test_ghost(self):
994
self.builder.build_snapshot('revid1', None, [])
995
rev_set = ['ghost', 'revid1']
996
self.assertParentIds(['initial'], rev_set)
998
def test_ghost_parent(self):
999
self.builder.build_snapshot('revid1', None, [])
1000
self.builder.build_snapshot('revid2', ['revid1', 'ghost'], [])
1001
rev_set = ['revid2', 'revid1']
1002
self.assertParentIds(['ghost', 'initial'], rev_set)
1004
def test_righthand_parent(self):
1005
self.builder.build_snapshot('revid1', None, [])
1006
self.builder.build_snapshot('revid2a', ['revid1'], [])
1007
self.builder.build_snapshot('revid2b', ['revid1'], [])
1008
self.builder.build_snapshot('revid3', ['revid2a', 'revid2b'], [])
1009
rev_set = ['revid3', 'revid2a']
1010
self.assertParentIds(['revid1', 'revid2b'], rev_set)
668
inv = repo.get_inventory_weave()
670
self.assertRaises(errors.OutSideTransaction,
671
inv.add_lines, 'foo', [], [])
1013
674
class TestWithBrokenRepo(TestCaseWithTransport):
1093
752
broken_repo = self.make_broken_repository()
1094
753
empty_repo = self.make_repository('empty-repo')
1096
empty_repo.fetch(broken_repo)
1097
except (errors.RevisionNotPresent, errors.BzrCheckError):
1098
# Test successful: compression parent not being copied leads to
1101
empty_repo.lock_read()
754
stream = broken_repo.get_data_stream(['rev1a', 'rev2', 'rev3'])
755
empty_repo.lock_write()
1102
756
self.addCleanup(empty_repo.unlock)
1103
text = empty_repo.texts.get_record_stream(
1104
[('file2-id', 'rev3')], 'topological', True).next()
1105
self.assertEqual('line\n', text.get_bytes_as('fulltext'))
757
empty_repo.start_write_group()
760
errors.KnitCorrupt, empty_repo.insert_data_stream, stream)
762
empty_repo.abort_write_group()
765
class TestKnitPackNoSubtrees(TestCaseWithTransport):
767
def get_format(self):
768
return bzrdir.format_registry.make_bzrdir('knitpack-experimental')
770
def test_disk_layout(self):
771
format = self.get_format()
772
repo = self.make_repository('.', format=format)
773
# in case of side effects of locking.
776
t = repo.bzrdir.get_repository_transport(None)
778
# XXX: no locks left when unlocked at the moment
779
# self.assertEqualDiff('', t.get('lock').read())
780
self.check_databases(t)
782
def check_format(self, t):
783
self.assertEqualDiff(
784
"Bazaar pack repository format 1 (needs bzr 0.92)\n",
785
t.get('format').read())
787
def assertHasKndx(self, t, knit_name):
788
"""Assert that knit_name exists on t."""
789
self.assertEqualDiff('# bzr knit index 8\n',
790
t.get(knit_name + '.kndx').read())
792
def assertHasNoKndx(self, t, knit_name):
793
"""Assert that knit_name has no index on t."""
794
self.assertFalse(t.has(knit_name + '.kndx'))
796
def assertHasNoKnit(self, t, knit_name):
797
"""Assert that knit_name exists on t."""
799
self.assertFalse(t.has(knit_name + '.knit'))
801
def check_databases(self, t):
802
"""check knit content for a repository."""
803
# check conversion worked
804
self.assertHasNoKndx(t, 'inventory')
805
self.assertHasNoKnit(t, 'inventory')
806
self.assertHasNoKndx(t, 'revisions')
807
self.assertHasNoKnit(t, 'revisions')
808
self.assertHasNoKndx(t, 'signatures')
809
self.assertHasNoKnit(t, 'signatures')
810
self.assertFalse(t.has('knits'))
811
# revision-indexes file-container directory
813
list(GraphIndex(t, 'pack-names', None).iter_all_entries()))
814
self.assertTrue(S_ISDIR(t.stat('packs').st_mode))
815
self.assertTrue(S_ISDIR(t.stat('upload').st_mode))
816
self.assertTrue(S_ISDIR(t.stat('indices').st_mode))
817
self.assertTrue(S_ISDIR(t.stat('obsolete_packs').st_mode))
819
def test_shared_disk_layout(self):
820
format = self.get_format()
821
repo = self.make_repository('.', shared=True, format=format)
823
t = repo.bzrdir.get_repository_transport(None)
825
# XXX: no locks left when unlocked at the moment
826
# self.assertEqualDiff('', t.get('lock').read())
827
# We should have a 'shared-storage' marker file.
828
self.assertEqualDiff('', t.get('shared-storage').read())
829
self.check_databases(t)
831
def test_shared_no_tree_disk_layout(self):
832
format = self.get_format()
833
repo = self.make_repository('.', shared=True, format=format)
834
repo.set_make_working_trees(False)
836
t = repo.bzrdir.get_repository_transport(None)
838
# XXX: no locks left when unlocked at the moment
839
# self.assertEqualDiff('', t.get('lock').read())
840
# We should have a 'shared-storage' marker file.
841
self.assertEqualDiff('', t.get('shared-storage').read())
842
# We should have a marker for the no-working-trees flag.
843
self.assertEqualDiff('', t.get('no-working-trees').read())
844
# The marker should go when we toggle the setting.
845
repo.set_make_working_trees(True)
846
self.assertFalse(t.has('no-working-trees'))
847
self.check_databases(t)
849
def test_adding_revision_creates_pack_indices(self):
850
format = self.get_format()
851
tree = self.make_branch_and_tree('.', format=format)
852
trans = tree.branch.repository.bzrdir.get_repository_transport(None)
854
list(GraphIndex(trans, 'pack-names', None).iter_all_entries()))
855
tree.commit('foobarbaz')
856
index = GraphIndex(trans, 'pack-names', None)
857
index_nodes = list(index.iter_all_entries())
858
self.assertEqual(1, len(index_nodes))
859
node = index_nodes[0]
861
# the pack sizes should be listed in the index
863
sizes = [int(digits) for digits in pack_value.split(' ')]
864
for size, suffix in zip(sizes, ['.rix', '.iix', '.tix', '.six']):
865
stat = trans.stat('indices/%s%s' % (name, suffix))
866
self.assertEqual(size, stat.st_size)
868
def test_pulling_nothing_leads_to_no_new_names(self):
869
format = self.get_format()
870
tree1 = self.make_branch_and_tree('1', format=format)
871
tree2 = self.make_branch_and_tree('2', format=format)
872
tree1.branch.repository.fetch(tree2.branch.repository)
873
trans = tree1.branch.repository.bzrdir.get_repository_transport(None)
875
list(GraphIndex(trans, 'pack-names', None).iter_all_entries()))
877
def test_commit_across_pack_shape_boundary_autopacks(self):
878
format = self.get_format()
879
tree = self.make_branch_and_tree('.', format=format)
880
trans = tree.branch.repository.bzrdir.get_repository_transport(None)
881
# This test could be a little cheaper by replacing the packs
882
# attribute on the repository to allow a different pack distribution
883
# and max packs policy - so we are checking the policy is honoured
884
# in the test. But for now 11 commits is not a big deal in a single
887
tree.commit('commit %s' % x)
888
# there should be 9 packs:
889
index = GraphIndex(trans, 'pack-names', None)
890
self.assertEqual(9, len(list(index.iter_all_entries())))
891
# committing one more should coalesce to 1 of 10.
892
tree.commit('commit triggering pack')
893
index = GraphIndex(trans, 'pack-names', None)
894
self.assertEqual(1, len(list(index.iter_all_entries())))
895
# packing should not damage data
896
tree = tree.bzrdir.open_workingtree()
897
check_result = tree.branch.repository.check(
898
[tree.branch.last_revision()])
899
# XXX: Todo check packs obsoleted correctly - old packs and indices
900
# in the obsolete_packs directory.
901
large_pack_name = list(index.iter_all_entries())[0][1][0]
902
# finally, committing again should not touch the large pack.
903
tree.commit('commit not triggering pack')
904
index = GraphIndex(trans, 'pack-names', None)
905
self.assertEqual(2, len(list(index.iter_all_entries())))
906
pack_names = [node[1][0] for node in index.iter_all_entries()]
907
self.assertTrue(large_pack_name in pack_names)
909
def test_pack_after_two_commits_packs_everything(self):
910
format = self.get_format()
911
tree = self.make_branch_and_tree('.', format=format)
912
trans = tree.branch.repository.bzrdir.get_repository_transport(None)
914
tree.commit('more work')
915
tree.branch.repository.pack()
916
# there should be 1 pack:
917
index = GraphIndex(trans, 'pack-names', None)
918
self.assertEqual(1, len(list(index.iter_all_entries())))
919
self.assertEqual(2, len(tree.branch.repository.all_revision_ids()))
921
def test_pack_repositories_support_multiple_write_locks(self):
922
format = self.get_format()
923
self.make_repository('.', shared=True, format=format)
924
r1 = repository.Repository.open('.')
925
r2 = repository.Repository.open('.')
927
self.addCleanup(r1.unlock)
931
def _add_text(self, repo, fileid):
932
"""Add a text to the repository within a write group."""
933
vf =repo.weave_store.get_weave(fileid, repo.get_transaction())
934
vf.add_lines('samplerev+' + fileid, [], [])
936
def test_concurrent_writers_merge_new_packs(self):
937
format = self.get_format()
938
self.make_repository('.', shared=True, format=format)
939
r1 = repository.Repository.open('.')
940
r2 = repository.Repository.open('.')
943
# access enough data to load the names list
944
list(r1.all_revision_ids())
947
# access enough data to load the names list
948
list(r2.all_revision_ids())
949
r1.start_write_group()
951
r2.start_write_group()
953
self._add_text(r1, 'fileidr1')
954
self._add_text(r2, 'fileidr2')
956
r2.abort_write_group()
959
r1.abort_write_group()
961
# both r1 and r2 have open write groups with data in them
962
# created while the other's write group was open.
963
# Commit both which requires a merge to the pack-names.
965
r1.commit_write_group()
967
r1.abort_write_group()
968
r2.abort_write_group()
970
r2.commit_write_group()
971
# tell r1 to reload from disk
972
r1._pack_collection.reset()
973
# Now both repositories should know about both names
974
r1._pack_collection.ensure_loaded()
975
r2._pack_collection.ensure_loaded()
976
self.assertEqual(r1._pack_collection.names(), r2._pack_collection.names())
977
self.assertEqual(2, len(r1._pack_collection.names()))
983
def test_concurrent_writer_second_preserves_dropping_a_pack(self):
984
format = self.get_format()
985
self.make_repository('.', shared=True, format=format)
986
r1 = repository.Repository.open('.')
987
r2 = repository.Repository.open('.')
991
r1.start_write_group()
993
self._add_text(r1, 'fileidr1')
995
r1.abort_write_group()
998
r1.commit_write_group()
999
r1._pack_collection.ensure_loaded()
1000
name_to_drop = r1._pack_collection.all_packs()[0].name
1005
# access enough data to load the names list
1006
list(r1.all_revision_ids())
1009
# access enough data to load the names list
1010
list(r2.all_revision_ids())
1011
r1._pack_collection.ensure_loaded()
1013
r2.start_write_group()
1015
# in r1, drop the pack
1016
r1._pack_collection._remove_pack_from_memory(
1017
r1._pack_collection.get_pack_by_name(name_to_drop))
1019
self._add_text(r2, 'fileidr2')
1021
r2.abort_write_group()
1024
r1._pack_collection.reset()
1026
# r1 has a changed names list, and r2 an open write groups with
1028
# save r1, and then commit the r2 write group, which requires a
1029
# merge to the pack-names, which should not reinstate
1032
r1._pack_collection._save_pack_names()
1033
r1._pack_collection.reset()
1035
r2.abort_write_group()
1038
r2.commit_write_group()
1040
r2.abort_write_group()
1042
# Now both repositories should now about just one name.
1043
r1._pack_collection.ensure_loaded()
1044
r2._pack_collection.ensure_loaded()
1045
self.assertEqual(r1._pack_collection.names(), r2._pack_collection.names())
1046
self.assertEqual(1, len(r1._pack_collection.names()))
1047
self.assertFalse(name_to_drop in r1._pack_collection.names())
1053
def test_lock_write_does_not_physically_lock(self):
1054
repo = self.make_repository('.', format=self.get_format())
1056
self.addCleanup(repo.unlock)
1057
self.assertFalse(repo.get_physical_lock_status())
1059
def prepare_for_break_lock(self):
1060
# Setup the global ui factory state so that a break-lock method call
1061
# will find usable input in the input stream.
1062
old_factory = bzrlib.ui.ui_factory
1063
def restoreFactory():
1064
bzrlib.ui.ui_factory = old_factory
1065
self.addCleanup(restoreFactory)
1066
bzrlib.ui.ui_factory = bzrlib.ui.SilentUIFactory()
1067
bzrlib.ui.ui_factory.stdin = StringIO("y\n")
1069
def test_break_lock_breaks_physical_lock(self):
1070
repo = self.make_repository('.', format=self.get_format())
1071
repo._pack_collection.lock_names()
1072
repo2 = repository.Repository.open('.')
1073
self.assertTrue(repo.get_physical_lock_status())
1074
self.prepare_for_break_lock()
1076
self.assertFalse(repo.get_physical_lock_status())
1078
def test_broken_physical_locks_error_on__unlock_names_lock(self):
1079
repo = self.make_repository('.', format=self.get_format())
1080
repo._pack_collection.lock_names()
1081
self.assertTrue(repo.get_physical_lock_status())
1082
repo2 = repository.Repository.open('.')
1083
self.prepare_for_break_lock()
1085
self.assertRaises(errors.LockBroken, repo._pack_collection._unlock_names)
1087
def test_fetch_without_find_ghosts_ignores_ghosts(self):
1088
# we want two repositories at this point:
1089
# one with a revision that is a ghost in the other
1091
# 'ghost' is present in has_ghost, 'ghost' is absent in 'missing_ghost'.
1092
# 'references' is present in both repositories, and 'tip' is present
1093
# just in has_ghost.
1094
# has_ghost missing_ghost
1095
#------------------------------
1097
# 'references' 'references'
1099
# In this test we fetch 'tip' which should not fetch 'ghost'
1100
has_ghost = self.make_repository('has_ghost', format=self.get_format())
1101
missing_ghost = self.make_repository('missing_ghost',
1102
format=self.get_format())
1104
def add_commit(repo, revision_id, parent_ids):
1106
repo.start_write_group()
1107
inv = inventory.Inventory(revision_id=revision_id)
1108
inv.root.revision = revision_id
1109
root_id = inv.root.file_id
1110
sha1 = repo.add_inventory(revision_id, inv, [])
1111
vf = repo.weave_store.get_weave_or_empty(root_id,
1112
repo.get_transaction())
1113
vf.add_lines(revision_id, [], [])
1114
rev = bzrlib.revision.Revision(timestamp=0,
1116
committer="Foo Bar <foo@example.com>",
1118
inventory_sha1=sha1,
1119
revision_id=revision_id)
1120
rev.parent_ids = parent_ids
1121
repo.add_revision(revision_id, rev)
1122
repo.commit_write_group()
1124
add_commit(has_ghost, 'ghost', [])
1125
add_commit(has_ghost, 'references', ['ghost'])
1126
add_commit(missing_ghost, 'references', ['ghost'])
1127
add_commit(has_ghost, 'tip', ['references'])
1128
missing_ghost.fetch(has_ghost, 'tip')
1129
# missing ghost now has tip and not ghost.
1130
rev = missing_ghost.get_revision('tip')
1131
inv = missing_ghost.get_inventory('tip')
1132
self.assertRaises(errors.NoSuchRevision,
1133
missing_ghost.get_revision, 'ghost')
1134
self.assertRaises(errors.RevisionNotPresent,
1135
missing_ghost.get_inventory, 'ghost')
1138
class TestKnitPackSubtrees(TestKnitPackNoSubtrees):
1140
def get_format(self):
1141
return bzrdir.format_registry.make_bzrdir(
1142
'knitpack-subtree-experimental')
1144
def check_format(self, t):
1145
self.assertEqualDiff(
1146
"Bazaar pack repository format 1 with subtree support (needs bzr 0.92)\n",
1147
t.get('format').read())
1108
1150
class TestRepositoryPackCollection(TestCaseWithTransport):
1110
1152
def get_format(self):
1111
return bzrdir.format_registry.make_bzrdir('pack-0.92')
1113
def get_packs(self):
1114
format = self.get_format()
1115
repo = self.make_repository('.', format=format)
1116
return repo._pack_collection
1118
def make_packs_and_alt_repo(self, write_lock=False):
1119
"""Create a pack repo with 3 packs, and access it via a second repo."""
1120
tree = self.make_branch_and_tree('.', format=self.get_format())
1122
self.addCleanup(tree.unlock)
1123
rev1 = tree.commit('one')
1124
rev2 = tree.commit('two')
1125
rev3 = tree.commit('three')
1126
r = repository.Repository.open('.')
1131
self.addCleanup(r.unlock)
1132
packs = r._pack_collection
1133
packs.ensure_loaded()
1134
return tree, r, packs, [rev1, rev2, rev3]
1136
def test__clear_obsolete_packs(self):
1137
packs = self.get_packs()
1138
obsolete_pack_trans = packs.transport.clone('obsolete_packs')
1139
obsolete_pack_trans.put_bytes('a-pack.pack', 'content\n')
1140
obsolete_pack_trans.put_bytes('a-pack.rix', 'content\n')
1141
obsolete_pack_trans.put_bytes('a-pack.iix', 'content\n')
1142
obsolete_pack_trans.put_bytes('another-pack.pack', 'foo\n')
1143
obsolete_pack_trans.put_bytes('not-a-pack.rix', 'foo\n')
1144
res = packs._clear_obsolete_packs()
1145
self.assertEqual(['a-pack', 'another-pack'], sorted(res))
1146
self.assertEqual([], obsolete_pack_trans.list_dir('.'))
1148
def test__clear_obsolete_packs_preserve(self):
1149
packs = self.get_packs()
1150
obsolete_pack_trans = packs.transport.clone('obsolete_packs')
1151
obsolete_pack_trans.put_bytes('a-pack.pack', 'content\n')
1152
obsolete_pack_trans.put_bytes('a-pack.rix', 'content\n')
1153
obsolete_pack_trans.put_bytes('a-pack.iix', 'content\n')
1154
obsolete_pack_trans.put_bytes('another-pack.pack', 'foo\n')
1155
obsolete_pack_trans.put_bytes('not-a-pack.rix', 'foo\n')
1156
res = packs._clear_obsolete_packs(preserve=set(['a-pack']))
1157
self.assertEqual(['a-pack', 'another-pack'], sorted(res))
1158
self.assertEqual(['a-pack.iix', 'a-pack.pack', 'a-pack.rix'],
1159
sorted(obsolete_pack_trans.list_dir('.')))
1153
return bzrdir.format_registry.make_bzrdir('knitpack-experimental')
1161
1155
def test__max_pack_count(self):
1162
1156
"""The maximum pack count is a function of the number of revisions."""
1157
format = self.get_format()
1158
repo = self.make_repository('.', format=format)
1159
packs = repo._pack_collection
1163
1160
# no revisions - one pack, so that we can have a revision free repo
1164
1161
# without it blowing up
1165
packs = self.get_packs()
1166
1162
self.assertEqual(1, packs._max_pack_count(0))
1167
1163
# after that the sum of the digits, - check the first 1-9
1168
1164
self.assertEqual(1, packs._max_pack_count(1))
1338
1299
tree.lock_read()
1339
1300
self.addCleanup(tree.unlock)
1340
1301
packs = tree.branch.repository._pack_collection
1342
1302
packs.ensure_loaded()
1343
1303
name = packs.names()[0]
1344
1304
pack_1 = packs.get_pack_by_name(name)
1345
1305
# the pack should be correctly initialised
1346
sizes = packs._names[name]
1347
rev_index = GraphIndex(packs._index_transport, name + '.rix', sizes[0])
1348
inv_index = GraphIndex(packs._index_transport, name + '.iix', sizes[1])
1349
txt_index = GraphIndex(packs._index_transport, name + '.tix', sizes[2])
1350
sig_index = GraphIndex(packs._index_transport, name + '.six', sizes[3])
1306
rev_index = GraphIndex(packs._index_transport, name + '.rix',
1307
packs._names[name][0])
1308
inv_index = GraphIndex(packs._index_transport, name + '.iix',
1309
packs._names[name][1])
1310
txt_index = GraphIndex(packs._index_transport, name + '.tix',
1311
packs._names[name][2])
1312
sig_index = GraphIndex(packs._index_transport, name + '.six',
1313
packs._names[name][3])
1351
1314
self.assertEqual(pack_repo.ExistingPack(packs._pack_transport,
1352
1315
name, rev_index, inv_index, txt_index, sig_index), pack_1)
1353
1316
# and the same instance should be returned on successive calls.
1354
1317
self.assertTrue(pack_1 is packs.get_pack_by_name(name))
1356
def test_reload_pack_names_new_entry(self):
1357
tree, r, packs, revs = self.make_packs_and_alt_repo()
1358
names = packs.names()
1359
# Add a new pack file into the repository
1360
rev4 = tree.commit('four')
1361
new_names = tree.branch.repository._pack_collection.names()
1362
new_name = set(new_names).difference(names)
1363
self.assertEqual(1, len(new_name))
1364
new_name = new_name.pop()
1365
# The old collection hasn't noticed yet
1366
self.assertEqual(names, packs.names())
1367
self.assertTrue(packs.reload_pack_names())
1368
self.assertEqual(new_names, packs.names())
1369
# And the repository can access the new revision
1370
self.assertEqual({rev4:(revs[-1],)}, r.get_parent_map([rev4]))
1371
self.assertFalse(packs.reload_pack_names())
1373
def test_reload_pack_names_added_and_removed(self):
1374
tree, r, packs, revs = self.make_packs_and_alt_repo()
1375
names = packs.names()
1376
# Now repack the whole thing
1377
tree.branch.repository.pack()
1378
new_names = tree.branch.repository._pack_collection.names()
1379
# The other collection hasn't noticed yet
1380
self.assertEqual(names, packs.names())
1381
self.assertTrue(packs.reload_pack_names())
1382
self.assertEqual(new_names, packs.names())
1383
self.assertEqual({revs[-1]:(revs[-2],)}, r.get_parent_map([revs[-1]]))
1384
self.assertFalse(packs.reload_pack_names())
1386
def test_reload_pack_names_preserves_pending(self):
1387
# TODO: Update this to also test for pending-deleted names
1388
tree, r, packs, revs = self.make_packs_and_alt_repo(write_lock=True)
1389
# We will add one pack (via start_write_group + insert_record_stream),
1390
# and remove another pack (via _remove_pack_from_memory)
1391
orig_names = packs.names()
1392
orig_at_load = packs._packs_at_load
1393
to_remove_name = iter(orig_names).next()
1394
r.start_write_group()
1395
self.addCleanup(r.abort_write_group)
1396
r.texts.insert_record_stream([versionedfile.FulltextContentFactory(
1397
('text', 'rev'), (), None, 'content\n')])
1398
new_pack = packs._new_pack
1399
self.assertTrue(new_pack.data_inserted())
1401
packs.allocate(new_pack)
1402
packs._new_pack = None
1403
removed_pack = packs.get_pack_by_name(to_remove_name)
1404
packs._remove_pack_from_memory(removed_pack)
1405
names = packs.names()
1406
all_nodes, deleted_nodes, new_nodes, _ = packs._diff_pack_names()
1407
new_names = set([x[0][0] for x in new_nodes])
1408
self.assertEqual(names, sorted([x[0][0] for x in all_nodes]))
1409
self.assertEqual(set(names) - set(orig_names), new_names)
1410
self.assertEqual(set([new_pack.name]), new_names)
1411
self.assertEqual([to_remove_name],
1412
sorted([x[0][0] for x in deleted_nodes]))
1413
packs.reload_pack_names()
1414
reloaded_names = packs.names()
1415
self.assertEqual(orig_at_load, packs._packs_at_load)
1416
self.assertEqual(names, reloaded_names)
1417
all_nodes, deleted_nodes, new_nodes, _ = packs._diff_pack_names()
1418
new_names = set([x[0][0] for x in new_nodes])
1419
self.assertEqual(names, sorted([x[0][0] for x in all_nodes]))
1420
self.assertEqual(set(names) - set(orig_names), new_names)
1421
self.assertEqual(set([new_pack.name]), new_names)
1422
self.assertEqual([to_remove_name],
1423
sorted([x[0][0] for x in deleted_nodes]))
1425
def test_autopack_obsoletes_new_pack(self):
1426
tree, r, packs, revs = self.make_packs_and_alt_repo(write_lock=True)
1427
packs._max_pack_count = lambda x: 1
1428
packs.pack_distribution = lambda x: [10]
1429
r.start_write_group()
1430
r.revisions.insert_record_stream([versionedfile.FulltextContentFactory(
1431
('bogus-rev',), (), None, 'bogus-content\n')])
1432
# This should trigger an autopack, which will combine everything into a
1434
new_names = r.commit_write_group()
1435
names = packs.names()
1436
self.assertEqual(1, len(names))
1437
self.assertEqual([names[0] + '.pack'],
1438
packs._pack_transport.list_dir('.'))
1440
def test_autopack_reloads_and_stops(self):
1441
tree, r, packs, revs = self.make_packs_and_alt_repo(write_lock=True)
1442
# After we have determined what needs to be autopacked, trigger a
1443
# full-pack via the other repo which will cause us to re-evaluate and
1444
# decide we don't need to do anything
1445
orig_execute = packs._execute_pack_operations
1446
def _munged_execute_pack_ops(*args, **kwargs):
1447
tree.branch.repository.pack()
1448
return orig_execute(*args, **kwargs)
1449
packs._execute_pack_operations = _munged_execute_pack_ops
1450
packs._max_pack_count = lambda x: 1
1451
packs.pack_distribution = lambda x: [10]
1452
self.assertFalse(packs.autopack())
1453
self.assertEqual(1, len(packs.names()))
1454
self.assertEqual(tree.branch.repository._pack_collection.names(),
1457
def test__save_pack_names(self):
1458
tree, r, packs, revs = self.make_packs_and_alt_repo(write_lock=True)
1459
names = packs.names()
1460
pack = packs.get_pack_by_name(names[0])
1461
packs._remove_pack_from_memory(pack)
1462
packs._save_pack_names(obsolete_packs=[pack])
1463
cur_packs = packs._pack_transport.list_dir('.')
1464
self.assertEqual([n + '.pack' for n in names[1:]], sorted(cur_packs))
1465
# obsolete_packs will also have stuff like .rix and .iix present.
1466
obsolete_packs = packs.transport.list_dir('obsolete_packs')
1467
obsolete_names = set([osutils.splitext(n)[0] for n in obsolete_packs])
1468
self.assertEqual([pack.name], sorted(obsolete_names))
1470
def test__save_pack_names_already_obsoleted(self):
1471
tree, r, packs, revs = self.make_packs_and_alt_repo(write_lock=True)
1472
names = packs.names()
1473
pack = packs.get_pack_by_name(names[0])
1474
packs._remove_pack_from_memory(pack)
1475
# We are going to simulate a concurrent autopack by manually obsoleting
1476
# the pack directly.
1477
packs._obsolete_packs([pack])
1478
packs._save_pack_names(clear_obsolete_packs=True,
1479
obsolete_packs=[pack])
1480
cur_packs = packs._pack_transport.list_dir('.')
1481
self.assertEqual([n + '.pack' for n in names[1:]], sorted(cur_packs))
1482
# Note that while we set clear_obsolete_packs=True, it should not
1483
# delete a pack file that we have also scheduled for obsoletion.
1484
obsolete_packs = packs.transport.list_dir('obsolete_packs')
1485
obsolete_names = set([osutils.splitext(n)[0] for n in obsolete_packs])
1486
self.assertEqual([pack.name], sorted(obsolete_names))
1490
1320
class TestPack(TestCaseWithTransport):
1491
1321
"""Tests for the Pack object."""
1566
1387
self.assertEqual(20, len(pack.random_name))
1567
1388
self.assertIsInstance(pack.random_name, str)
1568
1389
self.assertIsInstance(pack.start_time, float)
1571
class TestPacker(TestCaseWithTransport):
1572
"""Tests for the packs repository Packer class."""
1574
def test_pack_optimizes_pack_order(self):
1575
builder = self.make_branch_builder('.', format="1.9")
1576
builder.start_series()
1577
builder.build_snapshot('A', None, [
1578
('add', ('', 'root-id', 'directory', None)),
1579
('add', ('f', 'f-id', 'file', 'content\n'))])
1580
builder.build_snapshot('B', ['A'],
1581
[('modify', ('f-id', 'new-content\n'))])
1582
builder.build_snapshot('C', ['B'],
1583
[('modify', ('f-id', 'third-content\n'))])
1584
builder.build_snapshot('D', ['C'],
1585
[('modify', ('f-id', 'fourth-content\n'))])
1586
b = builder.get_branch()
1588
builder.finish_series()
1589
self.addCleanup(b.unlock)
1590
# At this point, we should have 4 pack files available
1591
# Because of how they were built, they correspond to
1592
# ['D', 'C', 'B', 'A']
1593
packs = b.repository._pack_collection.packs
1594
packer = pack_repo.Packer(b.repository._pack_collection,
1596
revision_ids=['B', 'C'])
1597
# Now, when we are copying the B & C revisions, their pack files should
1598
# be moved to the front of the stack
1599
# The new ordering moves B & C to the front of the .packs attribute,
1600
# and leaves the others in the original order.
1601
new_packs = [packs[1], packs[2], packs[0], packs[3]]
1602
new_pack = packer.pack()
1603
self.assertEqual(new_packs, packer.packs)
1606
class TestOptimisingPacker(TestCaseWithTransport):
1607
"""Tests for the OptimisingPacker class."""
1609
def get_pack_collection(self):
1610
repo = self.make_repository('.')
1611
return repo._pack_collection
1613
def test_open_pack_will_optimise(self):
1614
packer = pack_repo.OptimisingPacker(self.get_pack_collection(),
1616
new_pack = packer.open_pack()
1617
self.addCleanup(new_pack.abort) # ensure cleanup
1618
self.assertIsInstance(new_pack, pack_repo.NewPack)
1619
self.assertTrue(new_pack.revision_index._optimize_for_size)
1620
self.assertTrue(new_pack.inventory_index._optimize_for_size)
1621
self.assertTrue(new_pack.text_index._optimize_for_size)
1622
self.assertTrue(new_pack.signature_index._optimize_for_size)
1625
class TestCrossFormatPacks(TestCaseWithTransport):
1627
def log_pack(self, hint=None):
1628
self.calls.append(('pack', hint))
1629
self.orig_pack(hint=hint)
1630
if self.expect_hint:
1631
self.assertTrue(hint)
1633
def run_stream(self, src_fmt, target_fmt, expect_pack_called):
1634
self.expect_hint = expect_pack_called
1636
source_tree = self.make_branch_and_tree('src', format=src_fmt)
1637
source_tree.lock_write()
1638
self.addCleanup(source_tree.unlock)
1639
tip = source_tree.commit('foo')
1640
target = self.make_repository('target', format=target_fmt)
1642
self.addCleanup(target.unlock)
1643
source = source_tree.branch.repository._get_source(target._format)
1644
self.orig_pack = target.pack
1645
target.pack = self.log_pack
1646
search = target.search_missing_revision_ids(
1647
source_tree.branch.repository, tip)
1648
stream = source.get_stream(search)
1649
from_format = source_tree.branch.repository._format
1650
sink = target._get_sink()
1651
sink.insert_stream(stream, from_format, [])
1652
if expect_pack_called:
1653
self.assertLength(1, self.calls)
1655
self.assertLength(0, self.calls)
1657
def run_fetch(self, src_fmt, target_fmt, expect_pack_called):
1658
self.expect_hint = expect_pack_called
1660
source_tree = self.make_branch_and_tree('src', format=src_fmt)
1661
source_tree.lock_write()
1662
self.addCleanup(source_tree.unlock)
1663
tip = source_tree.commit('foo')
1664
target = self.make_repository('target', format=target_fmt)
1666
self.addCleanup(target.unlock)
1667
source = source_tree.branch.repository
1668
self.orig_pack = target.pack
1669
target.pack = self.log_pack
1670
target.fetch(source)
1671
if expect_pack_called:
1672
self.assertLength(1, self.calls)
1674
self.assertLength(0, self.calls)
1676
def test_sink_format_hint_no(self):
1677
# When the target format says packing makes no difference, pack is not
1679
self.run_stream('1.9', 'rich-root-pack', False)
1681
def test_sink_format_hint_yes(self):
1682
# When the target format says packing makes a difference, pack is
1684
self.run_stream('1.9', '2a', True)
1686
def test_sink_format_same_no(self):
1687
# When the formats are the same, pack is not called.
1688
self.run_stream('2a', '2a', False)
1690
def test_IDS_format_hint_no(self):
1691
# When the target format says packing makes no difference, pack is not
1693
self.run_fetch('1.9', 'rich-root-pack', False)
1695
def test_IDS_format_hint_yes(self):
1696
# When the target format says packing makes a difference, pack is
1698
self.run_fetch('1.9', '2a', True)
1700
def test_IDS_format_same_no(self):
1701
# When the formats are the same, pack is not called.
1702
self.run_fetch('2a', '2a', False)
added
removed
bzrlib/tests/test_repository.py
