from gpu_rent.cloud import ensure_floating_ip, server_floating_ip, server_status
from gpu_rent.lock import SessionLock
from gpu_rent.paths import lock_path


class Obj:
    def __init__(self, **kwargs):
        self.__dict__.update(kwargs)


def test_server_floating_ip():
    server = Obj(
        status="ACTIVE",
        addresses={"gpu-rent": [{"addr": "192.168.77.10", "OS-EXT-IPS:type": "fixed"}, {"addr": "203.0.113.9", "OS-EXT-IPS:type": "floating"}]},
    )
    assert server_status(server) == "ACTIVE"
    assert server_floating_ip(server) == "203.0.113.9"


def test_lock_roundtrip():
    with SessionLock():
        assert lock_path().is_file()
    assert not lock_path().is_file()


def test_ensure_floating_ip_reuses_existing():
    class Net:
        def ports(self, device_id=None):
            return [Obj(id="p1")]

        def update_ip(self, existing_id, port_id=None):
            assert existing_id == "old-fip"
            assert port_id == "p1"
            return Obj(floating_ip_address="9.9.9.9")

        def create_ip(self, floating_network_id=None):
            raise AssertionError("must reuse FIP")

    conn = Obj(network=Net())
    server = Obj(id="s1", addresses={})
    ip, fip_id = ensure_floating_ip(conn, server, "old-fip", "9.9.9.9", lambda m: None)
    assert ip == "9.9.9.9"
    assert fip_id == "old-fip"