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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
|
"""Tests for the client."""
from __future__ import annotations
from typing import TYPE_CHECKING, Any
from aiohttp import ClientError, ClientSession
from aioresponses import CallbackResult, aioresponses
import pytest
from switchbot_api import SwitchBotAPI, SwitchBotConnectionError
from tests import load_fixture
from tests.const import MOCK_URL
if TYPE_CHECKING:
from syrupy import SnapshotAssertion
async def test_providing_session(
responses: aioresponses,
) -> None:
"""Test putting in own session."""
responses.get(
f"{MOCK_URL}/devices",
status=200,
body=load_fixture("device_list.json"),
)
async with ClientSession() as session:
client = SwitchBotAPI("abc", "def", session=session)
await client.list_devices()
assert client.session is not None
assert not client.session.closed
await client.close()
assert not client.session.closed
async def test_using_default_session(
responses: aioresponses,
) -> None:
"""Test creating own session."""
responses.get(
f"{MOCK_URL}/devices",
status=200,
body=load_fixture("device_list.json"),
)
client = SwitchBotAPI("abc", "def")
await client.list_devices()
assert client.session is not None
assert not client.session.closed
await client.close()
assert client.session.closed
async def test_client_error(
client: SwitchBotAPI,
responses: aioresponses,
) -> None:
"""Test client error."""
async def response_handler(_: str, **_kwargs: Any) -> CallbackResult:
"""Response handler for this test."""
raise ClientError
responses.get(
f"{MOCK_URL}/devices",
callback=response_handler,
)
with pytest.raises(SwitchBotConnectionError):
await client.list_devices()
async def test_device_list(
client: SwitchBotAPI,
responses: aioresponses,
snapshot: SnapshotAssertion,
) -> None:
"""Test listing devices."""
responses.get(
f"{MOCK_URL}/devices",
status=200,
body=load_fixture("device_list.json"),
)
assert await client.list_devices() == snapshot
@pytest.mark.parametrize(
"device_fixture",
[
"curtain",
"curtain3",
"hub_2",
],
)
async def test_device_status(
client: SwitchBotAPI,
responses: aioresponses,
snapshot: SnapshotAssertion,
device_fixture: str,
) -> None:
"""Test fetching device status."""
responses.get(
f"{MOCK_URL}/devices/abc/status",
status=200,
body=load_fixture(f"{device_fixture}.json"),
)
assert await client.get_status("abc") == snapshot
|
