1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
|
from __future__ import annotations
import socket
from typing import TYPE_CHECKING
from unittest.mock import Mock
from pytest import MonkeyPatch
from typing_extensions import override
import pyartnet.base.network as network_module
from pyartnet.base.network import MulticastNetworkTarget, UnicastNetworkTarget
if TYPE_CHECKING:
from types import TracebackType
class MockedSocket:
def __init__(self) -> None:
self.mp = MonkeyPatch()
def mock(self):
m_socket_obj = Mock(['sendto', 'setblocking', 'setsockopt', 'bind', 'close', 'family'], name='socket_obj')
m_socket_obj.sendto = m_sendto = Mock(name='socket_obj.sendto')
m_socket_obj.family = socket.AF_INET
constant_names = [
name for name in dir(socket)
if name.startswith(('AF_', 'SOCK_', 'SOL_', 'IPPROTO_', 'IP_', 'SO_',)) or
name in ('herror', 'gaierror')
]
m = Mock(['socket', 'gethostname', 'inet_pton', *constant_names], name='Mock socket package')
m.gethostname = socket.gethostname
m.socket = Mock([], return_value=m_socket_obj, name='Mock socket obj')
# Copy constants
for name in constant_names:
setattr(m, name, getattr(socket, name))
self.mp.setattr(network_module, 'socket', m)
return m_sendto
def undo(self) -> None:
self.mp.undo()
def __enter__(self):
return self.mock()
def __exit__(self, exc_type: type[BaseException] | None,
exc_val: BaseException | None, exc_tb: TracebackType | None) -> None:
self.undo()
class UnicastNetworkTestingTarget(UnicastNetworkTarget):
@override
async def resolve_hostname(self) -> None:
self._ip_v6 = False
return None
class MulticastTestingNetworkTarget(MulticastNetworkTarget):
@override
async def resolve_hostname(self) -> None:
self._ip_v6 = False
return None
|
