mirror of
https://github.com/NewFuture/DDNS.git
synced 2026-10-20 17:38:36 +00:00
90e04b70bd
* test(http): handle wrapped network timeouts Co-authored-by: Copilot App <223556219+Copilot@users.noreply.github.com> * test(http): tighten network error classification Co-authored-by: Copilot App <223556219+Copilot@users.noreply.github.com> --------- Co-authored-by: NewFuture <6290356+NewFuture@users.noreply.github.com> Co-authored-by: Copilot App <223556219+Copilot@users.noreply.github.com>
95 lines
2.8 KiB
Python
95 lines
2.8 KiB
Python
# coding=utf-8
|
|
"""
|
|
DDNS Tests Package
|
|
"""
|
|
|
|
import errno
|
|
import os
|
|
import socket
|
|
import ssl
|
|
import sys
|
|
import unittest
|
|
|
|
TEST_HTTP_TIMEOUT = 5
|
|
|
|
try:
|
|
_TIMEOUT_ERRORS = (socket.timeout, TimeoutError)
|
|
except NameError: # Python 2
|
|
_TIMEOUT_ERRORS = (socket.timeout,)
|
|
|
|
try:
|
|
_STRING_TYPES = (basestring,) # noqa: F821
|
|
except NameError: # Python 3
|
|
_STRING_TYPES = (str,)
|
|
|
|
_NETWORK_MESSAGE_ERRORS = (IOError, OSError, socket.error)
|
|
_SSL_ERRORS = (ssl.SSLError, getattr(ssl, "CertificateError", ssl.SSLError))
|
|
_NETWORK_ERRNOS = {
|
|
value
|
|
for value in (
|
|
getattr(errno, "ECONNABORTED", None),
|
|
getattr(errno, "ECONNREFUSED", None),
|
|
getattr(errno, "ECONNRESET", None),
|
|
getattr(errno, "EHOSTDOWN", None),
|
|
getattr(errno, "EHOSTUNREACH", None),
|
|
getattr(errno, "ENETDOWN", None),
|
|
getattr(errno, "ENETRESET", None),
|
|
getattr(errno, "ENETUNREACH", None),
|
|
getattr(errno, "EPIPE", None),
|
|
getattr(errno, "ETIMEDOUT", None),
|
|
10060, # WSAETIMEDOUT
|
|
)
|
|
if value is not None
|
|
}
|
|
|
|
|
|
def is_network_error(error, include_ssl=False):
|
|
"""Return whether an exception represents an unavailable test endpoint."""
|
|
if isinstance(error, _TIMEOUT_ERRORS):
|
|
return True
|
|
|
|
reason = getattr(error, "reason", None)
|
|
if reason is not None and reason is not error:
|
|
if isinstance(reason, _STRING_TYPES):
|
|
error = reason
|
|
elif is_network_error(reason, include_ssl):
|
|
return True
|
|
|
|
if isinstance(error, (socket.gaierror, socket.herror)):
|
|
return True
|
|
|
|
if getattr(error, "errno", None) in _NETWORK_ERRNOS:
|
|
return True
|
|
|
|
if include_ssl and isinstance(error, _SSL_ERRORS):
|
|
return True
|
|
|
|
if not isinstance(error, _NETWORK_MESSAGE_ERRORS + _STRING_TYPES):
|
|
return False
|
|
|
|
error_msg = str(error).lower()
|
|
network_keywords = ["timeout", "timed out", "connection", "resolution", "unreachable", "network"]
|
|
if include_ssl:
|
|
network_keywords.extend(["ssl", "certificate"])
|
|
return any(keyword in error_msg for keyword in network_keywords)
|
|
|
|
|
|
try:
|
|
from unittest import mock # type: ignore
|
|
from unittest.mock import patch, MagicMock, call
|
|
except ImportError: # Python 2
|
|
from mock import patch, MagicMock, call # type: ignore
|
|
import mock # type: ignore
|
|
|
|
__all__ = ["patch", "MagicMock", "unittest", "call", "mock", "TEST_HTTP_TIMEOUT", "is_network_error"]
|
|
|
|
# 添加当前目录到 Python 路径，这样就可以直接导入 test_base
|
|
current_dir = os.path.dirname(__file__)
|
|
if current_dir not in sys.path:
|
|
sys.path.insert(0, current_dir)
|
|
|
|
# 添加上级目录到 Python 路径，这样就可以导入 ddns 模块
|
|
parent_dir = os.path.dirname(current_dir)
|
|
if parent_dir not in sys.path:
|
|
sys.path.insert(0, parent_dir)
|