38 lines
924 B
Python
38 lines
924 B
Python
"""Tests for the dnsip integration."""
|
|
from __future__ import annotations
|
|
|
|
|
|
class QueryResult:
|
|
"""Return Query results."""
|
|
|
|
host = "1.2.3.4"
|
|
ttl = 60
|
|
|
|
|
|
class RetrieveDNS:
|
|
"""Return list of test information."""
|
|
|
|
def __init__(
|
|
self, nameservers: list[str] | None = None, error: Exception | None = None
|
|
) -> None:
|
|
"""Initialize DNS class."""
|
|
if nameservers:
|
|
self.nameservers = nameservers
|
|
self._nameservers = ["1.2.3.4"]
|
|
self.error = error
|
|
|
|
async def query(self, hostname, qtype) -> dict[str, str]:
|
|
"""Return information."""
|
|
if self.error:
|
|
raise self.error
|
|
return [QueryResult]
|
|
|
|
@property
|
|
def nameservers(self) -> list[str]:
|
|
"""Return nameserver."""
|
|
return self._nameservers
|
|
|
|
@nameservers.setter
|
|
def nameservers(self, value: list[str]) -> None:
|
|
self._nameservers = value
|