s031_create_erase_vm.py

Go to the documentation of this file.

# -*- coding: utf-8 -*-
# @COPYRIGHT_begin
#
# Copyright [2010-2014] Institute of Nuclear Physics PAN, Krakow, Poland
#
# Licensed under the Apache License, Version 2.0 (the "License");
#    you may not use this file except in compliance with the License.
#    You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
#    distributed under the License is distributed on an "AS IS" BASIS,
#    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#    See the License for the specific language governing permissions and
#    limitations under the License.
#
# @COPYRIGHT_end

from predefined.users import *
from predefined.actions import *
from predefined.macros import *
import settings

actions = []
actions.append(list_clusters)
actions.append(list_private_vm_images)
actions.append(list_templates)
actions.append(list_user_public_ip)

# Create VMs
vm_names = []
for i in range(10):
    actions.append({'u': u_new,
                   'fname': 'vm.user.create',
                   'args': [{'vnc': True,
                             'name': 'TestVM%d' % i,
                             'iso_list': [],
                             'image_id': MACRO_GET_PRIVATE_VM_IMAGE_ID(settings.PRIVATE_IMAGE),
                             'disk_list': [],
                             'template_id': MACRO_GET_TEMPLATE_ID("Small"),
                             'description': 'asd'}],
                   })
    vm_names.append("TestVM%d" % i)

# wailt until all are in init
actions.append({'u': u_new,
                'fname': 'vm.user.list',
                'args': [],
                'wait_response': 'len(filter(lambda x: x["state"] == 0 and x["name"] in %s, r)) > 0' % str(vm_names),
                'wait_max': 3})

# wait until all are in running state
for i in range(10):
    actions.append({'u': u_new,
                    'fname': 'vm.user.list',
                    'args': [],
                    'wait_response': 'len(filter(lambda x: x["state"] == 1 and x["name"] == "TestVM%d", r)) > 0' % i,
                    'wait_max': 120})

actions.append(list_user_vms)

# Generate list to kill them all
vm_ids = []
for i in range(10):
    vm_ids.append(MACRO_GET_USER_VM_ID("TestVM%d" % i))

# Kill them all
actions.append({'u': u_new,
                'fname': 'vm.user.destroy',
                'args': [vm_ids]})