def test_get_block_fail(self):
reader = make_reader(0)
splitter = UniformSplitter(self.project_id,
self.vault_id,
reader)
offset, block = splitter.get_block()
self.assertEqual(offset, 0)
self.assertIsNone(block)
def test_get_block(self):
reader = make_reader(10 * 1024 * 1024)
splitter = UniformSplitter(self.project_id,
self.vault_id,
reader)
running_offset = 0
for _ in range(2):
offset, block = splitter.get_block()
self.assertEqual(offset, running_offset)
self.assertIsInstance(block, api.Block)
self.assertEqual(splitter.chunk_size, len(block))
running_offset = running_offset + splitter.chunk_size
