mirror of
https://github.com/sabnzbd/sabnzbd.git
synced 2025-12-24 00:00:12 -05:00
84 lines
3.6 KiB
Python
84 lines
3.6 KiB
Python
#!/usr/bin/python3 -OO
|
|
# Copyright 2007-2025 by The SABnzbd-Team (sabnzbd.org)
|
|
#
|
|
# This program is free software; you can redistribute it and/or
|
|
# modify it under the terms of the GNU General Public License
|
|
# as published by the Free Software Foundation; either version 2
|
|
# of the License, or (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program; if not, write to the Free Software
|
|
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
|
|
"""
|
|
tests.test_get_addrinfo - Testing SABnzbd concurrent IP address testing implementation
|
|
"""
|
|
import os
|
|
import sys
|
|
import socket
|
|
|
|
import pytest
|
|
from flaky import flaky
|
|
|
|
from sabnzbd.get_addrinfo import get_fastest_addrinfo, IPV6_MAPPING
|
|
from sabnzbd.misc import is_ipv4_addr, is_ipv6_addr
|
|
|
|
|
|
@flaky
|
|
class TestConcurrentAddressTesting:
|
|
"""Tests of get_fastest_addrinfo() against various websites/servers
|
|
get_fastest_addrinfo() tests all IP addresses concurrently and returns the fastest one
|
|
after CONNECTION_CHECK interval, or None (if not resolvable, or not reachable)
|
|
"""
|
|
|
|
def test_google_http(self):
|
|
addrinfo = get_fastest_addrinfo("www.google.com", port=80)
|
|
assert is_ipv4_addr(addrinfo.ipaddress) or is_ipv6_addr(addrinfo.ipaddress)
|
|
assert "google" in addrinfo.canonname
|
|
|
|
def test_google_https(self):
|
|
addrinfo = get_fastest_addrinfo("www.google.com", port=443)
|
|
assert is_ipv4_addr(addrinfo.ipaddress) or is_ipv6_addr(addrinfo.ipaddress)
|
|
assert "google" in addrinfo.canonname
|
|
|
|
def test_google_http_want_ipv4(self):
|
|
addrinfo = get_fastest_addrinfo("www.google.com", port=80, family=socket.AF_INET)
|
|
assert is_ipv4_addr(addrinfo.ipaddress) and not is_ipv6_addr(addrinfo.ipaddress)
|
|
assert "google" in addrinfo.canonname
|
|
|
|
def test_google_http_want_ipv6(self):
|
|
# TODO: timeout needed for IPv4-only CI environment?
|
|
if addrinfo := get_fastest_addrinfo("www.google.com", port=80, timeout=2, family=socket.AF_INET6):
|
|
assert not is_ipv4_addr(addrinfo.ipaddress) and is_ipv6_addr(addrinfo.ipaddress)
|
|
assert "google" in addrinfo.canonname
|
|
|
|
def test_not_resolvable(self):
|
|
assert get_fastest_addrinfo("not.resolvable.invalid", port=80) is None
|
|
|
|
def test_ipv6_only(self):
|
|
if addrinfo := get_fastest_addrinfo("ipv6.google.com", port=443, timeout=2):
|
|
assert is_ipv6_addr(addrinfo.ipaddress)
|
|
assert "google" in addrinfo.canonname
|
|
|
|
def test_google_unreachable_port(self):
|
|
assert get_fastest_addrinfo("www.google.com", port=33333, timeout=1) is None
|
|
|
|
@pytest.mark.xfail(reason="CI sometimes blocks this")
|
|
def test_nntp(self):
|
|
if ip := get_fastest_addrinfo("news.newshosting.com", port=119).ipaddress:
|
|
assert is_ipv4_addr(ip) or is_ipv6_addr(ip)
|
|
|
|
@pytest.mark.skipif(sys.platform.startswith("darwin"), reason="Resolves strangely on macOS CI")
|
|
@pytest.mark.parametrize("hostname", IPV6_MAPPING.keys())
|
|
def test_ipv6_mapping(self, hostname):
|
|
# This test will let us remove hostnames from the mapping
|
|
# once the providers add IPv6 to their main hostname
|
|
with pytest.raises(socket.gaierror):
|
|
# Print results for us to see the new information
|
|
print(socket.getaddrinfo(hostname, 119, socket.AF_INET6, socket.SOCK_STREAM))
|