128 lines
4.1 KiB
Python
128 lines
4.1 KiB
Python
# Copyright (c) 2015 Thales Services SAS
|
|
#
|
|
# 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 fixtures
|
|
import netaddr
|
|
|
|
from neutron.agent.linux import ip_lib
|
|
from neutron.tests.common import net_helpers
|
|
from neutron.tests import tools
|
|
|
|
|
|
class Pinger(object):
|
|
def __init__(self, namespace, timeout=1, max_attempts=1):
|
|
self.namespace = namespace
|
|
self._timeout = timeout
|
|
self._max_attempts = max_attempts
|
|
|
|
def _ping_destination(self, dest_address):
|
|
ns_ip_wrapper = ip_lib.IPWrapper(self.namespace)
|
|
ipversion = netaddr.IPAddress(dest_address).version
|
|
ping_command = 'ping' if ipversion == 4 else 'ping6'
|
|
ns_ip_wrapper.netns.execute([ping_command, '-c', self._max_attempts,
|
|
'-W', self._timeout, dest_address])
|
|
|
|
def assert_ping(self, dst_ip):
|
|
self._ping_destination(dst_ip)
|
|
|
|
def assert_no_ping(self, dst_ip):
|
|
try:
|
|
self._ping_destination(dst_ip)
|
|
tools.fail("destination ip %(dst_ip)s is replying to ping "
|
|
"from namespace %(ns)s, but it shouldn't" %
|
|
{'ns': self.namespace, 'dst_ip': dst_ip})
|
|
except RuntimeError:
|
|
pass
|
|
|
|
|
|
class FakeMachine(fixtures.Fixture):
|
|
"""Create a fake machine.
|
|
|
|
:ivar bridge: bridge on which the fake machine is bound
|
|
:ivar ip_cidr: fake machine ip_cidr
|
|
:type ip_cidr: str
|
|
:ivar ip: fake machine ip
|
|
:type ip: str
|
|
:ivar gateway_ip: fake machine gateway ip
|
|
:type gateway_ip: str
|
|
|
|
:ivar namespace: namespace emulating the machine
|
|
:type namespace: str
|
|
:ivar port: port binding the namespace to the bridge
|
|
:type port: IPDevice
|
|
"""
|
|
|
|
def __init__(self, bridge, ip_cidr, gateway_ip=None):
|
|
super(FakeMachine, self).__init__()
|
|
self.bridge = bridge
|
|
self.ip_cidr = ip_cidr
|
|
self.ip = self.ip_cidr.partition('/')[0]
|
|
self.gateway_ip = gateway_ip
|
|
|
|
def setUp(self):
|
|
super(FakeMachine, self).setUp()
|
|
ns_fixture = self.useFixture(
|
|
net_helpers.NamespaceFixture())
|
|
self.namespace = ns_fixture.name
|
|
|
|
self.port = self.useFixture(
|
|
net_helpers.PortFixture.get(self.bridge, self.namespace)).port
|
|
self.port.addr.add(self.ip_cidr)
|
|
|
|
if self.gateway_ip:
|
|
net_helpers.set_namespace_gateway(self.port, self.gateway_ip)
|
|
|
|
def execute(self, *args, **kwargs):
|
|
ns_ip_wrapper = ip_lib.IPWrapper(self.namespace)
|
|
return ns_ip_wrapper.netns.execute(*args, **kwargs)
|
|
|
|
def assert_ping(self, dst_ip):
|
|
pinger = Pinger(self.namespace)
|
|
pinger.assert_ping(dst_ip)
|
|
|
|
def assert_no_ping(self, dst_ip):
|
|
pinger = Pinger(self.namespace)
|
|
pinger.assert_no_ping(dst_ip)
|
|
|
|
|
|
class PeerMachines(fixtures.Fixture):
|
|
"""Create 'amount' peered machines on an ip_cidr.
|
|
|
|
:ivar bridge: bridge on which peer machines are bound
|
|
:ivar ip_cidr: ip_cidr on which peer machines have ips
|
|
:type ip_cidr: str
|
|
:ivar machines: fake machines
|
|
:type machines: FakeMachine list
|
|
"""
|
|
|
|
AMOUNT = 2
|
|
CIDR = '192.168.0.1/24'
|
|
|
|
def __init__(self, bridge, ip_cidr=None, gateway_ip=None):
|
|
super(PeerMachines, self).__init__()
|
|
self.bridge = bridge
|
|
self.ip_cidr = ip_cidr or self.CIDR
|
|
self.gateway_ip = gateway_ip
|
|
|
|
def setUp(self):
|
|
super(PeerMachines, self).setUp()
|
|
self.machines = []
|
|
|
|
for index in range(self.AMOUNT):
|
|
ip_cidr = net_helpers.increment_ip_cidr(self.ip_cidr, index)
|
|
self.machines.append(
|
|
self.useFixture(
|
|
FakeMachine(self.bridge, ip_cidr, self.gateway_ip)))
|