5557.1.7
by John Arbash Meinel
Merge in the bzr.dev 5582 |
1 |
# Copyright (C) 2006-2011 Canonical Ltd
|
1563.2.1
by Robert Collins
Merge in a variation of the versionedfile api from versioned-file. |
2 |
#
|
3 |
# Authors:
|
|
4 |
# Johan Rydberg <jrydberg@gnu.org>
|
|
5 |
#
|
|
6 |
# This program is free software; you can redistribute it and/or modify
|
|
7 |
# it under the terms of the GNU General Public License as published by
|
|
8 |
# the Free Software Foundation; either version 2 of the License, or
|
|
9 |
# (at your option) any later version.
|
|
1887.1.1
by Adeodato Simó
Do not separate paragraphs in the copyright statement with blank lines, |
10 |
#
|
1563.2.1
by Robert Collins
Merge in a variation of the versionedfile api from versioned-file. |
11 |
# This program is distributed in the hope that it will be useful,
|
12 |
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
13 |
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
14 |
# GNU General Public License for more details.
|
|
1887.1.1
by Adeodato Simó
Do not separate paragraphs in the copyright statement with blank lines, |
15 |
#
|
1563.2.1
by Robert Collins
Merge in a variation of the versionedfile api from versioned-file. |
16 |
# You should have received a copy of the GNU General Public License
|
17 |
# along with this program; if not, write to the Free Software
|
|
4183.7.1
by Sabin Iacob
update FSF mailing address |
18 |
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
1563.2.1
by Robert Collins
Merge in a variation of the versionedfile api from versioned-file. |
19 |
|
20 |
||
1704.2.15
by Martin Pool
Remove TODO about knit testing printed from test suite |
21 |
# TODO: might be nice to create a versionedfile with some type of corruption
|
22 |
# considered typical and check that it can be detected/corrected.
|
|
23 |
||
5590.1.1
by John Arbash Meinel
Stop using tuned_gzip, it seems to give incorrect results on python 2.7 |
24 |
from gzip import GzipFile |
4005.3.3
by Robert Collins
Test NetworkRecordStream with delta'd texts. |
25 |
from itertools import chain, izip |
1664.2.9
by Aaron Bentley
Ported weave merge test to versionedfile |
26 |
from StringIO import StringIO |
27 |
||
2039.1.1
by Aaron Bentley
Clean up progress properly when interrupted during fetch (#54000) |
28 |
from bzrlib import ( |
29 |
errors, |
|
4593.5.35
by John Arbash Meinel
Start working on a per-vf implementation test of find_ancestry. |
30 |
graph as _mod_graph, |
3735.31.1
by John Arbash Meinel
Bring the groupcompress plugin into the brisbane-core branch. |
31 |
groupcompress, |
4241.4.1
by Ian Clatworthy
add sha generation support to versionedfiles |
32 |
knit as _mod_knit, |
2309.4.7
by John Arbash Meinel
Update VersionedFile tests to ensure that they can take Unicode, |
33 |
osutils, |
2039.1.1
by Aaron Bentley
Clean up progress properly when interrupted during fetch (#54000) |
34 |
progress, |
5273.1.7
by Vincent Ladeuil
No more use of the get_transport imported *symbol*, all uses are through |
35 |
transport, |
4332.3.26
by Robert Collins
Allow passing keys to check to VersionedFile.check(). |
36 |
ui, |
2039.1.1
by Aaron Bentley
Clean up progress properly when interrupted during fetch (#54000) |
37 |
)
|
1563.2.11
by Robert Collins
Consolidate reweave and join as we have no separate usage, make reweave tests apply to all versionedfile implementations and deprecate the old reweave apis. |
38 |
from bzrlib.errors import ( |
3316.2.3
by Robert Collins
Remove manual notification of transaction finishing on versioned files. |
39 |
RevisionNotPresent, |
1563.2.11
by Robert Collins
Consolidate reweave and join as we have no separate usage, make reweave tests apply to all versionedfile implementations and deprecate the old reweave apis. |
40 |
RevisionAlreadyPresent, |
41 |
)
|
|
2770.1.1
by Aaron Bentley
Initial implmentation of plain knit annotation |
42 |
from bzrlib.knit import ( |
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
43 |
cleanup_pack_knit, |
44 |
make_file_factory, |
|
45 |
make_pack_factory, |
|
2770.1.1
by Aaron Bentley
Initial implmentation of plain knit annotation |
46 |
)
|
3350.6.2
by Robert Collins
Prepare parameterised test environment. |
47 |
from bzrlib.tests import ( |
3518.1.1
by Jelmer Vernooij
Add VirtualVersionedFiles class. |
48 |
TestCase, |
3350.6.2
by Robert Collins
Prepare parameterised test environment. |
49 |
TestCaseWithMemoryTransport, |
4009.3.2
by Andrew Bennetts
Add test_insert_record_stream_delta_missing_basis_can_be_added_later. |
50 |
TestNotApplicable, |
3350.6.2
by Robert Collins
Prepare parameterised test environment. |
51 |
TestSkipped, |
52 |
)
|
|
2929.3.8
by Vincent Ladeuil
Rename bzrlib/test/HTTPTestUtils.py to bzrlib/tests/http_utils.py and fix uses. |
53 |
from bzrlib.tests.http_utils import TestCaseWithWebserver |
1563.2.13
by Robert Collins
InterVersionedFile implemented. |
54 |
from bzrlib.transport.memory import MemoryTransport |
1563.2.12
by Robert Collins
Checkpointing: created InterObject to factor out common inter object worker code, added InterVersionedFile and tests to allow making join work between any versionedfile. |
55 |
import bzrlib.versionedfile as versionedfile |
3350.6.2
by Robert Collins
Prepare parameterised test environment. |
56 |
from bzrlib.versionedfile import ( |
57 |
ConstantMapper, |
|
58 |
HashEscapedPrefixMapper, |
|
59 |
PrefixMapper, |
|
3518.1.1
by Jelmer Vernooij
Add VirtualVersionedFiles class. |
60 |
VirtualVersionedFiles, |
3350.6.2
by Robert Collins
Prepare parameterised test environment. |
61 |
make_versioned_files_factory, |
62 |
)
|
|
5582.10.46
by Jelmer Vernooij
Revert unnecessary changes. |
63 |
from bzrlib.weave import WeaveFile |
64 |
from bzrlib.weavefile import write_weave |
|
5559.2.2
by Martin Pool
Change to using standard load_tests_apply_scenarios. |
65 |
from bzrlib.tests.scenarios import load_tests_apply_scenarios |
66 |
||
67 |
||
68 |
load_tests = load_tests_apply_scenarios |
|
3350.6.2
by Robert Collins
Prepare parameterised test environment. |
69 |
|
70 |
||
3350.3.11
by Robert Collins
Test inserting a stream that overlaps the current content of a knit does not error. |
71 |
def get_diamond_vf(f, trailing_eol=True, left_only=False): |
3350.3.4
by Robert Collins
Finish adapters for annotated knits to unannotated knits and full texts. |
72 |
"""Get a diamond graph to exercise deltas and merges.
|
3943.8.1
by Marius Kruger
remove all trailing whitespace from bzr source |
73 |
|
3350.3.4
by Robert Collins
Finish adapters for annotated knits to unannotated knits and full texts. |
74 |
:param trailing_eol: If True end the last line with \n.
|
75 |
"""
|
|
76 |
parents = { |
|
77 |
'origin': (), |
|
78 |
'base': (('origin',),), |
|
79 |
'left': (('base',),), |
|
80 |
'right': (('base',),), |
|
81 |
'merged': (('left',), ('right',)), |
|
82 |
}
|
|
83 |
# insert a diamond graph to exercise deltas and merges.
|
|
84 |
if trailing_eol: |
|
85 |
last_char = '\n' |
|
86 |
else: |
|
87 |
last_char = '' |
|
88 |
f.add_lines('origin', [], ['origin' + last_char]) |
|
89 |
f.add_lines('base', ['origin'], ['base' + last_char]) |
|
90 |
f.add_lines('left', ['base'], ['base\n', 'left' + last_char]) |
|
3350.3.11
by Robert Collins
Test inserting a stream that overlaps the current content of a knit does not error. |
91 |
if not left_only: |
92 |
f.add_lines('right', ['base'], |
|
93 |
['base\n', 'right' + last_char]) |
|
94 |
f.add_lines('merged', ['left', 'right'], |
|
95 |
['base\n', 'left\n', 'right\n', 'merged' + last_char]) |
|
3350.3.4
by Robert Collins
Finish adapters for annotated knits to unannotated knits and full texts. |
96 |
return f, parents |
97 |
||
98 |
||
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
99 |
def get_diamond_files(files, key_length, trailing_eol=True, left_only=False, |
4241.4.1
by Ian Clatworthy
add sha generation support to versionedfiles |
100 |
nograph=False, nokeys=False): |
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
101 |
"""Get a diamond graph to exercise deltas and merges.
|
102 |
||
103 |
This creates a 5-node graph in files. If files supports 2-length keys two
|
|
104 |
graphs are made to exercise the support for multiple ids.
|
|
3943.8.1
by Marius Kruger
remove all trailing whitespace from bzr source |
105 |
|
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
106 |
:param trailing_eol: If True end the last line with \n.
|
107 |
:param key_length: The length of keys in files. Currently supports length 1
|
|
108 |
and 2 keys.
|
|
109 |
:param left_only: If True do not add the right and merged nodes.
|
|
110 |
:param nograph: If True, do not provide parents to the add_lines calls;
|
|
111 |
this is useful for tests that need inserted data but have graphless
|
|
112 |
stores.
|
|
4241.4.1
by Ian Clatworthy
add sha generation support to versionedfiles |
113 |
:param nokeys: If True, pass None is as the key for all insertions.
|
114 |
Currently implies nograph.
|
|
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
115 |
:return: The results of the add_lines calls.
|
116 |
"""
|
|
4241.4.1
by Ian Clatworthy
add sha generation support to versionedfiles |
117 |
if nokeys: |
118 |
nograph = True |
|
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
119 |
if key_length == 1: |
120 |
prefixes = [()] |
|
121 |
else: |
|
122 |
prefixes = [('FileA',), ('FileB',)] |
|
123 |
# insert a diamond graph to exercise deltas and merges.
|
|
124 |
if trailing_eol: |
|
125 |
last_char = '\n' |
|
126 |
else: |
|
127 |
last_char = '' |
|
128 |
result = [] |
|
129 |
def get_parents(suffix_list): |
|
130 |
if nograph: |
|
131 |
return () |
|
132 |
else: |
|
133 |
result = [prefix + suffix for suffix in suffix_list] |
|
134 |
return result |
|
4241.4.1
by Ian Clatworthy
add sha generation support to versionedfiles |
135 |
def get_key(suffix): |
136 |
if nokeys: |
|
137 |
return (None, ) |
|
138 |
else: |
|
139 |
return (suffix,) |
|
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
140 |
# we loop over each key because that spreads the inserts across prefixes,
|
141 |
# which is how commit operates.
|
|
142 |
for prefix in prefixes: |
|
4241.4.1
by Ian Clatworthy
add sha generation support to versionedfiles |
143 |
result.append(files.add_lines(prefix + get_key('origin'), (), |
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
144 |
['origin' + last_char])) |
145 |
for prefix in prefixes: |
|
4241.4.1
by Ian Clatworthy
add sha generation support to versionedfiles |
146 |
result.append(files.add_lines(prefix + get_key('base'), |
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
147 |
get_parents([('origin',)]), ['base' + last_char])) |
148 |
for prefix in prefixes: |
|
4241.4.1
by Ian Clatworthy
add sha generation support to versionedfiles |
149 |
result.append(files.add_lines(prefix + get_key('left'), |
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
150 |
get_parents([('base',)]), |
151 |
['base\n', 'left' + last_char])) |
|
152 |
if not left_only: |
|
153 |
for prefix in prefixes: |
|
4241.4.1
by Ian Clatworthy
add sha generation support to versionedfiles |
154 |
result.append(files.add_lines(prefix + get_key('right'), |
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
155 |
get_parents([('base',)]), |
156 |
['base\n', 'right' + last_char])) |
|
157 |
for prefix in prefixes: |
|
4241.4.1
by Ian Clatworthy
add sha generation support to versionedfiles |
158 |
result.append(files.add_lines(prefix + get_key('merged'), |
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
159 |
get_parents([('left',), ('right',)]), |
160 |
['base\n', 'left\n', 'right\n', 'merged' + last_char])) |
|
161 |
return result |
|
162 |
||
163 |
||
1563.2.1
by Robert Collins
Merge in a variation of the versionedfile api from versioned-file. |
164 |
class VersionedFileTestMixIn(object): |
165 |
"""A mixin test class for testing VersionedFiles.
|
|
166 |
||
167 |
This is not an adaptor-style test at this point because
|
|
168 |
theres no dynamic substitution of versioned file implementations,
|
|
169 |
they are strictly controlled by their owning repositories.
|
|
170 |
"""
|
|
171 |
||
3316.2.3
by Robert Collins
Remove manual notification of transaction finishing on versioned files. |
172 |
def get_transaction(self): |
173 |
if not hasattr(self, '_transaction'): |
|
174 |
self._transaction = None |
|
175 |
return self._transaction |
|
176 |
||
1563.2.1
by Robert Collins
Merge in a variation of the versionedfile api from versioned-file. |
177 |
def test_add(self): |
178 |
f = self.get_file() |
|
179 |
f.add_lines('r0', [], ['a\n', 'b\n']) |
|
180 |
f.add_lines('r1', ['r0'], ['b\n', 'c\n']) |
|
1563.2.9
by Robert Collins
Update versionedfile api tests to ensure that data is available after every operation. |
181 |
def verify_file(f): |
182 |
versions = f.versions() |
|
183 |
self.assertTrue('r0' in versions) |
|
184 |
self.assertTrue('r1' in versions) |
|
185 |
self.assertEquals(f.get_lines('r0'), ['a\n', 'b\n']) |
|
186 |
self.assertEquals(f.get_text('r0'), 'a\nb\n') |
|
187 |
self.assertEquals(f.get_lines('r1'), ['b\n', 'c\n']) |
|
1563.2.18
by Robert Collins
get knit repositories really using knits for text storage. |
188 |
self.assertEqual(2, len(f)) |
189 |
self.assertEqual(2, f.num_versions()) |
|
3943.8.1
by Marius Kruger
remove all trailing whitespace from bzr source |
190 |
|
1563.2.9
by Robert Collins
Update versionedfile api tests to ensure that data is available after every operation. |
191 |
self.assertRaises(RevisionNotPresent, |
192 |
f.add_lines, 'r2', ['foo'], []) |
|
193 |
self.assertRaises(RevisionAlreadyPresent, |
|
194 |
f.add_lines, 'r1', [], []) |
|
195 |
verify_file(f) |
|
1666.1.6
by Robert Collins
Make knit the default format. |
196 |
# this checks that reopen with create=True does not break anything.
|
197 |
f = self.reopen_file(create=True) |
|
1563.2.9
by Robert Collins
Update versionedfile api tests to ensure that data is available after every operation. |
198 |
verify_file(f) |
1563.2.1
by Robert Collins
Merge in a variation of the versionedfile api from versioned-file. |
199 |
|
1596.2.32
by Robert Collins
Reduce re-extraction of texts during weave to knit joins by providing a memoisation facility. |
200 |
def test_adds_with_parent_texts(self): |
201 |
f = self.get_file() |
|
202 |
parent_texts = {} |
|
2776.1.1
by Robert Collins
* The ``add_lines`` methods on ``VersionedFile`` implementations has changed |
203 |
_, _, parent_texts['r0'] = f.add_lines('r0', [], ['a\n', 'b\n']) |
1596.2.32
by Robert Collins
Reduce re-extraction of texts during weave to knit joins by providing a memoisation facility. |
204 |
try: |
2776.1.1
by Robert Collins
* The ``add_lines`` methods on ``VersionedFile`` implementations has changed |
205 |
_, _, parent_texts['r1'] = f.add_lines_with_ghosts('r1', |
206 |
['r0', 'ghost'], ['b\n', 'c\n'], parent_texts=parent_texts) |
|
1596.2.32
by Robert Collins
Reduce re-extraction of texts during weave to knit joins by providing a memoisation facility. |
207 |
except NotImplementedError: |
208 |
# if the format doesn't support ghosts, just add normally.
|
|
2776.1.1
by Robert Collins
* The ``add_lines`` methods on ``VersionedFile`` implementations has changed |
209 |
_, _, parent_texts['r1'] = f.add_lines('r1', |
210 |
['r0'], ['b\n', 'c\n'], parent_texts=parent_texts) |
|
1596.2.32
by Robert Collins
Reduce re-extraction of texts during weave to knit joins by providing a memoisation facility. |
211 |
f.add_lines('r2', ['r1'], ['c\n', 'd\n'], parent_texts=parent_texts) |
212 |
self.assertNotEqual(None, parent_texts['r0']) |
|
213 |
self.assertNotEqual(None, parent_texts['r1']) |
|
214 |
def verify_file(f): |
|
215 |
versions = f.versions() |
|
216 |
self.assertTrue('r0' in versions) |
|
217 |
self.assertTrue('r1' in versions) |
|
218 |
self.assertTrue('r2' in versions) |
|
219 |
self.assertEquals(f.get_lines('r0'), ['a\n', 'b\n']) |
|
220 |
self.assertEquals(f.get_lines('r1'), ['b\n', 'c\n']) |
|
221 |
self.assertEquals(f.get_lines('r2'), ['c\n', 'd\n']) |
|
222 |
self.assertEqual(3, f.num_versions()) |
|
223 |
origins = f.annotate('r1') |
|
224 |
self.assertEquals(origins[0][0], 'r0') |
|
225 |
self.assertEquals(origins[1][0], 'r1') |
|
226 |
origins = f.annotate('r2') |
|
227 |
self.assertEquals(origins[0][0], 'r1') |
|
228 |
self.assertEquals(origins[1][0], 'r2') |
|
229 |
||
230 |
verify_file(f) |
|
231 |
f = self.reopen_file() |
|
232 |
verify_file(f) |
|
233 |
||
2805.6.7
by Robert Collins
Review feedback. |
234 |
def test_add_unicode_content(self): |
3943.8.1
by Marius Kruger
remove all trailing whitespace from bzr source |
235 |
# unicode content is not permitted in versioned files.
|
2805.6.7
by Robert Collins
Review feedback. |
236 |
# versioned files version sequences of bytes only.
|
237 |
vf = self.get_file() |
|
238 |
self.assertRaises(errors.BzrBadParameterUnicode, |
|
239 |
vf.add_lines, 'a', [], ['a\n', u'b\n', 'c\n']) |
|
240 |
self.assertRaises( |
|
241 |
(errors.BzrBadParameterUnicode, NotImplementedError), |
|
242 |
vf.add_lines_with_ghosts, 'a', [], ['a\n', u'b\n', 'c\n']) |
|
243 |
||
2520.4.150
by Aaron Bentley
Test that non-Weave uses left_matching_blocks for add_lines |
244 |
def test_add_follows_left_matching_blocks(self): |
245 |
"""If we change left_matching_blocks, delta changes
|
|
246 |
||
247 |
Note: There are multiple correct deltas in this case, because
|
|
248 |
we start with 1 "a" and we get 3.
|
|
249 |
"""
|
|
250 |
vf = self.get_file() |
|
251 |
if isinstance(vf, WeaveFile): |
|
252 |
raise TestSkipped("WeaveFile ignores left_matching_blocks") |
|
253 |
vf.add_lines('1', [], ['a\n']) |
|
254 |
vf.add_lines('2', ['1'], ['a\n', 'a\n', 'a\n'], |
|
255 |
left_matching_blocks=[(0, 0, 1), (1, 3, 0)]) |
|
2794.1.2
by Robert Collins
Nuke versioned file add/get delta support, allowing easy simplification of unannotated Content, reducing memory copies and friction during commit on unannotated texts. |
256 |
self.assertEqual(['a\n', 'a\n', 'a\n'], vf.get_lines('2')) |
2520.4.150
by Aaron Bentley
Test that non-Weave uses left_matching_blocks for add_lines |
257 |
vf.add_lines('3', ['1'], ['a\n', 'a\n', 'a\n'], |
258 |
left_matching_blocks=[(0, 2, 1), (1, 3, 0)]) |
|
2794.1.2
by Robert Collins
Nuke versioned file add/get delta support, allowing easy simplification of unannotated Content, reducing memory copies and friction during commit on unannotated texts. |
259 |
self.assertEqual(['a\n', 'a\n', 'a\n'], vf.get_lines('3')) |
2520.4.150
by Aaron Bentley
Test that non-Weave uses left_matching_blocks for add_lines |
260 |
|
2805.6.7
by Robert Collins
Review feedback. |
261 |
def test_inline_newline_throws(self): |
262 |
# \r characters are not permitted in lines being added
|
|
263 |
vf = self.get_file() |
|
3943.8.1
by Marius Kruger
remove all trailing whitespace from bzr source |
264 |
self.assertRaises(errors.BzrBadParameterContainsNewline, |
2805.6.7
by Robert Collins
Review feedback. |
265 |
vf.add_lines, 'a', [], ['a\n\n']) |
266 |
self.assertRaises( |
|
267 |
(errors.BzrBadParameterContainsNewline, NotImplementedError), |
|
268 |
vf.add_lines_with_ghosts, 'a', [], ['a\n\n']) |
|
269 |
# but inline CR's are allowed
|
|
270 |
vf.add_lines('a', [], ['a\r\n']) |
|
271 |
try: |
|
272 |
vf.add_lines_with_ghosts('b', [], ['a\r\n']) |
|
273 |
except NotImplementedError: |
|
274 |
pass
|
|
275 |
||
2229.2.1
by Aaron Bentley
Reject reserved ids in versiondfile, tree, branch and repository |
276 |
def test_add_reserved(self): |
277 |
vf = self.get_file() |
|
278 |
self.assertRaises(errors.ReservedId, |
|
279 |
vf.add_lines, 'a:', [], ['a\n', 'b\n', 'c\n']) |
|
280 |
||
2794.1.1
by Robert Collins
Allow knits to be instructed not to add a text based on a sha, for commit. |
281 |
def test_add_lines_nostoresha(self): |
282 |
"""When nostore_sha is supplied using old content raises."""
|
|
283 |
vf = self.get_file() |
|
284 |
empty_text = ('a', []) |
|
285 |
sample_text_nl = ('b', ["foo\n", "bar\n"]) |
|
286 |
sample_text_no_nl = ('c', ["foo\n", "bar"]) |
|
287 |
shas = [] |
|
288 |
for version, lines in (empty_text, sample_text_nl, sample_text_no_nl): |
|
289 |
sha, _, _ = vf.add_lines(version, [], lines) |
|
290 |
shas.append(sha) |
|
291 |
# we now have a copy of all the lines in the vf.
|
|
292 |
for sha, (version, lines) in zip( |
|
293 |
shas, (empty_text, sample_text_nl, sample_text_no_nl)): |
|
294 |
self.assertRaises(errors.ExistingContent, |
|
295 |
vf.add_lines, version + "2", [], lines, |
|
296 |
nostore_sha=sha) |
|
297 |
# and no new version should have been added.
|
|
298 |
self.assertRaises(errors.RevisionNotPresent, vf.get_lines, |
|
299 |
version + "2") |
|
300 |
||
2803.1.1
by Robert Collins
Fix typo in ghosts version of test_add_lines_nostoresha. |
301 |
def test_add_lines_with_ghosts_nostoresha(self): |
2794.1.1
by Robert Collins
Allow knits to be instructed not to add a text based on a sha, for commit. |
302 |
"""When nostore_sha is supplied using old content raises."""
|
303 |
vf = self.get_file() |
|
304 |
empty_text = ('a', []) |
|
305 |
sample_text_nl = ('b', ["foo\n", "bar\n"]) |
|
306 |
sample_text_no_nl = ('c', ["foo\n", "bar"]) |
|
307 |
shas = [] |
|
308 |
for version, lines in (empty_text, sample_text_nl, sample_text_no_nl): |
|
309 |
sha, _, _ = vf.add_lines(version, [], lines) |
|
310 |
shas.append(sha) |
|
311 |
# we now have a copy of all the lines in the vf.
|
|
312 |
# is the test applicable to this vf implementation?
|
|
313 |
try: |
|
314 |
vf.add_lines_with_ghosts('d', [], []) |
|
315 |
except NotImplementedError: |
|
316 |
raise TestSkipped("add_lines_with_ghosts is optional") |
|
317 |
for sha, (version, lines) in zip( |
|
318 |
shas, (empty_text, sample_text_nl, sample_text_no_nl)): |
|
319 |
self.assertRaises(errors.ExistingContent, |
|
320 |
vf.add_lines_with_ghosts, version + "2", [], lines, |
|
321 |
nostore_sha=sha) |
|
322 |
# and no new version should have been added.
|
|
323 |
self.assertRaises(errors.RevisionNotPresent, vf.get_lines, |
|
324 |
version + "2") |
|
325 |
||
2776.1.1
by Robert Collins
* The ``add_lines`` methods on ``VersionedFile`` implementations has changed |
326 |
def test_add_lines_return_value(self): |
327 |
# add_lines should return the sha1 and the text size.
|
|
328 |
vf = self.get_file() |
|
329 |
empty_text = ('a', []) |
|
330 |
sample_text_nl = ('b', ["foo\n", "bar\n"]) |
|
331 |
sample_text_no_nl = ('c', ["foo\n", "bar"]) |
|
332 |
# check results for the three cases:
|
|
333 |
for version, lines in (empty_text, sample_text_nl, sample_text_no_nl): |
|
334 |
# the first two elements are the same for all versioned files:
|
|
335 |
# - the digest and the size of the text. For some versioned files
|
|
336 |
# additional data is returned in additional tuple elements.
|
|
337 |
result = vf.add_lines(version, [], lines) |
|
338 |
self.assertEqual(3, len(result)) |
|
339 |
self.assertEqual((osutils.sha_strings(lines), sum(map(len, lines))), |
|
340 |
result[0:2]) |
|
341 |
# parents should not affect the result:
|
|
342 |
lines = sample_text_nl[1] |
|
343 |
self.assertEqual((osutils.sha_strings(lines), sum(map(len, lines))), |
|
344 |
vf.add_lines('d', ['b', 'c'], lines)[0:2]) |
|
345 |
||
2229.2.1
by Aaron Bentley
Reject reserved ids in versiondfile, tree, branch and repository |
346 |
def test_get_reserved(self): |
347 |
vf = self.get_file() |
|
348 |
self.assertRaises(errors.ReservedId, vf.get_texts, ['b:']) |
|
349 |
self.assertRaises(errors.ReservedId, vf.get_lines, 'b:') |
|
350 |
self.assertRaises(errors.ReservedId, vf.get_text, 'b:') |
|
351 |
||
3468.2.4
by Martin Pool
Test and fix #234748 problems in trailing newline diffs |
352 |
def test_add_unchanged_last_line_noeol_snapshot(self): |
353 |
"""Add a text with an unchanged last line with no eol should work."""
|
|
354 |
# Test adding this in a number of chain lengths; because the interface
|
|
355 |
# for VersionedFile does not allow forcing a specific chain length, we
|
|
356 |
# just use a small base to get the first snapshot, then a much longer
|
|
357 |
# first line for the next add (which will make the third add snapshot)
|
|
358 |
# and so on. 20 has been chosen as an aribtrary figure - knits use 200
|
|
359 |
# as a capped delta length, but ideally we would have some way of
|
|
360 |
# tuning the test to the store (e.g. keep going until a snapshot
|
|
361 |
# happens).
|
|
362 |
for length in range(20): |
|
363 |
version_lines = {} |
|
364 |
vf = self.get_file('case-%d' % length) |
|
365 |
prefix = 'step-%d' |
|
366 |
parents = [] |
|
367 |
for step in range(length): |
|
368 |
version = prefix % step |
|
369 |
lines = (['prelude \n'] * step) + ['line'] |
|
370 |
vf.add_lines(version, parents, lines) |
|
371 |
version_lines[version] = lines |
|
372 |
parents = [version] |
|
373 |
vf.add_lines('no-eol', parents, ['line']) |
|
374 |
vf.get_texts(version_lines.keys()) |
|
375 |
self.assertEqualDiff('line', vf.get_text('no-eol')) |
|
376 |
||
377 |
def test_get_texts_eol_variation(self): |
|
378 |
# similar to the failure in <http://bugs.launchpad.net/234748>
|
|
379 |
vf = self.get_file() |
|
380 |
sample_text_nl = ["line\n"] |
|
381 |
sample_text_no_nl = ["line"] |
|
382 |
versions = [] |
|
383 |
version_lines = {} |
|
384 |
parents = [] |
|
385 |
for i in range(4): |
|
386 |
version = 'v%d' % i |
|
387 |
if i % 2: |
|
388 |
lines = sample_text_nl |
|
389 |
else: |
|
390 |
lines = sample_text_no_nl |
|
391 |
# left_matching blocks is an internal api; it operates on the
|
|
392 |
# *internal* representation for a knit, which is with *all* lines
|
|
393 |
# being normalised to end with \n - even the final line in a no_nl
|
|
394 |
# file. Using it here ensures that a broken internal implementation
|
|
395 |
# (which is what this test tests) will generate a correct line
|
|
396 |
# delta (which is to say, an empty delta).
|
|
397 |
vf.add_lines(version, parents, lines, |
|
398 |
left_matching_blocks=[(0, 0, 1)]) |
|
399 |
parents = [version] |
|
400 |
versions.append(version) |
|
401 |
version_lines[version] = lines |
|
402 |
vf.check() |
|
403 |
vf.get_texts(versions) |
|
404 |
vf.get_texts(reversed(versions)) |
|
405 |
||
3460.2.1
by Robert Collins
* Inserting a bundle which changes the contents of a file with no trailing |
406 |
def test_add_lines_with_matching_blocks_noeol_last_line(self): |
407 |
"""Add a text with an unchanged last line with no eol should work."""
|
|
408 |
from bzrlib import multiparent |
|
409 |
# Hand verified sha1 of the text we're adding.
|
|
410 |
sha1 = '6a1d115ec7b60afb664dc14890b5af5ce3c827a4' |
|
411 |
# Create a mpdiff which adds a new line before the trailing line, and
|
|
412 |
# reuse the last line unaltered (which can cause annotation reuse).
|
|
413 |
# Test adding this in two situations:
|
|
414 |
# On top of a new insertion
|
|
415 |
vf = self.get_file('fulltext') |
|
416 |
vf.add_lines('noeol', [], ['line']) |
|
417 |
vf.add_lines('noeol2', ['noeol'], ['newline\n', 'line'], |
|
418 |
left_matching_blocks=[(0, 1, 1)]) |
|
419 |
self.assertEqualDiff('newline\nline', vf.get_text('noeol2')) |
|
420 |
# On top of a delta
|
|
421 |
vf = self.get_file('delta') |
|
422 |
vf.add_lines('base', [], ['line']) |
|
423 |
vf.add_lines('noeol', ['base'], ['prelude\n', 'line']) |
|
424 |
vf.add_lines('noeol2', ['noeol'], ['newline\n', 'line'], |
|
425 |
left_matching_blocks=[(1, 1, 1)]) |
|
426 |
self.assertEqualDiff('newline\nline', vf.get_text('noeol2')) |
|
427 |
||
2520.4.85
by Aaron Bentley
Get all test passing (which just proves there aren't enough tests!) |
428 |
def test_make_mpdiffs(self): |
2520.4.3
by Aaron Bentley
Implement plain strategy for extracting and installing multiparent diffs |
429 |
from bzrlib import multiparent |
430 |
vf = self.get_file('foo') |
|
431 |
sha1s = self._setup_for_deltas(vf) |
|
432 |
new_vf = self.get_file('bar') |
|
433 |
for version in multiparent.topo_iter(vf): |
|
2520.4.85
by Aaron Bentley
Get all test passing (which just proves there aren't enough tests!) |
434 |
mpdiff = vf.make_mpdiffs([version])[0] |
3287.5.2
by Robert Collins
Deprecate VersionedFile.get_parents, breaking pulling from a ghost containing knit or pack repository to weaves, which improves correctness and allows simplification of core code. |
435 |
new_vf.add_mpdiffs([(version, vf.get_parent_map([version])[version], |
3350.8.3
by Robert Collins
VF.get_sha1s needed changing to be stackable. |
436 |
vf.get_sha1s([version])[version], mpdiff)]) |
2520.4.3
by Aaron Bentley
Implement plain strategy for extracting and installing multiparent diffs |
437 |
self.assertEqualDiff(vf.get_text(version), |
438 |
new_vf.get_text(version)) |
|
439 |
||
3453.3.2
by John Arbash Meinel
Add a test case for the first loop, unable to find a way to trigger the second loop |
440 |
def test_make_mpdiffs_with_ghosts(self): |
441 |
vf = self.get_file('foo') |
|
3453.3.4
by John Arbash Meinel
Skip the new test for old weave formats that don't support ghosts |
442 |
try: |
443 |
vf.add_lines_with_ghosts('text', ['ghost'], ['line\n']) |
|
444 |
except NotImplementedError: |
|
445 |
# old Weave formats do not allow ghosts
|
|
446 |
return
|
|
3453.3.2
by John Arbash Meinel
Add a test case for the first loop, unable to find a way to trigger the second loop |
447 |
self.assertRaises(errors.RevisionNotPresent, vf.make_mpdiffs, ['ghost']) |
448 |
||
1596.2.38
by Robert Collins
rollback from using deltas to using fulltexts - deltas need more work to be ready. |
449 |
def _setup_for_deltas(self, f): |
2794.1.2
by Robert Collins
Nuke versioned file add/get delta support, allowing easy simplification of unannotated Content, reducing memory copies and friction during commit on unannotated texts. |
450 |
self.assertFalse(f.has_version('base')) |
1596.2.36
by Robert Collins
add a get_delta api to versioned_file. |
451 |
# add texts that should trip the knit maximum delta chain threshold
|
452 |
# as well as doing parallel chains of data in knits.
|
|
453 |
# this is done by two chains of 25 insertions
|
|
454 |
f.add_lines('base', [], ['line\n']) |
|
1596.2.38
by Robert Collins
rollback from using deltas to using fulltexts - deltas need more work to be ready. |
455 |
f.add_lines('noeol', ['base'], ['line']) |
456 |
# detailed eol tests:
|
|
457 |
# shared last line with parent no-eol
|
|
458 |
f.add_lines('noeolsecond', ['noeol'], ['line\n', 'line']) |
|
459 |
# differing last line with parent, both no-eol
|
|
460 |
f.add_lines('noeolnotshared', ['noeolsecond'], ['line\n', 'phone']) |
|
461 |
# add eol following a noneol parent, change content
|
|
462 |
f.add_lines('eol', ['noeol'], ['phone\n']) |
|
463 |
# add eol following a noneol parent, no change content
|
|
464 |
f.add_lines('eolline', ['noeol'], ['line\n']) |
|
465 |
# noeol with no parents:
|
|
466 |
f.add_lines('noeolbase', [], ['line']) |
|
467 |
# noeol preceeding its leftmost parent in the output:
|
|
468 |
# this is done by making it a merge of two parents with no common
|
|
3943.8.1
by Marius Kruger
remove all trailing whitespace from bzr source |
469 |
# anestry: noeolbase and noeol with the
|
1596.2.38
by Robert Collins
rollback from using deltas to using fulltexts - deltas need more work to be ready. |
470 |
# later-inserted parent the leftmost.
|
471 |
f.add_lines('eolbeforefirstparent', ['noeolbase', 'noeol'], ['line']) |
|
472 |
# two identical eol texts
|
|
473 |
f.add_lines('noeoldup', ['noeol'], ['line']) |
|
1596.2.36
by Robert Collins
add a get_delta api to versioned_file. |
474 |
next_parent = 'base' |
475 |
text_name = 'chain1-' |
|
476 |
text = ['line\n'] |
|
477 |
sha1s = {0 :'da6d3141cb4a5e6f464bf6e0518042ddc7bfd079', |
|
478 |
1 :'45e21ea146a81ea44a821737acdb4f9791c8abe7', |
|
479 |
2 :'e1f11570edf3e2a070052366c582837a4fe4e9fa', |
|
480 |
3 :'26b4b8626da827088c514b8f9bbe4ebf181edda1', |
|
481 |
4 :'e28a5510be25ba84d31121cff00956f9970ae6f6', |
|
482 |
5 :'d63ec0ce22e11dcf65a931b69255d3ac747a318d', |
|
483 |
6 :'2c2888d288cb5e1d98009d822fedfe6019c6a4ea', |
|
484 |
7 :'95c14da9cafbf828e3e74a6f016d87926ba234ab', |
|
485 |
8 :'779e9a0b28f9f832528d4b21e17e168c67697272', |
|
486 |
9 :'1f8ff4e5c6ff78ac106fcfe6b1e8cb8740ff9a8f', |
|
487 |
10:'131a2ae712cf51ed62f143e3fbac3d4206c25a05', |
|
488 |
11:'c5a9d6f520d2515e1ec401a8f8a67e6c3c89f199', |
|
489 |
12:'31a2286267f24d8bedaa43355f8ad7129509ea85', |
|
490 |
13:'dc2a7fe80e8ec5cae920973973a8ee28b2da5e0a', |
|
491 |
14:'2c4b1736566b8ca6051e668de68650686a3922f2', |
|
492 |
15:'5912e4ecd9b0c07be4d013e7e2bdcf9323276cde', |
|
493 |
16:'b0d2e18d3559a00580f6b49804c23fea500feab3', |
|
494 |
17:'8e1d43ad72f7562d7cb8f57ee584e20eb1a69fc7', |
|
495 |
18:'5cf64a3459ae28efa60239e44b20312d25b253f3', |
|
496 |
19:'1ebed371807ba5935958ad0884595126e8c4e823', |
|
497 |
20:'2aa62a8b06fb3b3b892a3292a068ade69d5ee0d3', |
|
498 |
21:'01edc447978004f6e4e962b417a4ae1955b6fe5d', |
|
499 |
22:'d8d8dc49c4bf0bab401e0298bb5ad827768618bb', |
|
500 |
23:'c21f62b1c482862983a8ffb2b0c64b3451876e3f', |
|
501 |
24:'c0593fe795e00dff6b3c0fe857a074364d5f04fc', |
|
502 |
25:'dd1a1cf2ba9cc225c3aff729953e6364bf1d1855', |
|
503 |
}
|
|
504 |
for depth in range(26): |
|
505 |
new_version = text_name + '%s' % depth |
|
506 |
text = text + ['line\n'] |
|
507 |
f.add_lines(new_version, [next_parent], text) |
|
508 |
next_parent = new_version |
|
509 |
next_parent = 'base' |
|
510 |
text_name = 'chain2-' |
|
511 |
text = ['line\n'] |
|
512 |
for depth in range(26): |
|
513 |
new_version = text_name + '%s' % depth |
|
514 |
text = text + ['line\n'] |
|
515 |
f.add_lines(new_version, [next_parent], text) |
|
516 |
next_parent = new_version |
|
1596.2.38
by Robert Collins
rollback from using deltas to using fulltexts - deltas need more work to be ready. |
517 |
return sha1s |
1596.2.37
by Robert Collins
Switch to delta based content copying in the generic versioned file copier. |
518 |
|
1563.2.1
by Robert Collins
Merge in a variation of the versionedfile api from versioned-file. |
519 |
def test_ancestry(self): |
520 |
f = self.get_file() |
|
1563.2.29
by Robert Collins
Remove all but fetch references to repository.revision_store. |
521 |
self.assertEqual([], f.get_ancestry([])) |
1563.2.1
by Robert Collins
Merge in a variation of the versionedfile api from versioned-file. |
522 |
f.add_lines('r0', [], ['a\n', 'b\n']) |
523 |
f.add_lines('r1', ['r0'], ['b\n', 'c\n']) |
|
524 |
f.add_lines('r2', ['r0'], ['b\n', 'c\n']) |
|
525 |
f.add_lines('r3', ['r2'], ['b\n', 'c\n']) |
|
526 |
f.add_lines('rM', ['r1', 'r2'], ['b\n', 'c\n']) |
|
1563.2.29
by Robert Collins
Remove all but fetch references to repository.revision_store. |
527 |
self.assertEqual([], f.get_ancestry([])) |
1563.2.35
by Robert Collins
cleanup deprecation warnings and finish conversion so the inventory is knit based too. |
528 |
versions = f.get_ancestry(['rM']) |
529 |
# there are some possibilities:
|
|
530 |
# r0 r1 r2 rM r3
|
|
531 |
# r0 r1 r2 r3 rM
|
|
532 |
# etc
|
|
533 |
# so we check indexes
|
|
534 |
r0 = versions.index('r0') |
|
535 |
r1 = versions.index('r1') |
|
536 |
r2 = versions.index('r2') |
|
537 |
self.assertFalse('r3' in versions) |
|
538 |
rM = versions.index('rM') |
|
539 |
self.assertTrue(r0 < r1) |
|
540 |
self.assertTrue(r0 < r2) |
|
541 |
self.assertTrue(r1 < rM) |
|
542 |
self.assertTrue(r2 < rM) |
|
1563.2.1
by Robert Collins
Merge in a variation of the versionedfile api from versioned-file. |
543 |
|
544 |
self.assertRaises(RevisionNotPresent, |
|
545 |
f.get_ancestry, ['rM', 'rX']) |
|
1594.2.21
by Robert Collins
Teach versioned files to prevent mutation after finishing. |
546 |
|
2530.1.1
by Aaron Bentley
Make topological sorting optional for get_ancestry |
547 |
self.assertEqual(set(f.get_ancestry('rM')), |
548 |
set(f.get_ancestry('rM', topo_sorted=False))) |
|
549 |
||
1594.2.21
by Robert Collins
Teach versioned files to prevent mutation after finishing. |
550 |
def test_mutate_after_finish(self): |
3316.2.3
by Robert Collins
Remove manual notification of transaction finishing on versioned files. |
551 |
self._transaction = 'before' |
1594.2.21
by Robert Collins
Teach versioned files to prevent mutation after finishing. |
552 |
f = self.get_file() |
3316.2.3
by Robert Collins
Remove manual notification of transaction finishing on versioned files. |
553 |
self._transaction = 'after' |
1594.2.21
by Robert Collins
Teach versioned files to prevent mutation after finishing. |
554 |
self.assertRaises(errors.OutSideTransaction, f.add_lines, '', [], []) |
555 |
self.assertRaises(errors.OutSideTransaction, f.add_lines_with_ghosts, '', [], []) |
|
3943.8.1
by Marius Kruger
remove all trailing whitespace from bzr source |
556 |
|
1563.2.15
by Robert Collins
remove the weavestore assumptions about the number and nature of files it manages. |
557 |
def test_copy_to(self): |
558 |
f = self.get_file() |
|
559 |
f.add_lines('0', [], ['a\n']) |
|
560 |
t = MemoryTransport() |
|
561 |
f.copy_to('foo', t) |
|
3316.2.3
by Robert Collins
Remove manual notification of transaction finishing on versioned files. |
562 |
for suffix in self.get_factory().get_suffixes(): |
1563.2.15
by Robert Collins
remove the weavestore assumptions about the number and nature of files it manages. |
563 |
self.assertTrue(t.has('foo' + suffix)) |
564 |
||
565 |
def test_get_suffixes(self): |
|
566 |
f = self.get_file() |
|
567 |
# and should be a list
|
|
3316.2.3
by Robert Collins
Remove manual notification of transaction finishing on versioned files. |
568 |
self.assertTrue(isinstance(self.get_factory().get_suffixes(), list)) |
1563.2.15
by Robert Collins
remove the weavestore assumptions about the number and nature of files it manages. |
569 |
|
3287.5.1
by Robert Collins
Add VersionedFile.get_parent_map. |
570 |
def test_get_parent_map(self): |
571 |
f = self.get_file() |
|
572 |
f.add_lines('r0', [], ['a\n', 'b\n']) |
|
573 |
self.assertEqual( |
|
574 |
{'r0':()}, f.get_parent_map(['r0'])) |
|
575 |
f.add_lines('r1', ['r0'], ['a\n', 'b\n']) |
|
576 |
self.assertEqual( |
|
577 |
{'r1':('r0',)}, f.get_parent_map(['r1'])) |
|
578 |
self.assertEqual( |
|
579 |
{'r0':(), |
|
580 |
'r1':('r0',)}, |
|
581 |
f.get_parent_map(['r0', 'r1'])) |
|
582 |
f.add_lines('r2', [], ['a\n', 'b\n']) |
|
583 |
f.add_lines('r3', [], ['a\n', 'b\n']) |
|
584 |
f.add_lines('m', ['r0', 'r1', 'r2', 'r3'], ['a\n', 'b\n']) |
|
585 |
self.assertEqual( |
|
586 |
{'m':('r0', 'r1', 'r2', 'r3')}, f.get_parent_map(['m'])) |
|
587 |
self.assertEqual({}, f.get_parent_map('y')) |
|
588 |
self.assertEqual( |
|
589 |
{'r0':(), |
|
590 |
'r1':('r0',)}, |
|
591 |
f.get_parent_map(['r0', 'y', 'r1'])) |
|
592 |
||
1563.2.1
by Robert Collins
Merge in a variation of the versionedfile api from versioned-file. |
593 |
def test_annotate(self): |
594 |
f = self.get_file() |
|
595 |
f.add_lines('r0', [], ['a\n', 'b\n']) |
|
596 |
f.add_lines('r1', ['r0'], ['c\n', 'b\n']) |
|
597 |
origins = f.annotate('r1') |
|
598 |
self.assertEquals(origins[0][0], 'r1') |
|
599 |
self.assertEquals(origins[1][0], 'r0') |
|
600 |
||
601 |
self.assertRaises(RevisionNotPresent, |
|
602 |
f.annotate, 'foo') |
|
603 |
||
1563.2.6
by Robert Collins
Start check tests for knits (pending), and remove dead code. |
604 |
def test_detection(self): |
605 |
# Test weaves detect corruption.
|
|
606 |
#
|
|
607 |
# Weaves contain a checksum of their texts.
|
|
608 |
# When a text is extracted, this checksum should be
|
|
609 |
# verified.
|
|
610 |
||
611 |
w = self.get_file_corrupted_text() |
|
612 |
||
613 |
self.assertEqual('hello\n', w.get_text('v1')) |
|
5582.9.20
by Jelmer Vernooij
remove some of the weave changes. |
614 |
self.assertRaises(errors.WeaveInvalidChecksum, w.get_text, 'v2') |
615 |
self.assertRaises(errors.WeaveInvalidChecksum, w.get_lines, 'v2') |
|
616 |
self.assertRaises(errors.WeaveInvalidChecksum, w.check) |
|
1563.2.6
by Robert Collins
Start check tests for knits (pending), and remove dead code. |
617 |
|
618 |
w = self.get_file_corrupted_checksum() |
|
619 |
||
620 |
self.assertEqual('hello\n', w.get_text('v1')) |
|
5582.9.20
by Jelmer Vernooij
remove some of the weave changes. |
621 |
self.assertRaises(errors.WeaveInvalidChecksum, w.get_text, 'v2') |
622 |
self.assertRaises(errors.WeaveInvalidChecksum, w.get_lines, 'v2') |
|
623 |
self.assertRaises(errors.WeaveInvalidChecksum, w.check) |
|
1563.2.6
by Robert Collins
Start check tests for knits (pending), and remove dead code. |
624 |
|
625 |
def get_file_corrupted_text(self): |
|
626 |
"""Return a versioned file with corrupt text but valid metadata."""
|
|
627 |
raise NotImplementedError(self.get_file_corrupted_text) |
|
628 |
||
1563.2.9
by Robert Collins
Update versionedfile api tests to ensure that data is available after every operation. |
629 |
def reopen_file(self, name='foo'): |
630 |
"""Open the versioned file from disk again."""
|
|
631 |
raise NotImplementedError(self.reopen_file) |
|
632 |
||
1594.2.6
by Robert Collins
Introduce a api specifically for looking at lines in some versions of the inventory, for fileid_involved. |
633 |
def test_iter_lines_added_or_present_in_versions(self): |
634 |
# test that we get at least an equalset of the lines added by
|
|
3943.8.1
by Marius Kruger
remove all trailing whitespace from bzr source |
635 |
# versions in the weave
|
1594.2.6
by Robert Collins
Introduce a api specifically for looking at lines in some versions of the inventory, for fileid_involved. |
636 |
# the ordering here is to make a tree so that dumb searches have
|
637 |
# more changes to muck up.
|
|
2039.1.1
by Aaron Bentley
Clean up progress properly when interrupted during fetch (#54000) |
638 |
|
4961.2.9
by Martin Pool
Rip out most remaining uses of DummyProgressBar |
639 |
class InstrumentedProgress(progress.ProgressTask): |
2039.1.1
by Aaron Bentley
Clean up progress properly when interrupted during fetch (#54000) |
640 |
|
641 |
def __init__(self): |
|
4961.2.9
by Martin Pool
Rip out most remaining uses of DummyProgressBar |
642 |
progress.ProgressTask.__init__(self) |
2039.1.1
by Aaron Bentley
Clean up progress properly when interrupted during fetch (#54000) |
643 |
self.updates = [] |
644 |
||
645 |
def update(self, msg=None, current=None, total=None): |
|
646 |
self.updates.append((msg, current, total)) |
|
647 |
||
1594.2.6
by Robert Collins
Introduce a api specifically for looking at lines in some versions of the inventory, for fileid_involved. |
648 |
vf = self.get_file() |
649 |
# add a base to get included
|
|
650 |
vf.add_lines('base', [], ['base\n']) |
|
651 |
# add a ancestor to be included on one side
|
|
652 |
vf.add_lines('lancestor', [], ['lancestor\n']) |
|
653 |
# add a ancestor to be included on the other side
|
|
654 |
vf.add_lines('rancestor', ['base'], ['rancestor\n']) |
|
655 |
# add a child of rancestor with no eofile-nl
|
|
656 |
vf.add_lines('child', ['rancestor'], ['base\n', 'child\n']) |
|
657 |
# add a child of lancestor and base to join the two roots
|
|
658 |
vf.add_lines('otherchild', |
|
659 |
['lancestor', 'base'], |
|
660 |
['base\n', 'lancestor\n', 'otherchild\n']) |
|
2039.1.1
by Aaron Bentley
Clean up progress properly when interrupted during fetch (#54000) |
661 |
def iter_with_versions(versions, expected): |
1594.2.6
by Robert Collins
Introduce a api specifically for looking at lines in some versions of the inventory, for fileid_involved. |
662 |
# now we need to see what lines are returned, and how often.
|
2975.3.1
by Robert Collins
Change (without backwards compatibility) the |
663 |
lines = {} |
2039.1.1
by Aaron Bentley
Clean up progress properly when interrupted during fetch (#54000) |
664 |
progress = InstrumentedProgress() |
1594.2.6
by Robert Collins
Introduce a api specifically for looking at lines in some versions of the inventory, for fileid_involved. |
665 |
# iterate over the lines
|
2975.3.1
by Robert Collins
Change (without backwards compatibility) the |
666 |
for line in vf.iter_lines_added_or_present_in_versions(versions, |
2039.1.1
by Aaron Bentley
Clean up progress properly when interrupted during fetch (#54000) |
667 |
pb=progress): |
2975.3.1
by Robert Collins
Change (without backwards compatibility) the |
668 |
lines.setdefault(line, 0) |
1594.2.6
by Robert Collins
Introduce a api specifically for looking at lines in some versions of the inventory, for fileid_involved. |
669 |
lines[line] += 1 |
2975.3.1
by Robert Collins
Change (without backwards compatibility) the |
670 |
if []!= progress.updates: |
2039.1.2
by Aaron Bentley
Tweak test to avoid catching assert |
671 |
self.assertEqual(expected, progress.updates) |
1594.2.6
by Robert Collins
Introduce a api specifically for looking at lines in some versions of the inventory, for fileid_involved. |
672 |
return lines |
2147.1.3
by John Arbash Meinel
In knit.py we were re-using a variable in 2 loops, causing bogus progress messages to be generated. |
673 |
lines = iter_with_versions(['child', 'otherchild'], |
4103.3.4
by Martin Pool
Update test that depends on progress bar strings |
674 |
[('Walking content', 0, 2), |
675 |
('Walking content', 1, 2), |
|
676 |
('Walking content', 2, 2)]) |
|
1594.2.6
by Robert Collins
Introduce a api specifically for looking at lines in some versions of the inventory, for fileid_involved. |
677 |
# we must see child and otherchild
|
2975.3.1
by Robert Collins
Change (without backwards compatibility) the |
678 |
self.assertTrue(lines[('child\n', 'child')] > 0) |
679 |
self.assertTrue(lines[('otherchild\n', 'otherchild')] > 0) |
|
1594.2.6
by Robert Collins
Introduce a api specifically for looking at lines in some versions of the inventory, for fileid_involved. |
680 |
# we dont care if we got more than that.
|
3943.8.1
by Marius Kruger
remove all trailing whitespace from bzr source |
681 |
|
1594.2.6
by Robert Collins
Introduce a api specifically for looking at lines in some versions of the inventory, for fileid_involved. |
682 |
# test all lines
|
4103.3.4
by Martin Pool
Update test that depends on progress bar strings |
683 |
lines = iter_with_versions(None, [('Walking content', 0, 5), |
684 |
('Walking content', 1, 5), |
|
685 |
('Walking content', 2, 5), |
|
686 |
('Walking content', 3, 5), |
|
687 |
('Walking content', 4, 5), |
|
688 |
('Walking content', 5, 5)]) |
|
1594.2.6
by Robert Collins
Introduce a api specifically for looking at lines in some versions of the inventory, for fileid_involved. |
689 |
# all lines must be seen at least once
|
2975.3.1
by Robert Collins
Change (without backwards compatibility) the |
690 |
self.assertTrue(lines[('base\n', 'base')] > 0) |
691 |
self.assertTrue(lines[('lancestor\n', 'lancestor')] > 0) |
|
692 |
self.assertTrue(lines[('rancestor\n', 'rancestor')] > 0) |
|
693 |
self.assertTrue(lines[('child\n', 'child')] > 0) |
|
694 |
self.assertTrue(lines[('otherchild\n', 'otherchild')] > 0) |
|
1594.2.7
by Robert Collins
Add versionedfile.fix_parents api for correcting data post hoc. |
695 |
|
1594.2.8
by Robert Collins
add ghost aware apis to knits. |
696 |
def test_add_lines_with_ghosts(self): |
697 |
# some versioned file formats allow lines to be added with parent
|
|
698 |
# information that is > than that in the format. Formats that do
|
|
699 |
# not support this need to raise NotImplementedError on the
|
|
700 |
# add_lines_with_ghosts api.
|
|
701 |
vf = self.get_file() |
|
702 |
# add a revision with ghost parents
|
|
2249.5.12
by John Arbash Meinel
Change the APIs for VersionedFile, Store, and some of Repository into utf-8 |
703 |
# The preferred form is utf8, but we should translate when needed
|
704 |
parent_id_unicode = u'b\xbfse' |
|
705 |
parent_id_utf8 = parent_id_unicode.encode('utf8') |
|
1594.2.8
by Robert Collins
add ghost aware apis to knits. |
706 |
try: |
2309.4.7
by John Arbash Meinel
Update VersionedFile tests to ensure that they can take Unicode, |
707 |
vf.add_lines_with_ghosts('notbxbfse', [parent_id_utf8], []) |
1594.2.8
by Robert Collins
add ghost aware apis to knits. |
708 |
except NotImplementedError: |
709 |
# check the other ghost apis are also not implemented
|
|
710 |
self.assertRaises(NotImplementedError, vf.get_ancestry_with_ghosts, ['foo']) |
|
711 |
self.assertRaises(NotImplementedError, vf.get_parents_with_ghosts, 'foo') |
|
712 |
return
|
|
2150.2.1
by Robert Collins
Correctly decode utf8 revision ids from knits when parsing, fixes a regression where a unicode revision id is stored correctly, but then indexed by the utf8 value on the next invocation of bzr, rather than the unicode value. |
713 |
vf = self.reopen_file() |
1594.2.8
by Robert Collins
add ghost aware apis to knits. |
714 |
# test key graph related apis: getncestry, _graph, get_parents
|
715 |
# has_version
|
|
716 |
# - these are ghost unaware and must not be reflect ghosts
|
|
2249.5.12
by John Arbash Meinel
Change the APIs for VersionedFile, Store, and some of Repository into utf-8 |
717 |
self.assertEqual(['notbxbfse'], vf.get_ancestry('notbxbfse')) |
718 |
self.assertFalse(vf.has_version(parent_id_utf8)) |
|
1594.2.8
by Robert Collins
add ghost aware apis to knits. |
719 |
# we have _with_ghost apis to give us ghost information.
|
2249.5.12
by John Arbash Meinel
Change the APIs for VersionedFile, Store, and some of Repository into utf-8 |
720 |
self.assertEqual([parent_id_utf8, 'notbxbfse'], vf.get_ancestry_with_ghosts(['notbxbfse'])) |
721 |
self.assertEqual([parent_id_utf8], vf.get_parents_with_ghosts('notbxbfse')) |
|
1594.2.8
by Robert Collins
add ghost aware apis to knits. |
722 |
# if we add something that is a ghost of another, it should correct the
|
723 |
# results of the prior apis
|
|
2858.2.1
by Martin Pool
Remove most calls to safe_file_id and safe_revision_id. |
724 |
vf.add_lines(parent_id_utf8, [], []) |
2249.5.12
by John Arbash Meinel
Change the APIs for VersionedFile, Store, and some of Repository into utf-8 |
725 |
self.assertEqual([parent_id_utf8, 'notbxbfse'], vf.get_ancestry(['notbxbfse'])) |
3287.5.2
by Robert Collins
Deprecate VersionedFile.get_parents, breaking pulling from a ghost containing knit or pack repository to weaves, which improves correctness and allows simplification of core code. |
726 |
self.assertEqual({'notbxbfse':(parent_id_utf8,)}, |
727 |
vf.get_parent_map(['notbxbfse'])) |
|
2249.5.12
by John Arbash Meinel
Change the APIs for VersionedFile, Store, and some of Repository into utf-8 |
728 |
self.assertTrue(vf.has_version(parent_id_utf8)) |
1594.2.8
by Robert Collins
add ghost aware apis to knits. |
729 |
# we have _with_ghost apis to give us ghost information.
|
2858.2.1
by Martin Pool
Remove most calls to safe_file_id and safe_revision_id. |
730 |
self.assertEqual([parent_id_utf8, 'notbxbfse'], |
731 |
vf.get_ancestry_with_ghosts(['notbxbfse'])) |
|
2249.5.12
by John Arbash Meinel
Change the APIs for VersionedFile, Store, and some of Repository into utf-8 |
732 |
self.assertEqual([parent_id_utf8], vf.get_parents_with_ghosts('notbxbfse')) |
1594.2.8
by Robert Collins
add ghost aware apis to knits. |
733 |
|
1594.2.9
by Robert Collins
Teach Knit repositories how to handle ghosts without corrupting at all. |
734 |
def test_add_lines_with_ghosts_after_normal_revs(self): |
735 |
# some versioned file formats allow lines to be added with parent
|
|
736 |
# information that is > than that in the format. Formats that do
|
|
737 |
# not support this need to raise NotImplementedError on the
|
|
738 |
# add_lines_with_ghosts api.
|
|
739 |
vf = self.get_file() |
|
740 |
# probe for ghost support
|
|
741 |
try: |
|
3287.6.5
by Robert Collins
Deprecate VersionedFile.has_ghost. |
742 |
vf.add_lines_with_ghosts('base', [], ['line\n', 'line_b\n']) |
1594.2.9
by Robert Collins
Teach Knit repositories how to handle ghosts without corrupting at all. |
743 |
except NotImplementedError: |
744 |
return
|
|
745 |
vf.add_lines_with_ghosts('references_ghost', |
|
746 |
['base', 'a_ghost'], |
|
747 |
['line\n', 'line_b\n', 'line_c\n']) |
|
748 |
origins = vf.annotate('references_ghost') |
|
749 |
self.assertEquals(('base', 'line\n'), origins[0]) |
|
750 |
self.assertEquals(('base', 'line_b\n'), origins[1]) |
|
751 |
self.assertEquals(('references_ghost', 'line_c\n'), origins[2]) |
|
1594.2.23
by Robert Collins
Test versioned file storage handling of clean/dirty status for accessed versioned files. |
752 |
|
753 |
def test_readonly_mode(self): |
|
5609.9.4
by Vincent Ladeuil
Use self.get_transport instead of transport.get_transport where possible. |
754 |
t = self.get_transport() |
1594.2.23
by Robert Collins
Test versioned file storage handling of clean/dirty status for accessed versioned files. |
755 |
factory = self.get_factory() |
5273.1.7
by Vincent Ladeuil
No more use of the get_transport imported *symbol*, all uses are through |
756 |
vf = factory('id', t, 0777, create=True, access_mode='w') |
757 |
vf = factory('id', t, access_mode='r') |
|
1594.2.23
by Robert Collins
Test versioned file storage handling of clean/dirty status for accessed versioned files. |
758 |
self.assertRaises(errors.ReadOnlyError, vf.add_lines, 'base', [], []) |
759 |
self.assertRaises(errors.ReadOnlyError, |
|
760 |
vf.add_lines_with_ghosts, |
|
761 |
'base', |
|
762 |
[],
|
|
763 |
[])
|
|
3943.8.1
by Marius Kruger
remove all trailing whitespace from bzr source |
764 |
|
3316.2.9
by Robert Collins
* ``VersionedFile.get_sha1`` is deprecated, please use |
765 |
def test_get_sha1s(self): |
1666.1.6
by Robert Collins
Make knit the default format. |
766 |
# check the sha1 data is available
|
767 |
vf = self.get_file() |
|
768 |
# a simple file
|
|
769 |
vf.add_lines('a', [], ['a\n']) |
|
770 |
# the same file, different metadata
|
|
771 |
vf.add_lines('b', ['a'], ['a\n']) |
|
772 |
# a file differing only in last newline.
|
|
773 |
vf.add_lines('c', [], ['a']) |
|
3350.8.3
by Robert Collins
VF.get_sha1s needed changing to be stackable. |
774 |
self.assertEqual({ |
775 |
'a': '3f786850e387550fdab836ed7e6dc881de23001b', |
|
776 |
'c': '86f7e437faa5a7fce15d1ddcb9eaeaea377667b8', |
|
777 |
'b': '3f786850e387550fdab836ed7e6dc881de23001b', |
|
778 |
},
|
|
779 |
vf.get_sha1s(['a', 'c', 'b'])) |
|
3943.8.1
by Marius Kruger
remove all trailing whitespace from bzr source |
780 |
|
1563.2.1
by Robert Collins
Merge in a variation of the versionedfile api from versioned-file. |
781 |
|
2535.3.1
by Andrew Bennetts
Add get_format_signature to VersionedFile |
782 |
class TestWeave(TestCaseWithMemoryTransport, VersionedFileTestMixIn): |
1563.2.1
by Robert Collins
Merge in a variation of the versionedfile api from versioned-file. |
783 |
|
784 |
def get_file(self, name='foo'): |
|
5609.9.4
by Vincent Ladeuil
Use self.get_transport instead of transport.get_transport where possible. |
785 |
return WeaveFile(name, self.get_transport(), |
5273.1.7
by Vincent Ladeuil
No more use of the get_transport imported *symbol*, all uses are through |
786 |
create=True, |
787 |
get_scope=self.get_transaction) |
|
1563.2.1
by Robert Collins
Merge in a variation of the versionedfile api from versioned-file. |
788 |
|
1563.2.6
by Robert Collins
Start check tests for knits (pending), and remove dead code. |
789 |
def get_file_corrupted_text(self): |
5609.9.4
by Vincent Ladeuil
Use self.get_transport instead of transport.get_transport where possible. |
790 |
w = WeaveFile('foo', self.get_transport(), |
5273.1.7
by Vincent Ladeuil
No more use of the get_transport imported *symbol*, all uses are through |
791 |
create=True, |
792 |
get_scope=self.get_transaction) |
|
1563.2.13
by Robert Collins
InterVersionedFile implemented. |
793 |
w.add_lines('v1', [], ['hello\n']) |
794 |
w.add_lines('v2', ['v1'], ['hello\n', 'there\n']) |
|
3943.8.1
by Marius Kruger
remove all trailing whitespace from bzr source |
795 |
|
1563.2.6
by Robert Collins
Start check tests for knits (pending), and remove dead code. |
796 |
# We are going to invasively corrupt the text
|
797 |
# Make sure the internals of weave are the same
|
|
798 |
self.assertEqual([('{', 0) |
|
799 |
, 'hello\n' |
|
800 |
, ('}', None) |
|
801 |
, ('{', 1) |
|
802 |
, 'there\n' |
|
803 |
, ('}', None) |
|
804 |
], w._weave) |
|
3943.8.1
by Marius Kruger
remove all trailing whitespace from bzr source |
805 |
|
1563.2.6
by Robert Collins
Start check tests for knits (pending), and remove dead code. |
806 |
self.assertEqual(['f572d396fae9206628714fb2ce00f72e94f2258f' |
807 |
, '90f265c6e75f1c8f9ab76dcf85528352c5f215ef' |
|
808 |
], w._sha1s) |
|
809 |
w.check() |
|
3943.8.1
by Marius Kruger
remove all trailing whitespace from bzr source |
810 |
|
1563.2.6
by Robert Collins
Start check tests for knits (pending), and remove dead code. |
811 |
# Corrupted
|
812 |
w._weave[4] = 'There\n' |
|
813 |
return w |
|
814 |
||
815 |
def get_file_corrupted_checksum(self): |
|
816 |
w = self.get_file_corrupted_text() |
|
817 |
# Corrected
|
|
818 |
w._weave[4] = 'there\n' |
|
819 |
self.assertEqual('hello\nthere\n', w.get_text('v2')) |
|
3943.8.1
by Marius Kruger
remove all trailing whitespace from bzr source |
820 |
|
1563.2.6
by Robert Collins
Start check tests for knits (pending), and remove dead code. |
821 |
#Invalid checksum, first digit changed
|
822 |
w._sha1s[1] = 'f0f265c6e75f1c8f9ab76dcf85528352c5f215ef' |
|
823 |
return w |
|
824 |
||
1666.1.6
by Robert Collins
Make knit the default format. |
825 |
def reopen_file(self, name='foo', create=False): |
5609.9.4
by Vincent Ladeuil
Use self.get_transport instead of transport.get_transport where possible. |
826 |
return WeaveFile(name, self.get_transport(), |
5273.1.7
by Vincent Ladeuil
No more use of the get_transport imported *symbol*, all uses are through |
827 |
create=create, |
828 |
get_scope=self.get_transaction) |
|
1563.2.9
by Robert Collins
Update versionedfile api tests to ensure that data is available after every operation. |
829 |
|
1563.2.25
by Robert Collins
Merge in upstream. |
830 |
def test_no_implicit_create(self): |
831 |
self.assertRaises(errors.NoSuchFile, |
|
832 |
WeaveFile, |
|
833 |
'foo', |
|
5609.9.4
by Vincent Ladeuil
Use self.get_transport instead of transport.get_transport where possible. |
834 |
self.get_transport(), |
3316.2.3
by Robert Collins
Remove manual notification of transaction finishing on versioned files. |
835 |
get_scope=self.get_transaction) |
1563.2.25
by Robert Collins
Merge in upstream. |
836 |
|
1594.2.23
by Robert Collins
Test versioned file storage handling of clean/dirty status for accessed versioned files. |
837 |
def get_factory(self): |
838 |
return WeaveFile |
|
839 |
||
1563.2.1
by Robert Collins
Merge in a variation of the versionedfile api from versioned-file. |
840 |
|
3062.1.9
by Aaron Bentley
Move PlanMerge into merge and _PlanMergeVersionedFile into versionedfile |
841 |
class TestPlanMergeVersionedFile(TestCaseWithMemoryTransport): |
842 |
||
843 |
def setUp(self): |
|
844 |
TestCaseWithMemoryTransport.setUp(self) |
|
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
845 |
mapper = PrefixMapper() |
846 |
factory = make_file_factory(True, mapper) |
|
847 |
self.vf1 = factory(self.get_transport('root-1')) |
|
848 |
self.vf2 = factory(self.get_transport('root-2')) |
|
849 |
self.plan_merge_vf = versionedfile._PlanMergeVersionedFile('root') |
|
850 |
self.plan_merge_vf.fallback_versionedfiles.extend([self.vf1, self.vf2]) |
|
3062.1.9
by Aaron Bentley
Move PlanMerge into merge and _PlanMergeVersionedFile into versionedfile |
851 |
|
852 |
def test_add_lines(self): |
|
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
853 |
self.plan_merge_vf.add_lines(('root', 'a:'), [], []) |
854 |
self.assertRaises(ValueError, self.plan_merge_vf.add_lines, |
|
855 |
('root', 'a'), [], []) |
|
856 |
self.assertRaises(ValueError, self.plan_merge_vf.add_lines, |
|
857 |
('root', 'a:'), None, []) |
|
858 |
self.assertRaises(ValueError, self.plan_merge_vf.add_lines, |
|
859 |
('root', 'a:'), [], None) |
|
3062.1.9
by Aaron Bentley
Move PlanMerge into merge and _PlanMergeVersionedFile into versionedfile |
860 |
|
861 |
def setup_abcde(self): |
|
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
862 |
self.vf1.add_lines(('root', 'A'), [], ['a']) |
863 |
self.vf1.add_lines(('root', 'B'), [('root', 'A')], ['b']) |
|
864 |
self.vf2.add_lines(('root', 'C'), [], ['c']) |
|
865 |
self.vf2.add_lines(('root', 'D'), [('root', 'C')], ['d']) |
|
866 |
self.plan_merge_vf.add_lines(('root', 'E:'), |
|
867 |
[('root', 'B'), ('root', 'D')], ['e']) |
|
3062.1.9
by Aaron Bentley
Move PlanMerge into merge and _PlanMergeVersionedFile into versionedfile |
868 |
|
869 |
def test_get_parents(self): |
|
870 |
self.setup_abcde() |
|
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
871 |
self.assertEqual({('root', 'B'):(('root', 'A'),)}, |
872 |
self.plan_merge_vf.get_parent_map([('root', 'B')])) |
|
873 |
self.assertEqual({('root', 'D'):(('root', 'C'),)}, |
|
874 |
self.plan_merge_vf.get_parent_map([('root', 'D')])) |
|
875 |
self.assertEqual({('root', 'E:'):(('root', 'B'),('root', 'D'))}, |
|
876 |
self.plan_merge_vf.get_parent_map([('root', 'E:')])) |
|
877 |
self.assertEqual({}, |
|
878 |
self.plan_merge_vf.get_parent_map([('root', 'F')])) |
|
3287.5.2
by Robert Collins
Deprecate VersionedFile.get_parents, breaking pulling from a ghost containing knit or pack repository to weaves, which improves correctness and allows simplification of core code. |
879 |
self.assertEqual({ |
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
880 |
('root', 'B'):(('root', 'A'),), |
881 |
('root', 'D'):(('root', 'C'),), |
|
882 |
('root', 'E:'):(('root', 'B'),('root', 'D')), |
|
883 |
},
|
|
884 |
self.plan_merge_vf.get_parent_map( |
|
885 |
[('root', 'B'), ('root', 'D'), ('root', 'E:'), ('root', 'F')])) |
|
3062.1.9
by Aaron Bentley
Move PlanMerge into merge and _PlanMergeVersionedFile into versionedfile |
886 |
|
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
887 |
def test_get_record_stream(self): |
3062.1.9
by Aaron Bentley
Move PlanMerge into merge and _PlanMergeVersionedFile into versionedfile |
888 |
self.setup_abcde() |
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
889 |
def get_record(suffix): |
890 |
return self.plan_merge_vf.get_record_stream( |
|
891 |
[('root', suffix)], 'unordered', True).next() |
|
892 |
self.assertEqual('a', get_record('A').get_bytes_as('fulltext')) |
|
893 |
self.assertEqual('c', get_record('C').get_bytes_as('fulltext')) |
|
894 |
self.assertEqual('e', get_record('E:').get_bytes_as('fulltext')) |
|
895 |
self.assertEqual('absent', get_record('F').storage_kind) |
|
1666.1.1
by Robert Collins
Add trivial http-using test for versioned files. |
896 |
|
897 |
||
898 |
class TestReadonlyHttpMixin(object): |
|
899 |
||
3316.2.3
by Robert Collins
Remove manual notification of transaction finishing on versioned files. |
900 |
def get_transaction(self): |
901 |
return 1 |
|
902 |
||
1666.1.1
by Robert Collins
Add trivial http-using test for versioned files. |
903 |
def test_readonly_http_works(self): |
904 |
# we should be able to read from http with a versioned file.
|
|
905 |
vf = self.get_file() |
|
1666.1.6
by Robert Collins
Make knit the default format. |
906 |
# try an empty file access
|
5273.1.7
by Vincent Ladeuil
No more use of the get_transport imported *symbol*, all uses are through |
907 |
readonly_vf = self.get_factory()('foo', transport.get_transport( |
5229.1.10
by Vincent Ladeuil
Attempt to fix the last (so far :) windows failing test. |
908 |
self.get_readonly_url('.'))) |
1666.1.6
by Robert Collins
Make knit the default format. |
909 |
self.assertEqual([], readonly_vf.versions()) |
5229.1.13
by Vincent Ladeuil
Eager tests are bad, we really want two tests here. |
910 |
|
911 |
def test_readonly_http_works_with_feeling(self): |
|
912 |
# we should be able to read from http with a versioned file.
|
|
913 |
vf = self.get_file() |
|
1666.1.6
by Robert Collins
Make knit the default format. |
914 |
# now with feeling.
|
1666.1.1
by Robert Collins
Add trivial http-using test for versioned files. |
915 |
vf.add_lines('1', [], ['a\n']) |
916 |
vf.add_lines('2', ['1'], ['b\n', 'a\n']) |
|
5273.1.7
by Vincent Ladeuil
No more use of the get_transport imported *symbol*, all uses are through |
917 |
readonly_vf = self.get_factory()('foo', transport.get_transport( |
5229.1.10
by Vincent Ladeuil
Attempt to fix the last (so far :) windows failing test. |
918 |
self.get_readonly_url('.'))) |
1666.1.6
by Robert Collins
Make knit the default format. |
919 |
self.assertEqual(['1', '2'], vf.versions()) |
5229.1.14
by Vincent Ladeuil
Fixed as per jam's review :) |
920 |
self.assertEqual(['1', '2'], readonly_vf.versions()) |
1666.1.1
by Robert Collins
Add trivial http-using test for versioned files. |
921 |
for version in readonly_vf.versions(): |
922 |
readonly_vf.get_lines(version) |
|
923 |
||
924 |
||
925 |
class TestWeaveHTTP(TestCaseWithWebserver, TestReadonlyHttpMixin): |
|
926 |
||
927 |
def get_file(self): |
|
5609.9.4
by Vincent Ladeuil
Use self.get_transport instead of transport.get_transport where possible. |
928 |
return WeaveFile('foo', self.get_transport(), |
5273.1.7
by Vincent Ladeuil
No more use of the get_transport imported *symbol*, all uses are through |
929 |
create=True, |
930 |
get_scope=self.get_transaction) |
|
1666.1.1
by Robert Collins
Add trivial http-using test for versioned files. |
931 |
|
932 |
def get_factory(self): |
|
933 |
return WeaveFile |
|
934 |
||
935 |
||
1664.2.9
by Aaron Bentley
Ported weave merge test to versionedfile |
936 |
class MergeCasesMixin(object): |
937 |
||
938 |
def doMerge(self, base, a, b, mp): |
|
939 |
from cStringIO import StringIO |
|
940 |
from textwrap import dedent |
|
941 |
||
942 |
def addcrlf(x): |
|
943 |
return x + '\n' |
|
3943.8.1
by Marius Kruger
remove all trailing whitespace from bzr source |
944 |
|
1664.2.9
by Aaron Bentley
Ported weave merge test to versionedfile |
945 |
w = self.get_file() |
946 |
w.add_lines('text0', [], map(addcrlf, base)) |
|
947 |
w.add_lines('text1', ['text0'], map(addcrlf, a)) |
|
948 |
w.add_lines('text2', ['text0'], map(addcrlf, b)) |
|
949 |
||
950 |
self.log_contents(w) |
|
951 |
||
952 |
self.log('merge plan:') |
|
953 |
p = list(w.plan_merge('text1', 'text2')) |
|
954 |
for state, line in p: |
|
955 |
if line: |
|
956 |
self.log('%12s | %s' % (state, line[:-1])) |
|
957 |
||
958 |
self.log('merge:') |
|
959 |
mt = StringIO() |
|
960 |
mt.writelines(w.weave_merge(p)) |
|
961 |
mt.seek(0) |
|
962 |
self.log(mt.getvalue()) |
|
963 |
||
964 |
mp = map(addcrlf, mp) |
|
965 |
self.assertEqual(mt.readlines(), mp) |
|
3943.8.1
by Marius Kruger
remove all trailing whitespace from bzr source |
966 |
|
967 |
||
1664.2.9
by Aaron Bentley
Ported weave merge test to versionedfile |
968 |
def testOneInsert(self): |
969 |
self.doMerge([], |
|
970 |
['aa'], |
|
971 |
[],
|
|
972 |
['aa']) |
|
973 |
||
974 |
def testSeparateInserts(self): |
|
975 |
self.doMerge(['aaa', 'bbb', 'ccc'], |
|
976 |
['aaa', 'xxx', 'bbb', 'ccc'], |
|
977 |
['aaa', 'bbb', 'yyy', 'ccc'], |
|
978 |
['aaa', 'xxx', 'bbb', 'yyy', 'ccc']) |
|
979 |
||
980 |
def testSameInsert(self): |
|
981 |
self.doMerge(['aaa', 'bbb', 'ccc'], |
|
982 |
['aaa', 'xxx', 'bbb', 'ccc'], |
|
983 |
['aaa', 'xxx', 'bbb', 'yyy', 'ccc'], |
|
984 |
['aaa', 'xxx', 'bbb', 'yyy', 'ccc']) |
|
985 |
overlappedInsertExpected = ['aaa', 'xxx', 'yyy', 'bbb'] |
|
986 |
def testOverlappedInsert(self): |
|
987 |
self.doMerge(['aaa', 'bbb'], |
|
988 |
['aaa', 'xxx', 'yyy', 'bbb'], |
|
989 |
['aaa', 'xxx', 'bbb'], self.overlappedInsertExpected) |
|
990 |
||
3943.8.1
by Marius Kruger
remove all trailing whitespace from bzr source |
991 |
# really it ought to reduce this to
|
1664.2.9
by Aaron Bentley
Ported weave merge test to versionedfile |
992 |
# ['aaa', 'xxx', 'yyy', 'bbb']
|
993 |
||
994 |
||
995 |
def testClashReplace(self): |
|
996 |
self.doMerge(['aaa'], |
|
997 |
['xxx'], |
|
998 |
['yyy', 'zzz'], |
|
3943.8.1
by Marius Kruger
remove all trailing whitespace from bzr source |
999 |
['<<<<<<< ', 'xxx', '=======', 'yyy', 'zzz', |
1664.2.9
by Aaron Bentley
Ported weave merge test to versionedfile |
1000 |
'>>>>>>> ']) |
1001 |
||
1002 |
def testNonClashInsert1(self): |
|
1003 |
self.doMerge(['aaa'], |
|
1004 |
['xxx', 'aaa'], |
|
1005 |
['yyy', 'zzz'], |
|
3943.8.1
by Marius Kruger
remove all trailing whitespace from bzr source |
1006 |
['<<<<<<< ', 'xxx', 'aaa', '=======', 'yyy', 'zzz', |
1664.2.9
by Aaron Bentley
Ported weave merge test to versionedfile |
1007 |
'>>>>>>> ']) |
1008 |
||
1009 |
def testNonClashInsert2(self): |
|
1010 |
self.doMerge(['aaa'], |
|
1011 |
['aaa'], |
|
1012 |
['yyy', 'zzz'], |
|
1013 |
['yyy', 'zzz']) |
|
1014 |
||
1015 |
||
1016 |
def testDeleteAndModify(self): |
|
1017 |
"""Clashing delete and modification.
|
|
1018 |
||
1019 |
If one side modifies a region and the other deletes it then
|
|
1020 |
there should be a conflict with one side blank.
|
|
1021 |
"""
|
|
1022 |
||
1023 |
#######################################
|
|
1024 |
# skippd, not working yet
|
|
1025 |
return
|
|
3943.8.1
by Marius Kruger
remove all trailing whitespace from bzr source |
1026 |
|
1664.2.9
by Aaron Bentley
Ported weave merge test to versionedfile |
1027 |
self.doMerge(['aaa', 'bbb', 'ccc'], |
1028 |
['aaa', 'ddd', 'ccc'], |
|
1029 |
['aaa', 'ccc'], |
|
1030 |
['<<<<<<<< ', 'aaa', '=======', '>>>>>>> ', 'ccc']) |
|
1031 |
||
1032 |
def _test_merge_from_strings(self, base, a, b, expected): |
|
1033 |
w = self.get_file() |
|
1034 |
w.add_lines('text0', [], base.splitlines(True)) |
|
1035 |
w.add_lines('text1', ['text0'], a.splitlines(True)) |
|
1036 |
w.add_lines('text2', ['text0'], b.splitlines(True)) |
|
1037 |
self.log('merge plan:') |
|
1038 |
p = list(w.plan_merge('text1', 'text2')) |
|
1039 |
for state, line in p: |
|
1040 |
if line: |
|
1041 |
self.log('%12s | %s' % (state, line[:-1])) |
|
1042 |
self.log('merge result:') |
|
1043 |
result_text = ''.join(w.weave_merge(p)) |
|
1044 |
self.log(result_text) |
|
1045 |
self.assertEqualDiff(result_text, expected) |
|
1046 |
||
1047 |
def test_weave_merge_conflicts(self): |
|
1048 |
# does weave merge properly handle plans that end with unchanged?
|
|
1049 |
result = ''.join(self.get_file().weave_merge([('new-a', 'hello\n')])) |
|
1050 |
self.assertEqual(result, 'hello\n') |
|
1051 |
||
1052 |
def test_deletion_extended(self): |
|
1053 |
"""One side deletes, the other deletes more.
|
|
1054 |
"""
|
|
1055 |
base = """\ |
|
1056 |
line 1
|
|
1057 |
line 2
|
|
1058 |
line 3
|
|
1059 |
"""
|
|
1060 |
a = """\ |
|
1061 |
line 1
|
|
1062 |
line 2
|
|
1063 |
"""
|
|
1064 |
b = """\ |
|
1065 |
line 1
|
|
1066 |
"""
|
|
1067 |
result = """\ |
|
1068 |
line 1
|
|
4312.1.3
by John Arbash Meinel
Different sides deleting different amounts is now a conflict. |
1069 |
<<<<<<<\x20 |
1070 |
line 2
|
|
1071 |
=======
|
|
1072 |
>>>>>>>\x20 |
|
1664.2.9
by Aaron Bentley
Ported weave merge test to versionedfile |
1073 |
"""
|
1074 |
self._test_merge_from_strings(base, a, b, result) |
|
1075 |
||
1076 |
def test_deletion_overlap(self): |
|
1077 |
"""Delete overlapping regions with no other conflict.
|
|
1078 |
||
3943.8.1
by Marius Kruger
remove all trailing whitespace from bzr source |
1079 |
Arguably it'd be better to treat these as agreement, rather than
|
1664.2.9
by Aaron Bentley
Ported weave merge test to versionedfile |
1080 |
conflict, but for now conflict is safer.
|
1081 |
"""
|
|
1082 |
base = """\ |
|
1083 |
start context
|
|
1084 |
int a() {} |
|
1085 |
int b() {} |
|
1086 |
int c() {} |
|
1087 |
end context
|
|
1088 |
"""
|
|
1089 |
a = """\ |
|
1090 |
start context
|
|
1091 |
int a() {} |
|
1092 |
end context
|
|
1093 |
"""
|
|
1094 |
b = """\ |
|
1095 |
start context
|
|
1096 |
int c() {} |
|
1097 |
end context
|
|
1098 |
"""
|
|
1099 |
result = """\ |
|
1100 |
start context
|
|
3943.8.2
by Marius Kruger
fix tests relying on trailing whitespace by replacing it with \x20. |
1101 |
<<<<<<<\x20 |
1664.2.9
by Aaron Bentley
Ported weave merge test to versionedfile |
1102 |
int a() {} |
1103 |
=======
|
|
1104 |
int c() {} |
|
3943.8.2
by Marius Kruger
fix tests relying on trailing whitespace by replacing it with \x20. |
1105 |
>>>>>>>\x20 |
1664.2.9
by Aaron Bentley
Ported weave merge test to versionedfile |
1106 |
end context
|
1107 |
"""
|
|
1108 |
self._test_merge_from_strings(base, a, b, result) |
|
1109 |
||
1110 |
def test_agreement_deletion(self): |
|
1111 |
"""Agree to delete some lines, without conflicts."""
|
|
1112 |
base = """\ |
|
1113 |
start context
|
|
1114 |
base line 1
|
|
1115 |
base line 2
|
|
1116 |
end context
|
|
1117 |
"""
|
|
1118 |
a = """\ |
|
1119 |
start context
|
|
1120 |
base line 1
|
|
1121 |
end context
|
|
1122 |
"""
|
|
1123 |
b = """\ |
|
1124 |
start context
|
|
1125 |
base line 1
|
|
1126 |
end context
|
|
1127 |
"""
|
|
1128 |
result = """\ |
|
1129 |
start context
|
|
1130 |
base line 1
|
|
1131 |
end context
|
|
1132 |
"""
|
|
1133 |
self._test_merge_from_strings(base, a, b, result) |
|
1134 |
||
1135 |
def test_sync_on_deletion(self): |
|
1136 |
"""Specific case of merge where we can synchronize incorrectly.
|
|
3943.8.1
by Marius Kruger
remove all trailing whitespace from bzr source |
1137 |
|
1664.2.9
by Aaron Bentley
Ported weave merge test to versionedfile |
1138 |
A previous version of the weave merge concluded that the two versions
|
1139 |
agreed on deleting line 2, and this could be a synchronization point.
|
|
3943.8.1
by Marius Kruger
remove all trailing whitespace from bzr source |
1140 |
Line 1 was then considered in isolation, and thought to be deleted on
|
1664.2.9
by Aaron Bentley
Ported weave merge test to versionedfile |
1141 |
both sides.
|
1142 |
||
1143 |
It's better to consider the whole thing as a disagreement region.
|
|
1144 |
"""
|
|
1145 |
base = """\ |
|
1146 |
start context
|
|
1147 |
base line 1
|
|
1148 |
base line 2
|
|
1149 |
end context
|
|
1150 |
"""
|
|
1151 |
a = """\ |
|
1152 |
start context
|
|
1153 |
base line 1
|
|
1154 |
a's replacement line 2
|
|
1155 |
end context
|
|
1156 |
"""
|
|
1157 |
b = """\ |
|
1158 |
start context
|
|
1159 |
b replaces
|
|
1160 |
both lines
|
|
1161 |
end context
|
|
1162 |
"""
|
|
1163 |
result = """\ |
|
1164 |
start context
|
|
3943.8.2
by Marius Kruger
fix tests relying on trailing whitespace by replacing it with \x20. |
1165 |
<<<<<<<\x20 |
1664.2.9
by Aaron Bentley
Ported weave merge test to versionedfile |
1166 |
base line 1
|
1167 |
a's replacement line 2
|
|
1168 |
=======
|
|
1169 |
b replaces
|
|
1170 |
both lines
|
|
3943.8.2
by Marius Kruger
fix tests relying on trailing whitespace by replacing it with \x20. |
1171 |
>>>>>>>\x20 |
1664.2.9
by Aaron Bentley
Ported weave merge test to versionedfile |
1172 |
end context
|
1173 |
"""
|
|
1174 |
self._test_merge_from_strings(base, a, b, result) |
|
1175 |
||
1176 |
||
2535.3.1
by Andrew Bennetts
Add get_format_signature to VersionedFile |
1177 |
class TestWeaveMerge(TestCaseWithMemoryTransport, MergeCasesMixin): |
1664.2.9
by Aaron Bentley
Ported weave merge test to versionedfile |
1178 |
|
1179 |
def get_file(self, name='foo'): |
|
5609.9.4
by Vincent Ladeuil
Use self.get_transport instead of transport.get_transport where possible. |
1180 |
return WeaveFile(name, self.get_transport(), |
5273.1.7
by Vincent Ladeuil
No more use of the get_transport imported *symbol*, all uses are through |
1181 |
create=True) |
1664.2.9
by Aaron Bentley
Ported weave merge test to versionedfile |
1182 |
|
1183 |
def log_contents(self, w): |
|
1184 |
self.log('weave is:') |
|
1185 |
tmpf = StringIO() |
|
1186 |
write_weave(w, tmpf) |
|
1187 |
self.log(tmpf.getvalue()) |
|
1188 |
||
3943.8.1
by Marius Kruger
remove all trailing whitespace from bzr source |
1189 |
overlappedInsertExpected = ['aaa', '<<<<<<< ', 'xxx', 'yyy', '=======', |
1664.2.9
by Aaron Bentley
Ported weave merge test to versionedfile |
1190 |
'xxx', '>>>>>>> ', 'bbb'] |
3350.3.4
by Robert Collins
Finish adapters for annotated knits to unannotated knits and full texts. |
1191 |
|
1192 |
||
1193 |
class TestContentFactoryAdaption(TestCaseWithMemoryTransport): |
|
1194 |
||
1195 |
def test_select_adaptor(self): |
|
3350.3.7
by Robert Collins
Create a registry of versioned file record adapters. |
1196 |
"""Test expected adapters exist."""
|
1197 |
# One scenario for each lookup combination we expect to use.
|
|
1198 |
# Each is source_kind, requested_kind, adapter class
|
|
1199 |
scenarios = [ |
|
1200 |
('knit-delta-gz', 'fulltext', _mod_knit.DeltaPlainToFullText), |
|
1201 |
('knit-ft-gz', 'fulltext', _mod_knit.FTPlainToFullText), |
|
1202 |
('knit-annotated-delta-gz', 'knit-delta-gz', |
|
1203 |
_mod_knit.DeltaAnnotatedToUnannotated), |
|
1204 |
('knit-annotated-delta-gz', 'fulltext', |
|
1205 |
_mod_knit.DeltaAnnotatedToFullText), |
|
1206 |
('knit-annotated-ft-gz', 'knit-ft-gz', |
|
1207 |
_mod_knit.FTAnnotatedToUnannotated), |
|
1208 |
('knit-annotated-ft-gz', 'fulltext', |
|
1209 |
_mod_knit.FTAnnotatedToFullText), |
|
1210 |
]
|
|
1211 |
for source, requested, klass in scenarios: |
|
1212 |
adapter_factory = versionedfile.adapter_registry.get( |
|
1213 |
(source, requested)) |
|
1214 |
adapter = adapter_factory(None) |
|
1215 |
self.assertIsInstance(adapter, klass) |
|
3350.3.4
by Robert Collins
Finish adapters for annotated knits to unannotated knits and full texts. |
1216 |
|
3350.3.5
by Robert Collins
Create adapters from plain compressed knit content. |
1217 |
def get_knit(self, annotated=True): |
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
1218 |
mapper = ConstantMapper('knit') |
1219 |
transport = self.get_transport() |
|
1220 |
return make_file_factory(annotated, mapper)(transport) |
|
3350.3.4
by Robert Collins
Finish adapters for annotated knits to unannotated knits and full texts. |
1221 |
|
1222 |
def helpGetBytes(self, f, ft_adapter, delta_adapter): |
|
3350.3.22
by Robert Collins
Review feedback. |
1223 |
"""Grab the interested adapted texts for tests."""
|
3350.3.4
by Robert Collins
Finish adapters for annotated knits to unannotated knits and full texts. |
1224 |
# origin is a fulltext
|
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
1225 |
entries = f.get_record_stream([('origin',)], 'unordered', False) |
3350.3.4
by Robert Collins
Finish adapters for annotated knits to unannotated knits and full texts. |
1226 |
base = entries.next() |
4005.3.1
by Robert Collins
Change the signature on VersionedFiles adapters to allow less typing and more flexability inside adapters. |
1227 |
ft_data = ft_adapter.get_bytes(base) |
3350.3.4
by Robert Collins
Finish adapters for annotated knits to unannotated knits and full texts. |
1228 |
# merged is both a delta and multiple parents.
|
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
1229 |
entries = f.get_record_stream([('merged',)], 'unordered', False) |
3350.3.4
by Robert Collins
Finish adapters for annotated knits to unannotated knits and full texts. |
1230 |
merged = entries.next() |
4005.3.1
by Robert Collins
Change the signature on VersionedFiles adapters to allow less typing and more flexability inside adapters. |
1231 |
delta_data = delta_adapter.get_bytes(merged) |
3350.3.4
by Robert Collins
Finish adapters for annotated knits to unannotated knits and full texts. |
1232 |
return ft_data, delta_data |
1233 |
||
1234 |
def test_deannotation_noeol(self): |
|
1235 |
"""Test converting annotated knits to unannotated knits."""
|
|
1236 |
# we need a full text, and a delta
|
|
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
1237 |
f = self.get_knit() |
1238 |
get_diamond_files(f, 1, trailing_eol=False) |
|
3350.3.4
by Robert Collins
Finish adapters for annotated knits to unannotated knits and full texts. |
1239 |
ft_data, delta_data = self.helpGetBytes(f, |
3350.3.7
by Robert Collins
Create a registry of versioned file record adapters. |
1240 |
_mod_knit.FTAnnotatedToUnannotated(None), |
1241 |
_mod_knit.DeltaAnnotatedToUnannotated(None)) |
|
3350.3.4
by Robert Collins
Finish adapters for annotated knits to unannotated knits and full texts. |
1242 |
self.assertEqual( |
1243 |
'version origin 1 b284f94827db1fa2970d9e2014f080413b547a7e\n' |
|
1244 |
'origin\n' |
|
1245 |
'end origin\n', |
|
1246 |
GzipFile(mode='rb', fileobj=StringIO(ft_data)).read()) |
|
1247 |
self.assertEqual( |
|
1248 |
'version merged 4 32c2e79763b3f90e8ccde37f9710b6629c25a796\n' |
|
1249 |
'1,2,3\nleft\nright\nmerged\nend merged\n', |
|
1250 |
GzipFile(mode='rb', fileobj=StringIO(delta_data)).read()) |
|
1251 |
||
1252 |
def test_deannotation(self): |
|
1253 |
"""Test converting annotated knits to unannotated knits."""
|
|
1254 |
# we need a full text, and a delta
|
|
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
1255 |
f = self.get_knit() |
1256 |
get_diamond_files(f, 1) |
|
3350.3.4
by Robert Collins
Finish adapters for annotated knits to unannotated knits and full texts. |
1257 |
ft_data, delta_data = self.helpGetBytes(f, |
3350.3.7
by Robert Collins
Create a registry of versioned file record adapters. |
1258 |
_mod_knit.FTAnnotatedToUnannotated(None), |
1259 |
_mod_knit.DeltaAnnotatedToUnannotated(None)) |
|
3350.3.4
by Robert Collins
Finish adapters for annotated knits to unannotated knits and full texts. |
1260 |
self.assertEqual( |
1261 |
'version origin 1 00e364d235126be43292ab09cb4686cf703ddc17\n' |
|
1262 |
'origin\n' |
|
1263 |
'end origin\n', |
|
1264 |
GzipFile(mode='rb', fileobj=StringIO(ft_data)).read()) |
|
1265 |
self.assertEqual( |
|
1266 |
'version merged 3 ed8bce375198ea62444dc71952b22cfc2b09226d\n' |
|
1267 |
'2,2,2\nright\nmerged\nend merged\n', |
|
1268 |
GzipFile(mode='rb', fileobj=StringIO(delta_data)).read()) |
|
1269 |
||
1270 |
def test_annotated_to_fulltext_no_eol(self): |
|
1271 |
"""Test adapting annotated knits to full texts (for -> weaves)."""
|
|
1272 |
# we need a full text, and a delta
|
|
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
1273 |
f = self.get_knit() |
1274 |
get_diamond_files(f, 1, trailing_eol=False) |
|
3350.3.4
by Robert Collins
Finish adapters for annotated knits to unannotated knits and full texts. |
1275 |
# Reconstructing a full text requires a backing versioned file, and it
|
1276 |
# must have the base lines requested from it.
|
|
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
1277 |
logged_vf = versionedfile.RecordingVersionedFilesDecorator(f) |
3350.3.4
by Robert Collins
Finish adapters for annotated knits to unannotated knits and full texts. |
1278 |
ft_data, delta_data = self.helpGetBytes(f, |
3350.3.7
by Robert Collins
Create a registry of versioned file record adapters. |
1279 |
_mod_knit.FTAnnotatedToFullText(None), |
3350.3.4
by Robert Collins
Finish adapters for annotated knits to unannotated knits and full texts. |
1280 |
_mod_knit.DeltaAnnotatedToFullText(logged_vf)) |
1281 |
self.assertEqual('origin', ft_data) |
|
1282 |
self.assertEqual('base\nleft\nright\nmerged', delta_data) |
|
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
1283 |
self.assertEqual([('get_record_stream', [('left',)], 'unordered', |
1284 |
True)], logged_vf.calls) |
|
3350.3.4
by Robert Collins
Finish adapters for annotated knits to unannotated knits and full texts. |
1285 |
|
1286 |
def test_annotated_to_fulltext(self): |
|
1287 |
"""Test adapting annotated knits to full texts (for -> weaves)."""
|
|
1288 |
# we need a full text, and a delta
|
|
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
1289 |
f = self.get_knit() |
1290 |
get_diamond_files(f, 1) |
|
3350.3.4
by Robert Collins
Finish adapters for annotated knits to unannotated knits and full texts. |
1291 |
# Reconstructing a full text requires a backing versioned file, and it
|
1292 |
# must have the base lines requested from it.
|
|
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
1293 |
logged_vf = versionedfile.RecordingVersionedFilesDecorator(f) |
3350.3.4
by Robert Collins
Finish adapters for annotated knits to unannotated knits and full texts. |
1294 |
ft_data, delta_data = self.helpGetBytes(f, |
3350.3.7
by Robert Collins
Create a registry of versioned file record adapters. |
1295 |
_mod_knit.FTAnnotatedToFullText(None), |
3350.3.4
by Robert Collins
Finish adapters for annotated knits to unannotated knits and full texts. |
1296 |
_mod_knit.DeltaAnnotatedToFullText(logged_vf)) |
1297 |
self.assertEqual('origin\n', ft_data) |
|
1298 |
self.assertEqual('base\nleft\nright\nmerged\n', delta_data) |
|
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
1299 |
self.assertEqual([('get_record_stream', [('left',)], 'unordered', |
1300 |
True)], logged_vf.calls) |
|
3350.3.5
by Robert Collins
Create adapters from plain compressed knit content. |
1301 |
|
1302 |
def test_unannotated_to_fulltext(self): |
|
1303 |
"""Test adapting unannotated knits to full texts.
|
|
3943.8.1
by Marius Kruger
remove all trailing whitespace from bzr source |
1304 |
|
3350.3.5
by Robert Collins
Create adapters from plain compressed knit content. |
1305 |
This is used for -> weaves, and for -> annotated knits.
|
1306 |
"""
|
|
1307 |
# we need a full text, and a delta
|
|
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
1308 |
f = self.get_knit(annotated=False) |
1309 |
get_diamond_files(f, 1) |
|
3350.3.5
by Robert Collins
Create adapters from plain compressed knit content. |
1310 |
# Reconstructing a full text requires a backing versioned file, and it
|
1311 |
# must have the base lines requested from it.
|
|
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
1312 |
logged_vf = versionedfile.RecordingVersionedFilesDecorator(f) |
3350.3.5
by Robert Collins
Create adapters from plain compressed knit content. |
1313 |
ft_data, delta_data = self.helpGetBytes(f, |
3350.3.7
by Robert Collins
Create a registry of versioned file record adapters. |
1314 |
_mod_knit.FTPlainToFullText(None), |
3350.3.5
by Robert Collins
Create adapters from plain compressed knit content. |
1315 |
_mod_knit.DeltaPlainToFullText(logged_vf)) |
1316 |
self.assertEqual('origin\n', ft_data) |
|
1317 |
self.assertEqual('base\nleft\nright\nmerged\n', delta_data) |
|
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
1318 |
self.assertEqual([('get_record_stream', [('left',)], 'unordered', |
1319 |
True)], logged_vf.calls) |
|
3350.3.5
by Robert Collins
Create adapters from plain compressed knit content. |
1320 |
|
3350.3.6
by Robert Collins
Test EOL behaviour of plain knit record adapters. |
1321 |
def test_unannotated_to_fulltext_no_eol(self): |
1322 |
"""Test adapting unannotated knits to full texts.
|
|
3943.8.1
by Marius Kruger
remove all trailing whitespace from bzr source |
1323 |
|
3350.3.6
by Robert Collins
Test EOL behaviour of plain knit record adapters. |
1324 |
This is used for -> weaves, and for -> annotated knits.
|
1325 |
"""
|
|
1326 |
# we need a full text, and a delta
|
|
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
1327 |
f = self.get_knit(annotated=False) |
1328 |
get_diamond_files(f, 1, trailing_eol=False) |
|
3350.3.6
by Robert Collins
Test EOL behaviour of plain knit record adapters. |
1329 |
# Reconstructing a full text requires a backing versioned file, and it
|
1330 |
# must have the base lines requested from it.
|
|
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
1331 |
logged_vf = versionedfile.RecordingVersionedFilesDecorator(f) |
3350.3.6
by Robert Collins
Test EOL behaviour of plain knit record adapters. |
1332 |
ft_data, delta_data = self.helpGetBytes(f, |
3350.3.7
by Robert Collins
Create a registry of versioned file record adapters. |
1333 |
_mod_knit.FTPlainToFullText(None), |
3350.3.6
by Robert Collins
Test EOL behaviour of plain knit record adapters. |
1334 |
_mod_knit.DeltaPlainToFullText(logged_vf)) |
1335 |
self.assertEqual('origin', ft_data) |
|
1336 |
self.assertEqual('base\nleft\nright\nmerged', delta_data) |
|
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
1337 |
self.assertEqual([('get_record_stream', [('left',)], 'unordered', |
1338 |
True)], logged_vf.calls) |
|
3350.3.6
by Robert Collins
Test EOL behaviour of plain knit record adapters. |
1339 |
|
3350.6.1
by Robert Collins
* New ``versionedfile.KeyMapper`` interface to abstract out the access to |
1340 |
|
1341 |
class TestKeyMapper(TestCaseWithMemoryTransport): |
|
1342 |
"""Tests for various key mapping logic."""
|
|
1343 |
||
1344 |
def test_identity_mapper(self): |
|
1345 |
mapper = versionedfile.ConstantMapper("inventory") |
|
1346 |
self.assertEqual("inventory", mapper.map(('foo@ar',))) |
|
1347 |
self.assertEqual("inventory", mapper.map(('quux',))) |
|
1348 |
||
1349 |
def test_prefix_mapper(self): |
|
1350 |
#format5: plain
|
|
1351 |
mapper = versionedfile.PrefixMapper() |
|
1352 |
self.assertEqual("file-id", mapper.map(("file-id", "revision-id"))) |
|
1353 |
self.assertEqual("new-id", mapper.map(("new-id", "revision-id"))) |
|
1354 |
self.assertEqual(('file-id',), mapper.unmap("file-id")) |
|
1355 |
self.assertEqual(('new-id',), mapper.unmap("new-id")) |
|
1356 |
||
1357 |
def test_hash_prefix_mapper(self): |
|
1358 |
#format6: hash + plain
|
|
1359 |
mapper = versionedfile.HashPrefixMapper() |
|
1360 |
self.assertEqual("9b/file-id", mapper.map(("file-id", "revision-id"))) |
|
1361 |
self.assertEqual("45/new-id", mapper.map(("new-id", "revision-id"))) |
|
1362 |
self.assertEqual(('file-id',), mapper.unmap("9b/file-id")) |
|
1363 |
self.assertEqual(('new-id',), mapper.unmap("45/new-id")) |
|
1364 |
||
1365 |
def test_hash_escaped_mapper(self): |
|
1366 |
#knit1: hash + escaped
|
|
1367 |
mapper = versionedfile.HashEscapedPrefixMapper() |
|
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
1368 |
self.assertEqual("88/%2520", mapper.map((" ", "revision-id"))) |
1369 |
self.assertEqual("ed/fil%2545-%2549d", mapper.map(("filE-Id", |
|
1370 |
"revision-id"))) |
|
1371 |
self.assertEqual("88/ne%2557-%2549d", mapper.map(("neW-Id", |
|
1372 |
"revision-id"))) |
|
1373 |
self.assertEqual(('filE-Id',), mapper.unmap("ed/fil%2545-%2549d")) |
|
1374 |
self.assertEqual(('neW-Id',), mapper.unmap("88/ne%2557-%2549d")) |
|
3350.6.2
by Robert Collins
Prepare parameterised test environment. |
1375 |
|
1376 |
||
1377 |
class TestVersionedFiles(TestCaseWithMemoryTransport): |
|
1378 |
"""Tests for the multiple-file variant of VersionedFile."""
|
|
1379 |
||
5559.2.2
by Martin Pool
Change to using standard load_tests_apply_scenarios. |
1380 |
# We want to be sure of behaviour for:
|
1381 |
# weaves prefix layout (weave texts)
|
|
1382 |
# individually named weaves (weave inventories)
|
|
1383 |
# annotated knits - prefix|hash|hash-escape layout, we test the third only
|
|
1384 |
# as it is the most complex mapper.
|
|
1385 |
# individually named knits
|
|
1386 |
# individual no-graph knits in packs (signatures)
|
|
1387 |
# individual graph knits in packs (inventories)
|
|
1388 |
# individual graph nocompression knits in packs (revisions)
|
|
1389 |
# plain text knits in packs (texts)
|
|
1390 |
len_one_scenarios = [ |
|
1391 |
('weave-named', { |
|
1392 |
'cleanup':None, |
|
1393 |
'factory':make_versioned_files_factory(WeaveFile, |
|
1394 |
ConstantMapper('inventory')), |
|
1395 |
'graph':True, |
|
1396 |
'key_length':1, |
|
1397 |
'support_partial_insertion': False, |
|
1398 |
}),
|
|
1399 |
('named-knit', { |
|
1400 |
'cleanup':None, |
|
1401 |
'factory':make_file_factory(False, ConstantMapper('revisions')), |
|
1402 |
'graph':True, |
|
1403 |
'key_length':1, |
|
1404 |
'support_partial_insertion': False, |
|
1405 |
}),
|
|
1406 |
('named-nograph-nodelta-knit-pack', { |
|
1407 |
'cleanup':cleanup_pack_knit, |
|
1408 |
'factory':make_pack_factory(False, False, 1), |
|
1409 |
'graph':False, |
|
1410 |
'key_length':1, |
|
1411 |
'support_partial_insertion': False, |
|
1412 |
}),
|
|
1413 |
('named-graph-knit-pack', { |
|
1414 |
'cleanup':cleanup_pack_knit, |
|
1415 |
'factory':make_pack_factory(True, True, 1), |
|
1416 |
'graph':True, |
|
1417 |
'key_length':1, |
|
1418 |
'support_partial_insertion': True, |
|
1419 |
}),
|
|
1420 |
('named-graph-nodelta-knit-pack', { |
|
1421 |
'cleanup':cleanup_pack_knit, |
|
1422 |
'factory':make_pack_factory(True, False, 1), |
|
1423 |
'graph':True, |
|
1424 |
'key_length':1, |
|
1425 |
'support_partial_insertion': False, |
|
1426 |
}),
|
|
1427 |
('groupcompress-nograph', { |
|
1428 |
'cleanup':groupcompress.cleanup_pack_group, |
|
1429 |
'factory':groupcompress.make_pack_factory(False, False, 1), |
|
1430 |
'graph': False, |
|
1431 |
'key_length':1, |
|
1432 |
'support_partial_insertion':False, |
|
1433 |
}),
|
|
1434 |
]
|
|
1435 |
len_two_scenarios = [ |
|
1436 |
('weave-prefix', { |
|
1437 |
'cleanup':None, |
|
1438 |
'factory':make_versioned_files_factory(WeaveFile, |
|
1439 |
PrefixMapper()), |
|
1440 |
'graph':True, |
|
1441 |
'key_length':2, |
|
1442 |
'support_partial_insertion': False, |
|
1443 |
}),
|
|
1444 |
('annotated-knit-escape', { |
|
1445 |
'cleanup':None, |
|
1446 |
'factory':make_file_factory(True, HashEscapedPrefixMapper()), |
|
1447 |
'graph':True, |
|
1448 |
'key_length':2, |
|
1449 |
'support_partial_insertion': False, |
|
1450 |
}),
|
|
1451 |
('plain-knit-pack', { |
|
1452 |
'cleanup':cleanup_pack_knit, |
|
1453 |
'factory':make_pack_factory(True, True, 2), |
|
1454 |
'graph':True, |
|
1455 |
'key_length':2, |
|
1456 |
'support_partial_insertion': True, |
|
1457 |
}),
|
|
1458 |
('groupcompress', { |
|
1459 |
'cleanup':groupcompress.cleanup_pack_group, |
|
1460 |
'factory':groupcompress.make_pack_factory(True, False, 1), |
|
1461 |
'graph': True, |
|
1462 |
'key_length':1, |
|
1463 |
'support_partial_insertion':False, |
|
1464 |
}),
|
|
1465 |
]
|
|
1466 |
||
1467 |
scenarios = len_one_scenarios + len_two_scenarios |
|
1468 |
||
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
1469 |
def get_versionedfiles(self, relpath='files'): |
1470 |
transport = self.get_transport(relpath) |
|
1471 |
if relpath != '.': |
|
1472 |
transport.mkdir('.') |
|
1473 |
files = self.factory(transport) |
|
1474 |
if self.cleanup is not None: |
|
4985.2.1
by Vincent Ladeuil
Deploy addAttrCleanup on the whole test suite. |
1475 |
self.addCleanup(self.cleanup, files) |
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
1476 |
return files |
1477 |
||
4398.8.8
by John Arbash Meinel
Respond to Andrew's review comments. |
1478 |
def get_simple_key(self, suffix): |
1479 |
"""Return a key for the object under test."""
|
|
1480 |
if self.key_length == 1: |
|
1481 |
return (suffix,) |
|
1482 |
else: |
|
1483 |
return ('FileA',) + (suffix,) |
|
1484 |
||
5816.8.2
by Andrew Bennetts
Add test for without_fallbacks. |
1485 |
def test_add_fallback_implies_without_fallbacks(self): |
1486 |
f = self.get_versionedfiles('files') |
|
1487 |
if getattr(f, 'add_fallback_versioned_files', None) is None: |
|
1488 |
raise TestNotApplicable("%s doesn't support fallbacks" |
|
1489 |
% (f.__class__.__name__,)) |
|
1490 |
g = self.get_versionedfiles('fallback') |
|
1491 |
key_a = self.get_simple_key('a') |
|
1492 |
g.add_lines(key_a, [], ['\n']) |
|
1493 |
f.add_fallback_versioned_files(g) |
|
1494 |
self.assertTrue(key_a in f.get_parent_map([key_a])) |
|
1495 |
self.assertFalse(key_a in f.without_fallbacks().get_parent_map([key_a])) |
|
1496 |
||
4398.8.1
by John Arbash Meinel
Add a VersionedFile.add_text() api. |
1497 |
def test_add_lines(self): |
1498 |
f = self.get_versionedfiles() |
|
4398.8.8
by John Arbash Meinel
Respond to Andrew's review comments. |
1499 |
key0 = self.get_simple_key('r0') |
1500 |
key1 = self.get_simple_key('r1') |
|
1501 |
key2 = self.get_simple_key('r2') |
|
1502 |
keyf = self.get_simple_key('foo') |
|
4398.8.1
by John Arbash Meinel
Add a VersionedFile.add_text() api. |
1503 |
f.add_lines(key0, [], ['a\n', 'b\n']) |
1504 |
if self.graph: |
|
1505 |
f.add_lines(key1, [key0], ['b\n', 'c\n']) |
|
1506 |
else: |
|
1507 |
f.add_lines(key1, [], ['b\n', 'c\n']) |
|
1508 |
keys = f.keys() |
|
1509 |
self.assertTrue(key0 in keys) |
|
1510 |
self.assertTrue(key1 in keys) |
|
1511 |
records = [] |
|
1512 |
for record in f.get_record_stream([key0, key1], 'unordered', True): |
|
1513 |
records.append((record.key, record.get_bytes_as('fulltext'))) |
|
1514 |
records.sort() |
|
1515 |
self.assertEqual([(key0, 'a\nb\n'), (key1, 'b\nc\n')], records) |
|
1516 |
||
4398.8.6
by John Arbash Meinel
Switch the api from VF.add_text to VF._add_text and trim some extra 'features'. |
1517 |
def test__add_text(self): |
4398.8.1
by John Arbash Meinel
Add a VersionedFile.add_text() api. |
1518 |
f = self.get_versionedfiles() |
4398.8.8
by John Arbash Meinel
Respond to Andrew's review comments. |
1519 |
key0 = self.get_simple_key('r0') |
1520 |
key1 = self.get_simple_key('r1') |
|
1521 |
key2 = self.get_simple_key('r2') |
|
1522 |
keyf = self.get_simple_key('foo') |
|
4398.8.6
by John Arbash Meinel
Switch the api from VF.add_text to VF._add_text and trim some extra 'features'. |
1523 |
f._add_text(key0, [], 'a\nb\n') |
4398.8.1
by John Arbash Meinel
Add a VersionedFile.add_text() api. |
1524 |
if self.graph: |
4398.8.6
by John Arbash Meinel
Switch the api from VF.add_text to VF._add_text and trim some extra 'features'. |
1525 |
f._add_text(key1, [key0], 'b\nc\n') |
4398.8.1
by John Arbash Meinel
Add a VersionedFile.add_text() api. |
1526 |
else: |
4398.8.6
by John Arbash Meinel
Switch the api from VF.add_text to VF._add_text and trim some extra 'features'. |
1527 |
f._add_text(key1, [], 'b\nc\n') |
4398.8.1
by John Arbash Meinel
Add a VersionedFile.add_text() api. |
1528 |
keys = f.keys() |
1529 |
self.assertTrue(key0 in keys) |
|
1530 |
self.assertTrue(key1 in keys) |
|
1531 |
records = [] |
|
1532 |
for record in f.get_record_stream([key0, key1], 'unordered', True): |
|
1533 |
records.append((record.key, record.get_bytes_as('fulltext'))) |
|
1534 |
records.sort() |
|
1535 |
self.assertEqual([(key0, 'a\nb\n'), (key1, 'b\nc\n')], records) |
|
1536 |
||
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
1537 |
def test_annotate(self): |
1538 |
files = self.get_versionedfiles() |
|
1539 |
self.get_diamond_files(files) |
|
1540 |
if self.key_length == 1: |
|
1541 |
prefix = () |
|
1542 |
else: |
|
1543 |
prefix = ('FileA',) |
|
1544 |
# introduced full text
|
|
1545 |
origins = files.annotate(prefix + ('origin',)) |
|
1546 |
self.assertEqual([ |
|
1547 |
(prefix + ('origin',), 'origin\n')], |
|
1548 |
origins) |
|
1549 |
# a delta
|
|
1550 |
origins = files.annotate(prefix + ('base',)) |
|
1551 |
self.assertEqual([ |
|
1552 |
(prefix + ('base',), 'base\n')], |
|
1553 |
origins) |
|
1554 |
# a merge
|
|
1555 |
origins = files.annotate(prefix + ('merged',)) |
|
1556 |
if self.graph: |
|
1557 |
self.assertEqual([ |
|
1558 |
(prefix + ('base',), 'base\n'), |
|
1559 |
(prefix + ('left',), 'left\n'), |
|
1560 |
(prefix + ('right',), 'right\n'), |
|
1561 |
(prefix + ('merged',), 'merged\n') |
|
1562 |
],
|
|
1563 |
origins) |
|
1564 |
else: |
|
1565 |
# Without a graph everything is new.
|
|
1566 |
self.assertEqual([ |
|
1567 |
(prefix + ('merged',), 'base\n'), |
|
1568 |
(prefix + ('merged',), 'left\n'), |
|
1569 |
(prefix + ('merged',), 'right\n'), |
|
1570 |
(prefix + ('merged',), 'merged\n') |
|
1571 |
],
|
|
1572 |
origins) |
|
1573 |
self.assertRaises(RevisionNotPresent, |
|
1574 |
files.annotate, prefix + ('missing-key',)) |
|
1575 |
||
4332.3.26
by Robert Collins
Allow passing keys to check to VersionedFile.check(). |
1576 |
def test_check_no_parameters(self): |
1577 |
files = self.get_versionedfiles() |
|
1578 |
||
1579 |
def test_check_progressbar_parameter(self): |
|
1580 |
"""A progress bar can be supplied because check can be a generator."""
|
|
1581 |
pb = ui.ui_factory.nested_progress_bar() |
|
1582 |
self.addCleanup(pb.finished) |
|
1583 |
files = self.get_versionedfiles() |
|
1584 |
files.check(progress_bar=pb) |
|
1585 |
||
1586 |
def test_check_with_keys_becomes_generator(self): |
|
4454.3.65
by John Arbash Meinel
Tests that VF implementations support .get_annotator() |
1587 |
files = self.get_versionedfiles() |
1588 |
self.get_diamond_files(files) |
|
4332.3.26
by Robert Collins
Allow passing keys to check to VersionedFile.check(). |
1589 |
keys = files.keys() |
1590 |
entries = files.check(keys=keys) |
|
1591 |
seen = set() |
|
1592 |
# Texts output should be fulltexts.
|
|
1593 |
self.capture_stream(files, entries, seen.add, |
|
1594 |
files.get_parent_map(keys), require_fulltext=True) |
|
1595 |
# All texts should be output.
|
|
1596 |
self.assertEqual(set(keys), seen) |
|
4454.3.65
by John Arbash Meinel
Tests that VF implementations support .get_annotator() |
1597 |
|
4744.2.5
by John Arbash Meinel
Change to a generic 'VersionedFiles.clear_cache()' api. |
1598 |
def test_clear_cache(self): |
1599 |
files = self.get_versionedfiles() |
|
1600 |
files.clear_cache() |
|
1601 |
||
3350.6.2
by Robert Collins
Prepare parameterised test environment. |
1602 |
def test_construct(self): |
1603 |
"""Each parameterised test can be constructed on a transport."""
|
|
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
1604 |
files = self.get_versionedfiles() |
1605 |
||
4241.4.1
by Ian Clatworthy
add sha generation support to versionedfiles |
1606 |
def get_diamond_files(self, files, trailing_eol=True, left_only=False, |
1607 |
nokeys=False): |
|
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
1608 |
return get_diamond_files(files, self.key_length, |
1609 |
trailing_eol=trailing_eol, nograph=not self.graph, |
|
4241.4.1
by Ian Clatworthy
add sha generation support to versionedfiles |
1610 |
left_only=left_only, nokeys=nokeys) |
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
1611 |
|
4398.8.4
by John Arbash Meinel
Implement add_text for GroupCompressVersionedFiles |
1612 |
def _add_content_nostoresha(self, add_lines): |
4119.1.1
by John Arbash Meinel
Move the 'add_lines_nostoresha' to being tested against all VF implementations. |
1613 |
"""When nostore_sha is supplied using old content raises."""
|
1614 |
vf = self.get_versionedfiles() |
|
1615 |
empty_text = ('a', []) |
|
1616 |
sample_text_nl = ('b', ["foo\n", "bar\n"]) |
|
1617 |
sample_text_no_nl = ('c', ["foo\n", "bar"]) |
|
1618 |
shas = [] |
|
1619 |
for version, lines in (empty_text, sample_text_nl, sample_text_no_nl): |
|
4398.8.4
by John Arbash Meinel
Implement add_text for GroupCompressVersionedFiles |
1620 |
if add_lines: |
1621 |
sha, _, _ = vf.add_lines(self.get_simple_key(version), [], |
|
1622 |
lines) |
|
1623 |
else: |
|
4398.8.6
by John Arbash Meinel
Switch the api from VF.add_text to VF._add_text and trim some extra 'features'. |
1624 |
sha, _, _ = vf._add_text(self.get_simple_key(version), [], |
1625 |
''.join(lines)) |
|
4119.1.1
by John Arbash Meinel
Move the 'add_lines_nostoresha' to being tested against all VF implementations. |
1626 |
shas.append(sha) |
1627 |
# we now have a copy of all the lines in the vf.
|
|
1628 |
for sha, (version, lines) in zip( |
|
1629 |
shas, (empty_text, sample_text_nl, sample_text_no_nl)): |
|
1630 |
new_key = self.get_simple_key(version + "2") |
|
1631 |
self.assertRaises(errors.ExistingContent, |
|
1632 |
vf.add_lines, new_key, [], lines, |
|
1633 |
nostore_sha=sha) |
|
4398.8.4
by John Arbash Meinel
Implement add_text for GroupCompressVersionedFiles |
1634 |
self.assertRaises(errors.ExistingContent, |
4398.8.6
by John Arbash Meinel
Switch the api from VF.add_text to VF._add_text and trim some extra 'features'. |
1635 |
vf._add_text, new_key, [], ''.join(lines), |
4398.8.4
by John Arbash Meinel
Implement add_text for GroupCompressVersionedFiles |
1636 |
nostore_sha=sha) |
4119.1.1
by John Arbash Meinel
Move the 'add_lines_nostoresha' to being tested against all VF implementations. |
1637 |
# and no new version should have been added.
|
1638 |
record = vf.get_record_stream([new_key], 'unordered', True).next() |
|
1639 |
self.assertEqual('absent', record.storage_kind) |
|
1640 |
||
4398.8.4
by John Arbash Meinel
Implement add_text for GroupCompressVersionedFiles |
1641 |
def test_add_lines_nostoresha(self): |
1642 |
self._add_content_nostoresha(add_lines=True) |
|
1643 |
||
4398.8.6
by John Arbash Meinel
Switch the api from VF.add_text to VF._add_text and trim some extra 'features'. |
1644 |
def test__add_text_nostoresha(self): |
4398.8.4
by John Arbash Meinel
Implement add_text for GroupCompressVersionedFiles |
1645 |
self._add_content_nostoresha(add_lines=False) |
1646 |
||
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
1647 |
def test_add_lines_return(self): |
1648 |
files = self.get_versionedfiles() |
|
1649 |
# save code by using the stock data insertion helper.
|
|
1650 |
adds = self.get_diamond_files(files) |
|
1651 |
results = [] |
|
1652 |
# We can only validate the first 2 elements returned from add_lines.
|
|
1653 |
for add in adds: |
|
1654 |
self.assertEqual(3, len(add)) |
|
1655 |
results.append(add[:2]) |
|
1656 |
if self.key_length == 1: |
|
1657 |
self.assertEqual([ |
|
1658 |
('00e364d235126be43292ab09cb4686cf703ddc17', 7), |
|
1659 |
('51c64a6f4fc375daf0d24aafbabe4d91b6f4bb44', 5), |
|
1660 |
('a8478686da38e370e32e42e8a0c220e33ee9132f', 10), |
|
1661 |
('9ef09dfa9d86780bdec9219a22560c6ece8e0ef1', 11), |
|
1662 |
('ed8bce375198ea62444dc71952b22cfc2b09226d', 23)], |
|
1663 |
results) |
|
1664 |
elif self.key_length == 2: |
|
1665 |
self.assertEqual([ |
|
1666 |
('00e364d235126be43292ab09cb4686cf703ddc17', 7), |
|
1667 |
('00e364d235126be43292ab09cb4686cf703ddc17', 7), |
|
1668 |
('51c64a6f4fc375daf0d24aafbabe4d91b6f4bb44', 5), |
|
1669 |
('51c64a6f4fc375daf0d24aafbabe4d91b6f4bb44', 5), |
|
1670 |
('a8478686da38e370e32e42e8a0c220e33ee9132f', 10), |
|
1671 |
('a8478686da38e370e32e42e8a0c220e33ee9132f', 10), |
|
1672 |
('9ef09dfa9d86780bdec9219a22560c6ece8e0ef1', 11), |
|
1673 |
('9ef09dfa9d86780bdec9219a22560c6ece8e0ef1', 11), |
|
1674 |
('ed8bce375198ea62444dc71952b22cfc2b09226d', 23), |
|
1675 |
('ed8bce375198ea62444dc71952b22cfc2b09226d', 23)], |
|
1676 |
results) |
|
1677 |
||
4241.4.1
by Ian Clatworthy
add sha generation support to versionedfiles |
1678 |
def test_add_lines_no_key_generates_chk_key(self): |
1679 |
files = self.get_versionedfiles() |
|
1680 |
# save code by using the stock data insertion helper.
|
|
1681 |
adds = self.get_diamond_files(files, nokeys=True) |
|
1682 |
results = [] |
|
1683 |
# We can only validate the first 2 elements returned from add_lines.
|
|
1684 |
for add in adds: |
|
1685 |
self.assertEqual(3, len(add)) |
|
1686 |
results.append(add[:2]) |
|
1687 |
if self.key_length == 1: |
|
1688 |
self.assertEqual([ |
|
1689 |
('00e364d235126be43292ab09cb4686cf703ddc17', 7), |
|
1690 |
('51c64a6f4fc375daf0d24aafbabe4d91b6f4bb44', 5), |
|
1691 |
('a8478686da38e370e32e42e8a0c220e33ee9132f', 10), |
|
1692 |
('9ef09dfa9d86780bdec9219a22560c6ece8e0ef1', 11), |
|
1693 |
('ed8bce375198ea62444dc71952b22cfc2b09226d', 23)], |
|
1694 |
results) |
|
1695 |
# Check the added items got CHK keys.
|
|
1696 |
self.assertEqual(set([ |
|
1697 |
('sha1:00e364d235126be43292ab09cb4686cf703ddc17',), |
|
1698 |
('sha1:51c64a6f4fc375daf0d24aafbabe4d91b6f4bb44',), |
|
1699 |
('sha1:9ef09dfa9d86780bdec9219a22560c6ece8e0ef1',), |
|
1700 |
('sha1:a8478686da38e370e32e42e8a0c220e33ee9132f',), |
|
1701 |
('sha1:ed8bce375198ea62444dc71952b22cfc2b09226d',), |
|
1702 |
]),
|
|
1703 |
files.keys()) |
|
1704 |
elif self.key_length == 2: |
|
1705 |
self.assertEqual([ |
|
1706 |
('00e364d235126be43292ab09cb4686cf703ddc17', 7), |
|
1707 |
('00e364d235126be43292ab09cb4686cf703ddc17', 7), |
|
1708 |
('51c64a6f4fc375daf0d24aafbabe4d91b6f4bb44', 5), |
|
1709 |
('51c64a6f4fc375daf0d24aafbabe4d91b6f4bb44', 5), |
|
1710 |
('a8478686da38e370e32e42e8a0c220e33ee9132f', 10), |
|
1711 |
('a8478686da38e370e32e42e8a0c220e33ee9132f', 10), |
|
1712 |
('9ef09dfa9d86780bdec9219a22560c6ece8e0ef1', 11), |
|
1713 |
('9ef09dfa9d86780bdec9219a22560c6ece8e0ef1', 11), |
|
1714 |
('ed8bce375198ea62444dc71952b22cfc2b09226d', 23), |
|
1715 |
('ed8bce375198ea62444dc71952b22cfc2b09226d', 23)], |
|
1716 |
results) |
|
1717 |
# Check the added items got CHK keys.
|
|
1718 |
self.assertEqual(set([ |
|
1719 |
('FileA', 'sha1:00e364d235126be43292ab09cb4686cf703ddc17'), |
|
1720 |
('FileA', 'sha1:51c64a6f4fc375daf0d24aafbabe4d91b6f4bb44'), |
|
1721 |
('FileA', 'sha1:9ef09dfa9d86780bdec9219a22560c6ece8e0ef1'), |
|
1722 |
('FileA', 'sha1:a8478686da38e370e32e42e8a0c220e33ee9132f'), |
|
1723 |
('FileA', 'sha1:ed8bce375198ea62444dc71952b22cfc2b09226d'), |
|
1724 |
('FileB', 'sha1:00e364d235126be43292ab09cb4686cf703ddc17'), |
|
1725 |
('FileB', 'sha1:51c64a6f4fc375daf0d24aafbabe4d91b6f4bb44'), |
|
1726 |
('FileB', 'sha1:9ef09dfa9d86780bdec9219a22560c6ece8e0ef1'), |
|
1727 |
('FileB', 'sha1:a8478686da38e370e32e42e8a0c220e33ee9132f'), |
|
1728 |
('FileB', 'sha1:ed8bce375198ea62444dc71952b22cfc2b09226d'), |
|
1729 |
]),
|
|
1730 |
files.keys()) |
|
1731 |
||
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
1732 |
def test_empty_lines(self): |
1733 |
"""Empty files can be stored."""
|
|
1734 |
f = self.get_versionedfiles() |
|
1735 |
key_a = self.get_simple_key('a') |
|
1736 |
f.add_lines(key_a, [], []) |
|
1737 |
self.assertEqual('', |
|
1738 |
f.get_record_stream([key_a], 'unordered', True |
|
1739 |
).next().get_bytes_as('fulltext')) |
|
1740 |
key_b = self.get_simple_key('b') |
|
1741 |
f.add_lines(key_b, self.get_parents([key_a]), []) |
|
1742 |
self.assertEqual('', |
|
1743 |
f.get_record_stream([key_b], 'unordered', True |
|
1744 |
).next().get_bytes_as('fulltext')) |
|
1745 |
||
1746 |
def test_newline_only(self): |
|
1747 |
f = self.get_versionedfiles() |
|
1748 |
key_a = self.get_simple_key('a') |
|
1749 |
f.add_lines(key_a, [], ['\n']) |
|
1750 |
self.assertEqual('\n', |
|
1751 |
f.get_record_stream([key_a], 'unordered', True |
|
1752 |
).next().get_bytes_as('fulltext')) |
|
1753 |
key_b = self.get_simple_key('b') |
|
1754 |
f.add_lines(key_b, self.get_parents([key_a]), ['\n']) |
|
1755 |
self.assertEqual('\n', |
|
1756 |
f.get_record_stream([key_b], 'unordered', True |
|
1757 |
).next().get_bytes_as('fulltext')) |
|
1758 |
||
4593.5.35
by John Arbash Meinel
Start working on a per-vf implementation test of find_ancestry. |
1759 |
def test_get_known_graph_ancestry(self): |
1760 |
f = self.get_versionedfiles() |
|
4593.5.36
by John Arbash Meinel
a few more implementations of the interface. |
1761 |
if not self.graph: |
1762 |
raise TestNotApplicable('ancestry info only relevant with graph.') |
|
4593.5.35
by John Arbash Meinel
Start working on a per-vf implementation test of find_ancestry. |
1763 |
key_a = self.get_simple_key('a') |
1764 |
key_b = self.get_simple_key('b') |
|
1765 |
key_c = self.get_simple_key('c') |
|
1766 |
# A
|
|
1767 |
# |\
|
|
1768 |
# | B
|
|
1769 |
# |/
|
|
1770 |
# C
|
|
1771 |
f.add_lines(key_a, [], ['\n']) |
|
1772 |
f.add_lines(key_b, [key_a], ['\n']) |
|
1773 |
f.add_lines(key_c, [key_a, key_b], ['\n']) |
|
1774 |
kg = f.get_known_graph_ancestry([key_c]) |
|
1775 |
self.assertIsInstance(kg, _mod_graph.KnownGraph) |
|
1776 |
self.assertEqual([key_a, key_b, key_c], list(kg.topo_sort())) |
|
1777 |
||
4634.11.2
by John Arbash Meinel
Teach VF.get_known_graph_ancestry to go to fallbacks (bug #419241) |
1778 |
def test_known_graph_with_fallbacks(self): |
1779 |
f = self.get_versionedfiles('files') |
|
1780 |
if not self.graph: |
|
1781 |
raise TestNotApplicable('ancestry info only relevant with graph.') |
|
1782 |
if getattr(f, 'add_fallback_versioned_files', None) is None: |
|
1783 |
raise TestNotApplicable("%s doesn't support fallbacks" |
|
1784 |
% (f.__class__.__name__,)) |
|
1785 |
key_a = self.get_simple_key('a') |
|
1786 |
key_b = self.get_simple_key('b') |
|
1787 |
key_c = self.get_simple_key('c') |
|
1788 |
# A only in fallback
|
|
1789 |
# |\
|
|
1790 |
# | B
|
|
1791 |
# |/
|
|
1792 |
# C
|
|
1793 |
g = self.get_versionedfiles('fallback') |
|
1794 |
g.add_lines(key_a, [], ['\n']) |
|
1795 |
f.add_fallback_versioned_files(g) |
|
1796 |
f.add_lines(key_b, [key_a], ['\n']) |
|
1797 |
f.add_lines(key_c, [key_a, key_b], ['\n']) |
|
1798 |
kg = f.get_known_graph_ancestry([key_c]) |
|
1799 |
self.assertEqual([key_a, key_b, key_c], list(kg.topo_sort())) |
|
1800 |
||
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
1801 |
def test_get_record_stream_empty(self): |
1802 |
"""An empty stream can be requested without error."""
|
|
1803 |
f = self.get_versionedfiles() |
|
1804 |
entries = f.get_record_stream([], 'unordered', False) |
|
1805 |
self.assertEqual([], list(entries)) |
|
1806 |
||
1807 |
def assertValidStorageKind(self, storage_kind): |
|
1808 |
"""Assert that storage_kind is a valid storage_kind."""
|
|
1809 |
self.assertSubset([storage_kind], |
|
1810 |
['mpdiff', 'knit-annotated-ft', 'knit-annotated-delta', |
|
3890.2.1
by John Arbash Meinel
Start working on a ChunkedContentFactory. |
1811 |
'knit-ft', 'knit-delta', 'chunked', 'fulltext', |
1812 |
'knit-annotated-ft-gz', 'knit-annotated-delta-gz', 'knit-ft-gz', |
|
4005.3.6
by Robert Collins
Support delta_closure=True with NetworkRecordStream to transmit deltas over the wire when full text extraction is required on the far end. |
1813 |
'knit-delta-gz', |
3735.32.12
by John Arbash Meinel
Add groupcompress-block[-ref] as valid stream types. |
1814 |
'knit-delta-closure', 'knit-delta-closure-ref', |
1815 |
'groupcompress-block', 'groupcompress-block-ref']) |
|
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
1816 |
|
4332.3.26
by Robert Collins
Allow passing keys to check to VersionedFile.check(). |
1817 |
def capture_stream(self, f, entries, on_seen, parents, |
1818 |
require_fulltext=False): |
|
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
1819 |
"""Capture a stream for testing."""
|
1820 |
for factory in entries: |
|
1821 |
on_seen(factory.key) |
|
1822 |
self.assertValidStorageKind(factory.storage_kind) |
|
4241.4.1
by Ian Clatworthy
add sha generation support to versionedfiles |
1823 |
if factory.sha1 is not None: |
1824 |
self.assertEqual(f.get_sha1s([factory.key])[factory.key], |
|
1825 |
factory.sha1) |
|
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
1826 |
self.assertEqual(parents[factory.key], factory.parents) |
1827 |
self.assertIsInstance(factory.get_bytes_as(factory.storage_kind), |
|
1828 |
str) |
|
4332.3.26
by Robert Collins
Allow passing keys to check to VersionedFile.check(). |
1829 |
if require_fulltext: |
1830 |
factory.get_bytes_as('fulltext') |
|
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
1831 |
|
1832 |
def test_get_record_stream_interface(self): |
|
1833 |
"""each item in a stream has to provide a regular interface."""
|
|
1834 |
files = self.get_versionedfiles() |
|
1835 |
self.get_diamond_files(files) |
|
1836 |
keys, _ = self.get_keys_and_sort_order() |
|
1837 |
parent_map = files.get_parent_map(keys) |
|
1838 |
entries = files.get_record_stream(keys, 'unordered', False) |
|
1839 |
seen = set() |
|
1840 |
self.capture_stream(files, entries, seen.add, parent_map) |
|
1841 |
self.assertEqual(set(keys), seen) |
|
1842 |
||
1843 |
def get_keys_and_sort_order(self): |
|
1844 |
"""Get diamond test keys list, and their sort ordering."""
|
|
1845 |
if self.key_length == 1: |
|
1846 |
keys = [('merged',), ('left',), ('right',), ('base',)] |
|
1847 |
sort_order = {('merged',):2, ('left',):1, ('right',):1, ('base',):0} |
|
1848 |
else: |
|
1849 |
keys = [ |
|
1850 |
('FileA', 'merged'), ('FileA', 'left'), ('FileA', 'right'), |
|
1851 |
('FileA', 'base'), |
|
1852 |
('FileB', 'merged'), ('FileB', 'left'), ('FileB', 'right'), |
|
1853 |
('FileB', 'base'), |
|
1854 |
]
|
|
1855 |
sort_order = { |
|
1856 |
('FileA', 'merged'):2, ('FileA', 'left'):1, ('FileA', 'right'):1, |
|
1857 |
('FileA', 'base'):0, |
|
1858 |
('FileB', 'merged'):2, ('FileB', 'left'):1, ('FileB', 'right'):1, |
|
1859 |
('FileB', 'base'):0, |
|
1860 |
}
|
|
1861 |
return keys, sort_order |
|
1862 |
||
4111.1.1
by Robert Collins
Add a groupcompress sort order. |
1863 |
def get_keys_and_groupcompress_sort_order(self): |
1864 |
"""Get diamond test keys list, and their groupcompress sort ordering."""
|
|
1865 |
if self.key_length == 1: |
|
1866 |
keys = [('merged',), ('left',), ('right',), ('base',)] |
|
1867 |
sort_order = {('merged',):0, ('left',):1, ('right',):1, ('base',):2} |
|
1868 |
else: |
|
1869 |
keys = [ |
|
1870 |
('FileA', 'merged'), ('FileA', 'left'), ('FileA', 'right'), |
|
1871 |
('FileA', 'base'), |
|
1872 |
('FileB', 'merged'), ('FileB', 'left'), ('FileB', 'right'), |
|
1873 |
('FileB', 'base'), |
|
1874 |
]
|
|
1875 |
sort_order = { |
|
1876 |
('FileA', 'merged'):0, ('FileA', 'left'):1, ('FileA', 'right'):1, |
|
1877 |
('FileA', 'base'):2, |
|
1878 |
('FileB', 'merged'):3, ('FileB', 'left'):4, ('FileB', 'right'):4, |
|
1879 |
('FileB', 'base'):5, |
|
1880 |
}
|
|
1881 |
return keys, sort_order |
|
1882 |
||
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
1883 |
def test_get_record_stream_interface_ordered(self): |
1884 |
"""each item in a stream has to provide a regular interface."""
|
|
1885 |
files = self.get_versionedfiles() |
|
1886 |
self.get_diamond_files(files) |
|
1887 |
keys, sort_order = self.get_keys_and_sort_order() |
|
1888 |
parent_map = files.get_parent_map(keys) |
|
1889 |
entries = files.get_record_stream(keys, 'topological', False) |
|
1890 |
seen = [] |
|
1891 |
self.capture_stream(files, entries, seen.append, parent_map) |
|
1892 |
self.assertStreamOrder(sort_order, seen, keys) |
|
1893 |
||
1894 |
def test_get_record_stream_interface_ordered_with_delta_closure(self): |
|
1895 |
"""each item must be accessible as a fulltext."""
|
|
1896 |
files = self.get_versionedfiles() |
|
1897 |
self.get_diamond_files(files) |
|
1898 |
keys, sort_order = self.get_keys_and_sort_order() |
|
1899 |
parent_map = files.get_parent_map(keys) |
|
1900 |
entries = files.get_record_stream(keys, 'topological', True) |
|
1901 |
seen = [] |
|
1902 |
for factory in entries: |
|
1903 |
seen.append(factory.key) |
|
1904 |
self.assertValidStorageKind(factory.storage_kind) |
|
3350.8.3
by Robert Collins
VF.get_sha1s needed changing to be stackable. |
1905 |
self.assertSubset([factory.sha1], |
1906 |
[None, files.get_sha1s([factory.key])[factory.key]]) |
|
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
1907 |
self.assertEqual(parent_map[factory.key], factory.parents) |
1908 |
# self.assertEqual(files.get_text(factory.key),
|
|
3890.2.1
by John Arbash Meinel
Start working on a ChunkedContentFactory. |
1909 |
ft_bytes = factory.get_bytes_as('fulltext') |
1910 |
self.assertIsInstance(ft_bytes, str) |
|
1911 |
chunked_bytes = factory.get_bytes_as('chunked') |
|
1912 |
self.assertEqualDiff(ft_bytes, ''.join(chunked_bytes)) |
|
1913 |
||
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
1914 |
self.assertStreamOrder(sort_order, seen, keys) |
1915 |
||
4111.1.1
by Robert Collins
Add a groupcompress sort order. |
1916 |
def test_get_record_stream_interface_groupcompress(self): |
1917 |
"""each item in a stream has to provide a regular interface."""
|
|
1918 |
files = self.get_versionedfiles() |
|
1919 |
self.get_diamond_files(files) |
|
1920 |
keys, sort_order = self.get_keys_and_groupcompress_sort_order() |
|
1921 |
parent_map = files.get_parent_map(keys) |
|
1922 |
entries = files.get_record_stream(keys, 'groupcompress', False) |
|
1923 |
seen = [] |
|
1924 |
self.capture_stream(files, entries, seen.append, parent_map) |
|
1925 |
self.assertStreamOrder(sort_order, seen, keys) |
|
1926 |
||
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
1927 |
def assertStreamOrder(self, sort_order, seen, keys): |
1928 |
self.assertEqual(len(set(seen)), len(keys)) |
|
1929 |
if self.key_length == 1: |
|
1930 |
lows = {():0} |
|
1931 |
else: |
|
1932 |
lows = {('FileA',):0, ('FileB',):0} |
|
1933 |
if not self.graph: |
|
1934 |
self.assertEqual(set(keys), set(seen)) |
|
1935 |
else: |
|
1936 |
for key in seen: |
|
1937 |
sort_pos = sort_order[key] |
|
1938 |
self.assertTrue(sort_pos >= lows[key[:-1]], |
|
1939 |
"Out of order in sorted stream: %r, %r" % (key, seen)) |
|
1940 |
lows[key[:-1]] = sort_pos |
|
1941 |
||
1942 |
def test_get_record_stream_unknown_storage_kind_raises(self): |
|
1943 |
"""Asking for a storage kind that the stream cannot supply raises."""
|
|
1944 |
files = self.get_versionedfiles() |
|
1945 |
self.get_diamond_files(files) |
|
1946 |
if self.key_length == 1: |
|
1947 |
keys = [('merged',), ('left',), ('right',), ('base',)] |
|
1948 |
else: |
|
1949 |
keys = [ |
|
1950 |
('FileA', 'merged'), ('FileA', 'left'), ('FileA', 'right'), |
|
1951 |
('FileA', 'base'), |
|
1952 |
('FileB', 'merged'), ('FileB', 'left'), ('FileB', 'right'), |
|
1953 |
('FileB', 'base'), |
|
1954 |
]
|
|
1955 |
parent_map = files.get_parent_map(keys) |
|
1956 |
entries = files.get_record_stream(keys, 'unordered', False) |
|
1957 |
# We track the contents because we should be able to try, fail a
|
|
1958 |
# particular kind and then ask for one that works and continue.
|
|
1959 |
seen = set() |
|
1960 |
for factory in entries: |
|
1961 |
seen.add(factory.key) |
|
1962 |
self.assertValidStorageKind(factory.storage_kind) |
|
4241.4.1
by Ian Clatworthy
add sha generation support to versionedfiles |
1963 |
if factory.sha1 is not None: |
1964 |
self.assertEqual(files.get_sha1s([factory.key])[factory.key], |
|
1965 |
factory.sha1) |
|
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
1966 |
self.assertEqual(parent_map[factory.key], factory.parents) |
1967 |
# currently no stream emits mpdiff
|
|
1968 |
self.assertRaises(errors.UnavailableRepresentation, |
|
1969 |
factory.get_bytes_as, 'mpdiff') |
|
1970 |
self.assertIsInstance(factory.get_bytes_as(factory.storage_kind), |
|
1971 |
str) |
|
1972 |
self.assertEqual(set(keys), seen) |
|
1973 |
||
1974 |
def test_get_record_stream_missing_records_are_absent(self): |
|
1975 |
files = self.get_versionedfiles() |
|
1976 |
self.get_diamond_files(files) |
|
1977 |
if self.key_length == 1: |
|
1978 |
keys = [('merged',), ('left',), ('right',), ('absent',), ('base',)] |
|
1979 |
else: |
|
1980 |
keys = [ |
|
1981 |
('FileA', 'merged'), ('FileA', 'left'), ('FileA', 'right'), |
|
1982 |
('FileA', 'absent'), ('FileA', 'base'), |
|
1983 |
('FileB', 'merged'), ('FileB', 'left'), ('FileB', 'right'), |
|
1984 |
('FileB', 'absent'), ('FileB', 'base'), |
|
1985 |
('absent', 'absent'), |
|
1986 |
]
|
|
1987 |
parent_map = files.get_parent_map(keys) |
|
1988 |
entries = files.get_record_stream(keys, 'unordered', False) |
|
1989 |
self.assertAbsentRecord(files, keys, parent_map, entries) |
|
1990 |
entries = files.get_record_stream(keys, 'topological', False) |
|
1991 |
self.assertAbsentRecord(files, keys, parent_map, entries) |
|
1992 |
||
4005.3.3
by Robert Collins
Test NetworkRecordStream with delta'd texts. |
1993 |
def assertRecordHasContent(self, record, bytes): |
1994 |
"""Assert that record has the bytes bytes."""
|
|
1995 |
self.assertEqual(bytes, record.get_bytes_as('fulltext')) |
|
4005.3.7
by Robert Collins
Review feedback. |
1996 |
self.assertEqual(bytes, ''.join(record.get_bytes_as('chunked'))) |
4005.3.3
by Robert Collins
Test NetworkRecordStream with delta'd texts. |
1997 |
|
4005.3.2
by Robert Collins
First passing NetworkRecordStream test - a fulltext from any record type which isn't a chunked or fulltext can be serialised and deserialised successfully. |
1998 |
def test_get_record_stream_native_formats_are_wire_ready_one_ft(self): |
1999 |
files = self.get_versionedfiles() |
|
2000 |
key = self.get_simple_key('foo') |
|
2001 |
files.add_lines(key, (), ['my text\n', 'content']) |
|
2002 |
stream = files.get_record_stream([key], 'unordered', False) |
|
2003 |
record = stream.next() |
|
2004 |
if record.storage_kind in ('chunked', 'fulltext'): |
|
2005 |
# chunked and fulltext representations are for direct use not wire
|
|
4005.3.3
by Robert Collins
Test NetworkRecordStream with delta'd texts. |
2006 |
# serialisation: check they are able to be used directly. To send
|
2007 |
# such records over the wire translation will be needed.
|
|
2008 |
self.assertRecordHasContent(record, "my text\ncontent") |
|
4005.3.2
by Robert Collins
First passing NetworkRecordStream test - a fulltext from any record type which isn't a chunked or fulltext can be serialised and deserialised successfully. |
2009 |
else: |
2010 |
bytes = [record.get_bytes_as(record.storage_kind)] |
|
2011 |
network_stream = versionedfile.NetworkRecordStream(bytes).read() |
|
2012 |
source_record = record |
|
2013 |
records = [] |
|
2014 |
for record in network_stream: |
|
2015 |
records.append(record) |
|
2016 |
self.assertEqual(source_record.storage_kind, |
|
2017 |
record.storage_kind) |
|
2018 |
self.assertEqual(source_record.parents, record.parents) |
|
2019 |
self.assertEqual( |
|
2020 |
source_record.get_bytes_as(source_record.storage_kind), |
|
2021 |
record.get_bytes_as(record.storage_kind)) |
|
2022 |
self.assertEqual(1, len(records)) |
|
2023 |
||
4005.3.5
by Robert Collins
Interface level test for using delta_closure=True over the network. |
2024 |
def assertStreamMetaEqual(self, records, expected, stream): |
2025 |
"""Assert that streams expected and stream have the same records.
|
|
4032.1.2
by John Arbash Meinel
Track down a few more files that have trailing whitespace. |
2026 |
|
4005.3.5
by Robert Collins
Interface level test for using delta_closure=True over the network. |
2027 |
:param records: A list to collect the seen records.
|
2028 |
:return: A generator of the records in stream.
|
|
2029 |
"""
|
|
2030 |
# We make assertions during copying to catch things early for
|
|
2031 |
# easier debugging.
|
|
2032 |
for record, ref_record in izip(stream, expected): |
|
2033 |
records.append(record) |
|
2034 |
self.assertEqual(ref_record.key, record.key) |
|
2035 |
self.assertEqual(ref_record.storage_kind, record.storage_kind) |
|
2036 |
self.assertEqual(ref_record.parents, record.parents) |
|
2037 |
yield record |
|
2038 |
||
2039 |
def stream_to_bytes_or_skip_counter(self, skipped_records, full_texts, |
|
2040 |
stream): |
|
2041 |
"""Convert a stream to a bytes iterator.
|
|
2042 |
||
2043 |
:param skipped_records: A list with one element to increment when a
|
|
2044 |
record is skipped.
|
|
4032.1.2
by John Arbash Meinel
Track down a few more files that have trailing whitespace. |
2045 |
:param full_texts: A dict from key->fulltext representation, for
|
4005.3.5
by Robert Collins
Interface level test for using delta_closure=True over the network. |
2046 |
checking chunked or fulltext stored records.
|
2047 |
:param stream: A record_stream.
|
|
2048 |
:return: An iterator over the bytes of each record.
|
|
2049 |
"""
|
|
2050 |
for record in stream: |
|
2051 |
if record.storage_kind in ('chunked', 'fulltext'): |
|
2052 |
skipped_records[0] += 1 |
|
2053 |
# check the content is correct for direct use.
|
|
2054 |
self.assertRecordHasContent(record, full_texts[record.key]) |
|
2055 |
else: |
|
2056 |
yield record.get_bytes_as(record.storage_kind) |
|
2057 |
||
4005.3.3
by Robert Collins
Test NetworkRecordStream with delta'd texts. |
2058 |
def test_get_record_stream_native_formats_are_wire_ready_ft_delta(self): |
2059 |
files = self.get_versionedfiles() |
|
2060 |
target_files = self.get_versionedfiles('target') |
|
2061 |
key = self.get_simple_key('ft') |
|
2062 |
key_delta = self.get_simple_key('delta') |
|
2063 |
files.add_lines(key, (), ['my text\n', 'content']) |
|
2064 |
if self.graph: |
|
2065 |
delta_parents = (key,) |
|
2066 |
else: |
|
2067 |
delta_parents = () |
|
2068 |
files.add_lines(key_delta, delta_parents, ['different\n', 'content\n']) |
|
2069 |
local = files.get_record_stream([key, key_delta], 'unordered', False) |
|
2070 |
ref = files.get_record_stream([key, key_delta], 'unordered', False) |
|
2071 |
skipped_records = [0] |
|
4005.3.5
by Robert Collins
Interface level test for using delta_closure=True over the network. |
2072 |
full_texts = { |
2073 |
key: "my text\ncontent", |
|
2074 |
key_delta: "different\ncontent\n", |
|
2075 |
}
|
|
2076 |
byte_stream = self.stream_to_bytes_or_skip_counter( |
|
2077 |
skipped_records, full_texts, local) |
|
4005.3.3
by Robert Collins
Test NetworkRecordStream with delta'd texts. |
2078 |
network_stream = versionedfile.NetworkRecordStream(byte_stream).read() |
2079 |
records = [] |
|
2080 |
# insert the stream from the network into a versioned files object so we can
|
|
2081 |
# check the content was carried across correctly without doing delta
|
|
2082 |
# inspection.
|
|
4005.3.5
by Robert Collins
Interface level test for using delta_closure=True over the network. |
2083 |
target_files.insert_record_stream( |
2084 |
self.assertStreamMetaEqual(records, ref, network_stream)) |
|
4005.3.3
by Robert Collins
Test NetworkRecordStream with delta'd texts. |
2085 |
# No duplicates on the wire thank you!
|
2086 |
self.assertEqual(2, len(records) + skipped_records[0]) |
|
2087 |
if len(records): |
|
4005.3.4
by Robert Collins
Test copying just a delta over the wire. |
2088 |
# if any content was copied it all must have all been.
|
2089 |
self.assertIdenticalVersionedFile(files, target_files) |
|
2090 |
||
2091 |
def test_get_record_stream_native_formats_are_wire_ready_delta(self): |
|
2092 |
# copy a delta over the wire
|
|
2093 |
files = self.get_versionedfiles() |
|
2094 |
target_files = self.get_versionedfiles('target') |
|
2095 |
key = self.get_simple_key('ft') |
|
2096 |
key_delta = self.get_simple_key('delta') |
|
2097 |
files.add_lines(key, (), ['my text\n', 'content']) |
|
2098 |
if self.graph: |
|
2099 |
delta_parents = (key,) |
|
2100 |
else: |
|
2101 |
delta_parents = () |
|
2102 |
files.add_lines(key_delta, delta_parents, ['different\n', 'content\n']) |
|
4005.3.5
by Robert Collins
Interface level test for using delta_closure=True over the network. |
2103 |
# Copy the basis text across so we can reconstruct the delta during
|
2104 |
# insertion into target.
|
|
4005.3.4
by Robert Collins
Test copying just a delta over the wire. |
2105 |
target_files.insert_record_stream(files.get_record_stream([key], |
2106 |
'unordered', False)) |
|
2107 |
local = files.get_record_stream([key_delta], 'unordered', False) |
|
2108 |
ref = files.get_record_stream([key_delta], 'unordered', False) |
|
2109 |
skipped_records = [0] |
|
4005.3.5
by Robert Collins
Interface level test for using delta_closure=True over the network. |
2110 |
full_texts = { |
2111 |
key_delta: "different\ncontent\n", |
|
2112 |
}
|
|
2113 |
byte_stream = self.stream_to_bytes_or_skip_counter( |
|
2114 |
skipped_records, full_texts, local) |
|
4005.3.4
by Robert Collins
Test copying just a delta over the wire. |
2115 |
network_stream = versionedfile.NetworkRecordStream(byte_stream).read() |
2116 |
records = [] |
|
2117 |
# insert the stream from the network into a versioned files object so we can
|
|
2118 |
# check the content was carried across correctly without doing delta
|
|
2119 |
# inspection during check_stream.
|
|
4005.3.5
by Robert Collins
Interface level test for using delta_closure=True over the network. |
2120 |
target_files.insert_record_stream( |
2121 |
self.assertStreamMetaEqual(records, ref, network_stream)) |
|
4005.3.4
by Robert Collins
Test copying just a delta over the wire. |
2122 |
# No duplicates on the wire thank you!
|
2123 |
self.assertEqual(1, len(records) + skipped_records[0]) |
|
2124 |
if len(records): |
|
2125 |
# if any content was copied it all must have all been
|
|
4005.3.3
by Robert Collins
Test NetworkRecordStream with delta'd texts. |
2126 |
self.assertIdenticalVersionedFile(files, target_files) |
2127 |
||
4005.3.5
by Robert Collins
Interface level test for using delta_closure=True over the network. |
2128 |
def test_get_record_stream_wire_ready_delta_closure_included(self): |
2129 |
# copy a delta over the wire with the ability to get its full text.
|
|
2130 |
files = self.get_versionedfiles() |
|
2131 |
key = self.get_simple_key('ft') |
|
2132 |
key_delta = self.get_simple_key('delta') |
|
2133 |
files.add_lines(key, (), ['my text\n', 'content']) |
|
2134 |
if self.graph: |
|
2135 |
delta_parents = (key,) |
|
2136 |
else: |
|
2137 |
delta_parents = () |
|
2138 |
files.add_lines(key_delta, delta_parents, ['different\n', 'content\n']) |
|
2139 |
local = files.get_record_stream([key_delta], 'unordered', True) |
|
2140 |
ref = files.get_record_stream([key_delta], 'unordered', True) |
|
2141 |
skipped_records = [0] |
|
2142 |
full_texts = { |
|
2143 |
key_delta: "different\ncontent\n", |
|
2144 |
}
|
|
2145 |
byte_stream = self.stream_to_bytes_or_skip_counter( |
|
2146 |
skipped_records, full_texts, local) |
|
2147 |
network_stream = versionedfile.NetworkRecordStream(byte_stream).read() |
|
2148 |
records = [] |
|
2149 |
# insert the stream from the network into a versioned files object so we can
|
|
2150 |
# check the content was carried across correctly without doing delta
|
|
2151 |
# inspection during check_stream.
|
|
2152 |
for record in self.assertStreamMetaEqual(records, ref, network_stream): |
|
2153 |
# we have to be able to get the full text out:
|
|
2154 |
self.assertRecordHasContent(record, full_texts[record.key]) |
|
2155 |
# No duplicates on the wire thank you!
|
|
2156 |
self.assertEqual(1, len(records) + skipped_records[0]) |
|
2157 |
||
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
2158 |
def assertAbsentRecord(self, files, keys, parents, entries): |
2159 |
"""Helper for test_get_record_stream_missing_records_are_absent."""
|
|
2160 |
seen = set() |
|
2161 |
for factory in entries: |
|
2162 |
seen.add(factory.key) |
|
2163 |
if factory.key[-1] == 'absent': |
|
2164 |
self.assertEqual('absent', factory.storage_kind) |
|
2165 |
self.assertEqual(None, factory.sha1) |
|
2166 |
self.assertEqual(None, factory.parents) |
|
2167 |
else: |
|
2168 |
self.assertValidStorageKind(factory.storage_kind) |
|
4241.4.1
by Ian Clatworthy
add sha generation support to versionedfiles |
2169 |
if factory.sha1 is not None: |
2170 |
sha1 = files.get_sha1s([factory.key])[factory.key] |
|
2171 |
self.assertEqual(sha1, factory.sha1) |
|
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
2172 |
self.assertEqual(parents[factory.key], factory.parents) |
2173 |
self.assertIsInstance(factory.get_bytes_as(factory.storage_kind), |
|
2174 |
str) |
|
2175 |
self.assertEqual(set(keys), seen) |
|
2176 |
||
2177 |
def test_filter_absent_records(self): |
|
2178 |
"""Requested missing records can be filter trivially."""
|
|
2179 |
files = self.get_versionedfiles() |
|
2180 |
self.get_diamond_files(files) |
|
2181 |
keys, _ = self.get_keys_and_sort_order() |
|
2182 |
parent_map = files.get_parent_map(keys) |
|
2183 |
# Add an absent record in the middle of the present keys. (We don't ask
|
|
2184 |
# for just absent keys to ensure that content before and after the
|
|
2185 |
# absent keys is still delivered).
|
|
2186 |
present_keys = list(keys) |
|
2187 |
if self.key_length == 1: |
|
2188 |
keys.insert(2, ('extra',)) |
|
2189 |
else: |
|
2190 |
keys.insert(2, ('extra', 'extra')) |
|
2191 |
entries = files.get_record_stream(keys, 'unordered', False) |
|
2192 |
seen = set() |
|
2193 |
self.capture_stream(files, versionedfile.filter_absent(entries), seen.add, |
|
2194 |
parent_map) |
|
2195 |
self.assertEqual(set(present_keys), seen) |
|
2196 |
||
2197 |
def get_mapper(self): |
|
2198 |
"""Get a mapper suitable for the key length of the test interface."""
|
|
2199 |
if self.key_length == 1: |
|
2200 |
return ConstantMapper('source') |
|
2201 |
else: |
|
2202 |
return HashEscapedPrefixMapper() |
|
2203 |
||
2204 |
def get_parents(self, parents): |
|
2205 |
"""Get parents, taking self.graph into consideration."""
|
|
2206 |
if self.graph: |
|
2207 |
return parents |
|
2208 |
else: |
|
2209 |
return None |
|
2210 |
||
4332.3.32
by Robert Collins
Merge bzr.dev. |
2211 |
def test_get_annotator(self): |
2212 |
files = self.get_versionedfiles() |
|
2213 |
self.get_diamond_files(files) |
|
2214 |
origin_key = self.get_simple_key('origin') |
|
2215 |
base_key = self.get_simple_key('base') |
|
2216 |
left_key = self.get_simple_key('left') |
|
2217 |
right_key = self.get_simple_key('right') |
|
2218 |
merged_key = self.get_simple_key('merged') |
|
2219 |
# annotator = files.get_annotator()
|
|
2220 |
# introduced full text
|
|
2221 |
origins, lines = files.get_annotator().annotate(origin_key) |
|
2222 |
self.assertEqual([(origin_key,)], origins) |
|
2223 |
self.assertEqual(['origin\n'], lines) |
|
2224 |
# a delta
|
|
2225 |
origins, lines = files.get_annotator().annotate(base_key) |
|
2226 |
self.assertEqual([(base_key,)], origins) |
|
2227 |
# a merge
|
|
2228 |
origins, lines = files.get_annotator().annotate(merged_key) |
|
2229 |
if self.graph: |
|
2230 |
self.assertEqual([ |
|
2231 |
(base_key,), |
|
2232 |
(left_key,), |
|
2233 |
(right_key,), |
|
2234 |
(merged_key,), |
|
2235 |
], origins) |
|
2236 |
else: |
|
2237 |
# Without a graph everything is new.
|
|
2238 |
self.assertEqual([ |
|
2239 |
(merged_key,), |
|
2240 |
(merged_key,), |
|
2241 |
(merged_key,), |
|
2242 |
(merged_key,), |
|
2243 |
], origins) |
|
2244 |
self.assertRaises(RevisionNotPresent, |
|
2245 |
files.get_annotator().annotate, self.get_simple_key('missing-key')) |
|
2246 |
||
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
2247 |
def test_get_parent_map(self): |
2248 |
files = self.get_versionedfiles() |
|
2249 |
if self.key_length == 1: |
|
2250 |
parent_details = [ |
|
2251 |
(('r0',), self.get_parents(())), |
|
2252 |
(('r1',), self.get_parents((('r0',),))), |
|
2253 |
(('r2',), self.get_parents(())), |
|
2254 |
(('r3',), self.get_parents(())), |
|
2255 |
(('m',), self.get_parents((('r0',),('r1',),('r2',),('r3',)))), |
|
2256 |
]
|
|
2257 |
else: |
|
2258 |
parent_details = [ |
|
2259 |
(('FileA', 'r0'), self.get_parents(())), |
|
2260 |
(('FileA', 'r1'), self.get_parents((('FileA', 'r0'),))), |
|
2261 |
(('FileA', 'r2'), self.get_parents(())), |
|
2262 |
(('FileA', 'r3'), self.get_parents(())), |
|
2263 |
(('FileA', 'm'), self.get_parents((('FileA', 'r0'), |
|
2264 |
('FileA', 'r1'), ('FileA', 'r2'), ('FileA', 'r3')))), |
|
2265 |
]
|
|
2266 |
for key, parents in parent_details: |
|
2267 |
files.add_lines(key, parents, []) |
|
2268 |
# immediately after adding it should be queryable.
|
|
2269 |
self.assertEqual({key:parents}, files.get_parent_map([key])) |
|
2270 |
# We can ask for an empty set
|
|
2271 |
self.assertEqual({}, files.get_parent_map([])) |
|
2272 |
# We can ask for many keys
|
|
2273 |
all_parents = dict(parent_details) |
|
2274 |
self.assertEqual(all_parents, files.get_parent_map(all_parents.keys())) |
|
2275 |
# Absent keys are just not included in the result.
|
|
2276 |
keys = all_parents.keys() |
|
2277 |
if self.key_length == 1: |
|
2278 |
keys.insert(1, ('missing',)) |
|
2279 |
else: |
|
2280 |
keys.insert(1, ('missing', 'missing')) |
|
2281 |
# Absent keys are just ignored
|
|
2282 |
self.assertEqual(all_parents, files.get_parent_map(keys)) |
|
2283 |
||
2284 |
def test_get_sha1s(self): |
|
2285 |
files = self.get_versionedfiles() |
|
2286 |
self.get_diamond_files(files) |
|
2287 |
if self.key_length == 1: |
|
2288 |
keys = [('base',), ('origin',), ('left',), ('merged',), ('right',)] |
|
2289 |
else: |
|
2290 |
# ask for shas from different prefixes.
|
|
2291 |
keys = [ |
|
2292 |
('FileA', 'base'), ('FileB', 'origin'), ('FileA', 'left'), |
|
2293 |
('FileA', 'merged'), ('FileB', 'right'), |
|
2294 |
]
|
|
3350.8.3
by Robert Collins
VF.get_sha1s needed changing to be stackable. |
2295 |
self.assertEqual({ |
2296 |
keys[0]: '51c64a6f4fc375daf0d24aafbabe4d91b6f4bb44', |
|
2297 |
keys[1]: '00e364d235126be43292ab09cb4686cf703ddc17', |
|
2298 |
keys[2]: 'a8478686da38e370e32e42e8a0c220e33ee9132f', |
|
2299 |
keys[3]: 'ed8bce375198ea62444dc71952b22cfc2b09226d', |
|
2300 |
keys[4]: '9ef09dfa9d86780bdec9219a22560c6ece8e0ef1', |
|
2301 |
},
|
|
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
2302 |
files.get_sha1s(keys)) |
3943.8.1
by Marius Kruger
remove all trailing whitespace from bzr source |
2303 |
|
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
2304 |
def test_insert_record_stream_empty(self): |
2305 |
"""Inserting an empty record stream should work."""
|
|
2306 |
files = self.get_versionedfiles() |
|
2307 |
files.insert_record_stream([]) |
|
2308 |
||
2309 |
def assertIdenticalVersionedFile(self, expected, actual): |
|
2310 |
"""Assert that left and right have the same contents."""
|
|
2311 |
self.assertEqual(set(actual.keys()), set(expected.keys())) |
|
2312 |
actual_parents = actual.get_parent_map(actual.keys()) |
|
2313 |
if self.graph: |
|
2314 |
self.assertEqual(actual_parents, expected.get_parent_map(expected.keys())) |
|
2315 |
else: |
|
2316 |
for key, parents in actual_parents.items(): |
|
2317 |
self.assertEqual(None, parents) |
|
2318 |
for key in actual.keys(): |
|
2319 |
actual_text = actual.get_record_stream( |
|
2320 |
[key], 'unordered', True).next().get_bytes_as('fulltext') |
|
2321 |
expected_text = expected.get_record_stream( |
|
2322 |
[key], 'unordered', True).next().get_bytes_as('fulltext') |
|
2323 |
self.assertEqual(actual_text, expected_text) |
|
2324 |
||
2325 |
def test_insert_record_stream_fulltexts(self): |
|
2326 |
"""Any file should accept a stream of fulltexts."""
|
|
2327 |
files = self.get_versionedfiles() |
|
2328 |
mapper = self.get_mapper() |
|
2329 |
source_transport = self.get_transport('source') |
|
2330 |
source_transport.mkdir('.') |
|
2331 |
# weaves always output fulltexts.
|
|
2332 |
source = make_versioned_files_factory(WeaveFile, mapper)( |
|
2333 |
source_transport) |
|
2334 |
self.get_diamond_files(source, trailing_eol=False) |
|
2335 |
stream = source.get_record_stream(source.keys(), 'topological', |
|
2336 |
False) |
|
2337 |
files.insert_record_stream(stream) |
|
2338 |
self.assertIdenticalVersionedFile(source, files) |
|
2339 |
||
2340 |
def test_insert_record_stream_fulltexts_noeol(self): |
|
2341 |
"""Any file should accept a stream of fulltexts."""
|
|
2342 |
files = self.get_versionedfiles() |
|
2343 |
mapper = self.get_mapper() |
|
2344 |
source_transport = self.get_transport('source') |
|
2345 |
source_transport.mkdir('.') |
|
2346 |
# weaves always output fulltexts.
|
|
2347 |
source = make_versioned_files_factory(WeaveFile, mapper)( |
|
2348 |
source_transport) |
|
2349 |
self.get_diamond_files(source, trailing_eol=False) |
|
2350 |
stream = source.get_record_stream(source.keys(), 'topological', |
|
2351 |
False) |
|
2352 |
files.insert_record_stream(stream) |
|
2353 |
self.assertIdenticalVersionedFile(source, files) |
|
2354 |
||
2355 |
def test_insert_record_stream_annotated_knits(self): |
|
2356 |
"""Any file should accept a stream from plain knits."""
|
|
2357 |
files = self.get_versionedfiles() |
|
2358 |
mapper = self.get_mapper() |
|
2359 |
source_transport = self.get_transport('source') |
|
2360 |
source_transport.mkdir('.') |
|
2361 |
source = make_file_factory(True, mapper)(source_transport) |
|
2362 |
self.get_diamond_files(source) |
|
2363 |
stream = source.get_record_stream(source.keys(), 'topological', |
|
2364 |
False) |
|
2365 |
files.insert_record_stream(stream) |
|
2366 |
self.assertIdenticalVersionedFile(source, files) |
|
2367 |
||
2368 |
def test_insert_record_stream_annotated_knits_noeol(self): |
|
2369 |
"""Any file should accept a stream from plain knits."""
|
|
2370 |
files = self.get_versionedfiles() |
|
2371 |
mapper = self.get_mapper() |
|
2372 |
source_transport = self.get_transport('source') |
|
2373 |
source_transport.mkdir('.') |
|
2374 |
source = make_file_factory(True, mapper)(source_transport) |
|
2375 |
self.get_diamond_files(source, trailing_eol=False) |
|
2376 |
stream = source.get_record_stream(source.keys(), 'topological', |
|
2377 |
False) |
|
2378 |
files.insert_record_stream(stream) |
|
2379 |
self.assertIdenticalVersionedFile(source, files) |
|
2380 |
||
2381 |
def test_insert_record_stream_plain_knits(self): |
|
2382 |
"""Any file should accept a stream from plain knits."""
|
|
2383 |
files = self.get_versionedfiles() |
|
2384 |
mapper = self.get_mapper() |
|
2385 |
source_transport = self.get_transport('source') |
|
2386 |
source_transport.mkdir('.') |
|
2387 |
source = make_file_factory(False, mapper)(source_transport) |
|
2388 |
self.get_diamond_files(source) |
|
2389 |
stream = source.get_record_stream(source.keys(), 'topological', |
|
2390 |
False) |
|
2391 |
files.insert_record_stream(stream) |
|
2392 |
self.assertIdenticalVersionedFile(source, files) |
|
2393 |
||
2394 |
def test_insert_record_stream_plain_knits_noeol(self): |
|
2395 |
"""Any file should accept a stream from plain knits."""
|
|
2396 |
files = self.get_versionedfiles() |
|
2397 |
mapper = self.get_mapper() |
|
2398 |
source_transport = self.get_transport('source') |
|
2399 |
source_transport.mkdir('.') |
|
2400 |
source = make_file_factory(False, mapper)(source_transport) |
|
2401 |
self.get_diamond_files(source, trailing_eol=False) |
|
2402 |
stream = source.get_record_stream(source.keys(), 'topological', |
|
2403 |
False) |
|
2404 |
files.insert_record_stream(stream) |
|
2405 |
self.assertIdenticalVersionedFile(source, files) |
|
2406 |
||
2407 |
def test_insert_record_stream_existing_keys(self): |
|
2408 |
"""Inserting keys already in a file should not error."""
|
|
2409 |
files = self.get_versionedfiles() |
|
2410 |
source = self.get_versionedfiles('source') |
|
2411 |
self.get_diamond_files(source) |
|
2412 |
# insert some keys into f.
|
|
2413 |
self.get_diamond_files(files, left_only=True) |
|
2414 |
stream = source.get_record_stream(source.keys(), 'topological', |
|
2415 |
False) |
|
2416 |
files.insert_record_stream(stream) |
|
2417 |
self.assertIdenticalVersionedFile(source, files) |
|
2418 |
||
2419 |
def test_insert_record_stream_missing_keys(self): |
|
2420 |
"""Inserting a stream with absent keys should raise an error."""
|
|
2421 |
files = self.get_versionedfiles() |
|
2422 |
source = self.get_versionedfiles('source') |
|
2423 |
stream = source.get_record_stream([('missing',) * self.key_length], |
|
2424 |
'topological', False) |
|
2425 |
self.assertRaises(errors.RevisionNotPresent, files.insert_record_stream, |
|
2426 |
stream) |
|
2427 |
||
2428 |
def test_insert_record_stream_out_of_order(self): |
|
2429 |
"""An out of order stream can either error or work."""
|
|
2430 |
files = self.get_versionedfiles() |
|
2431 |
source = self.get_versionedfiles('source') |
|
2432 |
self.get_diamond_files(source) |
|
2433 |
if self.key_length == 1: |
|
2434 |
origin_keys = [('origin',)] |
|
2435 |
end_keys = [('merged',), ('left',)] |
|
2436 |
start_keys = [('right',), ('base',)] |
|
2437 |
else: |
|
2438 |
origin_keys = [('FileA', 'origin'), ('FileB', 'origin')] |
|
2439 |
end_keys = [('FileA', 'merged',), ('FileA', 'left',), |
|
2440 |
('FileB', 'merged',), ('FileB', 'left',)] |
|
2441 |
start_keys = [('FileA', 'right',), ('FileA', 'base',), |
|
2442 |
('FileB', 'right',), ('FileB', 'base',)] |
|
2443 |
origin_entries = source.get_record_stream(origin_keys, 'unordered', False) |
|
2444 |
end_entries = source.get_record_stream(end_keys, 'topological', False) |
|
2445 |
start_entries = source.get_record_stream(start_keys, 'topological', False) |
|
2446 |
entries = chain(origin_entries, end_entries, start_entries) |
|
2447 |
try: |
|
2448 |
files.insert_record_stream(entries) |
|
2449 |
except RevisionNotPresent: |
|
2450 |
# Must not have corrupted the file.
|
|
2451 |
files.check() |
|
2452 |
else: |
|
2453 |
self.assertIdenticalVersionedFile(source, files) |
|
2454 |
||
4634.84.2
by Andrew Bennetts
Add test. |
2455 |
def test_insert_record_stream_long_parent_chain_out_of_order(self): |
2456 |
"""An out of order stream can either error or work."""
|
|
2457 |
if not self.graph: |
|
2458 |
raise TestNotApplicable('ancestry info only relevant with graph.') |
|
2459 |
# Create a reasonably long chain of records based on each other, where
|
|
2460 |
# most will be deltas.
|
|
2461 |
source = self.get_versionedfiles('source') |
|
2462 |
parents = () |
|
2463 |
keys = [] |
|
2464 |
content = [('same same %d\n' % n) for n in range(500)] |
|
2465 |
for letter in 'abcdefghijklmnopqrstuvwxyz': |
|
2466 |
key = ('key-' + letter,) |
|
2467 |
if self.key_length == 2: |
|
2468 |
key = ('prefix',) + key |
|
2469 |
content.append('content for ' + letter + '\n') |
|
2470 |
source.add_lines(key, parents, content) |
|
2471 |
keys.append(key) |
|
2472 |
parents = (key,) |
|
2473 |
# Create a stream of these records, excluding the first record that the
|
|
2474 |
# rest ultimately depend upon, and insert it into a new vf.
|
|
2475 |
streams = [] |
|
2476 |
for key in reversed(keys): |
|
2477 |
streams.append(source.get_record_stream([key], 'unordered', False)) |
|
2478 |
deltas = chain(*streams[:-1]) |
|
2479 |
files = self.get_versionedfiles() |
|
2480 |
try: |
|
2481 |
files.insert_record_stream(deltas) |
|
2482 |
except RevisionNotPresent: |
|
2483 |
# Must not have corrupted the file.
|
|
2484 |
files.check() |
|
2485 |
else: |
|
2486 |
# Must only report either just the first key as a missing parent,
|
|
2487 |
# no key as missing (for nodelta scenarios).
|
|
2488 |
missing = set(files.get_missing_compression_parent_keys()) |
|
2489 |
missing.discard(keys[0]) |
|
2490 |
self.assertEqual(set(), missing) |
|
2491 |
||
4009.3.2
by Andrew Bennetts
Add test_insert_record_stream_delta_missing_basis_can_be_added_later. |
2492 |
def get_knit_delta_source(self): |
2493 |
"""Get a source that can produce a stream with knit delta records,
|
|
2494 |
regardless of this test's scenario.
|
|
2495 |
"""
|
|
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
2496 |
mapper = self.get_mapper() |
2497 |
source_transport = self.get_transport('source') |
|
2498 |
source_transport.mkdir('.') |
|
2499 |
source = make_file_factory(False, mapper)(source_transport) |
|
4009.3.1
by Andrew Bennetts
Fix test_insert_record_stream_delta_missing_basis_no_corruption to test what it claims to, and fix KnitVersionedFiles.get_record_stream to match the expected exception. |
2500 |
get_diamond_files(source, self.key_length, trailing_eol=True, |
2501 |
nograph=False, left_only=False) |
|
4009.3.2
by Andrew Bennetts
Add test_insert_record_stream_delta_missing_basis_can_be_added_later. |
2502 |
return source |
2503 |
||
2504 |
def test_insert_record_stream_delta_missing_basis_no_corruption(self): |
|
2505 |
"""Insertion where a needed basis is not included notifies the caller
|
|
2506 |
of the missing basis. In the meantime a record missing its basis is
|
|
2507 |
not added.
|
|
2508 |
"""
|
|
2509 |
source = self.get_knit_delta_source() |
|
4009.3.7
by Andrew Bennetts
Most tests passing. |
2510 |
keys = [self.get_simple_key('origin'), self.get_simple_key('merged')] |
2511 |
entries = source.get_record_stream(keys, 'unordered', False) |
|
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
2512 |
files = self.get_versionedfiles() |
4009.3.2
by Andrew Bennetts
Add test_insert_record_stream_delta_missing_basis_can_be_added_later. |
2513 |
if self.support_partial_insertion: |
4009.3.12
by Robert Collins
Polish on inserting record streams with missing compression parents. |
2514 |
self.assertEqual([], |
2515 |
list(files.get_missing_compression_parent_keys())) |
|
4009.3.2
by Andrew Bennetts
Add test_insert_record_stream_delta_missing_basis_can_be_added_later. |
2516 |
files.insert_record_stream(entries) |
2517 |
missing_bases = files.get_missing_compression_parent_keys() |
|
2518 |
self.assertEqual(set([self.get_simple_key('left')]), |
|
2519 |
set(missing_bases)) |
|
4009.3.7
by Andrew Bennetts
Most tests passing. |
2520 |
self.assertEqual(set(keys), set(files.get_parent_map(keys))) |
4009.3.2
by Andrew Bennetts
Add test_insert_record_stream_delta_missing_basis_can_be_added_later. |
2521 |
else: |
2522 |
self.assertRaises( |
|
2523 |
errors.RevisionNotPresent, files.insert_record_stream, entries) |
|
4009.3.7
by Andrew Bennetts
Most tests passing. |
2524 |
files.check() |
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
2525 |
|
4009.3.2
by Andrew Bennetts
Add test_insert_record_stream_delta_missing_basis_can_be_added_later. |
2526 |
def test_insert_record_stream_delta_missing_basis_can_be_added_later(self): |
2527 |
"""Insertion where a needed basis is not included notifies the caller
|
|
2528 |
of the missing basis. That basis can be added in a second
|
|
2529 |
insert_record_stream call that does not need to repeat records present
|
|
4009.3.3
by Andrew Bennetts
Add docstrings. |
2530 |
in the previous stream. The record(s) that required that basis are
|
2531 |
fully inserted once their basis is no longer missing.
|
|
4009.3.2
by Andrew Bennetts
Add test_insert_record_stream_delta_missing_basis_can_be_added_later. |
2532 |
"""
|
2533 |
if not self.support_partial_insertion: |
|
2534 |
raise TestNotApplicable( |
|
2535 |
'versioned file scenario does not support partial insertion') |
|
2536 |
source = self.get_knit_delta_source() |
|
2537 |
entries = source.get_record_stream([self.get_simple_key('origin'), |
|
2538 |
self.get_simple_key('merged')], 'unordered', False) |
|
2539 |
files = self.get_versionedfiles() |
|
2540 |
files.insert_record_stream(entries) |
|
2541 |
missing_bases = files.get_missing_compression_parent_keys() |
|
2542 |
self.assertEqual(set([self.get_simple_key('left')]), |
|
2543 |
set(missing_bases)) |
|
4009.3.7
by Andrew Bennetts
Most tests passing. |
2544 |
# 'merged' is inserted (although a commit of a write group involving
|
2545 |
# this versionedfiles would fail).
|
|
4009.3.2
by Andrew Bennetts
Add test_insert_record_stream_delta_missing_basis_can_be_added_later. |
2546 |
merged_key = self.get_simple_key('merged') |
4009.3.7
by Andrew Bennetts
Most tests passing. |
2547 |
self.assertEqual( |
2548 |
[merged_key], files.get_parent_map([merged_key]).keys()) |
|
2549 |
# Add the full delta closure of the missing records
|
|
4009.3.2
by Andrew Bennetts
Add test_insert_record_stream_delta_missing_basis_can_be_added_later. |
2550 |
missing_entries = source.get_record_stream( |
4009.3.7
by Andrew Bennetts
Most tests passing. |
2551 |
missing_bases, 'unordered', True) |
4009.3.2
by Andrew Bennetts
Add test_insert_record_stream_delta_missing_basis_can_be_added_later. |
2552 |
files.insert_record_stream(missing_entries) |
4009.3.7
by Andrew Bennetts
Most tests passing. |
2553 |
# Now 'merged' is fully inserted (and a commit would succeed).
|
4009.3.2
by Andrew Bennetts
Add test_insert_record_stream_delta_missing_basis_can_be_added_later. |
2554 |
self.assertEqual([], list(files.get_missing_compression_parent_keys())) |
2555 |
self.assertEqual( |
|
2556 |
[merged_key], files.get_parent_map([merged_key]).keys()) |
|
4009.3.7
by Andrew Bennetts
Most tests passing. |
2557 |
files.check() |
4009.3.2
by Andrew Bennetts
Add test_insert_record_stream_delta_missing_basis_can_be_added_later. |
2558 |
|
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
2559 |
def test_iter_lines_added_or_present_in_keys(self): |
2560 |
# test that we get at least an equalset of the lines added by
|
|
2561 |
# versions in the store.
|
|
2562 |
# the ordering here is to make a tree so that dumb searches have
|
|
2563 |
# more changes to muck up.
|
|
2564 |
||
4961.2.9
by Martin Pool
Rip out most remaining uses of DummyProgressBar |
2565 |
class InstrumentedProgress(progress.ProgressTask): |
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
2566 |
|
2567 |
def __init__(self): |
|
4961.2.9
by Martin Pool
Rip out most remaining uses of DummyProgressBar |
2568 |
progress.ProgressTask.__init__(self) |
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
2569 |
self.updates = [] |
2570 |
||
2571 |
def update(self, msg=None, current=None, total=None): |
|
2572 |
self.updates.append((msg, current, total)) |
|
2573 |
||
2574 |
files = self.get_versionedfiles() |
|
2575 |
# add a base to get included
|
|
2576 |
files.add_lines(self.get_simple_key('base'), (), ['base\n']) |
|
2577 |
# add a ancestor to be included on one side
|
|
2578 |
files.add_lines(self.get_simple_key('lancestor'), (), ['lancestor\n']) |
|
2579 |
# add a ancestor to be included on the other side
|
|
2580 |
files.add_lines(self.get_simple_key('rancestor'), |
|
2581 |
self.get_parents([self.get_simple_key('base')]), ['rancestor\n']) |
|
2582 |
# add a child of rancestor with no eofile-nl
|
|
2583 |
files.add_lines(self.get_simple_key('child'), |
|
2584 |
self.get_parents([self.get_simple_key('rancestor')]), |
|
2585 |
['base\n', 'child\n']) |
|
2586 |
# add a child of lancestor and base to join the two roots
|
|
2587 |
files.add_lines(self.get_simple_key('otherchild'), |
|
2588 |
self.get_parents([self.get_simple_key('lancestor'), |
|
2589 |
self.get_simple_key('base')]), |
|
2590 |
['base\n', 'lancestor\n', 'otherchild\n']) |
|
2591 |
def iter_with_keys(keys, expected): |
|
2592 |
# now we need to see what lines are returned, and how often.
|
|
2593 |
lines = {} |
|
2594 |
progress = InstrumentedProgress() |
|
2595 |
# iterate over the lines
|
|
2596 |
for line in files.iter_lines_added_or_present_in_keys(keys, |
|
2597 |
pb=progress): |
|
2598 |
lines.setdefault(line, 0) |
|
2599 |
lines[line] += 1 |
|
2600 |
if []!= progress.updates: |
|
2601 |
self.assertEqual(expected, progress.updates) |
|
2602 |
return lines |
|
2603 |
lines = iter_with_keys( |
|
2604 |
[self.get_simple_key('child'), self.get_simple_key('otherchild')], |
|
4103.3.4
by Martin Pool
Update test that depends on progress bar strings |
2605 |
[('Walking content', 0, 2), |
2606 |
('Walking content', 1, 2), |
|
2607 |
('Walking content', 2, 2)]) |
|
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
2608 |
# we must see child and otherchild
|
2609 |
self.assertTrue(lines[('child\n', self.get_simple_key('child'))] > 0) |
|
2610 |
self.assertTrue( |
|
2611 |
lines[('otherchild\n', self.get_simple_key('otherchild'))] > 0) |
|
2612 |
# we dont care if we got more than that.
|
|
3943.8.1
by Marius Kruger
remove all trailing whitespace from bzr source |
2613 |
|
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
2614 |
# test all lines
|
2615 |
lines = iter_with_keys(files.keys(), |
|
4103.3.4
by Martin Pool
Update test that depends on progress bar strings |
2616 |
[('Walking content', 0, 5), |
2617 |
('Walking content', 1, 5), |
|
2618 |
('Walking content', 2, 5), |
|
2619 |
('Walking content', 3, 5), |
|
2620 |
('Walking content', 4, 5), |
|
2621 |
('Walking content', 5, 5)]) |
|
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
2622 |
# all lines must be seen at least once
|
2623 |
self.assertTrue(lines[('base\n', self.get_simple_key('base'))] > 0) |
|
2624 |
self.assertTrue( |
|
2625 |
lines[('lancestor\n', self.get_simple_key('lancestor'))] > 0) |
|
2626 |
self.assertTrue( |
|
2627 |
lines[('rancestor\n', self.get_simple_key('rancestor'))] > 0) |
|
2628 |
self.assertTrue(lines[('child\n', self.get_simple_key('child'))] > 0) |
|
2629 |
self.assertTrue( |
|
2630 |
lines[('otherchild\n', self.get_simple_key('otherchild'))] > 0) |
|
2631 |
||
2632 |
def test_make_mpdiffs(self): |
|
2633 |
from bzrlib import multiparent |
|
2634 |
files = self.get_versionedfiles('source') |
|
2635 |
# add texts that should trip the knit maximum delta chain threshold
|
|
2636 |
# as well as doing parallel chains of data in knits.
|
|
2637 |
# this is done by two chains of 25 insertions
|
|
2638 |
files.add_lines(self.get_simple_key('base'), [], ['line\n']) |
|
2639 |
files.add_lines(self.get_simple_key('noeol'), |
|
2640 |
self.get_parents([self.get_simple_key('base')]), ['line']) |
|
2641 |
# detailed eol tests:
|
|
2642 |
# shared last line with parent no-eol
|
|
2643 |
files.add_lines(self.get_simple_key('noeolsecond'), |
|
2644 |
self.get_parents([self.get_simple_key('noeol')]), |
|
2645 |
['line\n', 'line']) |
|
2646 |
# differing last line with parent, both no-eol
|
|
2647 |
files.add_lines(self.get_simple_key('noeolnotshared'), |
|
2648 |
self.get_parents([self.get_simple_key('noeolsecond')]), |
|
2649 |
['line\n', 'phone']) |
|
2650 |
# add eol following a noneol parent, change content
|
|
2651 |
files.add_lines(self.get_simple_key('eol'), |
|
2652 |
self.get_parents([self.get_simple_key('noeol')]), ['phone\n']) |
|
2653 |
# add eol following a noneol parent, no change content
|
|
2654 |
files.add_lines(self.get_simple_key('eolline'), |
|
2655 |
self.get_parents([self.get_simple_key('noeol')]), ['line\n']) |
|
2656 |
# noeol with no parents:
|
|
2657 |
files.add_lines(self.get_simple_key('noeolbase'), [], ['line']) |
|
2658 |
# noeol preceeding its leftmost parent in the output:
|
|
2659 |
# this is done by making it a merge of two parents with no common
|
|
3943.8.1
by Marius Kruger
remove all trailing whitespace from bzr source |
2660 |
# anestry: noeolbase and noeol with the
|
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
2661 |
# later-inserted parent the leftmost.
|
2662 |
files.add_lines(self.get_simple_key('eolbeforefirstparent'), |
|
2663 |
self.get_parents([self.get_simple_key('noeolbase'), |
|
2664 |
self.get_simple_key('noeol')]), |
|
2665 |
['line']) |
|
2666 |
# two identical eol texts
|
|
2667 |
files.add_lines(self.get_simple_key('noeoldup'), |
|
2668 |
self.get_parents([self.get_simple_key('noeol')]), ['line']) |
|
2669 |
next_parent = self.get_simple_key('base') |
|
2670 |
text_name = 'chain1-' |
|
2671 |
text = ['line\n'] |
|
2672 |
sha1s = {0 :'da6d3141cb4a5e6f464bf6e0518042ddc7bfd079', |
|
2673 |
1 :'45e21ea146a81ea44a821737acdb4f9791c8abe7', |
|
2674 |
2 :'e1f11570edf3e2a070052366c582837a4fe4e9fa', |
|
2675 |
3 :'26b4b8626da827088c514b8f9bbe4ebf181edda1', |
|
2676 |
4 :'e28a5510be25ba84d31121cff00956f9970ae6f6', |
|
2677 |
5 :'d63ec0ce22e11dcf65a931b69255d3ac747a318d', |
|
2678 |
6 :'2c2888d288cb5e1d98009d822fedfe6019c6a4ea', |
|
2679 |
7 :'95c14da9cafbf828e3e74a6f016d87926ba234ab', |
|
2680 |
8 :'779e9a0b28f9f832528d4b21e17e168c67697272', |
|
2681 |
9 :'1f8ff4e5c6ff78ac106fcfe6b1e8cb8740ff9a8f', |
|
2682 |
10:'131a2ae712cf51ed62f143e3fbac3d4206c25a05', |
|
2683 |
11:'c5a9d6f520d2515e1ec401a8f8a67e6c3c89f199', |
|
2684 |
12:'31a2286267f24d8bedaa43355f8ad7129509ea85', |
|
2685 |
13:'dc2a7fe80e8ec5cae920973973a8ee28b2da5e0a', |
|
2686 |
14:'2c4b1736566b8ca6051e668de68650686a3922f2', |
|
2687 |
15:'5912e4ecd9b0c07be4d013e7e2bdcf9323276cde', |
|
2688 |
16:'b0d2e18d3559a00580f6b49804c23fea500feab3', |
|
2689 |
17:'8e1d43ad72f7562d7cb8f57ee584e20eb1a69fc7', |
|
2690 |
18:'5cf64a3459ae28efa60239e44b20312d25b253f3', |
|
2691 |
19:'1ebed371807ba5935958ad0884595126e8c4e823', |
|
2692 |
20:'2aa62a8b06fb3b3b892a3292a068ade69d5ee0d3', |
|
2693 |
21:'01edc447978004f6e4e962b417a4ae1955b6fe5d', |
|
2694 |
22:'d8d8dc49c4bf0bab401e0298bb5ad827768618bb', |
|
2695 |
23:'c21f62b1c482862983a8ffb2b0c64b3451876e3f', |
|
2696 |
24:'c0593fe795e00dff6b3c0fe857a074364d5f04fc', |
|
2697 |
25:'dd1a1cf2ba9cc225c3aff729953e6364bf1d1855', |
|
2698 |
}
|
|
2699 |
for depth in range(26): |
|
2700 |
new_version = self.get_simple_key(text_name + '%s' % depth) |
|
2701 |
text = text + ['line\n'] |
|
2702 |
files.add_lines(new_version, self.get_parents([next_parent]), text) |
|
2703 |
next_parent = new_version |
|
2704 |
next_parent = self.get_simple_key('base') |
|
2705 |
text_name = 'chain2-' |
|
2706 |
text = ['line\n'] |
|
2707 |
for depth in range(26): |
|
2708 |
new_version = self.get_simple_key(text_name + '%s' % depth) |
|
2709 |
text = text + ['line\n'] |
|
2710 |
files.add_lines(new_version, self.get_parents([next_parent]), text) |
|
2711 |
next_parent = new_version |
|
2712 |
target = self.get_versionedfiles('target') |
|
2713 |
for key in multiparent.topo_iter_keys(files, files.keys()): |
|
2714 |
mpdiff = files.make_mpdiffs([key])[0] |
|
2715 |
parents = files.get_parent_map([key])[key] or [] |
|
2716 |
target.add_mpdiffs( |
|
3350.8.3
by Robert Collins
VF.get_sha1s needed changing to be stackable. |
2717 |
[(key, parents, files.get_sha1s([key])[key], mpdiff)]) |
3350.6.4
by Robert Collins
First cut at pluralised VersionedFiles. Some rather massive API incompatabilities, primarily because of the difficulty of coherence among competing stores. |
2718 |
self.assertEqualDiff( |
2719 |
files.get_record_stream([key], 'unordered', |
|
2720 |
True).next().get_bytes_as('fulltext'), |
|
2721 |
target.get_record_stream([key], 'unordered', |
|
2722 |
True).next().get_bytes_as('fulltext') |
|
2723 |
)
|
|
2724 |
||
2725 |
def test_keys(self): |
|
2726 |
# While use is discouraged, versions() is still needed by aspects of
|
|
2727 |
# bzr.
|
|
2728 |
files = self.get_versionedfiles() |
|
2729 |
self.assertEqual(set(), set(files.keys())) |
|
2730 |
if self.key_length == 1: |
|
2731 |
key = ('foo',) |
|
2732 |
else: |
|
2733 |
key = ('foo', 'bar',) |
|
2734 |
files.add_lines(key, (), []) |
|
2735 |
self.assertEqual(set([key]), set(files.keys())) |
|
3518.1.1
by Jelmer Vernooij
Add VirtualVersionedFiles class. |
2736 |
|
2737 |
||
2738 |
class VirtualVersionedFilesTests(TestCase): |
|
2739 |
"""Basic tests for the VirtualVersionedFiles implementations."""
|
|
2740 |
||
2741 |
def _get_parent_map(self, keys): |
|
2742 |
ret = {} |
|
2743 |
for k in keys: |
|
2744 |
if k in self._parent_map: |
|
2745 |
ret[k] = self._parent_map[k] |
|
2746 |
return ret |
|
2747 |
||
2748 |
def setUp(self): |
|
2749 |
TestCase.setUp(self) |
|
2750 |
self._lines = {} |
|
2751 |
self._parent_map = {} |
|
3943.8.1
by Marius Kruger
remove all trailing whitespace from bzr source |
2752 |
self.texts = VirtualVersionedFiles(self._get_parent_map, |
3518.1.1
by Jelmer Vernooij
Add VirtualVersionedFiles class. |
2753 |
self._lines.get) |
2754 |
||
2755 |
def test_add_lines(self): |
|
3943.8.1
by Marius Kruger
remove all trailing whitespace from bzr source |
2756 |
self.assertRaises(NotImplementedError, |
3518.1.1
by Jelmer Vernooij
Add VirtualVersionedFiles class. |
2757 |
self.texts.add_lines, "foo", [], []) |
2758 |
||
2759 |
def test_add_mpdiffs(self): |
|
3943.8.1
by Marius Kruger
remove all trailing whitespace from bzr source |
2760 |
self.assertRaises(NotImplementedError, |
3518.1.1
by Jelmer Vernooij
Add VirtualVersionedFiles class. |
2761 |
self.texts.add_mpdiffs, []) |
2762 |
||
4332.3.26
by Robert Collins
Allow passing keys to check to VersionedFile.check(). |
2763 |
def test_check_noerrors(self): |
2764 |
self.texts.check() |
|
3518.1.1
by Jelmer Vernooij
Add VirtualVersionedFiles class. |
2765 |
|
2766 |
def test_insert_record_stream(self): |
|
2767 |
self.assertRaises(NotImplementedError, self.texts.insert_record_stream, |
|
2768 |
[])
|
|
2769 |
||
3518.1.2
by Jelmer Vernooij
Fix some stylistic issues pointed out by Ian. |
2770 |
def test_get_sha1s_nonexistent(self): |
2771 |
self.assertEquals({}, self.texts.get_sha1s([("NONEXISTENT",)])) |
|
3518.1.1
by Jelmer Vernooij
Add VirtualVersionedFiles class. |
2772 |
|
2773 |
def test_get_sha1s(self): |
|
2774 |
self._lines["key"] = ["dataline1", "dataline2"] |
|
2775 |
self.assertEquals({("key",): osutils.sha_strings(self._lines["key"])}, |
|
2776 |
self.texts.get_sha1s([("key",)])) |
|
2777 |
||
2778 |
def test_get_parent_map(self): |
|
2779 |
self._parent_map = {"G": ("A", "B")} |
|
3943.8.1
by Marius Kruger
remove all trailing whitespace from bzr source |
2780 |
self.assertEquals({("G",): (("A",),("B",))}, |
3518.1.1
by Jelmer Vernooij
Add VirtualVersionedFiles class. |
2781 |
self.texts.get_parent_map([("G",), ("L",)])) |
2782 |
||
2783 |
def test_get_record_stream(self): |
|
2784 |
self._lines["A"] = ["FOO", "BAR"] |
|
2785 |
it = self.texts.get_record_stream([("A",)], "unordered", True) |
|
2786 |
record = it.next() |
|
3890.2.2
by John Arbash Meinel
Change the signature to report the storage kind as 'chunked' |
2787 |
self.assertEquals("chunked", record.storage_kind) |
3518.1.1
by Jelmer Vernooij
Add VirtualVersionedFiles class. |
2788 |
self.assertEquals("FOOBAR", record.get_bytes_as("fulltext")) |
3890.2.2
by John Arbash Meinel
Change the signature to report the storage kind as 'chunked' |
2789 |
self.assertEquals(["FOO", "BAR"], record.get_bytes_as("chunked")) |
3518.1.1
by Jelmer Vernooij
Add VirtualVersionedFiles class. |
2790 |
|
2791 |
def test_get_record_stream_absent(self): |
|
2792 |
it = self.texts.get_record_stream([("A",)], "unordered", True) |
|
2793 |
record = it.next() |
|
2794 |
self.assertEquals("absent", record.storage_kind) |
|
2795 |
||
3949.4.1
by Jelmer Vernooij
Implement VirtualVersionedFiles.iter_lines_added_or_present_in_keys. |
2796 |
def test_iter_lines_added_or_present_in_keys(self): |
2797 |
self._lines["A"] = ["FOO", "BAR"] |
|
2798 |
self._lines["B"] = ["HEY"] |
|
2799 |
self._lines["C"] = ["Alberta"] |
|
2800 |
it = self.texts.iter_lines_added_or_present_in_keys([("A",), ("B",)]) |
|
4032.1.2
by John Arbash Meinel
Track down a few more files that have trailing whitespace. |
2801 |
self.assertEquals(sorted([("FOO", "A"), ("BAR", "A"), ("HEY", "B")]), |
3949.4.1
by Jelmer Vernooij
Implement VirtualVersionedFiles.iter_lines_added_or_present_in_keys. |
2802 |
sorted(list(it))) |
2803 |
||
3871.4.1
by John Arbash Meinel
Add a VFDecorator that can yield records in a specified order |
2804 |
|
2805 |
class TestOrderingVersionedFilesDecorator(TestCaseWithMemoryTransport): |
|
2806 |
||
2807 |
def get_ordering_vf(self, key_priority): |
|
2808 |
builder = self.make_branch_builder('test') |
|
2809 |
builder.start_series() |
|
2810 |
builder.build_snapshot('A', None, [ |
|
2811 |
('add', ('', 'TREE_ROOT', 'directory', None))]) |
|
2812 |
builder.build_snapshot('B', ['A'], []) |
|
2813 |
builder.build_snapshot('C', ['B'], []) |
|
2814 |
builder.build_snapshot('D', ['C'], []) |
|
2815 |
builder.finish_series() |
|
2816 |
b = builder.get_branch() |
|
2817 |
b.lock_read() |
|
2818 |
self.addCleanup(b.unlock) |
|
2819 |
vf = b.repository.inventories |
|
2820 |
return versionedfile.OrderingVersionedFilesDecorator(vf, key_priority) |
|
2821 |
||
2822 |
def test_get_empty(self): |
|
2823 |
vf = self.get_ordering_vf({}) |
|
2824 |
self.assertEqual([], vf.calls) |
|
2825 |
||
2826 |
def test_get_record_stream_topological(self): |
|
2827 |
vf = self.get_ordering_vf({('A',): 3, ('B',): 2, ('C',): 4, ('D',): 1}) |
|
2828 |
request_keys = [('B',), ('C',), ('D',), ('A',)] |
|
2829 |
keys = [r.key for r in vf.get_record_stream(request_keys, |
|
2830 |
'topological', False)] |
|
2831 |
# We should have gotten the keys in topological order
|
|
2832 |
self.assertEqual([('A',), ('B',), ('C',), ('D',)], keys) |
|
2833 |
# And recorded that the request was made
|
|
2834 |
self.assertEqual([('get_record_stream', request_keys, 'topological', |
|
2835 |
False)], vf.calls) |
|
2836 |
||
2837 |
def test_get_record_stream_ordered(self): |
|
2838 |
vf = self.get_ordering_vf({('A',): 3, ('B',): 2, ('C',): 4, ('D',): 1}) |
|
2839 |
request_keys = [('B',), ('C',), ('D',), ('A',)] |
|
2840 |
keys = [r.key for r in vf.get_record_stream(request_keys, |
|
2841 |
'unordered', False)] |
|
2842 |
# They should be returned based on their priority
|
|
2843 |
self.assertEqual([('D',), ('B',), ('A',), ('C',)], keys) |
|
2844 |
# And the request recorded
|
|
2845 |
self.assertEqual([('get_record_stream', request_keys, 'unordered', |
|
2846 |
False)], vf.calls) |
|
2847 |
||
2848 |
def test_get_record_stream_implicit_order(self): |
|
2849 |
vf = self.get_ordering_vf({('B',): 2, ('D',): 1}) |
|
2850 |
request_keys = [('B',), ('C',), ('D',), ('A',)] |
|
2851 |
keys = [r.key for r in vf.get_record_stream(request_keys, |
|
2852 |
'unordered', False)] |
|
2853 |
# A and C are not in the map, so they get sorted to the front. A comes
|
|
2854 |
# before C alphabetically, so it comes back first
|
|
2855 |
self.assertEqual([('A',), ('C',), ('D',), ('B',)], keys) |
|
2856 |
# And the request recorded
|
|
2857 |
self.assertEqual([('get_record_stream', request_keys, 'unordered', |
|
2858 |
False)], vf.calls) |