mirror of
https://github.com/jdejaegh/irm-kmi-api.git
synced 2025-06-26 20:05:40 +02:00
23 lines
646 B
Python
23 lines
646 B
Python
"""Fixtures for the IRM KMI integration tests."""
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
from unittest.mock import MagicMock
|
|
|
|
from irm_kmi_api.api import IrmKmiApiClientHa
|
|
from tests.const import IRM_KMI_TO_HA_CONDITION_MAP
|
|
|
|
|
|
def load_fixture(fixture):
|
|
with open(f"tests/fixtures/{fixture}", 'rb') as f:
|
|
return f.read()
|
|
|
|
|
|
def get_api_data(fixture: str) -> dict:
|
|
return json.loads(load_fixture(fixture))
|
|
|
|
|
|
def get_api_with_data(fixture: str) -> IrmKmiApiClientHa:
|
|
api = IrmKmiApiClientHa(session=MagicMock(), user_agent='', cdt_map=IRM_KMI_TO_HA_CONDITION_MAP)
|
|
api._api_data = get_api_data(fixture)
|
|
return api
|