from ipaddress import IPv4Address

import pytest

from primaite.simulator.network.container import Network
from primaite.simulator.network.hardware.nodes.computer import Computer
from primaite.simulator.network.hardware.nodes.server import Server
from primaite.simulator.network.protocols.ntp import NTPPacket
from primaite.simulator.system.services.ntp.ntp_client import NTPClient
from primaite.simulator.system.services.ntp.ntp_server import NTPServer
from primaite.simulator.system.services.service import ServiceOperatingState

# Create simple network for testing


def create_ntp_network() -> Network:
    """
    +------------+            +------------+
    |    ntp     |            |     ntp    |
    |  client_1  +------------+  server_1  |
    |            |            |            |
    +------------+            +------------+

    """

    network = Network()
    ntp_server = Server(
        hostname="ntp_server", ip_address="192.168.1.2", subnet_mask="255.255.255.0", default_gateway="192.168.1.1"
    )
    ntp_server.power_on()

    ntp_client = Computer(
        hostname="ntp_client", ip_address="192.168.1.3", subnet_mask="255.255.255.0", default_gateway="192.168.1.1"
    )
    ntp_client.power_on()
    network.connect(endpoint_b=ntp_server.ethernet_port[1], endpoint_a=ntp_client.ethernet_port[1])

    return network


# @pytest.fixture()
# def create_network():
#     return create_ntp_network()

# Define one node to be an NTP server and another node to be a NTP Client.


def test_ntp_client_server():
    network = create_ntp_network()
    server: Server = network.get_node_by_hostname("ntp_server")
    client: Computer = network.get_node_by_hostname("ntp_client")

    ntp_server: NTPServer = server.software_manager.software["NTP_Server"]
    ntp_client: NTPClient = client.software_manager.software["NTP_Client"]

    assert ntp_server.operating_state == ServiceOperatingState.RUNNING
    assert ntp_client.operating_state == ServiceOperatingState.RUNNING

    ntp_client.send(payload=NTPPacket())
    assert ntp_server.receive() is True
    assert ntp_client.receive() is True

    assert ntp_client.apply_timestep(1) is True


# Test ntp client behaviour when ntp server is unavailable.
def test_ntp_server_failure():
    network = create_ntp_network()
    server: Server = network.get_node_by_hostname("ntp_server")
    client: Computer = network.get_node_by_hostname("ntp_client")

    ntp_server: NTPServer = server.software_manager.software["NTP_Server"]
    ntp_client: NTPClient = client.software_manager.software["NTP_Client"]

    assert ntp_client.operating_state == ServiceOperatingState.RUNNING

    # Turn off ntp server.
    ntp_server.stop()
    assert ntp_server.operating_state == ServiceOperatingState.STOPPED
    assert ntp_client.receive() is False

    # Restart ntp server.
    ntp_server.start()
    assert ntp_server.operating_state == ServiceOperatingState.RUNNING