Files
Mattias Walström 34cb229965 test: Update topologies and test texts
All mgmt links should be lightgrey, all mgmt ports should be named mgmt
links between duts should be named link.
2024-10-16 10:51:01 +02:00

139 lines
5.9 KiB
Python
Executable File

#!/usr/bin/env python3
#
# Add join the group 224.1.1.1 from data1 on the host. Send to that
# group from `msend`, verify that `mrecv` receives it and that `!memb`
# does not.
#
"""
IGMP basic
Verify that all multicast get flooded when no IGMP join exists in the system and
the flooding stops as soon a join arrives
.1
.---------------------------.
| DUT |
'-data1-----data2-----data3-'
| | |
| | | 10.0.0.0/24
| | |
.-data1-. .-data2-. .-data3-.
| msend | | mrecv | | !memb |
'-------' '-------' '-------'
.2 .3 .4
HOST
"""
import infamy
import time
import infamy.multicast as mcast
query_interval = 4
with infamy.Test() as test:
with test.step("Set up topology and attach to target DUTs"):
env = infamy.Env()
target = env.attach("target", "mgmt")
_, msend = env.ltop.xlate("target", "data1")
_, mreceive = env.ltop.xlate("target", "data2")
_, nojoin = env.ltop.xlate("target", "data3")
with test.step("Configure device"):
target.put_config_dict("ietf-interfaces",
{
"interfaces": {
"interface": [
{
"name": msend,
"enabled": True,
"infix-interfaces:bridge-port": {
"bridge": "br0"
}
},
{
"name": mreceive,
"enabled": True,
"infix-interfaces:bridge-port": {
"bridge": "br0"
}
},
{
"name": nojoin,
"enabled": True,
"infix-interfaces:bridge-port": {
"bridge": "br0"
}
},
{
"name": "br0",
"enabled": True,
"type": "infix-if-type:bridge",
"ipv4": {
"address": [
{
"ip": "10.0.0.1",
"prefix-length": 24
}
]
},
"bridge": {
"multicast": {
"query-interval": query_interval,
"snooping": True
}
}
}
]
}
})
_, hsend = env.ltop.xlate("host", "data1")
_, hreceive = env.ltop.xlate("host", "data2")
_, hnojoin = env.ltop.xlate("host", "data3")
with infamy.IsolatedMacVlan(hsend) as send_ns, \
infamy.IsolatedMacVlan(hreceive) as receive_ns, \
infamy.IsolatedMacVlan(hnojoin) as nojoin_ns:
with test.step("Start multicast sender on host:data0, group 224.1.1.1"):
send_ns.addip("10.0.0.2")
receive_ns.addip("10.0.0.3")
nojoin_ns.addip("10.0.0.4")
send_ns.must_reach("10.0.0.1")
receive_ns.must_reach("10.0.0.1")
nojoin_ns.must_reach("10.0.0.1")
with mcast.MCastSender(send_ns, "224.1.1.1"):
with test.step("Verify that the group 224.1.1.1 is flooded on host:data2 and host:data3"):
infamy.parallel(lambda: receive_ns.must_receive("ip dst 224.1.1.1"),
lambda: nojoin_ns.must_receive("ip dst 224.1.1.1"))
with test.step("Join multicast group 224.1.1.1 on host:data2"):
with mcast.MCastReceiver(receive_ns, "224.1.1.1"):
with test.step("Verify group 224.1.1.1 is received on host:data2"):
receive_ns.must_receive("ip dst 224.1.1.1")
with test.step("Verify that the group 224.1.1.1 is no longer received on host:data3"):
attempt = 0
# This retry loop exists to handle the case where the first query is lost due
# to the network just starting up. In particular there's a case where a newly
# created bridge uses the "NOOP" scheduler (noop_enqueue()) for a split second
# while it's starting up, which might drop the first query msg.
while attempt <= query_interval:
try:
nojoin_ns.must_not_receive("ip dst 224.1.1.1")
break
except Exception as e:
attempt += 1
if attempt > query_interval:
test.fail()
else:
print(f"Got mcast flood, retrying ({attempt}/{query_interval})")
test.succeed()