evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Cleanup common mqtt tests (#120446)

Browse source
This commit is contained in:
Jan Bouwhuis 2024-06-25 19:15:11 +02:00 • committed by GitHub
parent 185e79fa1b
commit cbcf29720d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
28 changed files with 266 additions and 904 deletions
Download patch file Download diff file Expand all files Collapse all files

48
tests/components/mqtt/test_button.py
View file

@ -252,17 +252,11 @@ async def test_update_with_json_attrs_bad_json(
async def test_discovery_update_attr(
hass: HomeAssistant,
mqtt_mock_entry: MqttMockHAClientGenerator,
caplog: pytest.LogCaptureFixture,
hass: HomeAssistant, mqtt_mock_entry: MqttMockHAClientGenerator
) -> None:
"""Test update of discovered MQTTAttributes."""
await help_test_discovery_update_attr(
hass,
mqtt_mock_entry,
caplog,
button.DOMAIN,
DEFAULT_CONFIG,
hass, mqtt_mock_entry, button.DOMAIN, DEFAULT_CONFIG
)
@ -295,21 +289,15 @@ async def test_unique_id(
async def test_discovery_removal_button(
hass: HomeAssistant,
mqtt_mock_entry: MqttMockHAClientGenerator,
caplog: pytest.LogCaptureFixture,
hass: HomeAssistant, mqtt_mock_entry: MqttMockHAClientGenerator
) -> None:
"""Test removal of discovered button."""
data = '{ "name": "test", "command_topic": "test_topic" }'
await help_test_discovery_removal(
hass, mqtt_mock_entry, caplog, button.DOMAIN, data
)
await help_test_discovery_removal(hass, mqtt_mock_entry, button.DOMAIN, data)
async def test_discovery_update_button(
hass: HomeAssistant,
mqtt_mock_entry: MqttMockHAClientGenerator,
caplog: pytest.LogCaptureFixture,
hass: HomeAssistant, mqtt_mock_entry: MqttMockHAClientGenerator
) -> None:
"""Test update of discovered button."""
config1 = copy.deepcopy(DEFAULT_CONFIG[mqtt.DOMAIN][button.DOMAIN])
@ -318,19 +306,12 @@ async def test_discovery_update_button(
config2["name"] = "Milk"
await help_test_discovery_update(
hass,
mqtt_mock_entry,
caplog,
button.DOMAIN,
config1,
config2,
hass, mqtt_mock_entry, button.DOMAIN, config1, config2
)
async def test_discovery_update_unchanged_button(
hass: HomeAssistant,
mqtt_mock_entry: MqttMockHAClientGenerator,
caplog: pytest.LogCaptureFixture,
hass: HomeAssistant, mqtt_mock_entry: MqttMockHAClientGenerator
) -> None:
"""Test update of discovered button."""
data1 = (
@ -342,27 +323,18 @@ async def test_discovery_update_unchanged_button(
"homeassistant.components.mqtt.button.MqttButton.discovery_update"
) as discovery_update:
await help_test_discovery_update_unchanged(
hass,
mqtt_mock_entry,
caplog,
button.DOMAIN,
data1,
discovery_update,
hass, mqtt_mock_entry, button.DOMAIN, data1, discovery_update
)
@pytest.mark.no_fail_on_log_exception
async def test_discovery_broken(
hass: HomeAssistant,
mqtt_mock_entry: MqttMockHAClientGenerator,
caplog: pytest.LogCaptureFixture,
hass: HomeAssistant, mqtt_mock_entry: MqttMockHAClientGenerator
) -> None:
"""Test handling of bad discovery message."""
data1 = '{ "name": "Beer" }'
data2 = '{ "name": "Milk", "command_topic": "test_topic" }'
await help_test_discovery_broken(
hass, mqtt_mock_entry, caplog, button.DOMAIN, data1, data2
)
await help_test_discovery_broken(hass, mqtt_mock_entry, button.DOMAIN, data1, data2)
async def test_entity_device_info_with_connection(