#!/usr/bin/env python3
"""
Set hostname

Verify that it is possible to change hostname both normal
and using format %h-%m.

The %h-%m format expands to <default hostname>-<MAC>,
where MAC is the last three bytes of the base MAC address.

e.g. ix-01-01-01.
"""
import random
import string
import re
import infamy

with infamy.Test() as test:
    with test.step("Set up topology and attach to target DUT"):
        env = infamy.Env()
        target = env.attach("target", "mgmt")
        tgtssh = env.attach("target", "mgmt", "ssh")
        fmt = "%h-%m"
        new="h0stn4m3"

    with test.step("Set hostname to 'h0stn4m3'"):
        target.put_config_dict("ietf-system", {
            "system": {
                "hostname": new,
            }
        })

    with test.step("Verify new hostname 'h0stn4m3'"):
        running = target.get_config_dict("/ietf-system:system")
        assert running["system"]["hostname"] == new

    with test.step("Set hostname to '%h-%m'"):
        target.put_config_dict("ietf-system", {
            "system": {
                "hostname": fmt,
            }
        })

    with test.step("Verify hostname is  %h-%m in running configuration"):
        running = target.get_config_dict("/ietf-system:system")
        if running["system"]["hostname"] != fmt:
            test.fail()

    with test.step("Verify hostname format in operational, according to format"):
        cmd = tgtssh.runsh("sed -n s/^DEFAULT_HOSTNAME=//p /etc/os-release")
        default = cmd.stdout.rstrip()

        oper = target.get_data("/ietf-system:system")
        name = oper["system"]["hostname"]
        pattern = rf'^{default}-([0-9a-fA-F]{{2}}-){{2}}[0-9a-fA-F]{{2}}$'
        regex = re.compile(pattern)
        if not regex.match(name):
            test.fail()

    test.succeed()