def test_create_configdrive_partition(mocker, node, disk_attrs):
name = 'disk1'
node.mock_add_spec(['get_attribute'])
node.data = {"id": "1"}
node.get_attribute.return_value = disk_attrs
mock_create_part = mocker.patch("octane.util.disk.create_partition")
if disk_attrs:
disk_util.create_configdrive_partition(node)
mock_create_part.assert_called_once_with(
name, magic_consts.CONFIGDRIVE_PART_SIZE, node)
else:
with pytest.raises(disk_util.NoDisksInfoError):
disk_util.create_configdrive_partition(node)
def move_nodes(env, nodes, provision=True, roles=None):
env_id = env.data['id']
cmd = ["move", "node"]
if not provision:
cmd += ['--no-provision']
if roles:
cmd += ['--roles', ','.join(roles)]
cmd.append(str(env_id))
for node in nodes:
node_id = node.data['id']
cmd.append(str(node_id))
if provision and incompatible_provision_method(env):
disk.create_configdrive_partition(node)
fuel2_env_call(cmd)
if provision:
LOG.info("Nodes provision started. Please wait...")
wait_for_nodes(nodes, "provisioned")