/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar
2052.3.2 by John Arbash Meinel
Change Copyright .. by Canonical to Copyright ... Canonical
1
# Copyright (C) 2005 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
18
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
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
1664.2.9 by Aaron Bentley
Ported weave merge test to versionedfile
24
from StringIO import StringIO
25
1563.2.6 by Robert Collins
Start check tests for knits (pending), and remove dead code.
26
import bzrlib
2039.1.1 by Aaron Bentley
Clean up progress properly when interrupted during fetch (#54000)
27
from bzrlib import (
28
    errors,
2309.4.7 by John Arbash Meinel
Update VersionedFile tests to ensure that they can take Unicode,
29
    osutils,
2039.1.1 by Aaron Bentley
Clean up progress properly when interrupted during fetch (#54000)
30
    progress,
31
    )
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.
32
from bzrlib.errors import (
33
                           RevisionNotPresent, 
34
                           RevisionAlreadyPresent,
35
                           WeaveParentMismatch
36
                           )
2770.1.1 by Aaron Bentley
Initial implmentation of plain knit annotation
37
from bzrlib.knit import (
38
    KnitVersionedFile,
39
    KnitAnnotateFactory,
2770.1.10 by Aaron Bentley
Merge bzr.dev
40
    KnitPlainFactory,
2770.1.1 by Aaron Bentley
Initial implmentation of plain knit annotation
41
    )
3287.5.4 by Robert Collins
Bump the deprecation to 1.4.
42
from bzrlib.symbol_versioning import one_four
2535.3.48 by Andrew Bennetts
Merge from bzr.dev.
43
from bzrlib.tests import TestCaseWithMemoryTransport, TestSkipped
3102.1.1 by Vincent Ladeuil
Rename bzrlib/test/HTTPTestUtils.py to bzrlib/tests/http_utils.py and fix
44
from bzrlib.tests.http_utils import TestCaseWithWebserver
1563.2.1 by Robert Collins
Merge in a variation of the versionedfile api from versioned-file.
45
from bzrlib.trace import mutter
1563.2.16 by Robert Collins
Change WeaveStore into VersionedFileStore and make its versoined file class parameterisable.
46
from bzrlib.transport import get_transport
1563.2.13 by Robert Collins
InterVersionedFile implemented.
47
from bzrlib.transport.memory import MemoryTransport
1684.3.1 by Robert Collins
Fix versioned file joins with empty targets.
48
from bzrlib.tsort import topo_sort
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.
49
import bzrlib.versionedfile as versionedfile
1563.2.9 by Robert Collins
Update versionedfile api tests to ensure that data is available after every operation.
50
from bzrlib.weave import WeaveFile
1664.2.9 by Aaron Bentley
Ported weave merge test to versionedfile
51
from bzrlib.weavefile import read_weave, write_weave
1563.2.1 by Robert Collins
Merge in a variation of the versionedfile api from versioned-file.
52
53
54
class VersionedFileTestMixIn(object):
55
    """A mixin test class for testing VersionedFiles.
56
57
    This is not an adaptor-style test at this point because
58
    theres no dynamic substitution of versioned file implementations,
59
    they are strictly controlled by their owning repositories.
60
    """
61
62
    def test_add(self):
63
        f = self.get_file()
64
        f.add_lines('r0', [], ['a\n', 'b\n'])
65
        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.
66
        def verify_file(f):
67
            versions = f.versions()
68
            self.assertTrue('r0' in versions)
69
            self.assertTrue('r1' in versions)
70
            self.assertEquals(f.get_lines('r0'), ['a\n', 'b\n'])
71
            self.assertEquals(f.get_text('r0'), 'a\nb\n')
72
            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.
73
            self.assertEqual(2, len(f))
74
            self.assertEqual(2, f.num_versions())
1563.2.9 by Robert Collins
Update versionedfile api tests to ensure that data is available after every operation.
75
    
76
            self.assertRaises(RevisionNotPresent,
77
                f.add_lines, 'r2', ['foo'], [])
78
            self.assertRaises(RevisionAlreadyPresent,
79
                f.add_lines, 'r1', [], [])
80
        verify_file(f)
1666.1.6 by Robert Collins
Make knit the default format.
81
        # this checks that reopen with create=True does not break anything.
82
        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.
83
        verify_file(f)
1563.2.1 by Robert Collins
Merge in a variation of the versionedfile api from versioned-file.
84
1596.2.32 by Robert Collins
Reduce re-extraction of texts during weave to knit joins by providing a memoisation facility.
85
    def test_adds_with_parent_texts(self):
86
        f = self.get_file()
87
        parent_texts = {}
2776.1.1 by Robert Collins
* The ``add_lines`` methods on ``VersionedFile`` implementations has changed
88
        _, _, 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.
89
        try:
2776.1.1 by Robert Collins
* The ``add_lines`` methods on ``VersionedFile`` implementations has changed
90
            _, _, parent_texts['r1'] = f.add_lines_with_ghosts('r1',
91
                ['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.
92
        except NotImplementedError:
93
            # 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
94
            _, _, parent_texts['r1'] = f.add_lines('r1',
95
                ['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.
96
        f.add_lines('r2', ['r1'], ['c\n', 'd\n'], parent_texts=parent_texts)
97
        self.assertNotEqual(None, parent_texts['r0'])
98
        self.assertNotEqual(None, parent_texts['r1'])
99
        def verify_file(f):
100
            versions = f.versions()
101
            self.assertTrue('r0' in versions)
102
            self.assertTrue('r1' in versions)
103
            self.assertTrue('r2' in versions)
104
            self.assertEquals(f.get_lines('r0'), ['a\n', 'b\n'])
105
            self.assertEquals(f.get_lines('r1'), ['b\n', 'c\n'])
106
            self.assertEquals(f.get_lines('r2'), ['c\n', 'd\n'])
107
            self.assertEqual(3, f.num_versions())
108
            origins = f.annotate('r1')
109
            self.assertEquals(origins[0][0], 'r0')
110
            self.assertEquals(origins[1][0], 'r1')
111
            origins = f.annotate('r2')
112
            self.assertEquals(origins[0][0], 'r1')
113
            self.assertEquals(origins[1][0], 'r2')
114
115
        verify_file(f)
116
        f = self.reopen_file()
117
        verify_file(f)
118
2805.6.7 by Robert Collins
Review feedback.
119
    def test_add_unicode_content(self):
120
        # unicode content is not permitted in versioned files. 
121
        # versioned files version sequences of bytes only.
122
        vf = self.get_file()
123
        self.assertRaises(errors.BzrBadParameterUnicode,
124
            vf.add_lines, 'a', [], ['a\n', u'b\n', 'c\n'])
125
        self.assertRaises(
126
            (errors.BzrBadParameterUnicode, NotImplementedError),
127
            vf.add_lines_with_ghosts, 'a', [], ['a\n', u'b\n', 'c\n'])
128
2520.4.150 by Aaron Bentley
Test that non-Weave uses left_matching_blocks for add_lines
129
    def test_add_follows_left_matching_blocks(self):
130
        """If we change left_matching_blocks, delta changes
131
132
        Note: There are multiple correct deltas in this case, because
133
        we start with 1 "a" and we get 3.
134
        """
135
        vf = self.get_file()
136
        if isinstance(vf, WeaveFile):
137
            raise TestSkipped("WeaveFile ignores left_matching_blocks")
138
        vf.add_lines('1', [], ['a\n'])
139
        vf.add_lines('2', ['1'], ['a\n', 'a\n', 'a\n'],
140
                     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.
141
        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
142
        vf.add_lines('3', ['1'], ['a\n', 'a\n', 'a\n'],
143
                     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.
144
        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
145
2805.6.7 by Robert Collins
Review feedback.
146
    def test_inline_newline_throws(self):
147
        # \r characters are not permitted in lines being added
148
        vf = self.get_file()
149
        self.assertRaises(errors.BzrBadParameterContainsNewline, 
150
            vf.add_lines, 'a', [], ['a\n\n'])
151
        self.assertRaises(
152
            (errors.BzrBadParameterContainsNewline, NotImplementedError),
153
            vf.add_lines_with_ghosts, 'a', [], ['a\n\n'])
154
        # but inline CR's are allowed
155
        vf.add_lines('a', [], ['a\r\n'])
156
        try:
157
            vf.add_lines_with_ghosts('b', [], ['a\r\n'])
158
        except NotImplementedError:
159
            pass
160
2229.2.1 by Aaron Bentley
Reject reserved ids in versiondfile, tree, branch and repository
161
    def test_add_reserved(self):
162
        vf = self.get_file()
163
        self.assertRaises(errors.ReservedId,
164
            vf.add_lines, 'a:', [], ['a\n', 'b\n', 'c\n'])
165
2794.1.1 by Robert Collins
Allow knits to be instructed not to add a text based on a sha, for commit.
166
    def test_add_lines_nostoresha(self):
167
        """When nostore_sha is supplied using old content raises."""
168
        vf = self.get_file()
169
        empty_text = ('a', [])
170
        sample_text_nl = ('b', ["foo\n", "bar\n"])
171
        sample_text_no_nl = ('c', ["foo\n", "bar"])
172
        shas = []
173
        for version, lines in (empty_text, sample_text_nl, sample_text_no_nl):
174
            sha, _, _ = vf.add_lines(version, [], lines)
175
            shas.append(sha)
176
        # we now have a copy of all the lines in the vf.
177
        for sha, (version, lines) in zip(
178
            shas, (empty_text, sample_text_nl, sample_text_no_nl)):
179
            self.assertRaises(errors.ExistingContent,
180
                vf.add_lines, version + "2", [], lines,
181
                nostore_sha=sha)
182
            # and no new version should have been added.
183
            self.assertRaises(errors.RevisionNotPresent, vf.get_lines,
184
                version + "2")
185
2803.1.1 by Robert Collins
Fix typo in ghosts version of test_add_lines_nostoresha.
186
    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.
187
        """When nostore_sha is supplied using old content raises."""
188
        vf = self.get_file()
189
        empty_text = ('a', [])
190
        sample_text_nl = ('b', ["foo\n", "bar\n"])
191
        sample_text_no_nl = ('c', ["foo\n", "bar"])
192
        shas = []
193
        for version, lines in (empty_text, sample_text_nl, sample_text_no_nl):
194
            sha, _, _ = vf.add_lines(version, [], lines)
195
            shas.append(sha)
196
        # we now have a copy of all the lines in the vf.
197
        # is the test applicable to this vf implementation?
198
        try:
199
            vf.add_lines_with_ghosts('d', [], [])
200
        except NotImplementedError:
201
            raise TestSkipped("add_lines_with_ghosts is optional")
202
        for sha, (version, lines) in zip(
203
            shas, (empty_text, sample_text_nl, sample_text_no_nl)):
204
            self.assertRaises(errors.ExistingContent,
205
                vf.add_lines_with_ghosts, version + "2", [], lines,
206
                nostore_sha=sha)
207
            # and no new version should have been added.
208
            self.assertRaises(errors.RevisionNotPresent, vf.get_lines,
209
                version + "2")
210
2776.1.1 by Robert Collins
* The ``add_lines`` methods on ``VersionedFile`` implementations has changed
211
    def test_add_lines_return_value(self):
212
        # add_lines should return the sha1 and the text size.
213
        vf = self.get_file()
214
        empty_text = ('a', [])
215
        sample_text_nl = ('b', ["foo\n", "bar\n"])
216
        sample_text_no_nl = ('c', ["foo\n", "bar"])
217
        # check results for the three cases:
218
        for version, lines in (empty_text, sample_text_nl, sample_text_no_nl):
219
            # the first two elements are the same for all versioned files:
220
            # - the digest and the size of the text. For some versioned files
221
            #   additional data is returned in additional tuple elements.
222
            result = vf.add_lines(version, [], lines)
223
            self.assertEqual(3, len(result))
224
            self.assertEqual((osutils.sha_strings(lines), sum(map(len, lines))),
225
                result[0:2])
226
        # parents should not affect the result:
227
        lines = sample_text_nl[1]
228
        self.assertEqual((osutils.sha_strings(lines), sum(map(len, lines))),
229
            vf.add_lines('d', ['b', 'c'], lines)[0:2])
230
2229.2.1 by Aaron Bentley
Reject reserved ids in versiondfile, tree, branch and repository
231
    def test_get_reserved(self):
232
        vf = self.get_file()
233
        self.assertRaises(errors.ReservedId, vf.get_texts, ['b:'])
234
        self.assertRaises(errors.ReservedId, vf.get_lines, 'b:')
235
        self.assertRaises(errors.ReservedId, vf.get_text, 'b:')
236
2520.4.85 by Aaron Bentley
Get all test passing (which just proves there aren't enough tests!)
237
    def test_make_mpdiffs(self):
2520.4.3 by Aaron Bentley
Implement plain strategy for extracting and installing multiparent diffs
238
        from bzrlib import multiparent
239
        vf = self.get_file('foo')
240
        sha1s = self._setup_for_deltas(vf)
241
        new_vf = self.get_file('bar')
242
        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!)
243
            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.
244
            new_vf.add_mpdiffs([(version, vf.get_parent_map([version])[version],
2520.4.85 by Aaron Bentley
Get all test passing (which just proves there aren't enough tests!)
245
                                 vf.get_sha1(version), mpdiff)])
2520.4.3 by Aaron Bentley
Implement plain strategy for extracting and installing multiparent diffs
246
            self.assertEqualDiff(vf.get_text(version),
247
                                 new_vf.get_text(version))
248
1596.2.38 by Robert Collins
rollback from using deltas to using fulltexts - deltas need more work to be ready.
249
    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.
250
        self.assertFalse(f.has_version('base'))
1596.2.36 by Robert Collins
add a get_delta api to versioned_file.
251
        # add texts that should trip the knit maximum delta chain threshold
252
        # as well as doing parallel chains of data in knits.
253
        # this is done by two chains of 25 insertions
254
        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.
255
        f.add_lines('noeol', ['base'], ['line'])
256
        # detailed eol tests:
257
        # shared last line with parent no-eol
258
        f.add_lines('noeolsecond', ['noeol'], ['line\n', 'line'])
259
        # differing last line with parent, both no-eol
260
        f.add_lines('noeolnotshared', ['noeolsecond'], ['line\n', 'phone'])
261
        # add eol following a noneol parent, change content
262
        f.add_lines('eol', ['noeol'], ['phone\n'])
263
        # add eol following a noneol parent, no change content
264
        f.add_lines('eolline', ['noeol'], ['line\n'])
265
        # noeol with no parents:
266
        f.add_lines('noeolbase', [], ['line'])
267
        # noeol preceeding its leftmost parent in the output:
268
        # this is done by making it a merge of two parents with no common
269
        # anestry: noeolbase and noeol with the 
270
        # later-inserted parent the leftmost.
271
        f.add_lines('eolbeforefirstparent', ['noeolbase', 'noeol'], ['line'])
272
        # two identical eol texts
273
        f.add_lines('noeoldup', ['noeol'], ['line'])
1596.2.36 by Robert Collins
add a get_delta api to versioned_file.
274
        next_parent = 'base'
275
        text_name = 'chain1-'
276
        text = ['line\n']
277
        sha1s = {0 :'da6d3141cb4a5e6f464bf6e0518042ddc7bfd079',
278
                 1 :'45e21ea146a81ea44a821737acdb4f9791c8abe7',
279
                 2 :'e1f11570edf3e2a070052366c582837a4fe4e9fa',
280
                 3 :'26b4b8626da827088c514b8f9bbe4ebf181edda1',
281
                 4 :'e28a5510be25ba84d31121cff00956f9970ae6f6',
282
                 5 :'d63ec0ce22e11dcf65a931b69255d3ac747a318d',
283
                 6 :'2c2888d288cb5e1d98009d822fedfe6019c6a4ea',
284
                 7 :'95c14da9cafbf828e3e74a6f016d87926ba234ab',
285
                 8 :'779e9a0b28f9f832528d4b21e17e168c67697272',
286
                 9 :'1f8ff4e5c6ff78ac106fcfe6b1e8cb8740ff9a8f',
287
                 10:'131a2ae712cf51ed62f143e3fbac3d4206c25a05',
288
                 11:'c5a9d6f520d2515e1ec401a8f8a67e6c3c89f199',
289
                 12:'31a2286267f24d8bedaa43355f8ad7129509ea85',
290
                 13:'dc2a7fe80e8ec5cae920973973a8ee28b2da5e0a',
291
                 14:'2c4b1736566b8ca6051e668de68650686a3922f2',
292
                 15:'5912e4ecd9b0c07be4d013e7e2bdcf9323276cde',
293
                 16:'b0d2e18d3559a00580f6b49804c23fea500feab3',
294
                 17:'8e1d43ad72f7562d7cb8f57ee584e20eb1a69fc7',
295
                 18:'5cf64a3459ae28efa60239e44b20312d25b253f3',
296
                 19:'1ebed371807ba5935958ad0884595126e8c4e823',
297
                 20:'2aa62a8b06fb3b3b892a3292a068ade69d5ee0d3',
298
                 21:'01edc447978004f6e4e962b417a4ae1955b6fe5d',
299
                 22:'d8d8dc49c4bf0bab401e0298bb5ad827768618bb',
300
                 23:'c21f62b1c482862983a8ffb2b0c64b3451876e3f',
301
                 24:'c0593fe795e00dff6b3c0fe857a074364d5f04fc',
302
                 25:'dd1a1cf2ba9cc225c3aff729953e6364bf1d1855',
303
                 }
304
        for depth in range(26):
305
            new_version = text_name + '%s' % depth
306
            text = text + ['line\n']
307
            f.add_lines(new_version, [next_parent], text)
308
            next_parent = new_version
309
        next_parent = 'base'
310
        text_name = 'chain2-'
311
        text = ['line\n']
312
        for depth in range(26):
313
            new_version = text_name + '%s' % depth
314
            text = text + ['line\n']
315
            f.add_lines(new_version, [next_parent], text)
316
            next_parent = new_version
1596.2.38 by Robert Collins
rollback from using deltas to using fulltexts - deltas need more work to be ready.
317
        return sha1s
1596.2.37 by Robert Collins
Switch to delta based content copying in the generic versioned file copier.
318
1563.2.1 by Robert Collins
Merge in a variation of the versionedfile api from versioned-file.
319
    def test_ancestry(self):
320
        f = self.get_file()
1563.2.29 by Robert Collins
Remove all but fetch references to repository.revision_store.
321
        self.assertEqual([], f.get_ancestry([]))
1563.2.1 by Robert Collins
Merge in a variation of the versionedfile api from versioned-file.
322
        f.add_lines('r0', [], ['a\n', 'b\n'])
323
        f.add_lines('r1', ['r0'], ['b\n', 'c\n'])
324
        f.add_lines('r2', ['r0'], ['b\n', 'c\n'])
325
        f.add_lines('r3', ['r2'], ['b\n', 'c\n'])
326
        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.
327
        self.assertEqual([], f.get_ancestry([]))
1563.2.35 by Robert Collins
cleanup deprecation warnings and finish conversion so the inventory is knit based too.
328
        versions = f.get_ancestry(['rM'])
329
        # there are some possibilities:
330
        # r0 r1 r2 rM r3
331
        # r0 r1 r2 r3 rM
332
        # etc
333
        # so we check indexes
334
        r0 = versions.index('r0')
335
        r1 = versions.index('r1')
336
        r2 = versions.index('r2')
337
        self.assertFalse('r3' in versions)
338
        rM = versions.index('rM')
339
        self.assertTrue(r0 < r1)
340
        self.assertTrue(r0 < r2)
341
        self.assertTrue(r1 < rM)
342
        self.assertTrue(r2 < rM)
1563.2.1 by Robert Collins
Merge in a variation of the versionedfile api from versioned-file.
343
344
        self.assertRaises(RevisionNotPresent,
345
            f.get_ancestry, ['rM', 'rX'])
1594.2.21 by Robert Collins
Teach versioned files to prevent mutation after finishing.
346
2530.1.1 by Aaron Bentley
Make topological sorting optional for get_ancestry
347
        self.assertEqual(set(f.get_ancestry('rM')),
348
            set(f.get_ancestry('rM', topo_sorted=False)))
349
1594.2.21 by Robert Collins
Teach versioned files to prevent mutation after finishing.
350
    def test_mutate_after_finish(self):
351
        f = self.get_file()
352
        f.transaction_finished()
353
        self.assertRaises(errors.OutSideTransaction, f.add_lines, '', [], [])
354
        self.assertRaises(errors.OutSideTransaction, f.add_lines_with_ghosts, '', [], [])
355
        self.assertRaises(errors.OutSideTransaction, f.join, '')
1594.2.24 by Robert Collins
Make use of the transaction finalisation warning support to implement in-knit caching.
356
        self.assertRaises(errors.OutSideTransaction, f.clone_text, 'base', 'bar', ['foo'])
1563.2.7 by Robert Collins
add versioned file clear_cache entry.
357
        
358
    def test_clear_cache(self):
359
        f = self.get_file()
360
        # on a new file it should not error
361
        f.clear_cache()
362
        # and after adding content, doing a clear_cache and a get should work.
363
        f.add_lines('0', [], ['a'])
364
        f.clear_cache()
365
        self.assertEqual(['a'], f.get_lines('0'))
1563.2.1 by Robert Collins
Merge in a variation of the versionedfile api from versioned-file.
366
367
    def test_clone_text(self):
368
        f = self.get_file()
369
        f.add_lines('r0', [], ['a\n', 'b\n'])
1563.2.5 by Robert Collins
Remove unused transaction references from knit.py and the versionedfile interface.
370
        f.clone_text('r1', 'r0', ['r0'])
1563.2.9 by Robert Collins
Update versionedfile api tests to ensure that data is available after every operation.
371
        def verify_file(f):
372
            self.assertEquals(f.get_lines('r1'), f.get_lines('r0'))
373
            self.assertEquals(f.get_lines('r1'), ['a\n', 'b\n'])
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.
374
            self.assertEqual({'r1':('r0',)}, f.get_parent_map(['r1']))
1563.2.9 by Robert Collins
Update versionedfile api tests to ensure that data is available after every operation.
375
            self.assertRaises(RevisionNotPresent,
376
                f.clone_text, 'r2', 'rX', [])
377
            self.assertRaises(RevisionAlreadyPresent,
378
                f.clone_text, 'r1', 'r0', [])
379
        verify_file(f)
380
        verify_file(self.reopen_file())
1563.2.1 by Robert Collins
Merge in a variation of the versionedfile api from versioned-file.
381
1563.2.13 by Robert Collins
InterVersionedFile implemented.
382
    def test_create_empty(self):
383
        f = self.get_file()
384
        f.add_lines('0', [], ['a\n'])
385
        new_f = f.create_empty('t', MemoryTransport())
386
        # smoke test, specific types should check it is honoured correctly for
387
        # non type attributes
388
        self.assertEqual([], new_f.versions())
389
        self.assertTrue(isinstance(new_f, f.__class__))
390
1563.2.15 by Robert Collins
remove the weavestore assumptions about the number and nature of files it manages.
391
    def test_copy_to(self):
392
        f = self.get_file()
393
        f.add_lines('0', [], ['a\n'])
394
        t = MemoryTransport()
395
        f.copy_to('foo', t)
396
        for suffix in f.__class__.get_suffixes():
397
            self.assertTrue(t.has('foo' + suffix))
398
399
    def test_get_suffixes(self):
400
        f = self.get_file()
401
        # should be the same
402
        self.assertEqual(f.__class__.get_suffixes(), f.__class__.get_suffixes())
403
        # and should be a list
404
        self.assertTrue(isinstance(f.__class__.get_suffixes(), list))
405
1684.3.1 by Robert Collins
Fix versioned file joins with empty targets.
406
    def build_graph(self, file, graph):
407
        for node in topo_sort(graph.items()):
408
            file.add_lines(node, graph[node], [])
409
1563.2.13 by Robert Collins
InterVersionedFile implemented.
410
    def test_get_graph(self):
411
        f = self.get_file()
1684.3.1 by Robert Collins
Fix versioned file joins with empty targets.
412
        graph = {
2592.3.43 by Robert Collins
A knit iter_parents API.
413
            'v1': (),
414
            'v2': ('v1', ),
415
            'v3': ('v2', )}
1684.3.1 by Robert Collins
Fix versioned file joins with empty targets.
416
        self.build_graph(f, graph)
417
        self.assertEqual(graph, f.get_graph())
418
    
419
    def test_get_graph_partial(self):
420
        f = self.get_file()
421
        complex_graph = {}
422
        simple_a = {
2592.3.43 by Robert Collins
A knit iter_parents API.
423
            'c': (),
424
            'b': ('c', ),
425
            'a': ('b', ),
1684.3.1 by Robert Collins
Fix versioned file joins with empty targets.
426
            }
427
        complex_graph.update(simple_a)
428
        simple_b = {
2592.3.43 by Robert Collins
A knit iter_parents API.
429
            'c': (),
430
            'b': ('c', ),
1684.3.1 by Robert Collins
Fix versioned file joins with empty targets.
431
            }
432
        complex_graph.update(simple_b)
433
        simple_gam = {
2592.3.43 by Robert Collins
A knit iter_parents API.
434
            'c': (),
435
            'oo': (),
436
            'bar': ('oo', 'c'),
437
            'gam': ('bar', ),
1684.3.1 by Robert Collins
Fix versioned file joins with empty targets.
438
            }
439
        complex_graph.update(simple_gam)
440
        simple_b_gam = {}
441
        simple_b_gam.update(simple_gam)
442
        simple_b_gam.update(simple_b)
443
        self.build_graph(f, complex_graph)
444
        self.assertEqual(simple_a, f.get_graph(['a']))
445
        self.assertEqual(simple_b, f.get_graph(['b']))
446
        self.assertEqual(simple_gam, f.get_graph(['gam']))
447
        self.assertEqual(simple_b_gam, f.get_graph(['b', 'gam']))
1563.2.13 by Robert Collins
InterVersionedFile implemented.
448
1563.2.1 by Robert Collins
Merge in a variation of the versionedfile api from versioned-file.
449
    def test_get_parents(self):
450
        f = self.get_file()
451
        f.add_lines('r0', [], ['a\n', 'b\n'])
452
        f.add_lines('r1', [], ['a\n', 'b\n'])
453
        f.add_lines('r2', [], ['a\n', 'b\n'])
454
        f.add_lines('r3', [], ['a\n', 'b\n'])
455
        f.add_lines('m', ['r0', 'r1', 'r2', 'r3'], ['a\n', 'b\n'])
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.
456
        self.assertEqual(['r0', 'r1', 'r2', 'r3'],
3287.5.4 by Robert Collins
Bump the deprecation to 1.4.
457
            self.applyDeprecated(one_four, f.get_parents, 'm'))
1563.2.1 by Robert Collins
Merge in a variation of the versionedfile api from versioned-file.
458
        self.assertRaises(RevisionNotPresent,
3287.5.4 by Robert Collins
Bump the deprecation to 1.4.
459
            self.applyDeprecated, one_four, f.get_parents, 'y')
1563.2.1 by Robert Collins
Merge in a variation of the versionedfile api from versioned-file.
460
3287.5.1 by Robert Collins
Add VersionedFile.get_parent_map.
461
    def test_get_parent_map(self):
462
        f = self.get_file()
463
        f.add_lines('r0', [], ['a\n', 'b\n'])
464
        self.assertEqual(
465
            {'r0':()}, f.get_parent_map(['r0']))
466
        f.add_lines('r1', ['r0'], ['a\n', 'b\n'])
467
        self.assertEqual(
468
            {'r1':('r0',)}, f.get_parent_map(['r1']))
469
        self.assertEqual(
470
            {'r0':(),
471
             'r1':('r0',)},
472
            f.get_parent_map(['r0', 'r1']))
473
        f.add_lines('r2', [], ['a\n', 'b\n'])
474
        f.add_lines('r3', [], ['a\n', 'b\n'])
475
        f.add_lines('m', ['r0', 'r1', 'r2', 'r3'], ['a\n', 'b\n'])
476
        self.assertEqual(
477
            {'m':('r0', 'r1', 'r2', 'r3')}, f.get_parent_map(['m']))
478
        self.assertEqual({}, f.get_parent_map('y'))
479
        self.assertEqual(
480
            {'r0':(),
481
             'r1':('r0',)},
482
            f.get_parent_map(['r0', 'y', 'r1']))
483
1563.2.1 by Robert Collins
Merge in a variation of the versionedfile api from versioned-file.
484
    def test_annotate(self):
485
        f = self.get_file()
486
        f.add_lines('r0', [], ['a\n', 'b\n'])
487
        f.add_lines('r1', ['r0'], ['c\n', 'b\n'])
488
        origins = f.annotate('r1')
489
        self.assertEquals(origins[0][0], 'r1')
490
        self.assertEquals(origins[1][0], 'r0')
491
492
        self.assertRaises(RevisionNotPresent,
493
            f.annotate, 'foo')
494
1563.2.6 by Robert Collins
Start check tests for knits (pending), and remove dead code.
495
    def test_detection(self):
496
        # Test weaves detect corruption.
497
        #
498
        # Weaves contain a checksum of their texts.
499
        # When a text is extracted, this checksum should be
500
        # verified.
501
502
        w = self.get_file_corrupted_text()
503
504
        self.assertEqual('hello\n', w.get_text('v1'))
505
        self.assertRaises(errors.WeaveInvalidChecksum, w.get_text, 'v2')
506
        self.assertRaises(errors.WeaveInvalidChecksum, w.get_lines, 'v2')
507
        self.assertRaises(errors.WeaveInvalidChecksum, w.check)
508
509
        w = self.get_file_corrupted_checksum()
510
511
        self.assertEqual('hello\n', w.get_text('v1'))
512
        self.assertRaises(errors.WeaveInvalidChecksum, w.get_text, 'v2')
513
        self.assertRaises(errors.WeaveInvalidChecksum, w.get_lines, 'v2')
514
        self.assertRaises(errors.WeaveInvalidChecksum, w.check)
515
516
    def get_file_corrupted_text(self):
517
        """Return a versioned file with corrupt text but valid metadata."""
518
        raise NotImplementedError(self.get_file_corrupted_text)
519
1563.2.9 by Robert Collins
Update versionedfile api tests to ensure that data is available after every operation.
520
    def reopen_file(self, name='foo'):
521
        """Open the versioned file from disk again."""
522
        raise NotImplementedError(self.reopen_file)
523
2592.3.43 by Robert Collins
A knit iter_parents API.
524
    def test_iter_parents(self):
525
        """iter_parents returns the parents for many nodes."""
526
        f = self.get_file()
527
        # sample data:
528
        # no parents
529
        f.add_lines('r0', [], ['a\n', 'b\n'])
530
        # 1 parents
531
        f.add_lines('r1', ['r0'], ['a\n', 'b\n'])
532
        # 2 parents
533
        f.add_lines('r2', ['r1', 'r0'], ['a\n', 'b\n'])
534
        # XXX TODO a ghost
535
        # cases: each sample data individually:
536
        self.assertEqual(set([('r0', ())]),
537
            set(f.iter_parents(['r0'])))
538
        self.assertEqual(set([('r1', ('r0', ))]),
539
            set(f.iter_parents(['r1'])))
540
        self.assertEqual(set([('r2', ('r1', 'r0'))]),
541
            set(f.iter_parents(['r2'])))
542
        # no nodes returned for a missing node
543
        self.assertEqual(set(),
544
            set(f.iter_parents(['missing'])))
545
        # 1 node returned with missing nodes skipped
546
        self.assertEqual(set([('r1', ('r0', ))]),
547
            set(f.iter_parents(['ghost1', 'r1', 'ghost'])))
548
        # 2 nodes returned
549
        self.assertEqual(set([('r0', ()), ('r1', ('r0', ))]),
550
            set(f.iter_parents(['r0', 'r1'])))
551
        # 2 nodes returned, missing skipped
552
        self.assertEqual(set([('r0', ()), ('r1', ('r0', ))]),
553
            set(f.iter_parents(['a', 'r0', 'b', 'r1', 'c'])))
554
1594.2.6 by Robert Collins
Introduce a api specifically for looking at lines in some versions of the inventory, for fileid_involved.
555
    def test_iter_lines_added_or_present_in_versions(self):
556
        # test that we get at least an equalset of the lines added by
557
        # versions in the weave 
558
        # the ordering here is to make a tree so that dumb searches have
559
        # more changes to muck up.
2039.1.1 by Aaron Bentley
Clean up progress properly when interrupted during fetch (#54000)
560
561
        class InstrumentedProgress(progress.DummyProgress):
562
563
            def __init__(self):
564
565
                progress.DummyProgress.__init__(self)
566
                self.updates = []
567
568
            def update(self, msg=None, current=None, total=None):
569
                self.updates.append((msg, current, total))
570
1594.2.6 by Robert Collins
Introduce a api specifically for looking at lines in some versions of the inventory, for fileid_involved.
571
        vf = self.get_file()
572
        # add a base to get included
573
        vf.add_lines('base', [], ['base\n'])
574
        # add a ancestor to be included on one side
575
        vf.add_lines('lancestor', [], ['lancestor\n'])
576
        # add a ancestor to be included on the other side
577
        vf.add_lines('rancestor', ['base'], ['rancestor\n'])
578
        # add a child of rancestor with no eofile-nl
579
        vf.add_lines('child', ['rancestor'], ['base\n', 'child\n'])
580
        # add a child of lancestor and base to join the two roots
581
        vf.add_lines('otherchild',
582
                     ['lancestor', 'base'],
583
                     ['base\n', 'lancestor\n', 'otherchild\n'])
2039.1.1 by Aaron Bentley
Clean up progress properly when interrupted during fetch (#54000)
584
        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.
585
            # now we need to see what lines are returned, and how often.
2975.3.1 by Robert Collins
Change (without backwards compatibility) the
586
            lines = {}
2039.1.1 by Aaron Bentley
Clean up progress properly when interrupted during fetch (#54000)
587
            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.
588
            # iterate over the lines
2975.3.1 by Robert Collins
Change (without backwards compatibility) the
589
            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)
590
                pb=progress):
2975.3.1 by Robert Collins
Change (without backwards compatibility) the
591
                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.
592
                lines[line] += 1
2975.3.1 by Robert Collins
Change (without backwards compatibility) the
593
            if []!= progress.updates:
2039.1.2 by Aaron Bentley
Tweak test to avoid catching assert
594
                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.
595
            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.
596
        lines = iter_with_versions(['child', 'otherchild'],
597
                                   [('Walking content.', 0, 2),
598
                                    ('Walking content.', 1, 2),
2039.1.1 by Aaron Bentley
Clean up progress properly when interrupted during fetch (#54000)
599
                                    ('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.
600
        # we must see child and otherchild
2975.3.1 by Robert Collins
Change (without backwards compatibility) the
601
        self.assertTrue(lines[('child\n', 'child')] > 0)
602
        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.
603
        # we dont care if we got more than that.
604
        
605
        # test all 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.
606
        lines = iter_with_versions(None, [('Walking content.', 0, 5),
607
                                          ('Walking content.', 1, 5),
608
                                          ('Walking content.', 2, 5),
609
                                          ('Walking content.', 3, 5),
610
                                          ('Walking content.', 4, 5),
2039.1.1 by Aaron Bentley
Clean up progress properly when interrupted during fetch (#54000)
611
                                          ('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.
612
        # all lines must be seen at least once
2975.3.1 by Robert Collins
Change (without backwards compatibility) the
613
        self.assertTrue(lines[('base\n', 'base')] > 0)
614
        self.assertTrue(lines[('lancestor\n', 'lancestor')] > 0)
615
        self.assertTrue(lines[('rancestor\n', 'rancestor')] > 0)
616
        self.assertTrue(lines[('child\n', 'child')] > 0)
617
        self.assertTrue(lines[('otherchild\n', 'otherchild')] > 0)
1594.2.7 by Robert Collins
Add versionedfile.fix_parents api for correcting data post hoc.
618
1594.2.8 by Robert Collins
add ghost aware apis to knits.
619
    def test_add_lines_with_ghosts(self):
620
        # some versioned file formats allow lines to be added with parent
621
        # information that is > than that in the format. Formats that do
622
        # not support this need to raise NotImplementedError on the
623
        # add_lines_with_ghosts api.
624
        vf = self.get_file()
625
        # 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
626
        # The preferred form is utf8, but we should translate when needed
627
        parent_id_unicode = u'b\xbfse'
628
        parent_id_utf8 = parent_id_unicode.encode('utf8')
1594.2.8 by Robert Collins
add ghost aware apis to knits.
629
        try:
2309.4.7 by John Arbash Meinel
Update VersionedFile tests to ensure that they can take Unicode,
630
            vf.add_lines_with_ghosts('notbxbfse', [parent_id_utf8], [])
1594.2.8 by Robert Collins
add ghost aware apis to knits.
631
        except NotImplementedError:
632
            # check the other ghost apis are also not implemented
633
            self.assertRaises(NotImplementedError, vf.get_ancestry_with_ghosts, ['foo'])
634
            self.assertRaises(NotImplementedError, vf.get_parents_with_ghosts, 'foo')
635
            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.
636
        vf = self.reopen_file()
1594.2.8 by Robert Collins
add ghost aware apis to knits.
637
        # test key graph related apis: getncestry, _graph, get_parents
638
        # has_version
639
        # - 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
640
        self.assertEqual(['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.
641
        self.assertEqual([],
3287.5.4 by Robert Collins
Bump the deprecation to 1.4.
642
            self.applyDeprecated(one_four, vf.get_parents, 'notbxbfse'))
2592.3.43 by Robert Collins
A knit iter_parents API.
643
        self.assertEqual({'notbxbfse':()}, vf.get_graph())
2249.5.12 by John Arbash Meinel
Change the APIs for VersionedFile, Store, and some of Repository into utf-8
644
        self.assertFalse(vf.has_version(parent_id_utf8))
1594.2.8 by Robert Collins
add ghost aware apis to knits.
645
        # 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
646
        self.assertEqual([parent_id_utf8, 'notbxbfse'], vf.get_ancestry_with_ghosts(['notbxbfse']))
647
        self.assertEqual([parent_id_utf8], vf.get_parents_with_ghosts('notbxbfse'))
3287.6.7 by Robert Collins
* ``VersionedFile.get_graph_with_ghosts`` is deprecated, with no
648
        self.assertEqual({'notbxbfse':(parent_id_utf8,)},
649
            self.applyDeprecated(one_four, vf.get_graph_with_ghosts))
3287.6.5 by Robert Collins
Deprecate VersionedFile.has_ghost.
650
        self.assertTrue(self.applyDeprecated(one_four, vf.has_ghost,
651
            parent_id_utf8))
1594.2.8 by Robert Collins
add ghost aware apis to knits.
652
        # if we add something that is a ghost of another, it should correct the
653
        # results of the prior apis
2858.2.1 by Martin Pool
Remove most calls to safe_file_id and safe_revision_id.
654
        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
655
        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.
656
        self.assertEqual({'notbxbfse':(parent_id_utf8,)},
657
            vf.get_parent_map(['notbxbfse']))
2592.3.43 by Robert Collins
A knit iter_parents API.
658
        self.assertEqual({parent_id_utf8:(),
659
                          'notbxbfse':(parent_id_utf8, ),
1594.2.8 by Robert Collins
add ghost aware apis to knits.
660
                          },
661
                         vf.get_graph())
2249.5.12 by John Arbash Meinel
Change the APIs for VersionedFile, Store, and some of Repository into utf-8
662
        self.assertTrue(vf.has_version(parent_id_utf8))
1594.2.8 by Robert Collins
add ghost aware apis to knits.
663
        # 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.
664
        self.assertEqual([parent_id_utf8, 'notbxbfse'],
665
            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
666
        self.assertEqual([parent_id_utf8], vf.get_parents_with_ghosts('notbxbfse'))
3287.5.5 by Robert Collins
Refactor internals of knit implementations to implement get_parents_with_ghosts in terms of get_parent_map.
667
        self.assertEqual({parent_id_utf8:(),
668
                          'notbxbfse':(parent_id_utf8,),
1594.2.8 by Robert Collins
add ghost aware apis to knits.
669
                          },
3287.6.7 by Robert Collins
* ``VersionedFile.get_graph_with_ghosts`` is deprecated, with no
670
            self.applyDeprecated(one_four, vf.get_graph_with_ghosts))
3287.6.5 by Robert Collins
Deprecate VersionedFile.has_ghost.
671
        self.assertFalse(self.applyDeprecated(one_four, vf.has_ghost,
672
            parent_id_utf8))
1594.2.8 by Robert Collins
add ghost aware apis to knits.
673
1594.2.9 by Robert Collins
Teach Knit repositories how to handle ghosts without corrupting at all.
674
    def test_add_lines_with_ghosts_after_normal_revs(self):
675
        # some versioned file formats allow lines to be added with parent
676
        # information that is > than that in the format. Formats that do
677
        # not support this need to raise NotImplementedError on the
678
        # add_lines_with_ghosts api.
679
        vf = self.get_file()
680
        # probe for ghost support
681
        try:
3287.6.5 by Robert Collins
Deprecate VersionedFile.has_ghost.
682
            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.
683
        except NotImplementedError:
684
            return
685
        vf.add_lines_with_ghosts('references_ghost',
686
                                 ['base', 'a_ghost'],
687
                                 ['line\n', 'line_b\n', 'line_c\n'])
688
        origins = vf.annotate('references_ghost')
689
        self.assertEquals(('base', 'line\n'), origins[0])
690
        self.assertEquals(('base', 'line_b\n'), origins[1])
691
        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.
692
693
    def test_readonly_mode(self):
694
        transport = get_transport(self.get_url('.'))
695
        factory = self.get_factory()
696
        vf = factory('id', transport, 0777, create=True, access_mode='w')
697
        vf = factory('id', transport, access_mode='r')
698
        self.assertRaises(errors.ReadOnlyError, vf.add_lines, 'base', [], [])
699
        self.assertRaises(errors.ReadOnlyError,
700
                          vf.add_lines_with_ghosts,
701
                          'base',
702
                          [],
703
                          [])
704
        self.assertRaises(errors.ReadOnlyError, vf.join, 'base')
1594.2.24 by Robert Collins
Make use of the transaction finalisation warning support to implement in-knit caching.
705
        self.assertRaises(errors.ReadOnlyError, vf.clone_text, 'base', 'bar', ['foo'])
1666.1.6 by Robert Collins
Make knit the default format.
706
    
707
    def test_get_sha1(self):
708
        # check the sha1 data is available
709
        vf = self.get_file()
710
        # a simple file
711
        vf.add_lines('a', [], ['a\n'])
712
        # the same file, different metadata
713
        vf.add_lines('b', ['a'], ['a\n'])
714
        # a file differing only in last newline.
715
        vf.add_lines('c', [], ['a'])
716
        self.assertEqual(
717
            '3f786850e387550fdab836ed7e6dc881de23001b', vf.get_sha1('a'))
718
        self.assertEqual(
719
            '3f786850e387550fdab836ed7e6dc881de23001b', vf.get_sha1('b'))
720
        self.assertEqual(
721
            '86f7e437faa5a7fce15d1ddcb9eaeaea377667b8', vf.get_sha1('c'))
2520.4.89 by Aaron Bentley
Add get_sha1s to weaves
722
723
        self.assertEqual(['3f786850e387550fdab836ed7e6dc881de23001b',
724
                          '86f7e437faa5a7fce15d1ddcb9eaeaea377667b8',
725
                          '3f786850e387550fdab836ed7e6dc881de23001b'],
726
                          vf.get_sha1s(['a', 'c', 'b']))
1594.2.9 by Robert Collins
Teach Knit repositories how to handle ghosts without corrupting at all.
727
        
1563.2.1 by Robert Collins
Merge in a variation of the versionedfile api from versioned-file.
728
2535.3.1 by Andrew Bennetts
Add get_format_signature to VersionedFile
729
class TestWeave(TestCaseWithMemoryTransport, VersionedFileTestMixIn):
1563.2.1 by Robert Collins
Merge in a variation of the versionedfile api from versioned-file.
730
731
    def get_file(self, name='foo'):
1563.2.25 by Robert Collins
Merge in upstream.
732
        return WeaveFile(name, get_transport(self.get_url('.')), create=True)
1563.2.1 by Robert Collins
Merge in a variation of the versionedfile api from versioned-file.
733
1563.2.6 by Robert Collins
Start check tests for knits (pending), and remove dead code.
734
    def get_file_corrupted_text(self):
1563.2.25 by Robert Collins
Merge in upstream.
735
        w = WeaveFile('foo', get_transport(self.get_url('.')), create=True)
1563.2.13 by Robert Collins
InterVersionedFile implemented.
736
        w.add_lines('v1', [], ['hello\n'])
737
        w.add_lines('v2', ['v1'], ['hello\n', 'there\n'])
1563.2.6 by Robert Collins
Start check tests for knits (pending), and remove dead code.
738
        
739
        # We are going to invasively corrupt the text
740
        # Make sure the internals of weave are the same
741
        self.assertEqual([('{', 0)
742
                        , 'hello\n'
743
                        , ('}', None)
744
                        , ('{', 1)
745
                        , 'there\n'
746
                        , ('}', None)
747
                        ], w._weave)
748
        
749
        self.assertEqual(['f572d396fae9206628714fb2ce00f72e94f2258f'
750
                        , '90f265c6e75f1c8f9ab76dcf85528352c5f215ef'
751
                        ], w._sha1s)
752
        w.check()
753
        
754
        # Corrupted
755
        w._weave[4] = 'There\n'
756
        return w
757
758
    def get_file_corrupted_checksum(self):
759
        w = self.get_file_corrupted_text()
760
        # Corrected
761
        w._weave[4] = 'there\n'
762
        self.assertEqual('hello\nthere\n', w.get_text('v2'))
763
        
764
        #Invalid checksum, first digit changed
765
        w._sha1s[1] =  'f0f265c6e75f1c8f9ab76dcf85528352c5f215ef'
766
        return w
767
1666.1.6 by Robert Collins
Make knit the default format.
768
    def reopen_file(self, name='foo', create=False):
769
        return WeaveFile(name, get_transport(self.get_url('.')), create=create)
1563.2.9 by Robert Collins
Update versionedfile api tests to ensure that data is available after every operation.
770
1563.2.25 by Robert Collins
Merge in upstream.
771
    def test_no_implicit_create(self):
772
        self.assertRaises(errors.NoSuchFile,
773
                          WeaveFile,
774
                          'foo',
775
                          get_transport(self.get_url('.')))
776
1594.2.23 by Robert Collins
Test versioned file storage handling of clean/dirty status for accessed versioned files.
777
    def get_factory(self):
778
        return WeaveFile
779
1563.2.1 by Robert Collins
Merge in a variation of the versionedfile api from versioned-file.
780
2535.3.1 by Andrew Bennetts
Add get_format_signature to VersionedFile
781
class TestKnit(TestCaseWithMemoryTransport, VersionedFileTestMixIn):
1563.2.4 by Robert Collins
First cut at including the knit implementation of versioned_file.
782
783
    def get_file(self, name='foo'):
2770.1.1 by Aaron Bentley
Initial implmentation of plain knit annotation
784
        return self.get_factory()(name, get_transport(self.get_url('.')),
785
                                  delta=True, create=True)
1563.2.6 by Robert Collins
Start check tests for knits (pending), and remove dead code.
786
1594.2.23 by Robert Collins
Test versioned file storage handling of clean/dirty status for accessed versioned files.
787
    def get_factory(self):
788
        return KnitVersionedFile
789
1563.2.6 by Robert Collins
Start check tests for knits (pending), and remove dead code.
790
    def get_file_corrupted_text(self):
791
        knit = self.get_file()
792
        knit.add_lines('v1', [], ['hello\n'])
793
        knit.add_lines('v2', ['v1'], ['hello\n', 'there\n'])
794
        return knit
1563.2.9 by Robert Collins
Update versionedfile api tests to ensure that data is available after every operation.
795
1666.1.6 by Robert Collins
Make knit the default format.
796
    def reopen_file(self, name='foo', create=False):
2770.1.1 by Aaron Bentley
Initial implmentation of plain knit annotation
797
        return self.get_factory()(name, get_transport(self.get_url('.')),
1666.1.6 by Robert Collins
Make knit the default format.
798
            delta=True,
799
            create=create)
1563.2.10 by Robert Collins
Change weave store to be a versioned store, using WeaveFiles which maintain integrity without needing explicit 'put' operations.
800
801
    def test_detection(self):
1563.2.19 by Robert Collins
stub out a check for knits.
802
        knit = self.get_file()
803
        knit.check()
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.
804
1563.2.25 by Robert Collins
Merge in upstream.
805
    def test_no_implicit_create(self):
806
        self.assertRaises(errors.NoSuchFile,
807
                          KnitVersionedFile,
808
                          'foo',
809
                          get_transport(self.get_url('.')))
810
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.
811
2770.1.1 by Aaron Bentley
Initial implmentation of plain knit annotation
812
class TestPlaintextKnit(TestKnit):
813
    """Test a knit with no cached annotations"""
814
815
    def _factory(self, name, transport, file_mode=None, access_mode=None,
816
                 delta=True, create=False):
817
        return KnitVersionedFile(name, transport, file_mode, access_mode,
818
                                 KnitPlainFactory(), delta=delta,
819
                                 create=create)
820
821
    def get_factory(self):
822
        return self._factory
823
824
3062.1.9 by Aaron Bentley
Move PlanMerge into merge and _PlanMergeVersionedFile into versionedfile
825
class TestPlanMergeVersionedFile(TestCaseWithMemoryTransport):
826
827
    def setUp(self):
828
        TestCaseWithMemoryTransport.setUp(self)
829
        self.vf1 = KnitVersionedFile('root', self.get_transport(), create=True)
830
        self.vf2 = KnitVersionedFile('root', self.get_transport(), create=True)
831
        self.plan_merge_vf = versionedfile._PlanMergeVersionedFile('root',
832
            [self.vf1, self.vf2])
833
834
    def test_add_lines(self):
835
        self.plan_merge_vf.add_lines('a:', [], [])
836
        self.assertRaises(ValueError, self.plan_merge_vf.add_lines, 'a', [],
837
                          [])
838
        self.assertRaises(ValueError, self.plan_merge_vf.add_lines, 'a:', None,
839
                          [])
840
        self.assertRaises(ValueError, self.plan_merge_vf.add_lines, 'a:', [],
841
                          None)
842
843
    def test_ancestry(self):
844
        self.vf1.add_lines('A', [], [])
845
        self.vf1.add_lines('B', ['A'], [])
846
        self.plan_merge_vf.add_lines('C:', ['B'], [])
847
        self.plan_merge_vf.add_lines('D:', ['C:'], [])
848
        self.assertEqual(set(['A', 'B', 'C:', 'D:']),
3062.1.14 by Aaron Bentley
Use topo_sorted=False with get_ancestry
849
            self.plan_merge_vf.get_ancestry('D:', topo_sorted=False))
3062.1.9 by Aaron Bentley
Move PlanMerge into merge and _PlanMergeVersionedFile into versionedfile
850
851
    def setup_abcde(self):
852
        self.vf1.add_lines('A', [], ['a'])
853
        self.vf1.add_lines('B', ['A'], ['b'])
854
        self.vf2.add_lines('C', [], ['c'])
855
        self.vf2.add_lines('D', ['C'], ['d'])
856
        self.plan_merge_vf.add_lines('E:', ['B', 'D'], ['e'])
857
858
    def test_ancestry_uses_all_versionedfiles(self):
859
        self.setup_abcde()
860
        self.assertEqual(set(['A', 'B', 'C', 'D', 'E:']),
3062.1.14 by Aaron Bentley
Use topo_sorted=False with get_ancestry
861
            self.plan_merge_vf.get_ancestry('E:', topo_sorted=False))
3062.1.9 by Aaron Bentley
Move PlanMerge into merge and _PlanMergeVersionedFile into versionedfile
862
863
    def test_ancestry_raises_revision_not_present(self):
864
        error = self.assertRaises(errors.RevisionNotPresent,
3062.1.14 by Aaron Bentley
Use topo_sorted=False with get_ancestry
865
                                  self.plan_merge_vf.get_ancestry, 'E:', False)
3062.1.9 by Aaron Bentley
Move PlanMerge into merge and _PlanMergeVersionedFile into versionedfile
866
        self.assertContainsRe(str(error), '{E:} not present in "root"')
867
868
    def test_get_parents(self):
869
        self.setup_abcde()
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.
870
        self.assertEqual({'B':('A',)}, self.plan_merge_vf.get_parent_map(['B']))
871
        self.assertEqual({'D':('C',)}, self.plan_merge_vf.get_parent_map(['D']))
872
        self.assertEqual({'E:':('B', 'D')},
873
            self.plan_merge_vf.get_parent_map(['E:']))
874
        self.assertEqual({}, self.plan_merge_vf.get_parent_map(['F']))
875
        self.assertEqual({
876
                'B':('A',),
877
                'D':('C',),
878
                'E:':('B', 'D'),
879
                }, self.plan_merge_vf.get_parent_map(['B', 'D', 'E:', 'F']))
3062.1.9 by Aaron Bentley
Move PlanMerge into merge and _PlanMergeVersionedFile into versionedfile
880
881
    def test_get_lines(self):
882
        self.setup_abcde()
883
        self.assertEqual(['a'], self.plan_merge_vf.get_lines('A'))
884
        self.assertEqual(['c'], self.plan_merge_vf.get_lines('C'))
885
        self.assertEqual(['e'], self.plan_merge_vf.get_lines('E:'))
886
        error = self.assertRaises(errors.RevisionNotPresent,
887
                                  self.plan_merge_vf.get_lines, 'F')
888
        self.assertContainsRe(str(error), '{F} not present in "root"')
889
890
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.
891
class InterString(versionedfile.InterVersionedFile):
892
    """An inter-versionedfile optimised code path for strings.
893
894
    This is for use during testing where we use strings as versionedfiles
895
    so that none of the default regsitered interversionedfile classes will
896
    match - which lets us test the match logic.
897
    """
898
899
    @staticmethod
900
    def is_compatible(source, target):
901
        """InterString is compatible with strings-as-versionedfiles."""
902
        return isinstance(source, str) and isinstance(target, str)
903
904
905
# TODO this and the InterRepository core logic should be consolidatable
906
# if we make the registry a separate class though we still need to 
907
# test the behaviour in the active registry to catch failure-to-handle-
908
# stange-objects
2535.3.1 by Andrew Bennetts
Add get_format_signature to VersionedFile
909
class TestInterVersionedFile(TestCaseWithMemoryTransport):
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.
910
911
    def test_get_default_inter_versionedfile(self):
912
        # test that the InterVersionedFile.get(a, b) probes
913
        # for a class where is_compatible(a, b) returns
914
        # true and returns a default interversionedfile otherwise.
915
        # This also tests that the default registered optimised interversionedfile
916
        # classes do not barf inappropriately when a surprising versionedfile type
917
        # is handed to them.
918
        dummy_a = "VersionedFile 1."
919
        dummy_b = "VersionedFile 2."
920
        self.assertGetsDefaultInterVersionedFile(dummy_a, dummy_b)
921
922
    def assertGetsDefaultInterVersionedFile(self, a, b):
923
        """Asserts that InterVersionedFile.get(a, b) -> the default."""
924
        inter = versionedfile.InterVersionedFile.get(a, b)
925
        self.assertEqual(versionedfile.InterVersionedFile,
926
                         inter.__class__)
927
        self.assertEqual(a, inter.source)
928
        self.assertEqual(b, inter.target)
929
930
    def test_register_inter_versionedfile_class(self):
931
        # test that a optimised code path provider - a
932
        # InterVersionedFile subclass can be registered and unregistered
933
        # and that it is correctly selected when given a versionedfile
934
        # pair that it returns true on for the is_compatible static method
935
        # check
936
        dummy_a = "VersionedFile 1."
937
        dummy_b = "VersionedFile 2."
938
        versionedfile.InterVersionedFile.register_optimiser(InterString)
939
        try:
940
            # we should get the default for something InterString returns False
941
            # to
942
            self.assertFalse(InterString.is_compatible(dummy_a, None))
943
            self.assertGetsDefaultInterVersionedFile(dummy_a, None)
944
            # and we should get an InterString for a pair it 'likes'
945
            self.assertTrue(InterString.is_compatible(dummy_a, dummy_b))
946
            inter = versionedfile.InterVersionedFile.get(dummy_a, dummy_b)
947
            self.assertEqual(InterString, inter.__class__)
948
            self.assertEqual(dummy_a, inter.source)
949
            self.assertEqual(dummy_b, inter.target)
950
        finally:
951
            versionedfile.InterVersionedFile.unregister_optimiser(InterString)
952
        # now we should get the default InterVersionedFile object again.
953
        self.assertGetsDefaultInterVersionedFile(dummy_a, dummy_b)
1666.1.1 by Robert Collins
Add trivial http-using test for versioned files.
954
955
956
class TestReadonlyHttpMixin(object):
957
958
    def test_readonly_http_works(self):
959
        # we should be able to read from http with a versioned file.
960
        vf = self.get_file()
1666.1.6 by Robert Collins
Make knit the default format.
961
        # try an empty file access
962
        readonly_vf = self.get_factory()('foo', get_transport(self.get_readonly_url('.')))
963
        self.assertEqual([], readonly_vf.versions())
964
        # now with feeling.
1666.1.1 by Robert Collins
Add trivial http-using test for versioned files.
965
        vf.add_lines('1', [], ['a\n'])
966
        vf.add_lines('2', ['1'], ['b\n', 'a\n'])
967
        readonly_vf = self.get_factory()('foo', get_transport(self.get_readonly_url('.')))
1666.1.6 by Robert Collins
Make knit the default format.
968
        self.assertEqual(['1', '2'], vf.versions())
1666.1.1 by Robert Collins
Add trivial http-using test for versioned files.
969
        for version in readonly_vf.versions():
970
            readonly_vf.get_lines(version)
971
972
973
class TestWeaveHTTP(TestCaseWithWebserver, TestReadonlyHttpMixin):
974
975
    def get_file(self):
976
        return WeaveFile('foo', get_transport(self.get_url('.')), create=True)
977
978
    def get_factory(self):
979
        return WeaveFile
980
981
982
class TestKnitHTTP(TestCaseWithWebserver, TestReadonlyHttpMixin):
983
984
    def get_file(self):
985
        return KnitVersionedFile('foo', get_transport(self.get_url('.')),
986
                                 delta=True, create=True)
987
988
    def get_factory(self):
989
        return KnitVersionedFile
1664.2.9 by Aaron Bentley
Ported weave merge test to versionedfile
990
991
992
class MergeCasesMixin(object):
993
994
    def doMerge(self, base, a, b, mp):
995
        from cStringIO import StringIO
996
        from textwrap import dedent
997
998
        def addcrlf(x):
999
            return x + '\n'
1000
        
1001
        w = self.get_file()
1002
        w.add_lines('text0', [], map(addcrlf, base))
1003
        w.add_lines('text1', ['text0'], map(addcrlf, a))
1004
        w.add_lines('text2', ['text0'], map(addcrlf, b))
1005
1006
        self.log_contents(w)
1007
1008
        self.log('merge plan:')
1009
        p = list(w.plan_merge('text1', 'text2'))
1010
        for state, line in p:
1011
            if line:
1012
                self.log('%12s | %s' % (state, line[:-1]))
1013
1014
        self.log('merge:')
1015
        mt = StringIO()
1016
        mt.writelines(w.weave_merge(p))
1017
        mt.seek(0)
1018
        self.log(mt.getvalue())
1019
1020
        mp = map(addcrlf, mp)
1021
        self.assertEqual(mt.readlines(), mp)
1022
        
1023
        
1024
    def testOneInsert(self):
1025
        self.doMerge([],
1026
                     ['aa'],
1027
                     [],
1028
                     ['aa'])
1029
1030
    def testSeparateInserts(self):
1031
        self.doMerge(['aaa', 'bbb', 'ccc'],
1032
                     ['aaa', 'xxx', 'bbb', 'ccc'],
1033
                     ['aaa', 'bbb', 'yyy', 'ccc'],
1034
                     ['aaa', 'xxx', 'bbb', 'yyy', 'ccc'])
1035
1036
    def testSameInsert(self):
1037
        self.doMerge(['aaa', 'bbb', 'ccc'],
1038
                     ['aaa', 'xxx', 'bbb', 'ccc'],
1039
                     ['aaa', 'xxx', 'bbb', 'yyy', 'ccc'],
1040
                     ['aaa', 'xxx', 'bbb', 'yyy', 'ccc'])
1041
    overlappedInsertExpected = ['aaa', 'xxx', 'yyy', 'bbb']
1042
    def testOverlappedInsert(self):
1043
        self.doMerge(['aaa', 'bbb'],
1044
                     ['aaa', 'xxx', 'yyy', 'bbb'],
1045
                     ['aaa', 'xxx', 'bbb'], self.overlappedInsertExpected)
1046
1047
        # really it ought to reduce this to 
1048
        # ['aaa', 'xxx', 'yyy', 'bbb']
1049
1050
1051
    def testClashReplace(self):
1052
        self.doMerge(['aaa'],
1053
                     ['xxx'],
1054
                     ['yyy', 'zzz'],
1055
                     ['<<<<<<< ', 'xxx', '=======', 'yyy', 'zzz', 
1056
                      '>>>>>>> '])
1057
1058
    def testNonClashInsert1(self):
1059
        self.doMerge(['aaa'],
1060
                     ['xxx', 'aaa'],
1061
                     ['yyy', 'zzz'],
1062
                     ['<<<<<<< ', 'xxx', 'aaa', '=======', 'yyy', 'zzz', 
1063
                      '>>>>>>> '])
1064
1065
    def testNonClashInsert2(self):
1066
        self.doMerge(['aaa'],
1067
                     ['aaa'],
1068
                     ['yyy', 'zzz'],
1069
                     ['yyy', 'zzz'])
1070
1071
1072
    def testDeleteAndModify(self):
1073
        """Clashing delete and modification.
1074
1075
        If one side modifies a region and the other deletes it then
1076
        there should be a conflict with one side blank.
1077
        """
1078
1079
        #######################################
1080
        # skippd, not working yet
1081
        return
1082
        
1083
        self.doMerge(['aaa', 'bbb', 'ccc'],
1084
                     ['aaa', 'ddd', 'ccc'],
1085
                     ['aaa', 'ccc'],
1086
                     ['<<<<<<<< ', 'aaa', '=======', '>>>>>>> ', 'ccc'])
1087
1088
    def _test_merge_from_strings(self, base, a, b, expected):
1089
        w = self.get_file()
1090
        w.add_lines('text0', [], base.splitlines(True))
1091
        w.add_lines('text1', ['text0'], a.splitlines(True))
1092
        w.add_lines('text2', ['text0'], b.splitlines(True))
1093
        self.log('merge plan:')
1094
        p = list(w.plan_merge('text1', 'text2'))
1095
        for state, line in p:
1096
            if line:
1097
                self.log('%12s | %s' % (state, line[:-1]))
1098
        self.log('merge result:')
1099
        result_text = ''.join(w.weave_merge(p))
1100
        self.log(result_text)
1101
        self.assertEqualDiff(result_text, expected)
1102
1103
    def test_weave_merge_conflicts(self):
1104
        # does weave merge properly handle plans that end with unchanged?
1105
        result = ''.join(self.get_file().weave_merge([('new-a', 'hello\n')]))
1106
        self.assertEqual(result, 'hello\n')
1107
1108
    def test_deletion_extended(self):
1109
        """One side deletes, the other deletes more.
1110
        """
1111
        base = """\
1112
            line 1
1113
            line 2
1114
            line 3
1115
            """
1116
        a = """\
1117
            line 1
1118
            line 2
1119
            """
1120
        b = """\
1121
            line 1
1122
            """
1123
        result = """\
1124
            line 1
1125
            """
1126
        self._test_merge_from_strings(base, a, b, result)
1127
1128
    def test_deletion_overlap(self):
1129
        """Delete overlapping regions with no other conflict.
1130
1131
        Arguably it'd be better to treat these as agreement, rather than 
1132
        conflict, but for now conflict is safer.
1133
        """
1134
        base = """\
1135
            start context
1136
            int a() {}
1137
            int b() {}
1138
            int c() {}
1139
            end context
1140
            """
1141
        a = """\
1142
            start context
1143
            int a() {}
1144
            end context
1145
            """
1146
        b = """\
1147
            start context
1148
            int c() {}
1149
            end context
1150
            """
1151
        result = """\
1152
            start context
1153
<<<<<<< 
1154
            int a() {}
1155
=======
1156
            int c() {}
1157
>>>>>>> 
1158
            end context
1159
            """
1160
        self._test_merge_from_strings(base, a, b, result)
1161
1162
    def test_agreement_deletion(self):
1163
        """Agree to delete some lines, without conflicts."""
1164
        base = """\
1165
            start context
1166
            base line 1
1167
            base line 2
1168
            end context
1169
            """
1170
        a = """\
1171
            start context
1172
            base line 1
1173
            end context
1174
            """
1175
        b = """\
1176
            start context
1177
            base line 1
1178
            end context
1179
            """
1180
        result = """\
1181
            start context
1182
            base line 1
1183
            end context
1184
            """
1185
        self._test_merge_from_strings(base, a, b, result)
1186
1187
    def test_sync_on_deletion(self):
1188
        """Specific case of merge where we can synchronize incorrectly.
1189
        
1190
        A previous version of the weave merge concluded that the two versions
1191
        agreed on deleting line 2, and this could be a synchronization point.
1192
        Line 1 was then considered in isolation, and thought to be deleted on 
1193
        both sides.
1194
1195
        It's better to consider the whole thing as a disagreement region.
1196
        """
1197
        base = """\
1198
            start context
1199
            base line 1
1200
            base line 2
1201
            end context
1202
            """
1203
        a = """\
1204
            start context
1205
            base line 1
1206
            a's replacement line 2
1207
            end context
1208
            """
1209
        b = """\
1210
            start context
1211
            b replaces
1212
            both lines
1213
            end context
1214
            """
1215
        result = """\
1216
            start context
1217
<<<<<<< 
1218
            base line 1
1219
            a's replacement line 2
1220
=======
1221
            b replaces
1222
            both lines
1223
>>>>>>> 
1224
            end context
1225
            """
1226
        self._test_merge_from_strings(base, a, b, result)
1227
1228
2535.3.1 by Andrew Bennetts
Add get_format_signature to VersionedFile
1229
class TestKnitMerge(TestCaseWithMemoryTransport, MergeCasesMixin):
1664.2.9 by Aaron Bentley
Ported weave merge test to versionedfile
1230
1231
    def get_file(self, name='foo'):
1232
        return KnitVersionedFile(name, get_transport(self.get_url('.')),
1233
                                 delta=True, create=True)
1234
1235
    def log_contents(self, w):
1236
        pass
1237
1238
2535.3.1 by Andrew Bennetts
Add get_format_signature to VersionedFile
1239
class TestWeaveMerge(TestCaseWithMemoryTransport, MergeCasesMixin):
1664.2.9 by Aaron Bentley
Ported weave merge test to versionedfile
1240
1241
    def get_file(self, name='foo'):
1242
        return WeaveFile(name, get_transport(self.get_url('.')), create=True)
1243
1244
    def log_contents(self, w):
1245
        self.log('weave is:')
1246
        tmpf = StringIO()
1247
        write_weave(w, tmpf)
1248
        self.log(tmpf.getvalue())
1249
1250
    overlappedInsertExpected = ['aaa', '<<<<<<< ', 'xxx', 'yyy', '=======', 
1251
                                'xxx', '>>>>>>> ', 'bbb']
2535.3.1 by Andrew Bennetts
Add get_format_signature to VersionedFile
1252
1253
1254
class TestFormatSignatures(TestCaseWithMemoryTransport):
1255
2535.3.17 by Andrew Bennetts
[broken] Closer to a working Repository.fetch_revisions smart request.
1256
    def get_knit_file(self, name, annotated):
2535.3.1 by Andrew Bennetts
Add get_format_signature to VersionedFile
1257
        if annotated:
1258
            factory = KnitAnnotateFactory()
1259
        else:
1260
            factory = KnitPlainFactory()
1261
        return KnitVersionedFile(
2535.3.17 by Andrew Bennetts
[broken] Closer to a working Repository.fetch_revisions smart request.
1262
            name, get_transport(self.get_url('.')), create=True,
2535.3.1 by Andrew Bennetts
Add get_format_signature to VersionedFile
1263
            factory=factory)
1264
1265
    def test_knit_format_signatures(self):
1266
        """Different formats of knit have different signature strings."""
2535.3.17 by Andrew Bennetts
[broken] Closer to a working Repository.fetch_revisions smart request.
1267
        knit = self.get_knit_file('a', True)
1268
        self.assertEqual('knit-annotated', knit.get_format_signature())
1269
        knit = self.get_knit_file('p', False)
1270
        self.assertEqual('knit-plain', knit.get_format_signature())
2535.3.1 by Andrew Bennetts
Add get_format_signature to VersionedFile
1271