216 lines
6.4 KiB
Python
216 lines
6.4 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 typing
|
|
|
|
import yaml
|
|
|
|
import tobiko
|
|
from tobiko import config
|
|
from tobiko.openstack import glance
|
|
from tobiko.openstack.stacks import _nova
|
|
from tobiko.openstack.stacks import _vlan
|
|
from tobiko.shell import sh
|
|
|
|
|
|
CONF = config.CONF
|
|
|
|
UBUNTU_IMAGE_VERSION = 'focal'
|
|
|
|
UBUNTU_IMAGE_VERSION_NUMBER = '20.04'
|
|
|
|
UBUNTU_MINIMAL_IMAGE_URL = (
|
|
'https://cloud-images.ubuntu.com/minimal/releases/'
|
|
f'{UBUNTU_IMAGE_VERSION}/release/'
|
|
f'ubuntu-{UBUNTU_IMAGE_VERSION_NUMBER}-minimal-cloudimg-amd64.img')
|
|
|
|
|
|
class UbuntuMinimalImageFixture(glance.URLGlanceImageFixture):
|
|
image_url = CONF.tobiko.ubuntu.image_url or UBUNTU_MINIMAL_IMAGE_URL
|
|
image_name = CONF.tobiko.ubuntu.image_name
|
|
image_file = CONF.tobiko.ubuntu.image_file
|
|
disk_format = CONF.tobiko.ubuntu.disk_format or "qcow2"
|
|
container_format = CONF.tobiko.ubuntu.container_format or "bare"
|
|
username = CONF.tobiko.ubuntu.username or 'ubuntu'
|
|
password = CONF.tobiko.ubuntu.password or 'ununtu'
|
|
connection_timeout = CONF.tobiko.ubuntu.connection_timeout or 600.
|
|
|
|
|
|
IPERF3_SERVICE_FILE = """
|
|
[Unit]
|
|
Description=iperf3 server on port %i
|
|
After=syslog.target network.target
|
|
|
|
[Service]
|
|
ExecStart=/usr/bin/iperf3 -s -p %i
|
|
Restart=always
|
|
RuntimeMaxSec=3600
|
|
User=root
|
|
|
|
[Install]
|
|
WantedBy=multi-user.target
|
|
DefaultInstance=5201
|
|
"""
|
|
|
|
|
|
class UbuntuImageFixture(UbuntuMinimalImageFixture,
|
|
glance.CustomizedGlanceImageFixture):
|
|
"""Ubuntu server image running an HTTP server
|
|
|
|
The server has additional installed packages compared to
|
|
the minimal one:
|
|
- iperf3
|
|
- ping
|
|
- ncat
|
|
- nginx
|
|
- vlan
|
|
|
|
The image will also have below running services:
|
|
- nginx HTTP server listening on TCP port 80
|
|
- iperf3 server listening on TCP port 5201
|
|
"""
|
|
|
|
@property
|
|
def firstboot_commands(self) -> typing.List[str]:
|
|
return super().firstboot_commands + [
|
|
'sh -c "hostname > /var/www/html/id"']
|
|
|
|
@property
|
|
def install_packages(self) -> typing.List[str]:
|
|
return super().install_packages + ['iperf3',
|
|
'iputils-ping',
|
|
'nano',
|
|
'ncat',
|
|
'nginx',
|
|
'vlan']
|
|
|
|
# port of running HTTP server
|
|
http_port = 80
|
|
|
|
# port of running Iperf3 server
|
|
iperf3_port = 5201
|
|
|
|
@property
|
|
def iperf3_service_name(self) -> str:
|
|
return f"iperf3-server@{self.iperf3_port}.service"
|
|
|
|
@property
|
|
def run_commands(self) -> typing.List[str]:
|
|
run_commands = super().run_commands
|
|
run_commands.append(
|
|
f'echo "{IPERF3_SERVICE_FILE}" '
|
|
'> /etc/systemd/system/iperf3-server@.service')
|
|
run_commands.append(
|
|
f"systemctl enable iperf3-server@{self.iperf3_port}")
|
|
run_commands.append('echo "8021q" >> /etc/modules')
|
|
return run_commands
|
|
|
|
@property
|
|
def ethernet_device(self) -> str:
|
|
return 'ens3'
|
|
|
|
@property
|
|
def vlan_id(self) -> int:
|
|
return tobiko.tobiko_config().neutron.vlan_id
|
|
|
|
@property
|
|
def vlan_device(self) -> str:
|
|
return f'vlan{self.vlan_id}'
|
|
|
|
@property
|
|
def vlan_config(self) -> typing.Dict[str, typing.Any]:
|
|
return {
|
|
'network': {
|
|
'version': 2,
|
|
'vlans': {
|
|
self.vlan_device: {
|
|
'link': self.ethernet_device,
|
|
'dhcp4': True,
|
|
'dhcp6': True,
|
|
'id': self.vlan_id,
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
@property
|
|
def write_files(self) -> typing.Dict[str, str]:
|
|
write_files = super().write_files
|
|
write_files['/etc/netplan/75-tobiko-vlan.yaml'] = yaml.dump(
|
|
self.vlan_config)
|
|
return write_files
|
|
|
|
|
|
class UbuntuFlavorStackFixture(_nova.FlavorStackFixture):
|
|
ram = 256
|
|
swap = 512
|
|
|
|
|
|
class UbuntuMinimalServerStackFixture(_nova.CloudInitServerStackFixture):
|
|
|
|
#: Glance image used to create a Nova server instance
|
|
image_fixture = tobiko.required_setup_fixture(UbuntuMinimalImageFixture)
|
|
|
|
#: Flavor used to create a Nova server instance
|
|
flavor_stack = tobiko.required_setup_fixture(UbuntuFlavorStackFixture)
|
|
|
|
|
|
class UbuntuServerStackFixture(UbuntuMinimalServerStackFixture,
|
|
_vlan.VlanServerStackFixture):
|
|
"""Ubuntu server running an HTTP server
|
|
|
|
The server has additional commands compared to the minimal one:
|
|
iperf3
|
|
ping
|
|
"""
|
|
|
|
#: Glance image used to create a Nova server instance
|
|
image_fixture = tobiko.required_setup_fixture(UbuntuImageFixture)
|
|
|
|
# port of running HTTP server
|
|
@property
|
|
def http_port(self) -> int:
|
|
return self.image_fixture.http_port
|
|
|
|
@property
|
|
def iperf3_port(self) -> int:
|
|
return self.image_fixture.iperf3_port
|
|
|
|
@property
|
|
def iperf3_service_name(self) -> str:
|
|
return self.image_fixture.iperf3_service_name
|
|
|
|
def wait_for_iperf3_server(self,
|
|
timeout: tobiko.Seconds = None,
|
|
interval: tobiko.Seconds = None):
|
|
return sh.wait_for_active_systemd_units(self.iperf3_service_name,
|
|
timeout=timeout,
|
|
interval=interval,
|
|
ssh_client=self.ssh_client)
|
|
|
|
@property
|
|
def vlan_id(self) -> int:
|
|
return self.image_fixture.vlan_id
|
|
|
|
@property
|
|
def vlan_device(self) -> str:
|
|
return self.image_fixture.vlan_device
|
|
|
|
|
|
class UbuntuExternalServerStackFixture(UbuntuServerStackFixture,
|
|
_nova.ExternalServerStackFixture):
|
|
"""Ubuntu server with port on special external network
|
|
"""
|