163 lines
5.9 KiB
Python
163 lines
5.9 KiB
Python
# Copyright (c) 2019 Red Hat
|
|
# All Rights Reserved.
|
|
#
|
|
# 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 pytest
|
|
import testtools
|
|
from oslo_log import log
|
|
|
|
import tobiko
|
|
from tobiko.openstack import keystone
|
|
from tobiko.openstack import neutron
|
|
from tobiko.openstack import octavia
|
|
from tobiko.openstack import stacks
|
|
from tobiko.shell import sh
|
|
from tobiko.shell import ssh
|
|
|
|
|
|
LOG = log.getLogger(__name__)
|
|
|
|
|
|
@keystone.skip_if_missing_service(name='octavia')
|
|
class OctaviaBasicTrafficScenarioTest(testtools.TestCase):
|
|
"""Octavia traffic scenario test.
|
|
|
|
Create a load balancer with 2 members that run a server application,
|
|
Create a client that is connected to the load balancer VIP port,
|
|
Generate network traffic from the client to the load balanacer.
|
|
"""
|
|
loadbalancer_stack = tobiko.required_setup_fixture(
|
|
stacks.OctaviaLoadbalancerStackFixture)
|
|
|
|
listener_stack = tobiko.required_setup_fixture(
|
|
stacks.OctaviaListenerStackFixture)
|
|
|
|
pool_stack = tobiko.required_setup_fixture(
|
|
stacks.OctaviaPoolStackFixture)
|
|
|
|
member1_stack = tobiko.required_setup_fixture(
|
|
stacks.OctaviaMemberServerStackFixture)
|
|
|
|
member2_stack = tobiko.required_setup_fixture(
|
|
stacks.OctaviaOtherMemberServerStackFixture)
|
|
|
|
def setUp(self):
|
|
# pylint: disable=no-member
|
|
super(OctaviaBasicTrafficScenarioTest, self).setUp()
|
|
|
|
# Wait for Octavia objects to be active
|
|
LOG.info(f'Waiting for {self.member1_stack.stack_name} and '
|
|
f'{self.member2_stack.stack_name} to be created...')
|
|
self.pool_stack.wait_for_active_members()
|
|
|
|
octavia.wait_for_octavia_service(
|
|
loadbalancer_id=self.loadbalancer_stack.loadbalancer_id)
|
|
|
|
octavia.wait_for_members_to_be_reachable(
|
|
members=[self.member1_stack, self.member2_stack],
|
|
lb_protocol=self.listener_stack.lb_protocol,
|
|
lb_port=self.listener_stack.lb_port
|
|
)
|
|
|
|
@pytest.mark.flaky(reruns=3)
|
|
def test_round_robin_traffic(self):
|
|
for attempt in tobiko.retry(timeout=30.):
|
|
try:
|
|
octavia.check_members_balanced(
|
|
pool_id=self.pool_stack.pool_id,
|
|
ip_address=self.loadbalancer_stack.floating_ip_address,
|
|
lb_algorithm=self.pool_stack.lb_algorithm,
|
|
protocol=self.listener_stack.lb_protocol,
|
|
port=self.listener_stack.lb_port)
|
|
|
|
break
|
|
|
|
except octavia.RoundRobinException as e:
|
|
if attempt.is_last:
|
|
raise e
|
|
|
|
|
|
@neutron.skip_unless_is_ovn()
|
|
@keystone.skip_if_missing_service(name='octavia')
|
|
class OctaviaOVNProviderTrafficTest(testtools.TestCase):
|
|
"""Octavia OVN provider traffic test.
|
|
|
|
Create an OVN provider load balancer with 2 members that run a server
|
|
application,
|
|
Create a client that is connected to the load balancer VIP port via FIP,
|
|
Generate network traffic from the client to the load balanacer via ssh.
|
|
"""
|
|
loadbalancer_stack = tobiko.required_setup_fixture(
|
|
stacks.OctaviaOvnProviderLoadbalancerStackFixture)
|
|
|
|
listener_stack = tobiko.required_setup_fixture(
|
|
stacks.OctaviaOvnProviderListenerStackFixture)
|
|
|
|
pool_stack = tobiko.required_setup_fixture(
|
|
stacks.OctaviaOvnProviderPoolStackFixture)
|
|
|
|
member1_stack = tobiko.required_setup_fixture(
|
|
stacks.OctaviaOvnProviderMemberServerStackFixture)
|
|
|
|
member2_stack = tobiko.required_setup_fixture(
|
|
stacks.OctaviaOvnProviderOtherMemberServerStackFixture)
|
|
|
|
def setUp(self):
|
|
# pylint: disable=no-member
|
|
super(OctaviaOVNProviderTrafficTest, self).setUp()
|
|
|
|
# Wait for Octavia objects to be active
|
|
LOG.info(f'Waiting for {self.member1_stack.stack_name} and '
|
|
f'{self.member2_stack.stack_name} to be created...')
|
|
self.pool_stack.wait_for_active_members()
|
|
|
|
octavia.wait_for_octavia_service(
|
|
loadbalancer_id=self.loadbalancer_stack.loadbalancer_id)
|
|
|
|
def test_ssh_traffic(self):
|
|
LOG.info('Trying to ssh each member and get its hostname.')
|
|
|
|
username: typing.Optional[str] = None
|
|
expected = set()
|
|
for member_stack in [self.member1_stack, self.member2_stack]:
|
|
ssh_client = member_stack.server_stack.ssh_client
|
|
expected.add(sh.get_hostname(ssh_client=ssh_client))
|
|
if username is None:
|
|
username = ssh_client.setup_connect_parameters()['username']
|
|
else:
|
|
self.assertEqual(
|
|
username,
|
|
ssh_client.setup_connect_parameters()['username'],
|
|
"Member servers don't have the same "
|
|
"username to login with")
|
|
|
|
ssh_client = ssh.ssh_client(
|
|
host=self.loadbalancer_stack.floating_ip_address,
|
|
username=username)
|
|
|
|
actual = set()
|
|
for attempt in tobiko.retry(timeout=120.):
|
|
with ssh_client: # disconnect after every loop
|
|
actual.add(sh.ssh_hostname(ssh_client=ssh_client))
|
|
unexpected = actual - expected
|
|
if unexpected:
|
|
self.fail(f'Unexpected hostname: {unexpected}')
|
|
elif expected == actual:
|
|
break
|
|
elif attempt.is_last:
|
|
self.fail(f'Unreached host(s): {expected - actual}')
|