def test_job(self):
with open(self.filename) as yaml_data:
alpha_data = yaml.load(yaml_data)
self.assertIn('protocols', alpha_data)
self.assertIn(VlandProtocol.name, alpha_data['protocols'])
with open(self.filename) as sample_job_data:
parser = JobParser()
job = parser.parse(sample_job_data,
self.device,
4212,
None,
"",
output_dir='/tmp/')
job.logger = DummyLogger()
description_ref = pipeline_reference('bbb-group-vland-alpha.yaml')
self.assertEqual(description_ref, job.pipeline.describe(False))
job.validate()
self.assertNotEqual([], [
protocol.name for protocol in job.protocols
if protocol.name == MultinodeProtocol.name
])
ret = {
"message": {
"kvm01": {
"vlan_name": "name",
"vlan_tag": 6
}
},
"response": "ack"
}
self.assertEqual(('name', 6), (
ret['message']['kvm01']['vlan_name'],
ret['message']['kvm01']['vlan_tag'],
))
self.assertIn('protocols', job.parameters)
self.assertIn(VlandProtocol.name, job.parameters['protocols'])
self.assertIn(MultinodeProtocol.name, job.parameters['protocols'])
vprotocol = [
vprotocol for vprotocol in job.protocols
if vprotocol.name == VlandProtocol.name
][0]
self.assertTrue(vprotocol.valid)
self.assertEqual(vprotocol.names, {'vlan_one': '4212vlanone'})
self.assertFalse(vprotocol.check_timeout(120, {'request': 'no call'}))
self.assertRaises(JobError, vprotocol.check_timeout, 60,
'deploy_vlans')
self.assertRaises(JobError, vprotocol.check_timeout, 60,
{'request': 'deploy_vlans'})
self.assertTrue(
vprotocol.check_timeout(120, {'request': 'deploy_vlans'}))
for vlan_name in job.parameters['protocols'][VlandProtocol.name]:
if vlan_name == 'yaml_line':
continue
self.assertIn(vlan_name, vprotocol.params)
self.assertIn('switch', vprotocol.params[vlan_name])
self.assertIn('port', vprotocol.params[vlan_name])
self.assertIn('iface', vprotocol.params[vlan_name])
params = job.parameters['protocols'][vprotocol.name]
names = []
for key, _ in params.items():
if key == 'yaml_line':
continue
names.append(",".join([key, vprotocol.params[key]['iface']]))
# this device only has one interface with interface tags
self.assertEqual(names, ['vlan_one,eth1'])
def test_deployment(self):
job_parser = JobParser()
cubie = NewDevice(
os.path.join(os.path.dirname(__file__), '../devices/cubie1.yaml'))
sample_job_file = os.path.join(
os.path.dirname(__file__), 'sample_jobs/cubietruck-removable.yaml')
with open(sample_job_file) as sample_job_data:
job = job_parser.parse(sample_job_data,
cubie,
4212,
None,
"",
output_dir='/tmp/')
job.logger = DummyLogger()
job.validate()
self.assertIn('usb', cubie['parameters']['media'].keys())
deploy_params = [
methods for methods in job.parameters['actions']
if 'deploy' in methods.keys()
][1]['deploy']
self.assertIn('device', deploy_params)
self.assertIn(deploy_params['device'],
cubie['parameters']['media']['usb'])
self.assertIn(
'uuid',
cubie['parameters']['media']['usb'][deploy_params['device']])
self.assertIn(
'device_id',
cubie['parameters']['media']['usb'][deploy_params['device']])
self.assertNotIn(
'boot_part',
cubie['parameters']['media']['usb'][deploy_params['device']])
deploy_action = [
action for action in job.pipeline.actions
if action.name == 'storage-deploy'
][0]
tftp_deploy_action = [
action for action in job.pipeline.actions
if action.name == 'tftp-deploy'
][0]
self.assertIsNotNone(deploy_action)
test_dir = deploy_action.get_namespace_data(
action='test',
label='results',
key='lava_test_results_dir',
parameters=tftp_deploy_action.parameters)
self.assertIsNotNone(test_dir)
self.assertIn('/lava-', test_dir)
self.assertIsInstance(deploy_action, MassStorage)
self.assertIn('image', deploy_action.parameters.keys())
dd_action = [
action for action in deploy_action.internal_pipeline.actions
if action.name == 'dd-image'
][0]
self.assertEqual(
dd_action.boot_params[dd_action.parameters['device']]['uuid'],
'usb-SanDisk_Ultra_20060775320F43006019-0:0')
self.assertIsNotNone(
dd_action.get_namespace_data(action=dd_action.name,
label='u-boot',
key='boot_part'))
self.assertIsNotNone(
dd_action.get_namespace_data(action='uboot-from-media',
label='uuid',
key='boot_part'))
self.assertEqual(
'0', '%s' % dd_action.get_namespace_data(
action=dd_action.name, label='u-boot', key='boot_part'))
self.assertIsInstance(
dd_action.get_namespace_data(action='uboot-from-media',
label='uuid',
key='boot_part'), str)
self.assertEqual(
'0:1',
dd_action.get_namespace_data(action='uboot-from-media',
label='uuid',
key='boot_part'))
def test_substitutions(self):
"""
Test substitution of secondary media values into u-boot commands
Unlike most u-boot calls, removable knows in advance all the values it needs to substitute
into the boot commands for the secondary deployment as these are fixed by the device config
and the image details from the job submission.
"""
job_parser = JobParser()
cubie = NewDevice(os.path.join(os.path.dirname(__file__), '../devices/cubie1.yaml'))
sample_job_file = os.path.join(os.path.dirname(__file__), 'sample_jobs/cubietruck-removable.yaml')
with open(sample_job_file) as sample_job_data:
job = job_parser.parse(sample_job_data, cubie, 4212, None, "", output_dir='/tmp/')
job.logger = DummyLogger()
job.validate()
boot_params = [
methods for methods in job.parameters['actions'] if 'boot' in methods.keys()][1]['boot']
self.assertIn('ramdisk', boot_params)
self.assertIn('kernel', boot_params)
self.assertIn('dtb', boot_params)
self.assertIn('root_uuid', boot_params)
self.assertIn('boot_part', boot_params)
self.assertNotIn('type', boot_params)
self.assertGreater(len(job.pipeline.actions), 1)
self.assertIsNotNone(job.pipeline.actions[1].internal_pipeline)
u_boot_action = [action for action in job.pipeline.actions if action.name == 'uboot-action'][1]
overlay = [action for action in u_boot_action.internal_pipeline.actions if action.name == 'bootloader-overlay'][0]
self.assertIsNotNone(overlay.get_namespace_data(action='storage-deploy', label='u-boot', key='device'))
methods = cubie['actions']['boot']['methods']
self.assertIn('u-boot', methods)
self.assertIn('usb', methods['u-boot'])
self.assertIn('commands', methods['u-boot']['usb'])
commands_list = methods['u-boot']['usb']['commands']
device_id = u_boot_action.get_namespace_data(action='storage-deploy', label='u-boot', key='device')
self.assertIsNotNone(device_id)
kernel_type = u_boot_action.parameters['kernel_type']
bootcommand = map_kernel_uboot(kernel_type, device_params=cubie.get('parameters', None))
substitutions = {
'{BOOTX}': "%s %s %s %s" % (
bootcommand,
cubie['parameters'][bootcommand]['kernel'],
cubie['parameters'][bootcommand]['ramdisk'],
cubie['parameters'][bootcommand]['dtb'],),
'{RAMDISK}': boot_params['ramdisk'],
'{KERNEL}': boot_params['kernel'],
'{DTB}': boot_params['dtb'],
'{ROOT}': boot_params['root_uuid'],
'{ROOT_PART}': "%s:%s" % (
cubie['parameters']['media']['usb'][device_id]['device_id'],
u_boot_action.parameters['boot_part']
)
}
self.assertEqual('bootz 0x42000000 0x43300000 0x43000000', substitutions['{BOOTX}'])
self.assertEqual('/boot/initrd.img-3.16.0-4-armmp-lpae.u-boot', substitutions['{RAMDISK}'])
commands = substitute(commands_list, substitutions)
self.assertEqual(
commands,
[
'usb start',
'usb info',
'setenv autoload no',
"setenv initrd_high '0xffffffff'",
"setenv fdt_high '0xffffffff'",
'setenv initrd_addr_r ${ramdisk_addr_r}',
"setenv loadkernel 'load usb 0:1 ${kernel_addr_r} /boot/vmlinuz-3.16.0-4-armmp-lpae'",
"setenv loadinitrd 'load usb 0:1 ${initrd_addr_r} /boot/initrd.img-3.16.0-4-armmp-lpae.u-boot; setenv initrd_size ${filesize}'",
"setenv loadfdt 'load usb 0:1 ${fdt_addr_r} /boot/dtb-3.16.0-4-armmp-lpae'",
"setenv bootargs 'console=ttyS0,115200n8 root=UUID=159d17cc-697c-4125-95a0-a3775e1deabe ip=dhcp'",
"setenv bootcmd 'run loadkernel; run loadinitrd; run loadfdt; bootz 0x42000000 0x43300000 0x43000000'", 'boot'
]
)
