119 lines
3.8 KiB
Python
119 lines
3.8 KiB
Python
# Copyright 2019 Red Hat
|
|
#
|
|
# 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.
|
|
from __future__ import absolute_import
|
|
|
|
import tobiko
|
|
from tobiko import config
|
|
from tobiko.openstack import glance
|
|
from tobiko.openstack.stacks import _nova
|
|
from tobiko.shell import sh
|
|
|
|
|
|
CONF = config.CONF
|
|
|
|
CIRROS_IMAGE_VERSION = '0.5.1'
|
|
|
|
CIRROS_IMAGE_URL = (
|
|
'http://download.cirros-cloud.net/{version}/'
|
|
'cirros-{version}-x86_64-disk.img').format(version=CIRROS_IMAGE_VERSION)
|
|
|
|
|
|
class CirrosImageFixture(glance.URLGlanceImageFixture):
|
|
|
|
image_url = CONF.tobiko.cirros.image_url or CIRROS_IMAGE_URL
|
|
image_name = CONF.tobiko.cirros.image_name
|
|
image_file = CONF.tobiko.cirros.image_file
|
|
container_format = CONF.tobiko.cirros.container_format or "bare"
|
|
disk_format = CONF.tobiko.cirros.disk_format or "qcow2"
|
|
username = CONF.tobiko.cirros.username or 'cirros'
|
|
password = CONF.tobiko.cirros.password or 'gocubsgo'
|
|
connection_timeout = CONF.tobiko.cirros.connection_timeout or 200.
|
|
|
|
|
|
class CirrosFlavorStackFixture(_nova.FlavorStackFixture):
|
|
ram = 128
|
|
|
|
|
|
class CirrosServerStackFixture(_nova.ServerStackFixture):
|
|
|
|
#: Glance image used to create a Nova server instance
|
|
image_fixture = tobiko.required_setup_fixture(CirrosImageFixture)
|
|
|
|
#: Flavor used to create a Nova server instance
|
|
flavor_stack = tobiko.required_setup_fixture(CirrosFlavorStackFixture)
|
|
|
|
#: CirrOS can't get IP addresses from config-drive
|
|
need_dhcp = True
|
|
|
|
|
|
class CirrosPeerServerStackFixture(CirrosServerStackFixture,
|
|
_nova.PeerServerStackFixture):
|
|
#: Peer server used to reach this one
|
|
peer_stack = tobiko.required_setup_fixture(CirrosServerStackFixture)
|
|
|
|
|
|
class CirrosSameHostServerStackFixture(
|
|
CirrosPeerServerStackFixture, _nova.SameHostServerStackFixture):
|
|
pass
|
|
|
|
|
|
class CirrosDifferentHostServerStackFixture(
|
|
CirrosPeerServerStackFixture, _nova.DifferentHostServerStackFixture):
|
|
pass
|
|
|
|
|
|
class RebootCirrosServerOperation(sh.RebootHostOperation):
|
|
|
|
stack = tobiko.required_setup_fixture(CirrosServerStackFixture)
|
|
|
|
@property
|
|
def ssh_client(self):
|
|
return self.stack.ssh_client
|
|
|
|
|
|
class EvacuableCirrosImageFixture(CirrosImageFixture):
|
|
|
|
tags = ['evacuable']
|
|
|
|
|
|
class EvacuableServerStackFixture(CirrosServerStackFixture):
|
|
|
|
#: Glance image used to create a Nova server instance
|
|
image_fixture = tobiko.required_setup_fixture(EvacuableCirrosImageFixture)
|
|
|
|
|
|
class CirrosExternalServerStackFixture(CirrosServerStackFixture,
|
|
_nova.ExternalServerStackFixture):
|
|
pass
|
|
|
|
|
|
class CirrosHttpServerStackFixture(CirrosPeerServerStackFixture,
|
|
_nova.HttpServerStackFixture):
|
|
|
|
@property
|
|
def user_data(self):
|
|
# Launch a webserver on port 80 that replies the server name to the
|
|
# client
|
|
# This webserver relies on the nc command which may fail if multiple
|
|
# clients connect at the same time. For concurrency testing,
|
|
# OctaviaCentosServerStackFixture is more suited to handle multiple
|
|
# requests.
|
|
reply = ("HTTP/1.1 200 OK\r\n"
|
|
"Content-Length:8\r\n"
|
|
"\r\n"
|
|
"$(hostname)")
|
|
command = f'nc -lk -p {self.http_server_port} -e echo -e "{reply}"'
|
|
return ("#!/bin/sh\n"
|
|
f"{command}")
|