← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_knit.py

  • Committer: Martin Pool
  • Date: 2006-06-20 07:55:43 UTC
  • mfrom: (1798 +trunk)
  • mto: This revision was merged to the branch mainline in revision 1799.
  • Revision ID: mbp@sourcefrog.net-20060620075543-b10f6575d4a4fa32
[merge] bzr.dev

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_knit.py
28
28
    WeaveToKnit)
29
29
from bzrlib.osutils import split_lines
30
30
from bzrlib.tests import TestCaseWithTransport
31
 
from bzrlib.transport import TransportLogger
32
 
from bzrlib.transport.local import LocalTransport
 
31
from bzrlib.transport import TransportLogger, get_transport
33
32
from bzrlib.transport.memory import MemoryTransport
34
33
from bzrlib.weave import Weave
35
34
 
42
41
            factory = KnitPlainFactory()
43
42
        else:
44
43
            factory = None
45
 
        return KnitVersionedFile('test', LocalTransport('.'), access_mode='w', factory=factory, create=True)
 
44
        return KnitVersionedFile('test', get_transport('.'), access_mode='w', factory=factory, create=True)
46
45
 
47
46
 
48
47
class BasicKnitTests(KnitTests):
67
66
        k = self.make_test_knit()
68
67
        k.add_lines('text-1', [], split_lines(TEXT_1))
69
68
        del k
70
 
        k2 = KnitVersionedFile('test', LocalTransport('.'), access_mode='r', factory=KnitPlainFactory(), create=True)
 
69
        k2 = KnitVersionedFile('test', get_transport('.'), access_mode='r', factory=KnitPlainFactory(), create=True)
71
70
        self.assertTrue(k2.has_version('text-1'))
72
71
        self.assertEqualDiff(''.join(k2.get_lines('text-1')), TEXT_1)
73
72
 
95
94
    def test_incomplete(self):
96
95
        """Test if texts without a ending line-end can be inserted and
97
96
        extracted."""
98
 
        k = KnitVersionedFile('test', LocalTransport('.'), delta=False, create=True)
 
97
        k = KnitVersionedFile('test', get_transport('.'), delta=False, create=True)
99
98
        k.add_lines('text-1', [], ['a\n',    'b'  ])
100
99
        k.add_lines('text-2', ['text-1'], ['a\rb\n', 'b\n'])
101
100
        # reopening ensures maximum room for confusion
102
 
        k = KnitVersionedFile('test', LocalTransport('.'), delta=False, create=True)
 
101
        k = KnitVersionedFile('test', get_transport('.'), delta=False, create=True)
103
102
        self.assertEquals(k.get_lines('text-1'), ['a\n',    'b'  ])
104
103
        self.assertEquals(k.get_lines('text-2'), ['a\rb\n', 'b\n'])
105
104
 
127
126
 
128
127
    def test_add_delta(self):
129
128
        """Store in knit with parents"""
130
 
        k = KnitVersionedFile('test', LocalTransport('.'), factory=KnitPlainFactory(),
 
129
        k = KnitVersionedFile('test', get_transport('.'), factory=KnitPlainFactory(),
131
130
            delta=True, create=True)
132
131
        self.add_stock_one_and_one_a(k)
133
132
        k.clear_cache()
135
134
 
136
135
    def test_annotate(self):
137
136
        """Annotations"""
138
 
        k = KnitVersionedFile('knit', LocalTransport('.'), factory=KnitAnnotateFactory(),
 
137
        k = KnitVersionedFile('knit', get_transport('.'), factory=KnitAnnotateFactory(),
139
138
            delta=True, create=True)
140
139
        self.insert_and_test_small_annotate(k)
141
140
 
150
149
 
151
150
    def test_annotate_fulltext(self):
152
151
        """Annotations"""
153
 
        k = KnitVersionedFile('knit', LocalTransport('.'), factory=KnitAnnotateFactory(),
 
152
        k = KnitVersionedFile('knit', get_transport('.'), factory=KnitAnnotateFactory(),
154
153
            delta=False, create=True)
155
154
        self.insert_and_test_small_annotate(k)
156
155
 
229
228
 
230
229
    def test_knit_join(self):
231
230
        """Store in knit with parents"""
232
 
        k1 = KnitVersionedFile('test1', LocalTransport('.'), factory=KnitPlainFactory(), create=True)
 
231
        k1 = KnitVersionedFile('test1', get_transport('.'), factory=KnitPlainFactory(), create=True)
233
232
        k1.add_lines('text-a', [], split_lines(TEXT_1))
234
233
        k1.add_lines('text-b', ['text-a'], split_lines(TEXT_1))
235
234
 
238
237
 
239
238
        k1.add_lines('text-m', ['text-b', 'text-d'], split_lines(TEXT_1))
240
239
 
241
 
        k2 = KnitVersionedFile('test2', LocalTransport('.'), factory=KnitPlainFactory(), create=True)
 
240
        k2 = KnitVersionedFile('test2', get_transport('.'), factory=KnitPlainFactory(), create=True)
242
241
        count = k2.join(k1, version_ids=['text-m'])
243
242
        self.assertEquals(count, 5)
244
243
        self.assertTrue(k2.has_version('text-a'))
245
244
        self.assertTrue(k2.has_version('text-c'))
246
245
 
247
246
    def test_reannotate(self):
248
 
        k1 = KnitVersionedFile('knit1', LocalTransport('.'),
 
247
        k1 = KnitVersionedFile('knit1', get_transport('.'),
249
248
                               factory=KnitAnnotateFactory(), create=True)
250
249
        # 0
251
250
        k1.add_lines('text-a', [], ['a\n', 'b\n'])
252
251
        # 1
253
252
        k1.add_lines('text-b', ['text-a'], ['a\n', 'c\n'])
254
253
 
255
 
        k2 = KnitVersionedFile('test2', LocalTransport('.'),
 
254
        k2 = KnitVersionedFile('test2', get_transport('.'),
256
255
                               factory=KnitAnnotateFactory(), create=True)
257
256
        k2.join(k1, version_ids=['text-b'])
258
257
 
363
362
            "\nrevid2 line-delta 84 82 0 :",
364
363
            'test.kndx')
365
364
        # we should be able to load this file again
366
 
        knit = KnitVersionedFile('test', LocalTransport('.'), access_mode='r')
 
365
        knit = KnitVersionedFile('test', get_transport('.'), access_mode='r')
367
366
        self.assertEqual(['revid', 'revid2'], knit.versions())
368
367
        # write a short write to the file and ensure that its ignored
369
368
        indexfile = file('test.kndx', 'at')
370
369
        indexfile.write('\nrevid3 line-delta 166 82 1 2 3 4 5 .phwoar:demo ')
371
370
        indexfile.close()
372
371
        # we should be able to load this file again
373
 
        knit = KnitVersionedFile('test', LocalTransport('.'), access_mode='w')
 
372
        knit = KnitVersionedFile('test', get_transport('.'), access_mode='w')
374
373
        self.assertEqual(['revid', 'revid2'], knit.versions())
375
374
        # and add a revision with the same id the failed write had
376
375
        knit.add_lines('revid3', ['revid2'], ['a\n'])
377
376
        # and when reading it revid3 should now appear.
378
 
        knit = KnitVersionedFile('test', LocalTransport('.'), access_mode='r')
 
377
        knit = KnitVersionedFile('test', get_transport('.'), access_mode='r')
379
378
        self.assertEqual(['revid', 'revid2', 'revid3'], knit.versions())
380
379
        self.assertEqual(['revid2'], knit.get_parents('revid3'))
381
380