def test_shell():
"""
Checks that the bash shell of a host sends a proper reply.
"""
platform = DockerPlatform(None, None)
platform.pre_build()
# Add node
node1 = Node(identifier='host1', type='host')
host1 = platform.add_node(node1)
# Add ports
p1 = BidirectionalPort(identifier='p1')
platform.add_biport(host1, p1)
p2 = BidirectionalPort(identifier='p2')
platform.add_biport(host1, p2)
p3 = BidirectionalPort(identifier='p3')
platform.add_biport(host1, p3)
platform.post_build()
reply = host1('echo "var"')
platform.destroy()
assert 'var' in reply
def test_shell():
"""
Checks that both bash shells (default and front_panel) of a host reply
properly.
"""
platform = DockerPlatform(None, None)
platform.pre_build()
# Add node
node1 = Node(identifier='host1', type='host')
host1 = platform.add_node(node1)
# Add ports
p1 = BidirectionalPort(identifier='p1')
platform.add_biport(host1, p1)
p2 = BidirectionalPort(identifier='p2')
platform.add_biport(host1, p2)
p3 = BidirectionalPort(identifier='p3')
platform.add_biport(host1, p3)
platform.post_build()
reply = host1('echo "var"')
reply_front_panel = host1('echo "var"', shell='bash')
platform.destroy()
assert 'var' in reply
assert 'var' in reply_front_panel
def test_add_port():
"""
Add ports and uses 'ip link list' to check they exist.
"""
platform = DockerPlatform(None, None)
platform.pre_build()
node1 = Node(identifier='host1', type='host')
host1 = platform.add_node(node1)
assert platform.nmlnode_node_map[node1.identifier] is not None
# Add ports
p1 = BidirectionalPort(identifier='p1')
platform.add_biport(node1, p1)
p2 = BidirectionalPort(identifier='p2')
platform.add_biport(node1, p2)
p3 = BidirectionalPort(identifier='p3')
platform.add_biport(node1, p3)
# Add link
link = BidirectionalLink(identifier='link')
platform.add_bilink((node1, p1), (node1, p2), link)
platform.post_build()
result = host1('ip link list')
platform.destroy()
assert 'p1: <BROADCAST,MULTICAST> ' in result
assert 'p2: <BROADCAST,MULTICAST> ' in result
assert 'p3: <BROADCAST,MULTICAST> ' in result
def test_vtysh():
"""
Checks that the vtysh shell of a host sends a proper reply.
"""
platform = DockerPlatform(None, None)
platform.pre_build()
# Add node
node1 = Node(identifier='switch1', type='openswitch')
switch1 = platform.add_node(node1)
# Add ports
p1 = BidirectionalPort(identifier='p1', label=1)
platform.add_biport(switch1, p1)
p2 = BidirectionalPort(identifier='p2', label=2)
platform.add_biport(switch1, p2)
p3 = BidirectionalPort(identifier='p3', label=3)
platform.add_biport(switch1, p3)
platform.post_build()
reply = switch1('show vlan')
platform.destroy()
assert 'vlan' in reply
def test_build_topology():
"""
Builds (and destroys) a basic topology consisting in one switch and one
host
"""
# Setup which shell to use
shell = 'bash'
platform = DockerPlatform(None, None)
platform.pre_build()
# Create node 1
node1 = Node(identifier='host1', type='host')
host1 = platform.add_node(node1)
assert platform.nmlnode_node_map[node1.identifier] is not None
# Create node 2
node2 = Node(identifier='host2', type='host')
host2 = platform.add_node(node2)
assert platform.nmlnode_node_map[node2.identifier] is not None
# Add a port to node 1
p1 = BidirectionalPort(identifier='p1')
platform.add_biport(node1, p1)
# Add a port to node 2
p2 = BidirectionalPort(identifier='p2')
platform.add_biport(node2, p2)
# Create a link between both ports
link = BidirectionalLink(identifier='link')
platform.add_bilink((node1, p1), (node2, p2), link)
platform.post_build()
# Configure network
host1('ip link set dev p1 up', shell=shell)
host1('ip addr add 10.1.1.1/24 dev p1', shell=shell)
host2('ip link set dev p2 up', shell=shell)
host2('ip addr add 10.1.1.2/24 dev p2', shell=shell)
# Test ping
ping_result = host2('ping -c 1 10.1.1.1', shell=shell)
platform.destroy()
assert '1 packets transmitted, 1 received' in ping_result
def test_add_port():
"""
Add ports and uses 'ip link list' to check they exist.
"""
# Setup which shell to use
shell = 'bash'
platform = DockerPlatform(None, None)
platform.pre_build()
node1 = Node(identifier='host1', type='host')
host1 = platform.add_node(node1)
node2 = Node(identifier='host2', type='host')
host2 = platform.add_node(node2)
assert platform.nmlnode_node_map[node1.identifier] is not None
assert platform.nmlnode_node_map[node2.identifier] is not None
# Add ports
p1 = BidirectionalPort(identifier='p1')
platform.add_biport(node1, p1)
p2 = BidirectionalPort(identifier='p2')
platform.add_biport(node2, p2)
p3 = BidirectionalPort(identifier='p3')
platform.add_biport(node1, p3)
# Add link
link = BidirectionalLink(identifier='link')
platform.add_bilink((node1, p1), (node2, p2), link)
platform.post_build()
result1 = host1('ip link list', shell=shell)
result2 = host2('ip link list', shell=shell)
platform.destroy()
# On newer kernels, ip link list will append the name of the peer
# interface of a veth link after an @, like this: p1@if34
assert search('p1.*: <BROADCAST,MULTICAST> ', result1)
assert search('p2.*: <BROADCAST,MULTICAST> ', result2)
assert 'p3: <BROADCAST,MULTICAST> ' in result1
def test_no_duplicate_biport_ids():
with pytest.raises(ValueError):
"""
Builds a topology with a repeated biport identifier and checks that the
correct exception (ValueError) is raised
"""
# Build topology
platform = DockerPlatform(None, None)
platform.pre_build()
h1 = Node(identifier='hs1', type='host')
h2 = Node(identifier='hs2', type='host')
hs1 = platform.add_node(h1)
hs2 = platform.add_node(h2)
s1p1 = BidirectionalPort(identifier='1')
s1p2 = BidirectionalPort(identifier='1')
platform.add_biport(hs1, s1p1)
platform.add_biport(hs2, s1p2)
def test_ping():
"""
Builds the topology described on the following schema and ping h2 from h1.
::
+------+ +------+
| | +------+ +------+ | |
| h1 <-----> s1 <-----> s2 <-----> h2 |
| | +------+ +------+ | |
+------+ +------+
"""
# Setup which shell to use
shell = 'bash'
# Build topology
platform = DockerPlatform(None, None)
platform.pre_build()
h1 = Node(identifier='hs1', type='host')
h2 = Node(identifier='hs2', type='host')
s1 = Node(identifier='sw1', type='host')
s2 = Node(identifier='sw2', type='host')
hs1 = platform.add_node(h1)
hs2 = platform.add_node(h2)
sw1 = platform.add_node(s1)
sw2 = platform.add_node(s2)
s1p1 = BidirectionalPort(identifier='sw1-3')
s1p2 = BidirectionalPort(identifier='sw1-4')
platform.add_biport(s1, s1p1)
platform.add_biport(s1, s1p2)
s2p1 = BidirectionalPort(identifier='sw2-3')
s2p2 = BidirectionalPort(identifier='sw2-4')
platform.add_biport(s2, s2p1)
platform.add_biport(s2, s2p2)
h1p1 = BidirectionalPort(identifier='hs1-1')
h2p1 = BidirectionalPort(identifier='hs2-1')
platform.add_biport(h1, h1p1)
platform.add_biport(h2, h2p1)
link1 = BidirectionalLink(identifier='link1')
platform.add_bilink((s1, s1p1), (h1, h1p1), link1)
link2 = BidirectionalLink(identifier='link2')
platform.add_bilink((s1, s1p2), (s2, s2p1), link2)
link3 = BidirectionalLink(identifier='link3')
platform.add_bilink((s2, s2p2), (h2, h2p1), link3)
platform.post_build()
# Ping test
###########
# Configure IP and bring UP host 1 interfaces
hs1('ip link set dev hs1-1 up', shell=shell)
hs1('ip addr add 10.0.10.1/24 dev hs1-1', shell=shell)
# Configure IP and bring UP host 2 interfaces
hs2('ip link set dev hs2-1 up', shell=shell)
hs2('ip addr add 10.0.30.1/24 dev hs2-1', shell=shell)
# Configure IP and bring UP switch 1 interfaces
sw1('ip link set dev sw1-3 up', shell=shell)
sw1('ip link set dev sw1-4 up', shell=shell)
sw1('ip addr add 10.0.10.2/24 dev sw1-3', shell=shell)
sw1('ip addr add 10.0.20.1/24 dev sw1-4', shell=shell)
# Configure IP and bring UP switch 2 interfaces
sw2('ip link set dev sw2-3 up', shell=shell)
sw2('ip addr add 10.0.20.2/24 dev sw2-3', shell=shell)
sw2('ip link set dev sw2-4 up', shell=shell)
sw2('ip addr add 10.0.30.2/24 dev sw2-4', shell=shell)
# Set static routes in switches
sw1('ip route add 10.0.30.0/24 via 10.0.20.2', shell=shell)
sw2('ip route add 10.0.10.0/24 via 10.0.20.1', shell=shell)
# Set gateway in hosts
hs1('ip route add default via 10.0.10.2', shell=shell)
hs2('ip route add default via 10.0.30.2', shell=shell)
ping_result = hs1('ping -c 1 10.0.30.1', shell=shell)
platform.destroy()
assert '1 packets transmitted, 1 received' in ping_result
def test_ping_openvswitch():
"""
Builds the topology described on the following schema and ping h2 from h1.
::
+------+ +------+
| | +------+ +------+ | |
| h1 <-----> s1 <-----> s2 <-----> h2 |
| | +------+ +------+ | |
+------+ +------+
"""
# Build topology
platform = DockerPlatform(None, None)
platform.pre_build()
h1 = Node(identifier='hs1', type='host')
h2 = Node(identifier='hs2', type='host')
s1 = Node(identifier='sw1', type='openvswitch')
s2 = Node(identifier='sw2', type='openvswitch')
hs1 = platform.add_node(h1)
hs2 = platform.add_node(h2)
sw1 = platform.add_node(s1)
sw2 = platform.add_node(s2)
s1p1 = BidirectionalPort(identifier='sw1-3')
s1p2 = BidirectionalPort(identifier='sw1-4')
platform.add_biport(s1, s1p1)
platform.add_biport(s1, s1p2)
s2p1 = BidirectionalPort(identifier='sw2-3')
s2p2 = BidirectionalPort(identifier='sw2-4')
platform.add_biport(s2, s2p1)
platform.add_biport(s2, s2p2)
h1p1 = BidirectionalPort(identifier='hs1-1')
h2p1 = BidirectionalPort(identifier='hs2-1')
platform.add_biport(h1, h1p1)
platform.add_biport(h2, h2p1)
link1 = BidirectionalLink(identifier='link1')
platform.add_bilink((s1, s1p1), (h1, h1p1), link1)
link2 = BidirectionalLink(identifier='link2')
platform.add_bilink((s1, s1p2), (s2, s2p1), link2)
link3 = BidirectionalLink(identifier='link3')
platform.add_bilink((s2, s2p2), (h2, h2p1), link3)
platform.post_build()
# Ping test
###########
# Configure IP and bring UP host 1 interfaces
commands = """
ip link set hs1-1 up
ip addr add 10.0.10.1/24 dev hs1-1
ip route add default via 10.0.10.2
"""
hs1.libs.common.assert_batch(commands)
# Configure IP and bring UP host 2 interfaces
commands = """
ip link set hs2-1 up
ip addr add 10.0.30.1/24 dev hs2-1
ip route add default via 10.0.30.2
"""
hs2.libs.common.assert_batch(commands)
# Configure IP and bring UP switch 1 interfaces
commands = """
ip link set sw1-3 up
ip link set sw1-4 up
ip addr add 10.0.10.2/24 dev sw1-3
ip addr add 10.0.20.1/24 dev sw1-4
ip route add 10.0.30.0/24 via 10.0.20.2
"""
sw1.libs.common.assert_batch(commands)
# Configure IP and bring UP switch 2 interfaces
commands = """
ip link set sw2-3 up
ip link set sw2-4 up
ip addr add 10.0.20.2/24 dev sw2-3
ip addr add 10.0.30.2/24 dev sw2-4
ip route add 10.0.10.0/24 via 10.0.20.1
"""
sw2.libs.common.assert_batch(commands)
ping_result = hs1.send_command('ping -c 1 10.0.30.1')
platform.destroy()
assert '1 packets transmitted, 1 received' in ping_result