154 lines
3.9 KiB
Python
154 lines
3.9 KiB
Python
"""Configuration for 17Track tests."""
|
|
|
|
from collections.abc import Generator
|
|
from typing import Optional
|
|
from unittest.mock import AsyncMock, patch
|
|
|
|
from py17track.package import Package
|
|
import pytest
|
|
|
|
from homeassistant.components.seventeentrack.const import (
|
|
DEFAULT_SHOW_ARCHIVED,
|
|
DEFAULT_SHOW_DELIVERED,
|
|
)
|
|
from homeassistant.components.seventeentrack.sensor import (
|
|
CONF_SHOW_ARCHIVED,
|
|
CONF_SHOW_DELIVERED,
|
|
)
|
|
from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
DEFAULT_SUMMARY = {
|
|
"Not Found": 0,
|
|
"In Transit": 0,
|
|
"Expired": 0,
|
|
"Ready to be Picked Up": 0,
|
|
"Undelivered": 0,
|
|
"Delivered": 0,
|
|
"Returned": 0,
|
|
}
|
|
|
|
ACCOUNT_ID = "1234"
|
|
|
|
NEW_SUMMARY_DATA = {
|
|
"Not Found": 1,
|
|
"In Transit": 1,
|
|
"Expired": 1,
|
|
"Ready to be Picked Up": 1,
|
|
"Undelivered": 1,
|
|
"Delivered": 1,
|
|
"Returned": 1,
|
|
}
|
|
|
|
VALID_CONFIG = {
|
|
CONF_USERNAME: "test",
|
|
CONF_PASSWORD: "test",
|
|
}
|
|
|
|
INVALID_CONFIG = {"notusername": "seventeentrack", "notpassword": "test"}
|
|
|
|
VALID_OPTIONS = {
|
|
CONF_SHOW_ARCHIVED: True,
|
|
CONF_SHOW_DELIVERED: True,
|
|
}
|
|
|
|
NO_DELIVERED_OPTIONS = {
|
|
CONF_SHOW_ARCHIVED: False,
|
|
CONF_SHOW_DELIVERED: False,
|
|
}
|
|
|
|
VALID_PLATFORM_CONFIG_FULL = {
|
|
"sensor": {
|
|
"platform": "seventeentrack",
|
|
CONF_USERNAME: "test",
|
|
CONF_PASSWORD: "test",
|
|
CONF_SHOW_ARCHIVED: True,
|
|
CONF_SHOW_DELIVERED: True,
|
|
}
|
|
}
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_setup_entry() -> Generator[AsyncMock, None, None]:
|
|
"""Override async_setup_entry."""
|
|
with patch(
|
|
"homeassistant.components.seventeentrack.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(
|
|
domain="seventeentrack",
|
|
data=VALID_CONFIG,
|
|
options=VALID_OPTIONS,
|
|
unique_id=ACCOUNT_ID,
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_config_entry_with_default_options() -> MockConfigEntry:
|
|
"""Return the default mocked config entry."""
|
|
return MockConfigEntry(
|
|
domain="seventeentrack",
|
|
data=VALID_CONFIG,
|
|
options={
|
|
CONF_SHOW_ARCHIVED: DEFAULT_SHOW_ARCHIVED,
|
|
CONF_SHOW_DELIVERED: DEFAULT_SHOW_DELIVERED,
|
|
},
|
|
unique_id=ACCOUNT_ID,
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_seventeentrack():
|
|
"""Build a fixture for the 17Track API."""
|
|
mock_seventeentrack_api = AsyncMock()
|
|
with (
|
|
patch(
|
|
"homeassistant.components.seventeentrack.SeventeenTrackClient",
|
|
return_value=mock_seventeentrack_api,
|
|
),
|
|
patch(
|
|
"homeassistant.components.seventeentrack.config_flow.SeventeenTrackClient",
|
|
return_value=mock_seventeentrack_api,
|
|
) as mock_seventeentrack_api,
|
|
):
|
|
mock_seventeentrack_api.return_value.profile.account_id = ACCOUNT_ID
|
|
mock_seventeentrack_api.return_value.profile.login.return_value = True
|
|
mock_seventeentrack_api.return_value.profile.packages.return_value = []
|
|
mock_seventeentrack_api.return_value.profile.summary.return_value = (
|
|
DEFAULT_SUMMARY
|
|
)
|
|
yield mock_seventeentrack_api
|
|
|
|
|
|
def get_package(
|
|
tracking_number: str = "456",
|
|
destination_country: int = 206,
|
|
friendly_name: Optional[str] = "friendly name 1",
|
|
info_text: str = "info text 1",
|
|
location: str = "location 1",
|
|
timestamp: str = "2020-08-10 10:32",
|
|
origin_country: int = 206,
|
|
package_type: int = 2,
|
|
status: int = 0,
|
|
tz: str = "UTC",
|
|
):
|
|
"""Build a Package of the 17Track API."""
|
|
return Package(
|
|
tracking_number=tracking_number,
|
|
destination_country=destination_country,
|
|
friendly_name=friendly_name,
|
|
info_text=info_text,
|
|
location=location,
|
|
timestamp=timestamp,
|
|
origin_country=origin_country,
|
|
package_type=package_type,
|
|
status=status,
|
|
tz=tz,
|
|
)
|