mirror of
https://github.com/Chia-Network/chia-blockchain.git
synced 2026-08-28 18:14:19 -05:00
Matt HauffandGitHub
40db4635a8
* Enable PEP604 Ruff rules * Fix harcoded signature in test * Hack CLVMStreamable test with note to fast follow
106 lines
3.6 KiB
Python
106 lines
3.6 KiB
Python
from __future__ import annotations
|
|
|
|
import os
|
|
import sys
|
|
from ipaddress import IPv4Address, IPv6Address
|
|
|
|
import pytest
|
|
|
|
from chia.util.ip_address import IPAddress
|
|
from chia.util.network import parse_host_port, resolve
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"host_port, expected_host, expected_port",
|
|
[
|
|
("127.0.0.1:8080", "127.0.0.1", 8080),
|
|
("[::1]:8080", "::1", 8080),
|
|
("example.com:8080", "example.com", 8080),
|
|
("localhost:8555", "localhost", 8555),
|
|
],
|
|
)
|
|
def test_parse_host_port(host_port: str, expected_host: str, expected_port: int) -> None:
|
|
host, port = parse_host_port(host_port)
|
|
assert host == expected_host
|
|
assert port == expected_port
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"host_port",
|
|
[
|
|
"127.0.0.1",
|
|
"::1",
|
|
"example.com",
|
|
"localhost",
|
|
"127.0.0.1:",
|
|
":8080",
|
|
],
|
|
)
|
|
def test_parse_host_port_invalid(host_port: str) -> None:
|
|
with pytest.raises(ValueError):
|
|
parse_host_port(host_port)
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_resolve4() -> None:
|
|
# Run these tests forcing IPv4 resolution
|
|
prefer_ipv6 = False
|
|
assert await resolve("127.0.0.1", prefer_ipv6=prefer_ipv6) == IPAddress.create("127.0.0.1")
|
|
assert await resolve("10.11.12.13", prefer_ipv6=prefer_ipv6) == IPAddress.create("10.11.12.13")
|
|
assert await resolve("localhost", prefer_ipv6=prefer_ipv6) == IPAddress.create("127.0.0.1")
|
|
example = await resolve("example.net", prefer_ipv6=prefer_ipv6)
|
|
assert example.is_v4
|
|
assert not example.is_private
|
|
|
|
|
|
@pytest.mark.anyio
|
|
@pytest.mark.skipif(
|
|
condition=("GITHUB_ACTIONS" in os.environ) and (sys.platform in {"darwin", "win32"}),
|
|
reason="macOS and Windows runners in GitHub Actions do not seem to support IPv6",
|
|
)
|
|
async def test_resolve6() -> None:
|
|
# Run these tests forcing IPv6 resolution
|
|
prefer_ipv6 = True
|
|
assert await resolve("::1", prefer_ipv6=prefer_ipv6) == IPAddress.create("::1")
|
|
assert await resolve("2000:1000::1234:abcd", prefer_ipv6=prefer_ipv6) == IPAddress.create("2000:1000::1234:abcd")
|
|
# ip6-localhost is not always available, and localhost is IPv4 only
|
|
# on some systems. Just test neither here.
|
|
# assert await resolve("ip6-localhost", prefer_ipv6=prefer_ipv6) == IPAddress.create("::1")
|
|
# assert await resolve("localhost", prefer_ipv6=prefer_ipv6) == IPAddress.create("::1")
|
|
example = await resolve("example.net", prefer_ipv6=prefer_ipv6)
|
|
assert example.is_v6
|
|
assert not example.is_private
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"address_string, expected_inner",
|
|
[
|
|
("::1", IPv6Address),
|
|
("2000:1000::1234:abcd", IPv6Address),
|
|
("127.0.0.1", IPv4Address),
|
|
("10.11.12.13", IPv4Address),
|
|
("93.184.216.34", IPv4Address),
|
|
],
|
|
)
|
|
def test_ip_address(address_string: str, expected_inner: type[IPv4Address | IPv6Address]) -> None:
|
|
inner = expected_inner(address_string)
|
|
ip = IPAddress.create(address_string)
|
|
# Helpers
|
|
assert ip.is_v4 == (expected_inner == IPv4Address)
|
|
assert ip.is_v6 == (expected_inner == IPv6Address)
|
|
# Overwritten dataclass methods
|
|
assert int(ip) == int(inner)
|
|
assert str(ip) == str(inner)
|
|
assert repr(ip) == repr(inner)
|
|
# Forwarded IPv4Address, IPV6Address properties
|
|
assert ip.packed == inner.packed
|
|
assert ip.is_private == inner.is_private
|
|
# Still use dataclass comparison
|
|
assert ip != inner
|
|
|
|
|
|
@pytest.mark.parametrize("address_string", ["10.11.12.13.14", "10,11.12.13", "0:::", "localhost", "invalid"])
|
|
def test_invalid_ip_addresses(address_string: str) -> None:
|
|
with pytest.raises(ValueError):
|
|
IPAddress.create(address_string)
|