def test_parse_schemes(
self, mock_init, mock_imgsch, mock_partsch,
mock_fs, mock_img, mock_loop):
mock_init.return_value = None
data = {
'image_data': IMAGE_DATA_SAMPLE,
'output': '/some/local/path',
}
driver = NailgunBuildImage()
driver.data = data
driver.parse_schemes()
mock_fs_expected_calls = []
mock_img_expected_calls = []
images = []
fss = []
data_length = len(data['image_data'].keys())
for mount, image in six.iteritems(data['image_data']):
filename = os.path.basename(urlsplit(image['uri']).path)
img_kwargs = {
'uri': 'file://' + os.path.join(data['output'], filename),
'format': image['format'],
'container': image['container'],
'target_device': None
}
mock_img_expected_calls.append(mock.call(**img_kwargs))
images.append(objects.Image(**img_kwargs))
fs_kwargs = {
'device': None,
'mount': mount,
'fs_type': image['format']
}
mock_fs_expected_calls.append(mock.call(**fs_kwargs))
fss.append(objects.Fs(**fs_kwargs))
if mount == '/':
metadata_filename = filename.split('.', 1)[0] + '.yaml'
mock_imgsch_instance = mock_imgsch.return_value
mock_imgsch_instance.images = images
mock_partsch_instance = mock_partsch.return_value
mock_partsch_instance.fss = fss
self.assertEqual(
driver.metadata_uri, 'file://' + os.path.join(
data['output'], metadata_filename))
self.assertEqual(mock_img_expected_calls,
mock_img.call_args_list[:data_length])
self.assertEqual(mock_fs_expected_calls,
mock_fs.call_args_list[:data_length])
self.assertEqual(driver.image_scheme.images, images)
self.assertEqual(driver.partition_scheme.fss, fss)
def test_do_build_image(self, mock_umount_target, mock_mount_target,
mock_yaml_dump, mock_mkdtemp, mock_open,
mock_shutil_move, mock_os, mock_utils, mock_fu,
mock_bu):
loops = [objects.Loop(), objects.Loop()]
self.mgr.driver.image_scheme = objects.ImageScheme([
objects.Image('file:///fake/img.img.gz', loops[0], 'ext4', 'gzip'),
objects.Image('file:///fake/img-boot.img.gz', loops[1], 'ext2',
'gzip')
])
self.mgr.driver.partition_scheme = objects.PartitionScheme()
self.mgr.driver.partition_scheme.add_fs(device=loops[0],
mount='/',
fs_type='ext4')
self.mgr.driver.partition_scheme.add_fs(device=loops[1],
mount='/boot',
fs_type='ext2')
self.mgr.driver.metadata_uri = 'file:///fake/img.yaml'
self.mgr.driver.operating_system = objects.Ubuntu(
repos=[
objects.DEBRepo('ubuntu',
'http://fakeubuntu',
'trusty',
'fakesection',
priority=900),
objects.DEBRepo('ubuntu_zero',
'http://fakeubuntu_zero',
'trusty',
'fakesection',
priority=None),
objects.DEBRepo('mos',
'http://fakemos',
'mosX.Y',
'fakesection',
priority=1000)
],
packages=['fakepackage1', 'fakepackage2'])
mock_os.path.exists.return_value = False
mock_os.path.join.return_value = '/tmp/imgdir/proc'
mock_os.path.basename.side_effect = ['img.img.gz', 'img-boot.img.gz']
mock_bu.create_sparse_tmp_file.side_effect = \
['/tmp/img', '/tmp/img-boot']
mock_bu.get_free_loop_device.side_effect = ['/dev/loop0', '/dev/loop1']
mock_mkdtemp.return_value = '/tmp/imgdir'
getsize_side = [20, 2, 10, 1]
mock_os.path.getsize.side_effect = getsize_side
md5_side = [
'fakemd5_raw', 'fakemd5_gzip', 'fakemd5_raw_boot',
'fakemd5_gzip_boot'
]
mock_utils.calculate_md5.side_effect = md5_side
mock_bu.containerize.side_effect = ['/tmp/img.gz', '/tmp/img-boot.gz']
mock_bu.stop_chrooted_processes.side_effect = [
False, True, False, True
]
self.mgr.do_build_image()
self.assertEqual([
mock.call('/fake/img.img.gz'),
mock.call('/fake/img-boot.img.gz')
], mock_os.path.exists.call_args_list)
self.assertEqual([
mock.call(dir=CONF.image_build_dir, suffix=CONF.image_build_suffix)
] * 2, mock_bu.create_sparse_tmp_file.call_args_list)
self.assertEqual([mock.call()] * 2,
mock_bu.get_free_loop_device.call_args_list)
self.assertEqual([
mock.call('/tmp/img', '/dev/loop0'),
mock.call('/tmp/img-boot', '/dev/loop1')
], mock_bu.attach_file_to_loop.call_args_list)
self.assertEqual([
mock.call(
fs_type='ext4', fs_options='', fs_label='', dev='/dev/loop0'),
mock.call(
fs_type='ext2', fs_options='', fs_label='', dev='/dev/loop1')
], mock_fu.make_fs.call_args_list)
mock_mkdtemp.assert_called_once_with(dir=CONF.image_build_dir,
suffix=CONF.image_build_suffix)
mock_mount_target.assert_called_once_with('/tmp/imgdir',
treat_mtab=False,
pseudo=False)
self.assertEqual([mock.call('/tmp/imgdir')] * 2,
mock_bu.suppress_services_start.call_args_list)
mock_bu.run_debootstrap.assert_called_once_with(
uri='http://fakeubuntu', suite='trusty', chroot='/tmp/imgdir')
mock_bu.set_apt_get_env.assert_called_once_with()
mock_bu.pre_apt_get.assert_called_once_with('/tmp/imgdir')
self.assertEqual([
mock.call(name='ubuntu',
uri='http://fakeubuntu',
suite='trusty',
section='fakesection',
chroot='/tmp/imgdir'),
mock.call(name='ubuntu_zero',
uri='http://fakeubuntu_zero',
suite='trusty',
section='fakesection',
chroot='/tmp/imgdir'),
mock.call(name='mos',
uri='http://fakemos',
suite='mosX.Y',
section='fakesection',
chroot='/tmp/imgdir')
], mock_bu.add_apt_source.call_args_list)
# we don't call add_apt_preference for ubuntu_zero
# because it has priority == None
self.assertEqual([
mock.call(name='ubuntu',
priority=900,
suite='trusty',
section='fakesection',
chroot='/tmp/imgdir',
uri='http://fakeubuntu'),
mock.call(name='mos',
priority=1000,
suite='mosX.Y',
section='fakesection',
chroot='/tmp/imgdir',
uri='http://fakemos')
], mock_bu.add_apt_preference.call_args_list)
mock_utils.makedirs_if_not_exists.assert_called_once_with(
'/tmp/imgdir/proc')
self.assertEqual([
mock.call('tune2fs', '-O', '^has_journal', '/dev/loop0'),
mock.call('tune2fs', '-O', 'has_journal', '/dev/loop0')
], mock_utils.execute.call_args_list)
mock_fu.mount_bind.assert_called_once_with('/tmp/imgdir', '/proc')
mock_bu.run_apt_get.assert_called_once_with(
'/tmp/imgdir', packages=['fakepackage1', 'fakepackage2'])
mock_bu.do_post_inst.assert_called_once_with('/tmp/imgdir')
signal_calls = mock_bu.stop_chrooted_processes.call_args_list
self.assertEqual(
2 * [
mock.call('/tmp/imgdir', signal=signal.SIGTERM),
mock.call('/tmp/imgdir', signal=signal.SIGKILL)
], signal_calls)
self.assertEqual([mock.call('/tmp/imgdir/proc')] * 2,
mock_fu.umount_fs.call_args_list)
self.assertEqual(
[mock.call('/tmp/imgdir', try_lazy_umount=False, pseudo=False)] *
2, mock_umount_target.call_args_list)
self.assertEqual([mock.call('/dev/loop0'),
mock.call('/dev/loop1')] * 2,
mock_bu.deattach_loop.call_args_list)
self.assertEqual([mock.call('/tmp/img'),
mock.call('/tmp/img-boot')],
mock_bu.shrink_sparse_file.call_args_list)
self.assertEqual([
mock.call('/tmp/img'),
mock.call('/fake/img.img.gz'),
mock.call('/tmp/img-boot'),
mock.call('/fake/img-boot.img.gz')
], mock_os.path.getsize.call_args_list)
self.assertEqual([
mock.call('/tmp/img', 20),
mock.call('/fake/img.img.gz', 2),
mock.call('/tmp/img-boot', 10),
mock.call('/fake/img-boot.img.gz', 1)
], mock_utils.calculate_md5.call_args_list)
self.assertEqual([
mock.call('/tmp/img', 'gzip'),
mock.call('/tmp/img-boot', 'gzip')
], mock_bu.containerize.call_args_list)
mock_open.assert_called_once_with('/fake/img.yaml', 'w')
self.assertEqual([
mock.call('/tmp/img.gz', '/fake/img.img.gz'),
mock.call('/tmp/img-boot.gz', '/fake/img-boot.img.gz')
], mock_shutil_move.call_args_list)
metadata = {}
for repo in self.mgr.driver.operating_system.repos:
metadata.setdefault('repos', []).append({
'type': 'deb',
'name': repo.name,
'uri': repo.uri,
'suite': repo.suite,
'section': repo.section,
'priority': repo.priority,
'meta': repo.meta
})
metadata['packages'] = self.mgr.driver.operating_system.packages
metadata['images'] = [{
'raw_md5':
md5_side[0],
'raw_size':
getsize_side[0],
'raw_name':
None,
'container_name':
os.path.basename(self.mgr.driver.image_scheme.images[0].uri.split(
'file://', 1)[1]),
'container_md5':
md5_side[1],
'container_size':
getsize_side[1],
'container':
self.mgr.driver.image_scheme.images[0].container,
'format':
self.mgr.driver.image_scheme.images[0].format
}, {
'raw_md5':
md5_side[2],
'raw_size':
getsize_side[2],
'raw_name':
None,
'container_name':
os.path.basename(self.mgr.driver.image_scheme.images[1].uri.split(
'file://', 1)[1]),
'container_md5':
md5_side[3],
'container_size':
getsize_side[3],
'container':
self.mgr.driver.image_scheme.images[1].container,
'format':
self.mgr.driver.image_scheme.images[1].format
}]
mock_yaml_dump.assert_called_once_with(metadata, stream=mock_open())