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
|
"""Fixtures for aioelectricitymaps tests."""
from collections.abc import AsyncGenerator, Generator
import re
import aiohttp
from aioresponses import aioresponses
import pytest
from aioelectricitymaps import ElectricityMaps
from . import load_fixture
@pytest.fixture(name="responses")
def aioresponses_fixture() -> Generator[aioresponses, None, None]:
"""Return aioresponses fixture."""
with aioresponses() as mocked_responses:
yield mocked_responses
@pytest.fixture(name="mock_response")
def _mock_response(responses: aioresponses) -> None:
"""Mock an API response."""
url_pattern = re.compile(
r"^https://api\.electricitymaps\.com/v3/home-assistant\?.*$",
)
responses.get(
url_pattern,
status=200,
headers={"Content-Type": "application/json"},
body=load_fixture("response.json"),
)
@pytest.fixture(name="mock_broken_response")
def _mock_broken_response(responses: aioresponses) -> None:
"""Mock a bad API response."""
responses.get(
"https://api.electricitymaps.com/v3/home-assistant",
status=200,
headers={"Content-Type": "application/json"},
body='{"status": "ok"',
)
@pytest.fixture(name="electricitymaps_client")
async def client() -> AsyncGenerator[ElectricityMaps, None]:
"""Return a ElectricityMaps client."""
async with (
aiohttp.ClientSession(trust_env=True) as session,
ElectricityMaps(
token="abc123",
session=session,
) as electricitymaps_client,
):
yield electricitymaps_client
|
