#!/usr/bin/env python3
# This is a very basic DHCP test that requests an IPv4 lease
# from a DHCP server and checks that the lease is set on the
# interface.

import time
import infamy, infamy.dhcp
import infamy.iface as iface
from infamy.util import until

with infamy.Test() as test:
    ADDRESS = '10.0.0.42'
    with test.step("Initialize"):
        env = infamy.Env(infamy.std_topology("1x2"))
        target = env.attach("target", "mgmt")
        _, host = env.ltop.xlate("host", "data")

    with infamy.IsolatedMacVlan(host) as netns:
        netns.addip("10.0.0.1")
        with infamy.dhcp.Server(netns, ip=ADDRESS):
            _, port = env.ltop.xlate("target", "data")
            config = {
                "dhcp-client": {
                    "client-if": [{
                        "if-name": f"{port}"
                    }]
                }
            }
            target.put_config_dict("infix-dhcp-client", config)

            with test.step(f"Waiting for client to set DHCP lease {ADDRESS}"):
                until(lambda: iface.address_exist(target, port, ADDRESS))

    test.succeed()