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
|
"""Ubiquiti AirOS test fixtures."""
from _collections_abc import AsyncGenerator, Generator
import asyncio
from unittest.mock import AsyncMock, MagicMock, patch
import aiohttp
import pytest
from airos.airos6 import AirOS6
from airos.airos8 import AirOS8
from airos.discovery import AirOSDiscoveryProtocol
# pylint: disable=redefined-outer-name, unnecessary-default-type-args
@pytest.fixture
def base_url() -> str:
"""Return a testing url."""
return "http://device.local"
@pytest.fixture
async def airos6_device(base_url: str) -> AsyncGenerator[AirOS6, None]:
"""AirOS6 device fixture."""
session = aiohttp.ClientSession(cookie_jar=aiohttp.CookieJar())
instance = AirOS6(base_url, "username", "password", session, use_ssl=False)
yield instance
await session.close()
@pytest.fixture
async def airos8_device(base_url: str) -> AsyncGenerator[AirOS8, None]:
"""AirOS8 device fixture."""
session = aiohttp.ClientSession(cookie_jar=aiohttp.CookieJar())
instance = AirOS8(base_url, "username", "password", session, use_ssl=False)
yield instance
await session.close()
@pytest.fixture
def mock_datagram_endpoint() -> Generator[
tuple[asyncio.DatagramTransport, AirOSDiscoveryProtocol], None, None
]:
"""Fixture to mock the creation of the UDP datagram endpoint."""
# Define the mock objects FIRST, so they are in scope
mock_transport = MagicMock(spec=asyncio.DatagramTransport)
mock_protocol_instance = MagicMock(spec=AirOSDiscoveryProtocol)
# Now, define the AsyncMock using the pre-defined variables
mock_create_datagram_endpoint = AsyncMock(
return_value=(mock_transport, mock_protocol_instance)
)
with (
patch("asyncio.get_running_loop") as mock_get_loop,
patch(
"airos.discovery.AirOSDiscoveryProtocol",
new=MagicMock(return_value=mock_protocol_instance),
),
):
mock_loop = mock_get_loop.return_value
mock_loop.create_datagram_endpoint = mock_create_datagram_endpoint
yield mock_transport, mock_protocol_instance
|
