infix/test/case/infix_services/services_basic/test.py

#!/usr/bin/env python3
"""
Services basic

Verify that basic services like mDNS and LLDP can be enabled and
disabled.  We verify operation and non-operation by using tcpdump.
"""
import time
import infamy


def toggle(updown):
    """Toggle port down/up to kick services"""
    _, port = env.ltop.xlate("target", "data")
    act = "UP" if updown else "DOWN"

    print(f"target: taking interface {port} {act} ...")
    target.put_config_dict("ietf-interfaces", {
        "interfaces": {
            "interface": [
                {
                    "name": port,
                    "enabled": updown
                }
            ]
        }
    })


def verify(enabled, sec):
    """Verify service traffic, or no traffic in case service not enabled"""
    _, hport = env.ltop.xlate("host", "data")

    with infamy.IsolatedMacVlan(hport) as netns:
        snif = infamy.Sniffer(netns, "port 5353 or ether proto 0x88cc")
        act = "enabling" if enabled else "disabling"

        netns.addip("10.0.0.1")
        netns.addroute("0.0.0.0/0", "10.0.0.1")

        # Prevent old-state traffic from being captured before reconf
        toggle(False)

        with snif:
            print("host: collecting network traffic ...")

            # Put service enable/disable before starting tcpdump,
            # because LLDP lingers and will send a final shutdown
            # message that otherwise would get in the capture for
            # disable.
            print(f"target: {act} LLDP and mDNS services ...")
            target.put_config_dict("infix-services", {
                "mdns": {
                    "enabled": enabled
                }
            })
            target.put_config_dict("ieee802-dot1ab-lldp", {
                "lldp": {
                    "enabled": enabled
                }
            })

            toggle(True)
            time.sleep(sec)

        return snif.output()


with infamy.Test() as test:
    with test.step("Initialize"):
        env = infamy.Env()
        target = env.attach("target", "mgmt")

    with test.step("Set static IPv4 address and disable services"):
        _, tport = env.ltop.xlate("target", "data")

        target.put_config_dict("ietf-interfaces", {
            "interfaces": {
                "interface": [
                    {
                        "name": tport,
                        "enabled": True,
                        "ipv4": {
                            "address": [
                                {
                                    "ip": "10.0.0.10",
                                    "prefix-length": 24
                                }
                            ]
                        }
                    }
                ]
            }
        })
        target.put_config_dict("infix-services", {
            "mdns": {
                "enabled": False
            }
        })
        target.put_config_dict("ieee802-dot1ab-lldp", {
            "lldp": {
                "enabled": False
            }
        })

    with test.step("Start sniffer and enable services on target ..."):
        rc = verify(True, 10)
        print(rc.stdout)
        # breakpoint()
        if "10.0.0.10.5353" not in rc.stdout:
            test.fail()
        if "LLDP" not in rc.stdout:
            test.fail()

    with test.step("Disable services on target, verify no longer running ..."):
        rc = verify(False, 10)
        print(rc.stdout)
        if "10.0.0.10.5353" in rc.stdout:
            test.fail()
        if "LLDP" in rc.stdout:
            test.fail()

    test.succeed()