125 lines
4.0 KiB
Python
125 lines
4.0 KiB
Python
"""Test samsungtv diagnostics."""
|
|
from unittest.mock import Mock
|
|
|
|
from aiohttp import ClientSession
|
|
import pytest
|
|
from samsungtvws.exceptions import HttpApiError
|
|
|
|
from homeassistant.components.diagnostics import REDACTED
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
from . import setup_samsungtv_entry
|
|
from .const import (
|
|
MOCK_ENTRYDATA_ENCRYPTED_WS,
|
|
SAMPLE_DEVICE_INFO_UE48JU6400,
|
|
SAMPLE_DEVICE_INFO_WIFI,
|
|
)
|
|
from .test_media_player import MOCK_ENTRY_WS_WITH_MAC
|
|
|
|
from tests.components.diagnostics import get_diagnostics_for_config_entry
|
|
|
|
|
|
@pytest.mark.usefixtures("remotews")
|
|
async def test_entry_diagnostics(
|
|
hass: HomeAssistant, hass_client: ClientSession
|
|
) -> None:
|
|
"""Test config entry diagnostics."""
|
|
config_entry = await setup_samsungtv_entry(hass, MOCK_ENTRY_WS_WITH_MAC)
|
|
|
|
assert await get_diagnostics_for_config_entry(hass, hass_client, config_entry) == {
|
|
"entry": {
|
|
"data": {
|
|
"host": "fake_host",
|
|
"ip_address": "test",
|
|
"mac": "aa:bb:cc:dd:ee:ff",
|
|
"method": "websocket",
|
|
"model": "82GXARRS",
|
|
"name": "fake",
|
|
"port": 8002,
|
|
"token": REDACTED,
|
|
},
|
|
"disabled_by": None,
|
|
"domain": "samsungtv",
|
|
"entry_id": "123456",
|
|
"options": {},
|
|
"pref_disable_new_entities": False,
|
|
"pref_disable_polling": False,
|
|
"source": "user",
|
|
"title": "Mock Title",
|
|
"unique_id": "any",
|
|
"version": 2,
|
|
},
|
|
"device_info": SAMPLE_DEVICE_INFO_WIFI,
|
|
}
|
|
|
|
|
|
@pytest.mark.usefixtures("remoteencws")
|
|
async def test_entry_diagnostics_encrypted(
|
|
hass: HomeAssistant, rest_api: Mock, hass_client: ClientSession
|
|
) -> None:
|
|
"""Test config entry diagnostics."""
|
|
rest_api.rest_device_info.return_value = SAMPLE_DEVICE_INFO_UE48JU6400
|
|
config_entry = await setup_samsungtv_entry(hass, MOCK_ENTRYDATA_ENCRYPTED_WS)
|
|
|
|
assert await get_diagnostics_for_config_entry(hass, hass_client, config_entry) == {
|
|
"entry": {
|
|
"data": {
|
|
"host": "fake_host",
|
|
"ip_address": "test",
|
|
"mac": "aa:bb:cc:dd:ee:ff",
|
|
"method": "encrypted",
|
|
"model": "UE48JU6400",
|
|
"name": "fake",
|
|
"port": 8000,
|
|
"token": REDACTED,
|
|
"session_id": REDACTED,
|
|
},
|
|
"disabled_by": None,
|
|
"domain": "samsungtv",
|
|
"entry_id": "123456",
|
|
"options": {},
|
|
"pref_disable_new_entities": False,
|
|
"pref_disable_polling": False,
|
|
"source": "user",
|
|
"title": "Mock Title",
|
|
"unique_id": "any",
|
|
"version": 2,
|
|
},
|
|
"device_info": SAMPLE_DEVICE_INFO_UE48JU6400,
|
|
}
|
|
|
|
|
|
@pytest.mark.usefixtures("remoteencws")
|
|
async def test_entry_diagnostics_encrypte_offline(
|
|
hass: HomeAssistant, rest_api: Mock, hass_client: ClientSession
|
|
) -> None:
|
|
"""Test config entry diagnostics."""
|
|
rest_api.rest_device_info.side_effect = HttpApiError
|
|
config_entry = await setup_samsungtv_entry(hass, MOCK_ENTRYDATA_ENCRYPTED_WS)
|
|
|
|
assert await get_diagnostics_for_config_entry(hass, hass_client, config_entry) == {
|
|
"entry": {
|
|
"data": {
|
|
"host": "fake_host",
|
|
"ip_address": "test",
|
|
"mac": "aa:bb:cc:dd:ee:ff",
|
|
"method": "encrypted",
|
|
"name": "fake",
|
|
"port": 8000,
|
|
"token": REDACTED,
|
|
"session_id": REDACTED,
|
|
},
|
|
"disabled_by": None,
|
|
"domain": "samsungtv",
|
|
"entry_id": "123456",
|
|
"options": {},
|
|
"pref_disable_new_entities": False,
|
|
"pref_disable_polling": False,
|
|
"source": "user",
|
|
"title": "Mock Title",
|
|
"unique_id": "any",
|
|
"version": 2,
|
|
},
|
|
"device_info": None,
|
|
}
|