Files
Tobias Waldekranz 3f2ae3f3fc test: case: igmp_basic: Speed up positive reception tests
Use must{,_not}_receive instead of the general sniffer, so that we can
continue the test as soon as we see the transmission at the receiver,
rather than always waiting the full 5s.

While we're here:
- Show the port mappings in the top block-comment
- Define a few more steps to show the user what is going on
2024-05-20 16:21:06 +02:00

112 lines
4.6 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.
#
# .1
# .---------------------------.
# | DUT |
# '-data0-----data1-----data2-'
# | | |
# | | | 10.0.0.0/24
# | | |
# .-data0-. .-data1-. .-data2-.
# | msend | | mrecv | | !memb |
# '-------' '-------' '-------'
# .2 .3 .4
import infamy
import time
import infamy.multicast as mcast
with infamy.Test() as test:
with test.step("Initialize"):
env = infamy.Env(infamy.std_topology("1x4"))
target = env.attach("target", "mgmt")
_, msend = env.ltop.xlate("target", "data0")
_, mreceive = env.ltop.xlate("target", "data1")
_, nojoin = env.ltop.xlate("target", "data2")
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": {
"snooping": True
}
}
}
]
}
})
_, hsend = env.ltop.xlate("host", "data0")
_, hreceive = env.ltop.xlate("host", "data1")
_, hnojoin = env.ltop.xlate("host", "data2")
with infamy.IsolatedMacVlan(hsend) as send_ns, \
infamy.IsolatedMacVlan(hreceive) as receive_ns, \
infamy.IsolatedMacVlan(hnojoin) as nojoin_ns:
with test.step("Setup sender and receivers"):
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")
print("Starting sender")
with mcast.MCastSender(send_ns, "224.1.1.1"):
with test.step("Verify that the group is flooded on all ports"):
infamy.parallel(lambda: receive_ns.must_receive("ip dst 224.1.1.1"),
lambda: nojoin_ns.must_receive("ip dst 224.1.1.1"))
print("Subscribe to the group")
with mcast.MCastReceiver(receive_ns, "224.1.1.1"):
with test.step("Verify that the group is still forwarded to the member"):
receive_ns.must_receive("ip dst 224.1.1.1")
with test.step("Verify that the group is no longer forwarded to the non-member"):
nojoin_ns.must_not_receive("ip dst 224.1.1.1")
test.succeed()