mirror of
https://github.com/home-assistant/core.git
synced 2025-04-23 16:57:53 +00:00
Add Ituran integration (#129067)
This commit is contained in:
parent
e91cb99512
commit
d3fab7d87a
@ -753,6 +753,8 @@ build.json @home-assistant/supervisor
|
||||
/tests/components/ista_ecotrend/ @tr4nt0r
|
||||
/homeassistant/components/isy994/ @bdraco @shbatm
|
||||
/tests/components/isy994/ @bdraco @shbatm
|
||||
/homeassistant/components/ituran/ @shmuelzon
|
||||
/tests/components/ituran/ @shmuelzon
|
||||
/homeassistant/components/izone/ @Swamp-Ig
|
||||
/tests/components/izone/ @Swamp-Ig
|
||||
/homeassistant/components/jellyfin/ @j-stienstra @ctalkington
|
||||
|
28
homeassistant/components/ituran/__init__.py
Normal file
28
homeassistant/components/ituran/__init__.py
Normal file
@ -0,0 +1,28 @@
|
||||
"""The Ituran integration."""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
from homeassistant.const import Platform
|
||||
from homeassistant.core import HomeAssistant
|
||||
|
||||
from .coordinator import IturanConfigEntry, IturanDataUpdateCoordinator
|
||||
|
||||
PLATFORMS: list[Platform] = [
|
||||
Platform.DEVICE_TRACKER,
|
||||
]
|
||||
|
||||
|
||||
async def async_setup_entry(hass: HomeAssistant, entry: IturanConfigEntry) -> bool:
|
||||
"""Set up Ituran from a config entry."""
|
||||
|
||||
coordinator = IturanDataUpdateCoordinator(hass, entry=entry)
|
||||
await coordinator.async_config_entry_first_refresh()
|
||||
entry.runtime_data = coordinator
|
||||
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
|
||||
|
||||
return True
|
||||
|
||||
|
||||
async def async_unload_entry(hass: HomeAssistant, entry: IturanConfigEntry) -> bool:
|
||||
"""Unload a config entry."""
|
||||
return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
|
109
homeassistant/components/ituran/config_flow.py
Normal file
109
homeassistant/components/ituran/config_flow.py
Normal file
@ -0,0 +1,109 @@
|
||||
"""Config flow for Ituran integration."""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import logging
|
||||
from typing import Any
|
||||
|
||||
from pyituran import Ituran
|
||||
from pyituran.exceptions import IturanApiError, IturanAuthError
|
||||
import voluptuous as vol
|
||||
|
||||
from homeassistant.config_entries import ConfigFlow, ConfigFlowResult
|
||||
|
||||
from .const import (
|
||||
CONF_ID_OR_PASSPORT,
|
||||
CONF_MOBILE_ID,
|
||||
CONF_OTP,
|
||||
CONF_PHONE_NUMBER,
|
||||
DOMAIN,
|
||||
)
|
||||
|
||||
_LOGGER = logging.getLogger(__name__)
|
||||
|
||||
STEP_USER_DATA_SCHEMA = vol.Schema(
|
||||
{
|
||||
vol.Required(CONF_ID_OR_PASSPORT): str,
|
||||
vol.Required(CONF_PHONE_NUMBER): str,
|
||||
}
|
||||
)
|
||||
|
||||
STEP_OTP_DATA_SCHEMA = vol.Schema(
|
||||
{
|
||||
vol.Required(CONF_OTP): str,
|
||||
}
|
||||
)
|
||||
|
||||
|
||||
class IturanConfigFlow(ConfigFlow, domain=DOMAIN):
|
||||
"""Handle a config flow for Ituran."""
|
||||
|
||||
_user_info: dict[str, Any]
|
||||
|
||||
async def async_step_user(
|
||||
self, user_input: dict[str, Any] | None = None
|
||||
) -> ConfigFlowResult:
|
||||
"""Handle the inial step."""
|
||||
errors: dict[str, str] = {}
|
||||
if user_input is not None:
|
||||
await self.async_set_unique_id(user_input[CONF_ID_OR_PASSPORT])
|
||||
self._abort_if_unique_id_configured()
|
||||
|
||||
ituran = Ituran(
|
||||
user_input[CONF_ID_OR_PASSPORT],
|
||||
user_input[CONF_PHONE_NUMBER],
|
||||
)
|
||||
user_input[CONF_MOBILE_ID] = ituran.mobile_id
|
||||
try:
|
||||
authenticated = await ituran.is_authenticated()
|
||||
if not authenticated:
|
||||
await ituran.request_otp()
|
||||
except IturanApiError:
|
||||
errors["base"] = "cannot_connect"
|
||||
except IturanAuthError:
|
||||
errors["base"] = "invalid_auth"
|
||||
except Exception:
|
||||
_LOGGER.exception("Unexpected exception")
|
||||
errors["base"] = "unknown"
|
||||
else:
|
||||
if authenticated:
|
||||
return self.async_create_entry(
|
||||
title=f"Ituran {user_input[CONF_ID_OR_PASSPORT]}",
|
||||
data=user_input,
|
||||
)
|
||||
self._user_info = user_input
|
||||
return await self.async_step_otp()
|
||||
|
||||
return self.async_show_form(
|
||||
step_id="user", data_schema=STEP_USER_DATA_SCHEMA, errors=errors
|
||||
)
|
||||
|
||||
async def async_step_otp(
|
||||
self, user_input: dict[str, Any] | None = None
|
||||
) -> ConfigFlowResult:
|
||||
"""Handle the inial step."""
|
||||
errors: dict[str, str] = {}
|
||||
if user_input is not None:
|
||||
ituran = Ituran(
|
||||
self._user_info[CONF_ID_OR_PASSPORT],
|
||||
self._user_info[CONF_PHONE_NUMBER],
|
||||
self._user_info[CONF_MOBILE_ID],
|
||||
)
|
||||
try:
|
||||
await ituran.authenticate(user_input[CONF_OTP])
|
||||
except IturanApiError:
|
||||
errors["base"] = "cannot_connect"
|
||||
except IturanAuthError:
|
||||
errors["base"] = "invalid_otp"
|
||||
except Exception:
|
||||
_LOGGER.exception("Unexpected exception")
|
||||
errors["base"] = "unknown"
|
||||
else:
|
||||
return self.async_create_entry(
|
||||
title=f"Ituran {self._user_info[CONF_ID_OR_PASSPORT]}",
|
||||
data=self._user_info,
|
||||
)
|
||||
|
||||
return self.async_show_form(
|
||||
step_id="otp", data_schema=STEP_OTP_DATA_SCHEMA, errors=errors
|
||||
)
|
13
homeassistant/components/ituran/const.py
Normal file
13
homeassistant/components/ituran/const.py
Normal file
@ -0,0 +1,13 @@
|
||||
"""Constants for the Ituran integration."""
|
||||
|
||||
from datetime import timedelta
|
||||
from typing import Final
|
||||
|
||||
DOMAIN = "ituran"
|
||||
|
||||
CONF_ID_OR_PASSPORT: Final = "id_or_passport"
|
||||
CONF_PHONE_NUMBER: Final = "phone_number"
|
||||
CONF_MOBILE_ID: Final = "mobile_id"
|
||||
CONF_OTP: Final = "otp"
|
||||
|
||||
UPDATE_INTERVAL = timedelta(seconds=300)
|
76
homeassistant/components/ituran/coordinator.py
Normal file
76
homeassistant/components/ituran/coordinator.py
Normal file
@ -0,0 +1,76 @@
|
||||
"""Coordinator for Ituran."""
|
||||
|
||||
import logging
|
||||
|
||||
from pyituran import Ituran, Vehicle
|
||||
from pyituran.exceptions import IturanApiError, IturanAuthError
|
||||
|
||||
from homeassistant.config_entries import ConfigEntry
|
||||
from homeassistant.core import HomeAssistant
|
||||
from homeassistant.exceptions import ConfigEntryError
|
||||
from homeassistant.helpers import device_registry as dr
|
||||
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
|
||||
|
||||
from .const import (
|
||||
CONF_ID_OR_PASSPORT,
|
||||
CONF_MOBILE_ID,
|
||||
CONF_PHONE_NUMBER,
|
||||
DOMAIN,
|
||||
UPDATE_INTERVAL,
|
||||
)
|
||||
|
||||
_LOGGER = logging.getLogger(__name__)
|
||||
|
||||
type IturanConfigEntry = ConfigEntry[IturanDataUpdateCoordinator]
|
||||
|
||||
|
||||
class IturanDataUpdateCoordinator(DataUpdateCoordinator[dict[str, Vehicle]]):
|
||||
"""Class to manage fetching Ituran data."""
|
||||
|
||||
config_entry: IturanConfigEntry
|
||||
|
||||
def __init__(self, hass: HomeAssistant, entry: IturanConfigEntry) -> None:
|
||||
"""Initialize account-wide Ituran data updater."""
|
||||
super().__init__(
|
||||
hass,
|
||||
_LOGGER,
|
||||
name=f"{DOMAIN}-{entry.data[CONF_ID_OR_PASSPORT]}",
|
||||
update_interval=UPDATE_INTERVAL,
|
||||
config_entry=entry,
|
||||
)
|
||||
self.ituran = Ituran(
|
||||
entry.data[CONF_ID_OR_PASSPORT],
|
||||
entry.data[CONF_PHONE_NUMBER],
|
||||
entry.data[CONF_MOBILE_ID],
|
||||
)
|
||||
|
||||
async def _async_update_data(self) -> dict[str, Vehicle]:
|
||||
"""Fetch data from Ituran."""
|
||||
|
||||
try:
|
||||
vehicles = await self.ituran.get_vehicles()
|
||||
except IturanApiError as e:
|
||||
raise UpdateFailed(
|
||||
translation_domain=DOMAIN, translation_key="api_error"
|
||||
) from e
|
||||
except IturanAuthError as e:
|
||||
raise ConfigEntryError(
|
||||
translation_domain=DOMAIN, translation_key="auth_error"
|
||||
) from e
|
||||
|
||||
updated_data = {vehicle.license_plate: vehicle for vehicle in vehicles}
|
||||
self._cleanup_removed_vehicles(updated_data)
|
||||
|
||||
return updated_data
|
||||
|
||||
def _cleanup_removed_vehicles(self, data: dict[str, Vehicle]) -> None:
|
||||
account_vehicles = {(DOMAIN, license_plate) for license_plate in data}
|
||||
device_registry = dr.async_get(self.hass)
|
||||
device_entries = dr.async_entries_for_config_entry(
|
||||
device_registry, config_entry_id=self.config_entry.entry_id
|
||||
)
|
||||
for device in device_entries:
|
||||
if not device.identifiers.intersection(account_vehicles):
|
||||
device_registry.async_update_device(
|
||||
device.id, remove_config_entry_id=self.config_entry.entry_id
|
||||
)
|
49
homeassistant/components/ituran/device_tracker.py
Normal file
49
homeassistant/components/ituran/device_tracker.py
Normal file
@ -0,0 +1,49 @@
|
||||
"""Device tracker for Ituran vehicles."""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
from homeassistant.components.device_tracker import TrackerEntity
|
||||
from homeassistant.core import HomeAssistant
|
||||
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
||||
|
||||
from . import IturanConfigEntry
|
||||
from .coordinator import IturanDataUpdateCoordinator
|
||||
from .entity import IturanBaseEntity
|
||||
|
||||
|
||||
async def async_setup_entry(
|
||||
hass: HomeAssistant,
|
||||
config_entry: IturanConfigEntry,
|
||||
async_add_entities: AddEntitiesCallback,
|
||||
) -> None:
|
||||
"""Set up the Ituran tracker from config entry."""
|
||||
coordinator = config_entry.runtime_data
|
||||
async_add_entities(
|
||||
IturanDeviceTracker(coordinator, license_plate)
|
||||
for license_plate in coordinator.data
|
||||
)
|
||||
|
||||
|
||||
class IturanDeviceTracker(IturanBaseEntity, TrackerEntity):
|
||||
"""Ituran device tracker."""
|
||||
|
||||
_attr_translation_key = "car"
|
||||
_attr_name = None
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
coordinator: IturanDataUpdateCoordinator,
|
||||
license_plate: str,
|
||||
) -> None:
|
||||
"""Initialize the device tracker."""
|
||||
super().__init__(coordinator, license_plate, "device_tracker")
|
||||
|
||||
@property
|
||||
def latitude(self) -> float | None:
|
||||
"""Return latitude value of the device."""
|
||||
return self.vehicle.gps_coordinates[0]
|
||||
|
||||
@property
|
||||
def longitude(self) -> float | None:
|
||||
"""Return longitude value of the device."""
|
||||
return self.vehicle.gps_coordinates[1]
|
47
homeassistant/components/ituran/entity.py
Normal file
47
homeassistant/components/ituran/entity.py
Normal file
@ -0,0 +1,47 @@
|
||||
"""Base for all turan entities."""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
from pyituran import Vehicle
|
||||
|
||||
from homeassistant.helpers.device_registry import DeviceInfo
|
||||
from homeassistant.helpers.update_coordinator import CoordinatorEntity
|
||||
|
||||
from .const import DOMAIN
|
||||
from .coordinator import IturanDataUpdateCoordinator
|
||||
|
||||
|
||||
class IturanBaseEntity(CoordinatorEntity[IturanDataUpdateCoordinator]):
|
||||
"""Common base for Ituran entities."""
|
||||
|
||||
_attr_has_entity_name = True
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
coordinator: IturanDataUpdateCoordinator,
|
||||
license_plate: str,
|
||||
unique_key: str,
|
||||
) -> None:
|
||||
"""Initialize the entity."""
|
||||
super().__init__(coordinator)
|
||||
|
||||
self._license_plate = license_plate
|
||||
self._attr_unique_id = f"{license_plate}-{unique_key}"
|
||||
|
||||
self._attr_device_info = DeviceInfo(
|
||||
identifiers={(DOMAIN, self.vehicle.license_plate)},
|
||||
manufacturer=self.vehicle.make,
|
||||
model=self.vehicle.model,
|
||||
name=self.vehicle.model,
|
||||
serial_number=self.vehicle.license_plate,
|
||||
)
|
||||
|
||||
@property
|
||||
def available(self) -> bool:
|
||||
"""Return True if vehicle is still included in the account."""
|
||||
return super().available and self._license_plate in self.coordinator.data
|
||||
|
||||
@property
|
||||
def vehicle(self) -> Vehicle:
|
||||
"""Return the vehicle information associated with this entity."""
|
||||
return self.coordinator.data[self._license_plate]
|
9
homeassistant/components/ituran/icons.json
Normal file
9
homeassistant/components/ituran/icons.json
Normal file
@ -0,0 +1,9 @@
|
||||
{
|
||||
"entity": {
|
||||
"device_tracker": {
|
||||
"car": {
|
||||
"default": "mdi:car"
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
10
homeassistant/components/ituran/manifest.json
Normal file
10
homeassistant/components/ituran/manifest.json
Normal file
@ -0,0 +1,10 @@
|
||||
{
|
||||
"domain": "ituran",
|
||||
"name": "Ituran",
|
||||
"codeowners": ["@shmuelzon"],
|
||||
"config_flow": true,
|
||||
"documentation": "https://www.home-assistant.io/integrations/ituran",
|
||||
"integration_type": "hub",
|
||||
"iot_class": "cloud_polling",
|
||||
"requirements": ["pyituran==0.1.3"]
|
||||
}
|
92
homeassistant/components/ituran/quality_scale.yaml
Normal file
92
homeassistant/components/ituran/quality_scale.yaml
Normal file
@ -0,0 +1,92 @@
|
||||
rules:
|
||||
# Bronze
|
||||
config-flow: done
|
||||
test-before-configure: done
|
||||
unique-config-entry: done
|
||||
config-flow-test-coverage: done
|
||||
runtime-data: done
|
||||
test-before-setup: done
|
||||
appropriate-polling: done
|
||||
entity-unique-id: done
|
||||
has-entity-name: done
|
||||
entity-event-setup:
|
||||
status: exempt
|
||||
comment: |
|
||||
Entities of this integration does not explicitly subscribe to events.
|
||||
dependency-transparency: done
|
||||
action-setup:
|
||||
status: exempt
|
||||
comment: |
|
||||
This integration does not provide additional actions.
|
||||
common-modules: done
|
||||
docs-high-level-description: done
|
||||
docs-installation-instructions: done
|
||||
docs-removal-instructions: done
|
||||
docs-actions:
|
||||
status: exempt
|
||||
comment: |
|
||||
This integration does not provide additional actions.
|
||||
brands: done
|
||||
# Silver
|
||||
config-entry-unloading: done
|
||||
log-when-unavailable: done
|
||||
entity-unavailable: done
|
||||
action-exceptions:
|
||||
status: exempt
|
||||
comment: |
|
||||
This integration does not provide additional actions.
|
||||
reauthentication-flow: todo
|
||||
parallel-updates:
|
||||
status: exempt
|
||||
comment: |
|
||||
Read only platforms and coordinator.
|
||||
test-coverage: done
|
||||
integration-owner: done
|
||||
docs-installation-parameters: done
|
||||
docs-configuration-parameters:
|
||||
status: exempt
|
||||
comment: |
|
||||
No options flow.
|
||||
# Gold
|
||||
entity-translations: done
|
||||
entity-device-class:
|
||||
status: exempt
|
||||
comment: |
|
||||
Only device_tracker platform.
|
||||
devices: done
|
||||
entity-category: todo
|
||||
entity-disabled-by-default:
|
||||
status: exempt
|
||||
comment: |
|
||||
No noisy entities
|
||||
discovery:
|
||||
status: exempt
|
||||
comment: |
|
||||
This integration cannot be discovered, it is a connecting to a service
|
||||
provider, which uses the users credentials to get the data.
|
||||
stale-devices: todo
|
||||
diagnostics: todo
|
||||
exception-translations: done
|
||||
icon-translations: done
|
||||
reconfiguration-flow: todo
|
||||
dynamic-devices: done
|
||||
discovery-update-info:
|
||||
status: exempt
|
||||
comment: |
|
||||
This integration cannot be discovered, it is a connecting to a service
|
||||
provider, which uses the users credentials to get the data.
|
||||
repair-issues:
|
||||
status: exempt
|
||||
comment: |
|
||||
No repairs/issues.
|
||||
docs-use-cases: todo
|
||||
docs-supported-devices: done
|
||||
docs-supported-functions: done
|
||||
docs-data-update: done
|
||||
docs-known-limitations: done
|
||||
docs-troubleshooting: todo
|
||||
docs-examples: todo
|
||||
# Platinum
|
||||
async-dependency: done
|
||||
inject-websession: todo
|
||||
strict-typing: todo
|
41
homeassistant/components/ituran/strings.json
Normal file
41
homeassistant/components/ituran/strings.json
Normal file
@ -0,0 +1,41 @@
|
||||
{
|
||||
"config": {
|
||||
"step": {
|
||||
"user": {
|
||||
"data": {
|
||||
"id_or_passport": "ID or passport number",
|
||||
"phone_number": "Mobile phone number"
|
||||
},
|
||||
"data_description": {
|
||||
"id_or_passport": "The goverment ID or passport number provided when registering with Ituran.",
|
||||
"phone_number": "The mobile phone number provided when registering with Ituran. A one-time password will be sent to this mobile number."
|
||||
}
|
||||
},
|
||||
"otp": {
|
||||
"data": {
|
||||
"otp": "OTP"
|
||||
},
|
||||
"data_description": {
|
||||
"otp": "A one-time-password sent as a text message to the mobile phone number provided before."
|
||||
}
|
||||
}
|
||||
},
|
||||
"error": {
|
||||
"cannot_connect": "[%key:common::config_flow::error::cannot_connect%]",
|
||||
"invalid_auth": "[%key:common::config_flow::error::invalid_auth%]",
|
||||
"invalid_otp": "OTP invalid",
|
||||
"unknown": "[%key:common::config_flow::error::unknown%]"
|
||||
},
|
||||
"abort": {
|
||||
"already_configured": "[%key:common::config_flow::abort::already_configured_account%]"
|
||||
}
|
||||
},
|
||||
"exceptions": {
|
||||
"api_error": {
|
||||
"message": "An error occured while communicating with the Ituran service."
|
||||
},
|
||||
"auth_error": {
|
||||
"message": "Failed authenticating with the Ituran service, please remove and re-add integration."
|
||||
}
|
||||
}
|
||||
}
|
@ -296,6 +296,7 @@ FLOWS = {
|
||||
"iss",
|
||||
"ista_ecotrend",
|
||||
"isy994",
|
||||
"ituran",
|
||||
"izone",
|
||||
"jellyfin",
|
||||
"jewish_calendar",
|
||||
|
@ -2983,6 +2983,12 @@
|
||||
"config_flow": true,
|
||||
"iot_class": "local_push"
|
||||
},
|
||||
"ituran": {
|
||||
"name": "Ituran",
|
||||
"integration_type": "hub",
|
||||
"config_flow": true,
|
||||
"iot_class": "cloud_polling"
|
||||
},
|
||||
"izone": {
|
||||
"name": "iZone",
|
||||
"integration_type": "hub",
|
||||
|
@ -1996,6 +1996,9 @@ pyisy==3.1.14
|
||||
# homeassistant.components.itach
|
||||
pyitachip2ir==0.0.7
|
||||
|
||||
# homeassistant.components.ituran
|
||||
pyituran==0.1.3
|
||||
|
||||
# homeassistant.components.jvc_projector
|
||||
pyjvcprojector==1.1.2
|
||||
|
||||
|
@ -1610,6 +1610,9 @@ pyiss==1.0.1
|
||||
# homeassistant.components.isy994
|
||||
pyisy==3.1.14
|
||||
|
||||
# homeassistant.components.ituran
|
||||
pyituran==0.1.3
|
||||
|
||||
# homeassistant.components.jvc_projector
|
||||
pyjvcprojector==1.1.2
|
||||
|
||||
|
13
tests/components/ituran/__init__.py
Normal file
13
tests/components/ituran/__init__.py
Normal file
@ -0,0 +1,13 @@
|
||||
"""Tests for the Ituran integration."""
|
||||
|
||||
from homeassistant.core import HomeAssistant
|
||||
|
||||
from tests.common import MockConfigEntry
|
||||
|
||||
|
||||
async def setup_integration(hass: HomeAssistant, config_entry: MockConfigEntry) -> None:
|
||||
"""Fixture for setting up the component."""
|
||||
config_entry.add_to_hass(hass)
|
||||
|
||||
await hass.config_entries.async_setup(config_entry.entry_id)
|
||||
await hass.async_block_till_done()
|
83
tests/components/ituran/conftest.py
Normal file
83
tests/components/ituran/conftest.py
Normal file
@ -0,0 +1,83 @@
|
||||
"""Mocks for the Ituran integration."""
|
||||
|
||||
from collections.abc import Generator
|
||||
from datetime import datetime
|
||||
from unittest.mock import AsyncMock, PropertyMock, patch
|
||||
|
||||
import pytest
|
||||
|
||||
from homeassistant.components.ituran.const import (
|
||||
CONF_ID_OR_PASSPORT,
|
||||
CONF_MOBILE_ID,
|
||||
CONF_PHONE_NUMBER,
|
||||
DOMAIN,
|
||||
)
|
||||
|
||||
from .const import MOCK_CONFIG_DATA
|
||||
|
||||
from tests.common import MockConfigEntry
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def mock_setup_entry() -> Generator[AsyncMock]:
|
||||
"""Override async_setup_entry."""
|
||||
with patch(
|
||||
"homeassistant.components.ituran.async_setup_entry",
|
||||
return_value=True,
|
||||
) as mock_setup_entry:
|
||||
yield mock_setup_entry
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def mock_config_entry() -> MockConfigEntry:
|
||||
"""Return the default mocked config entry."""
|
||||
return MockConfigEntry(
|
||||
title=f"Ituran {MOCK_CONFIG_DATA[CONF_ID_OR_PASSPORT]}",
|
||||
domain=DOMAIN,
|
||||
data={
|
||||
CONF_ID_OR_PASSPORT: MOCK_CONFIG_DATA[CONF_ID_OR_PASSPORT],
|
||||
CONF_PHONE_NUMBER: MOCK_CONFIG_DATA[CONF_PHONE_NUMBER],
|
||||
CONF_MOBILE_ID: MOCK_CONFIG_DATA[CONF_MOBILE_ID],
|
||||
},
|
||||
unique_id=MOCK_CONFIG_DATA[CONF_ID_OR_PASSPORT],
|
||||
)
|
||||
|
||||
|
||||
class MockVehicle:
|
||||
"""Mock vehicle."""
|
||||
|
||||
def __init__(self) -> None:
|
||||
"""Initialize mock vehicle."""
|
||||
self.license_plate = "12345678"
|
||||
self.make = "mock make"
|
||||
self.model = "mock model"
|
||||
self.mileage = 1000
|
||||
self.speed = 20
|
||||
self.gps_coordinates = (25.0, -71.0)
|
||||
self.address = "Bermuda Triangle"
|
||||
self.heading = 150
|
||||
self.last_update = datetime(2024, 1, 1, 0, 0, 0)
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def mock_ituran() -> Generator[AsyncMock]:
|
||||
"""Return a mocked PalazzettiClient."""
|
||||
with (
|
||||
patch(
|
||||
"homeassistant.components.ituran.coordinator.Ituran",
|
||||
autospec=True,
|
||||
) as ituran,
|
||||
patch(
|
||||
"homeassistant.components.ituran.config_flow.Ituran",
|
||||
new=ituran,
|
||||
),
|
||||
):
|
||||
mock_ituran = ituran.return_value
|
||||
mock_ituran.is_authenticated.return_value = False
|
||||
mock_ituran.authenticate.return_value = True
|
||||
mock_ituran.get_vehicles.return_value = [MockVehicle()]
|
||||
type(mock_ituran).mobile_id = PropertyMock(
|
||||
return_value=MOCK_CONFIG_DATA[CONF_MOBILE_ID]
|
||||
)
|
||||
|
||||
yield mock_ituran
|
24
tests/components/ituran/const.py
Normal file
24
tests/components/ituran/const.py
Normal file
@ -0,0 +1,24 @@
|
||||
"""Constants for tests of the Ituran component."""
|
||||
|
||||
from typing import Any
|
||||
|
||||
from homeassistant.components.ituran.const import (
|
||||
CONF_ID_OR_PASSPORT,
|
||||
CONF_MOBILE_ID,
|
||||
CONF_PHONE_NUMBER,
|
||||
DOMAIN,
|
||||
)
|
||||
|
||||
MOCK_CONFIG_DATA: dict[str, str] = {
|
||||
CONF_ID_OR_PASSPORT: "12345678",
|
||||
CONF_PHONE_NUMBER: "0501234567",
|
||||
CONF_MOBILE_ID: "0123456789abcdef",
|
||||
}
|
||||
|
||||
MOCK_CONFIG_ENTRY: dict[str, Any] = {
|
||||
"domain": DOMAIN,
|
||||
"entry_id": "1",
|
||||
"source": "user",
|
||||
"title": MOCK_CONFIG_DATA[CONF_ID_OR_PASSPORT],
|
||||
"data": MOCK_CONFIG_DATA,
|
||||
}
|
51
tests/components/ituran/snapshots/test_device_tracker.ambr
Normal file
51
tests/components/ituran/snapshots/test_device_tracker.ambr
Normal file
@ -0,0 +1,51 @@
|
||||
# serializer version: 1
|
||||
# name: test_device_tracker[device_tracker.mock_model-entry]
|
||||
EntityRegistryEntrySnapshot({
|
||||
'aliases': set({
|
||||
}),
|
||||
'area_id': None,
|
||||
'capabilities': None,
|
||||
'config_entry_id': <ANY>,
|
||||
'device_class': None,
|
||||
'device_id': <ANY>,
|
||||
'disabled_by': None,
|
||||
'domain': 'device_tracker',
|
||||
'entity_category': <EntityCategory.DIAGNOSTIC: 'diagnostic'>,
|
||||
'entity_id': 'device_tracker.mock_model',
|
||||
'has_entity_name': True,
|
||||
'hidden_by': None,
|
||||
'icon': None,
|
||||
'id': <ANY>,
|
||||
'labels': set({
|
||||
}),
|
||||
'name': None,
|
||||
'options': dict({
|
||||
}),
|
||||
'original_device_class': None,
|
||||
'original_icon': None,
|
||||
'original_name': None,
|
||||
'platform': 'ituran',
|
||||
'previous_unique_id': None,
|
||||
'supported_features': 0,
|
||||
'translation_key': 'car',
|
||||
'unique_id': '12345678-device_tracker',
|
||||
'unit_of_measurement': None,
|
||||
})
|
||||
# ---
|
||||
# name: test_device_tracker[device_tracker.mock_model-state]
|
||||
StateSnapshot({
|
||||
'attributes': ReadOnlyDict({
|
||||
'friendly_name': 'mock model',
|
||||
'gps_accuracy': 0,
|
||||
'latitude': 25.0,
|
||||
'longitude': -71.0,
|
||||
'source_type': <SourceType.GPS: 'gps'>,
|
||||
}),
|
||||
'context': <ANY>,
|
||||
'entity_id': 'device_tracker.mock_model',
|
||||
'last_changed': <ANY>,
|
||||
'last_reported': <ANY>,
|
||||
'last_updated': <ANY>,
|
||||
'state': 'not_home',
|
||||
})
|
||||
# ---
|
35
tests/components/ituran/snapshots/test_init.ambr
Normal file
35
tests/components/ituran/snapshots/test_init.ambr
Normal file
@ -0,0 +1,35 @@
|
||||
# serializer version: 1
|
||||
# name: test_device
|
||||
list([
|
||||
DeviceRegistryEntrySnapshot({
|
||||
'area_id': None,
|
||||
'config_entries': <ANY>,
|
||||
'configuration_url': None,
|
||||
'connections': set({
|
||||
}),
|
||||
'disabled_by': None,
|
||||
'entry_type': None,
|
||||
'hw_version': None,
|
||||
'id': <ANY>,
|
||||
'identifiers': set({
|
||||
tuple(
|
||||
'ituran',
|
||||
'12345678',
|
||||
),
|
||||
}),
|
||||
'is_new': False,
|
||||
'labels': set({
|
||||
}),
|
||||
'manufacturer': 'mock make',
|
||||
'model': 'mock model',
|
||||
'model_id': None,
|
||||
'name': 'mock model',
|
||||
'name_by_user': None,
|
||||
'primary_config_entry': <ANY>,
|
||||
'serial_number': '12345678',
|
||||
'suggested_area': None,
|
||||
'sw_version': None,
|
||||
'via_device_id': None,
|
||||
}),
|
||||
])
|
||||
# ---
|
211
tests/components/ituran/test_config_flow.py
Normal file
211
tests/components/ituran/test_config_flow.py
Normal file
@ -0,0 +1,211 @@
|
||||
"""Test the Ituran config flow."""
|
||||
|
||||
from unittest.mock import AsyncMock
|
||||
|
||||
from pyituran.exceptions import IturanApiError, IturanAuthError
|
||||
import pytest
|
||||
|
||||
from homeassistant.components.ituran.const import (
|
||||
CONF_ID_OR_PASSPORT,
|
||||
CONF_MOBILE_ID,
|
||||
CONF_OTP,
|
||||
CONF_PHONE_NUMBER,
|
||||
DOMAIN,
|
||||
)
|
||||
from homeassistant.config_entries import SOURCE_USER, ConfigFlowResult
|
||||
from homeassistant.core import HomeAssistant
|
||||
from homeassistant.data_entry_flow import FlowResultType
|
||||
|
||||
from .const import MOCK_CONFIG_DATA
|
||||
|
||||
|
||||
async def __do_successful_user_step(
|
||||
hass: HomeAssistant, result: ConfigFlowResult, mock_ituran: AsyncMock
|
||||
):
|
||||
result = await hass.config_entries.flow.async_configure(
|
||||
result["flow_id"],
|
||||
user_input={
|
||||
CONF_ID_OR_PASSPORT: MOCK_CONFIG_DATA[CONF_ID_OR_PASSPORT],
|
||||
CONF_PHONE_NUMBER: MOCK_CONFIG_DATA[CONF_PHONE_NUMBER],
|
||||
},
|
||||
)
|
||||
|
||||
assert result["type"] is FlowResultType.FORM
|
||||
assert result["step_id"] == "otp"
|
||||
assert result["errors"] == {}
|
||||
|
||||
return result
|
||||
|
||||
|
||||
async def __do_successful_otp_step(
|
||||
hass: HomeAssistant,
|
||||
result: ConfigFlowResult,
|
||||
mock_ituran: AsyncMock,
|
||||
):
|
||||
result = await hass.config_entries.flow.async_configure(
|
||||
result["flow_id"],
|
||||
user_input={
|
||||
CONF_OTP: "123456",
|
||||
},
|
||||
)
|
||||
|
||||
assert result["type"] is FlowResultType.CREATE_ENTRY
|
||||
assert result["title"] == f"Ituran {MOCK_CONFIG_DATA[CONF_ID_OR_PASSPORT]}"
|
||||
assert result["data"][CONF_ID_OR_PASSPORT] == MOCK_CONFIG_DATA[CONF_ID_OR_PASSPORT]
|
||||
assert result["data"][CONF_PHONE_NUMBER] == MOCK_CONFIG_DATA[CONF_PHONE_NUMBER]
|
||||
assert result["data"][CONF_MOBILE_ID] is not None
|
||||
assert result["result"].unique_id == MOCK_CONFIG_DATA[CONF_ID_OR_PASSPORT]
|
||||
assert len(mock_ituran.is_authenticated.mock_calls) > 0
|
||||
assert len(mock_ituran.authenticate.mock_calls) > 0
|
||||
|
||||
return result
|
||||
|
||||
|
||||
async def test_full_user_flow(
|
||||
hass: HomeAssistant, mock_ituran: AsyncMock, mock_setup_entry: AsyncMock
|
||||
) -> None:
|
||||
"""Test the full user configuration flow."""
|
||||
result = await hass.config_entries.flow.async_init(
|
||||
DOMAIN, context={"source": SOURCE_USER}
|
||||
)
|
||||
|
||||
assert result["type"] is FlowResultType.FORM
|
||||
assert result["step_id"] == "user"
|
||||
|
||||
result = await __do_successful_user_step(hass, result, mock_ituran)
|
||||
await __do_successful_otp_step(hass, result, mock_ituran)
|
||||
|
||||
|
||||
async def test_invalid_auth(
|
||||
hass: HomeAssistant, mock_ituran: AsyncMock, mock_setup_entry: AsyncMock
|
||||
) -> None:
|
||||
"""Test invalid credentials configuration flow."""
|
||||
result = await hass.config_entries.flow.async_init(
|
||||
DOMAIN, context={"source": SOURCE_USER}
|
||||
)
|
||||
|
||||
assert result["type"] is FlowResultType.FORM
|
||||
assert result["step_id"] == "user"
|
||||
|
||||
mock_ituran.request_otp.side_effect = IturanAuthError
|
||||
result = await hass.config_entries.flow.async_configure(
|
||||
result["flow_id"],
|
||||
user_input={
|
||||
CONF_ID_OR_PASSPORT: MOCK_CONFIG_DATA[CONF_ID_OR_PASSPORT],
|
||||
CONF_PHONE_NUMBER: MOCK_CONFIG_DATA[CONF_PHONE_NUMBER],
|
||||
},
|
||||
)
|
||||
|
||||
assert result["type"] is FlowResultType.FORM
|
||||
assert result["step_id"] == "user"
|
||||
assert result["errors"] == {"base": "invalid_auth"}
|
||||
|
||||
mock_ituran.request_otp.side_effect = None
|
||||
result = await __do_successful_user_step(hass, result, mock_ituran)
|
||||
await __do_successful_otp_step(hass, result, mock_ituran)
|
||||
|
||||
|
||||
async def test_invalid_otp(
|
||||
hass: HomeAssistant, mock_ituran: AsyncMock, mock_setup_entry: AsyncMock
|
||||
) -> None:
|
||||
"""Test invalid OTP configuration flow."""
|
||||
result = await hass.config_entries.flow.async_init(
|
||||
DOMAIN, context={"source": SOURCE_USER}
|
||||
)
|
||||
|
||||
assert result["type"] is FlowResultType.FORM
|
||||
assert result["step_id"] == "user"
|
||||
|
||||
result = await __do_successful_user_step(hass, result, mock_ituran)
|
||||
|
||||
mock_ituran.authenticate.side_effect = IturanAuthError
|
||||
result = await hass.config_entries.flow.async_configure(
|
||||
result["flow_id"],
|
||||
user_input={
|
||||
CONF_OTP: "123456",
|
||||
},
|
||||
)
|
||||
|
||||
assert result["type"] is FlowResultType.FORM
|
||||
assert result["errors"] == {"base": "invalid_otp"}
|
||||
|
||||
mock_ituran.authenticate.side_effect = None
|
||||
await __do_successful_otp_step(hass, result, mock_ituran)
|
||||
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
("exception", "expected_error"),
|
||||
[(IturanApiError, "cannot_connect"), (Exception, "unknown")],
|
||||
)
|
||||
async def test_errors(
|
||||
hass: HomeAssistant,
|
||||
mock_ituran: AsyncMock,
|
||||
mock_setup_entry: AsyncMock,
|
||||
exception: Exception,
|
||||
expected_error: str,
|
||||
) -> None:
|
||||
"""Test connection errors during configuration flow."""
|
||||
result = await hass.config_entries.flow.async_init(
|
||||
DOMAIN, context={"source": SOURCE_USER}
|
||||
)
|
||||
|
||||
assert result["type"] is FlowResultType.FORM
|
||||
assert result["step_id"] == "user"
|
||||
|
||||
mock_ituran.request_otp.side_effect = exception
|
||||
result = await hass.config_entries.flow.async_configure(
|
||||
result["flow_id"],
|
||||
user_input={
|
||||
CONF_ID_OR_PASSPORT: MOCK_CONFIG_DATA[CONF_ID_OR_PASSPORT],
|
||||
CONF_PHONE_NUMBER: MOCK_CONFIG_DATA[CONF_PHONE_NUMBER],
|
||||
},
|
||||
)
|
||||
|
||||
assert result["type"] is FlowResultType.FORM
|
||||
assert result["step_id"] == "user"
|
||||
assert result["errors"] == {"base": expected_error}
|
||||
|
||||
mock_ituran.request_otp.side_effect = None
|
||||
result = await __do_successful_user_step(hass, result, mock_ituran)
|
||||
|
||||
mock_ituran.authenticate.side_effect = exception
|
||||
result = await hass.config_entries.flow.async_configure(
|
||||
result["flow_id"],
|
||||
user_input={
|
||||
CONF_OTP: "123456",
|
||||
},
|
||||
)
|
||||
|
||||
assert result["type"] is FlowResultType.FORM
|
||||
assert result["errors"] == {"base": expected_error}
|
||||
|
||||
mock_ituran.authenticate.side_effect = None
|
||||
await __do_successful_otp_step(hass, result, mock_ituran)
|
||||
|
||||
|
||||
async def test_already_authenticated(
|
||||
hass: HomeAssistant, mock_ituran: AsyncMock, mock_setup_entry: AsyncMock
|
||||
) -> None:
|
||||
"""Test user already authenticated configuration flow."""
|
||||
result = await hass.config_entries.flow.async_init(
|
||||
DOMAIN, context={"source": SOURCE_USER}
|
||||
)
|
||||
|
||||
assert result["type"] is FlowResultType.FORM
|
||||
assert result["step_id"] == "user"
|
||||
|
||||
mock_ituran.is_authenticated.return_value = True
|
||||
result = await hass.config_entries.flow.async_configure(
|
||||
result["flow_id"],
|
||||
user_input={
|
||||
CONF_ID_OR_PASSPORT: MOCK_CONFIG_DATA[CONF_ID_OR_PASSPORT],
|
||||
CONF_PHONE_NUMBER: MOCK_CONFIG_DATA[CONF_PHONE_NUMBER],
|
||||
},
|
||||
)
|
||||
|
||||
assert result["type"] is FlowResultType.CREATE_ENTRY
|
||||
assert result["title"] == f"Ituran {MOCK_CONFIG_DATA[CONF_ID_OR_PASSPORT]}"
|
||||
assert result["data"][CONF_ID_OR_PASSPORT] == MOCK_CONFIG_DATA[CONF_ID_OR_PASSPORT]
|
||||
assert result["data"][CONF_PHONE_NUMBER] == MOCK_CONFIG_DATA[CONF_PHONE_NUMBER]
|
||||
assert result["data"][CONF_MOBILE_ID] == MOCK_CONFIG_DATA[CONF_MOBILE_ID]
|
||||
assert result["result"].unique_id == MOCK_CONFIG_DATA[CONF_ID_OR_PASSPORT]
|
61
tests/components/ituran/test_device_tracker.py
Normal file
61
tests/components/ituran/test_device_tracker.py
Normal file
@ -0,0 +1,61 @@
|
||||
"""Test the Ituran device_tracker."""
|
||||
|
||||
from unittest.mock import AsyncMock
|
||||
|
||||
from freezegun.api import FrozenDateTimeFactory
|
||||
from pyituran.exceptions import IturanApiError
|
||||
from syrupy.assertion import SnapshotAssertion
|
||||
|
||||
from homeassistant.components.ituran.const import UPDATE_INTERVAL
|
||||
from homeassistant.const import STATE_UNAVAILABLE
|
||||
from homeassistant.core import HomeAssistant
|
||||
from homeassistant.helpers import entity_registry as er
|
||||
|
||||
from . import setup_integration
|
||||
|
||||
from tests.common import MockConfigEntry, async_fire_time_changed, snapshot_platform
|
||||
|
||||
|
||||
async def test_device_tracker(
|
||||
hass: HomeAssistant,
|
||||
entity_registry: er.EntityRegistry,
|
||||
snapshot: SnapshotAssertion,
|
||||
mock_ituran: AsyncMock,
|
||||
mock_config_entry: MockConfigEntry,
|
||||
) -> None:
|
||||
"""Test state of device_tracker."""
|
||||
await setup_integration(hass, mock_config_entry)
|
||||
await snapshot_platform(hass, entity_registry, snapshot, mock_config_entry.entry_id)
|
||||
|
||||
|
||||
async def test_availability(
|
||||
hass: HomeAssistant,
|
||||
freezer: FrozenDateTimeFactory,
|
||||
mock_ituran: AsyncMock,
|
||||
mock_config_entry: MockConfigEntry,
|
||||
) -> None:
|
||||
"""Test device is marked as unavailable when we can't reach the Ituran service."""
|
||||
entity_id = "device_tracker.mock_model"
|
||||
await setup_integration(hass, mock_config_entry)
|
||||
|
||||
state = hass.states.get(entity_id)
|
||||
assert state
|
||||
assert state.state != STATE_UNAVAILABLE
|
||||
|
||||
mock_ituran.get_vehicles.side_effect = IturanApiError
|
||||
freezer.tick(UPDATE_INTERVAL)
|
||||
async_fire_time_changed(hass)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
state = hass.states.get(entity_id)
|
||||
assert state
|
||||
assert state.state == STATE_UNAVAILABLE
|
||||
|
||||
mock_ituran.get_vehicles.side_effect = None
|
||||
freezer.tick(UPDATE_INTERVAL)
|
||||
async_fire_time_changed(hass)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
state = hass.states.get(entity_id)
|
||||
assert state
|
||||
assert state.state != STATE_UNAVAILABLE
|
113
tests/components/ituran/test_init.py
Normal file
113
tests/components/ituran/test_init.py
Normal file
@ -0,0 +1,113 @@
|
||||
"""Tests for the Ituran integration."""
|
||||
|
||||
from unittest.mock import AsyncMock
|
||||
|
||||
from pyituran.exceptions import IturanApiError, IturanAuthError
|
||||
from syrupy.assertion import SnapshotAssertion
|
||||
|
||||
from homeassistant.config_entries import ConfigEntryState
|
||||
from homeassistant.core import HomeAssistant
|
||||
from homeassistant.helpers import device_registry as dr
|
||||
|
||||
from . import setup_integration
|
||||
|
||||
from tests.common import MockConfigEntry
|
||||
|
||||
|
||||
async def test_load_unload_config_entry(
|
||||
hass: HomeAssistant,
|
||||
mock_config_entry: MockConfigEntry,
|
||||
mock_ituran: AsyncMock,
|
||||
) -> None:
|
||||
"""Test the Ituran configuration entry loading/unloading."""
|
||||
await setup_integration(hass, mock_config_entry)
|
||||
|
||||
assert mock_config_entry.state is ConfigEntryState.LOADED
|
||||
|
||||
await hass.config_entries.async_unload(mock_config_entry.entry_id)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
assert mock_config_entry.state is ConfigEntryState.NOT_LOADED
|
||||
|
||||
|
||||
async def test_device(
|
||||
hass: HomeAssistant,
|
||||
mock_config_entry: MockConfigEntry,
|
||||
mock_ituran: AsyncMock,
|
||||
snapshot: SnapshotAssertion,
|
||||
device_registry: dr.DeviceRegistry,
|
||||
) -> None:
|
||||
"""Test the device information."""
|
||||
await setup_integration(hass, mock_config_entry)
|
||||
|
||||
device_entries = dr.async_entries_for_config_entry(
|
||||
device_registry, mock_config_entry.entry_id
|
||||
)
|
||||
assert device_entries == snapshot
|
||||
|
||||
|
||||
async def test_remove_stale_devices(
|
||||
hass: HomeAssistant,
|
||||
mock_config_entry: MockConfigEntry,
|
||||
mock_ituran: AsyncMock,
|
||||
device_registry: dr.DeviceRegistry,
|
||||
) -> None:
|
||||
"""Test that devices not returned by the service are removed."""
|
||||
await setup_integration(hass, mock_config_entry)
|
||||
device_entries = dr.async_entries_for_config_entry(
|
||||
device_registry, mock_config_entry.entry_id
|
||||
)
|
||||
|
||||
assert len(device_entries) == 1
|
||||
|
||||
mock_ituran.get_vehicles.return_value = []
|
||||
await mock_config_entry.runtime_data.async_refresh()
|
||||
await hass.async_block_till_done()
|
||||
device_entries = dr.async_entries_for_config_entry(
|
||||
device_registry, mock_config_entry.entry_id
|
||||
)
|
||||
|
||||
assert len(device_entries) == 0
|
||||
|
||||
|
||||
async def test_recover_from_errors(
|
||||
hass: HomeAssistant,
|
||||
mock_config_entry: MockConfigEntry,
|
||||
mock_ituran: AsyncMock,
|
||||
device_registry: dr.DeviceRegistry,
|
||||
) -> None:
|
||||
"""Verify we can recover from service Errors."""
|
||||
|
||||
await setup_integration(hass, mock_config_entry)
|
||||
device_entries = dr.async_entries_for_config_entry(
|
||||
device_registry, mock_config_entry.entry_id
|
||||
)
|
||||
|
||||
assert len(device_entries) == 1
|
||||
|
||||
mock_ituran.get_vehicles.side_effect = IturanApiError
|
||||
await mock_config_entry.runtime_data.async_refresh()
|
||||
await hass.async_block_till_done()
|
||||
device_entries = dr.async_entries_for_config_entry(
|
||||
device_registry, mock_config_entry.entry_id
|
||||
)
|
||||
|
||||
assert len(device_entries) == 1
|
||||
|
||||
mock_ituran.get_vehicles.side_effect = IturanAuthError
|
||||
await mock_config_entry.runtime_data.async_refresh()
|
||||
await hass.async_block_till_done()
|
||||
device_entries = dr.async_entries_for_config_entry(
|
||||
device_registry, mock_config_entry.entry_id
|
||||
)
|
||||
|
||||
assert len(device_entries) == 1
|
||||
|
||||
mock_ituran.get_vehicles.side_effect = None
|
||||
await mock_config_entry.runtime_data.async_refresh()
|
||||
await hass.async_block_till_done()
|
||||
device_entries = dr.async_entries_for_config_entry(
|
||||
device_registry, mock_config_entry.entry_id
|
||||
)
|
||||
|
||||
assert len(device_entries) == 1
|
Loading…
x
Reference in New Issue
Block a user