"""The driver with its transport replaced: what it sends, what it returns."""

from __future__ import annotations

import pathlib

import pytest
from napalm.base.exceptions import ConnectionException

from napalm_example import ExampleDriver

FIXTURES = pathlib.Path(__file__).parent / "fixtures"


class FakeTransport:
    def __init__(self, answers: dict[str, str]) -> None:
        self.answers = answers
        self.sent: list[str] = []
        self.closed = False

    def command(self, line: str) -> str:
        self.sent.append(line)
        return self.answers[line]

    def close(self) -> None:
        self.closed = True


@pytest.fixture
def driver() -> ExampleDriver:
    return ExampleDriver("192.0.2.10", "admin", "secret")


def test_get_facts(driver: ExampleDriver) -> None:
    show_version = (FIXTURES / "example24-1.0.0-show-version.txt").read_text()
    driver.transport = FakeTransport({"show version": show_version})

    facts = driver.get_facts()

    assert facts["vendor"] == "Example"
    assert facts["hostname"] == "sw-lab-01"
    assert facts["fqdn"] == "sw-lab-01"
    assert facts["interface_list"] == []
    assert facts["uptime"] == 273906.0


def test_a_getter_before_open_raises(driver: ExampleDriver) -> None:
    with pytest.raises(ConnectionException):
        driver.get_facts()


def test_close_closes_the_transport_once(driver: ExampleDriver) -> None:
    transport = FakeTransport({})
    driver.transport = transport

    driver.close()
    driver.close()

    assert transport.closed
    assert driver.transport is None


def test_the_driver_is_registered_under_its_name() -> None:
    from napalm import get_network_driver

    assert get_network_driver("example") is ExampleDriver