#!/usr/bin/env python3
#
# Verify that a simple web server container can be configured to run
# with host networking, on port 80.  Operation is verified using a
# simple GET request for index.html and checking for a key phrase.
#
# The RPC actions: stop + start, and restart are also verified.
#

import infamy
from   infamy.util import until

def _verify(server):
    url = infamy.Furl(f"http://[{server}]/index.html")
    return url.check("It works")

with infamy.Test() as test:
    NAME  = "web"

    with test.step("Set up topology and attach to target DUT"):
        env = infamy.Env(infamy.std_topology("1x2"))
        target = env.attach("target", "mgmt")
        addr = target.address()

    with test.step(f"Create {NAME} container from bundled OCI image"):
        target.put_config_dict("infix-services", {
            "web": {
                "enabled": False
            }
        })
        target.put_config_dict("infix-containers", {
            "containers": {
                "container": [
                    {
                        "name": f"{NAME}",
                        "image": f"oci-archive:{infamy.Container.IMAGE}",
                        "network": {
                            "host": True
                        }
                    }
                ]
            }
        })

    with test.step(f"Verify {NAME} continer has started"):
        c = infamy.Container(target)
        until(lambda: c.running(NAME), attempts=10)

    with test.step(f"Verify {NAME} container responds"):
        until(lambda: _verify(addr), attempts=10)

    with test.step(f"Verify {NAME} container can be stopped and restarted"):
        c = infamy.Container(target)
        c.action(NAME, "stop")
        until(lambda: not c.running(NAME), attempts=10)
        c.action(NAME, "restart")

    with test.step(f"Verify {NAME} container still responds"):
        # Wait for it to restart and respond, or fail
        until(lambda: _verify(addr), attempts=10)

    test.succeed()