320
320
"""Tests for getting and putting knit records."""
322
322
def test_add_raw_records(self):
323
"""add_raw_records adds records retrievable later."""
323
"""Add_raw_records adds records retrievable later."""
324
324
access = self.get_access()
325
memos = access.add_raw_records([(b'key', 10)], [b'1234567890'])
325
memos = access.add_raw_records([(b'key', 10)], b'1234567890')
326
326
self.assertEqual([b'1234567890'], list(access.get_raw_records(memos)))
328
def test_add_raw_record(self):
329
"""add_raw_record adds records retrievable later."""
330
access = self.get_access()
331
memos = access.add_raw_record(b'key', 10, [b'1234567890'])
332
self.assertEqual([b'1234567890'], list(access.get_raw_records([memos])))
334
328
def test_add_several_raw_records(self):
335
329
"""add_raw_records with many records and read some back."""
336
330
access = self.get_access()
337
331
memos = access.add_raw_records([(b'key', 10), (b'key2', 2), (b'key3', 5)],
338
[b'12345678901234567'])
332
b'12345678901234567')
339
333
self.assertEqual([b'1234567890', b'12', b'34567'],
340
334
list(access.get_raw_records(memos)))
341
335
self.assertEqual([b'1234567890'],
383
377
"""Create a pack file with 2 records."""
384
378
access, writer = self._get_access(packname='packname', index='foo')
386
memos.extend(access.add_raw_records([(b'key1', 10)], [b'1234567890']))
387
memos.extend(access.add_raw_records([(b'key2', 5)], [b'12345']))
380
memos.extend(access.add_raw_records([(b'key1', 10)], b'1234567890'))
381
memos.extend(access.add_raw_records([(b'key2', 5)], b'12345'))
500
494
def test_read_from_several_packs(self):
501
495
access, writer = self._get_access()
503
memos.extend(access.add_raw_records([(b'key', 10)], [b'1234567890']))
497
memos.extend(access.add_raw_records([(b'key', 10)], b'1234567890'))
505
499
access, writer = self._get_access('pack2', 'FOOBAR')
506
memos.extend(access.add_raw_records([(b'key', 5)], [b'12345']))
500
memos.extend(access.add_raw_records([(b'key', 5)], b'12345'))
508
502
access, writer = self._get_access('pack3', 'BAZ')
509
memos.extend(access.add_raw_records([(b'key', 5)], [b'alpha']))
503
memos.extend(access.add_raw_records([(b'key', 5)], b'alpha'))
511
505
transport = self.get_transport()
512
506
access = pack_repo._DirectPackAccess({"FOO": (transport, 'packfile'),
535
529
writer = pack.ContainerWriter(write_data)
537
531
access.set_writer(writer, index, (transport, packname))
538
memos = access.add_raw_records([(b'key', 10)], [b'1234567890'])
532
memos = access.add_raw_records([(b'key', 10)], b'1234567890')
540
534
self.assertEqual([b'1234567890'], list(access.get_raw_records(memos)))