def test_multivolume_image_size(self):
args = parseargs(['-i', '0:4G,sdcard:2G,1:4G', 'model.assertion'])
self.assertEqual(args.image_size, {
0: GiB(4),
'sdcard': GiB(2),
1: GiB(4),
})
def test_image_size_option_suffixes(self):
args = parseargs(['snap', '--image-size', '45G', 'model.assertion'])
self.assertEqual(args.image_size, GiB(45))
self.assertEqual(args.given_image_size, '45G')
args = parseargs(['snap', '--image-size', '45M', 'model.assertion'])
self.assertEqual(args.image_size, MiB(45))
self.assertEqual(args.given_image_size, '45M')
def test_volume_offset_write(self):
gadget_spec = parse("""\
volumes:
first-image:
schema: gpt
bootloader: u-boot
structure:
- type: 00000000-0000-0000-0000-0000deadbeef
size: 400M
offset-write: 1G
""")
volume0 = gadget_spec.volumes['first-image']
partition0 = volume0.structures[0]
self.assertEqual(partition0.offset_write, GiB(1))
def Size32bit(v):
"""Coerce size to being a 32 bit integer."""
return as_size(v, max=GiB(4))
def test_initialize(self):
image = Image(self.img, GiB(1.25))
self.assertTrue(os.path.exists(image.path))
# GiB == 1024**3; 1.25GiB == 1342177280 bytes.
self.assertEqual(os.stat(image.path).st_size, 1342177280)
def test_g(self):
self.assertEqual(as_size('30G'), GiB(30))