74f99e9ab3
move code around so useful methods are in a common place. create a clients singleton holding all required clients for testing. Change-Id: I65d0a8b494ef760b9d14f132f5611cfefde03dc7
63 lines
2.1 KiB
Python
63 lines
2.1 KiB
Python
# Copyright 2017 - Nokia
|
|
#
|
|
# 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.
|
|
import time
|
|
|
|
from vitrage_tempest_tests.tests.common import glance_utils
|
|
from vitrage_tempest_tests.tests.common import neutron_utils
|
|
from vitrage_tempest_tests.tests.common.tempest_clients import TempestClients
|
|
from vitrage_tempest_tests.tests.utils import wait_for_status
|
|
|
|
|
|
def create_instances(num_instances, set_public_network=False, name='vm'):
|
|
kwargs = {}
|
|
flavor = get_first_flavor()
|
|
image = glance_utils.get_first_image()
|
|
if set_public_network:
|
|
public_net = neutron_utils.get_public_network()
|
|
if public_net:
|
|
kwargs.update({"networks": [{'uuid': public_net['id']}]})
|
|
|
|
resources = [TempestClients.nova().servers.create(
|
|
name='%s-%s' % (name, index),
|
|
flavor=flavor,
|
|
image=image,
|
|
**kwargs) for index in range(num_instances)]
|
|
wait_for_status(30, _check_num_instances, num_instances=num_instances,
|
|
state='active')
|
|
time.sleep(2)
|
|
return resources
|
|
|
|
|
|
def delete_all_instances():
|
|
instances = TempestClients.nova().servers.list()
|
|
for instance in instances:
|
|
try:
|
|
TempestClients.nova().servers.delete(instance)
|
|
except Exception:
|
|
pass
|
|
wait_for_status(30, _check_num_instances, num_instances=0)
|
|
time.sleep(2)
|
|
|
|
|
|
def get_first_flavor():
|
|
return TempestClients.nova().flavors.list()[0]
|
|
|
|
|
|
def _check_num_instances(num_instances=0, state=''):
|
|
if len(TempestClients.nova().servers.list()) != num_instances:
|
|
return False
|
|
|
|
return all(instance.__dict__['status'].upper() == state.upper()
|
|
for instance in TempestClients.nova().servers.list())
|