3333from homeassistant .helpers .service_info .hassio import HassioServiceInfo
3434
3535from .common import (
36- MOCK_ALARM_CONTROL_PANEL_LOCAL_CODE_SUBENTRY_DATA_SINGLE ,
37- MOCK_ALARM_CONTROL_PANEL_REMOTE_CODE_SUBENTRY_DATA_SINGLE ,
38- MOCK_ALARM_CONTROL_PANEL_REMOTE_CODE_TEXT_SUBENTRY_DATA_SINGLE ,
39- MOCK_BINARY_SENSOR_SUBENTRY_DATA_SINGLE ,
40- MOCK_BUTTON_SUBENTRY_DATA_SINGLE ,
41- MOCK_CLIMATE_HIGH_LOW_SUBENTRY_DATA_SINGLE ,
42- MOCK_CLIMATE_NO_TARGET_TEMP_SUBENTRY_DATA_SINGLE ,
43- MOCK_CLIMATE_SUBENTRY_DATA_SINGLE ,
44- MOCK_COVER_SUBENTRY_DATA_SINGLE ,
45- MOCK_FAN_SUBENTRY_DATA_SINGLE ,
36+ MOCK_ALARM_CONTROL_PANEL_LOCAL_CODE_SUBENTRY_DATA ,
37+ MOCK_ALARM_CONTROL_PANEL_REMOTE_CODE_SUBENTRY_DATA ,
38+ MOCK_ALARM_CONTROL_PANEL_REMOTE_CODE_TEXT_SUBENTRY_DATA ,
39+ MOCK_BINARY_SENSOR_SUBENTRY_DATA ,
40+ MOCK_BUTTON_SUBENTRY_DATA ,
41+ MOCK_CLIMATE_HIGH_LOW_SUBENTRY_DATA ,
42+ MOCK_CLIMATE_NO_TARGET_TEMP_SUBENTRY_DATA ,
43+ MOCK_CLIMATE_SUBENTRY_DATA ,
44+ MOCK_COVER_SUBENTRY_DATA ,
45+ MOCK_FAN_SUBENTRY_DATA ,
4646 MOCK_IMAGE_SUBENTRY_DATA_IMAGE_DATA ,
4747 MOCK_IMAGE_SUBENTRY_DATA_IMAGE_URL ,
48- MOCK_LIGHT_BASIC_KELVIN_SUBENTRY_DATA_SINGLE ,
49- MOCK_LOCK_SUBENTRY_DATA_SINGLE ,
48+ MOCK_LIGHT_BASIC_KELVIN_SUBENTRY_DATA ,
49+ MOCK_LOCK_SUBENTRY_DATA ,
50+ MOCK_NOTIFY_SUBENTRY_DATA ,
5051 MOCK_NOTIFY_SUBENTRY_DATA_MULTI ,
5152 MOCK_NOTIFY_SUBENTRY_DATA_NO_NAME ,
52- MOCK_NOTIFY_SUBENTRY_DATA_SINGLE ,
5353 MOCK_NUMBER_SUBENTRY_DATA_CUSTOM_UNIT ,
5454 MOCK_NUMBER_SUBENTRY_DATA_DEVICE_CLASS_UNIT ,
5555 MOCK_NUMBER_SUBENTRY_DATA_NO_UNIT ,
5656 MOCK_SELECT_SUBENTRY_DATA ,
57- MOCK_SENSOR_SUBENTRY_DATA_SINGLE ,
58- MOCK_SENSOR_SUBENTRY_DATA_SINGLE_LAST_RESET_TEMPLATE ,
59- MOCK_SENSOR_SUBENTRY_DATA_SINGLE_STATE_CLASS ,
60- MOCK_SWITCH_SUBENTRY_DATA_SINGLE_STATE_CLASS ,
57+ MOCK_SENSOR_SUBENTRY_DATA ,
58+ MOCK_SENSOR_SUBENTRY_DATA_LAST_RESET_TEMPLATE ,
59+ MOCK_SENSOR_SUBENTRY_DATA_STATE_CLASS ,
60+ MOCK_SWITCH_SUBENTRY_DATA ,
6161)
6262
6363from tests .common import MockConfigEntry , MockMqttReasonCode , get_schema_suggested_value
@@ -2675,7 +2675,7 @@ async def test_migrate_of_incompatible_config_entry(
26752675 ),
26762676 [
26772677 pytest .param (
2678- MOCK_ALARM_CONTROL_PANEL_LOCAL_CODE_SUBENTRY_DATA_SINGLE ,
2678+ MOCK_ALARM_CONTROL_PANEL_LOCAL_CODE_SUBENTRY_DATA ,
26792679 {"name" : "Milk notifier" , "mqtt_settings" : {"qos" : 0 }},
26802680 {"name" : "Alarm" },
26812681 {
@@ -2723,7 +2723,7 @@ async def test_migrate_of_incompatible_config_entry(
27232723 id = "alarm_control_panel_local_code" ,
27242724 ),
27252725 pytest .param (
2726- MOCK_ALARM_CONTROL_PANEL_REMOTE_CODE_SUBENTRY_DATA_SINGLE ,
2726+ MOCK_ALARM_CONTROL_PANEL_REMOTE_CODE_SUBENTRY_DATA ,
27272727 {"name" : "Milk notifier" , "mqtt_settings" : {"qos" : 1 }},
27282728 {"name" : "Alarm" },
27292729 {
@@ -2754,7 +2754,7 @@ async def test_migrate_of_incompatible_config_entry(
27542754 id = "alarm_control_panel_remote_code" ,
27552755 ),
27562756 pytest .param (
2757- MOCK_ALARM_CONTROL_PANEL_REMOTE_CODE_TEXT_SUBENTRY_DATA_SINGLE ,
2757+ MOCK_ALARM_CONTROL_PANEL_REMOTE_CODE_TEXT_SUBENTRY_DATA ,
27582758 {"name" : "Milk notifier" , "mqtt_settings" : {"qos" : 2 }},
27592759 {"name" : "Alarm" },
27602760 {
@@ -2785,7 +2785,7 @@ async def test_migrate_of_incompatible_config_entry(
27852785 id = "alarm_control_panel_remote_code_text" ,
27862786 ),
27872787 pytest .param (
2788- MOCK_BINARY_SENSOR_SUBENTRY_DATA_SINGLE ,
2788+ MOCK_BINARY_SENSOR_SUBENTRY_DATA ,
27892789 {"name" : "Milk notifier" , "mqtt_settings" : {"qos" : 2 }},
27902790 {"name" : "Hatch" },
27912791 {"device_class" : "door" },
@@ -2805,7 +2805,7 @@ async def test_migrate_of_incompatible_config_entry(
28052805 id = "binary_sensor" ,
28062806 ),
28072807 pytest .param (
2808- MOCK_BUTTON_SUBENTRY_DATA_SINGLE ,
2808+ MOCK_BUTTON_SUBENTRY_DATA ,
28092809 {"name" : "Milk notifier" , "mqtt_settings" : {"qos" : 2 }},
28102810 {"name" : "Restart" },
28112811 {"device_class" : "restart" },
@@ -2826,7 +2826,7 @@ async def test_migrate_of_incompatible_config_entry(
28262826 id = "button" ,
28272827 ),
28282828 pytest .param (
2829- MOCK_CLIMATE_HIGH_LOW_SUBENTRY_DATA_SINGLE ,
2829+ MOCK_CLIMATE_HIGH_LOW_SUBENTRY_DATA ,
28302830 {"name" : "Milk notifier" , "mqtt_settings" : {"qos" : 0 }},
28312831 {"name" : "Cooler" },
28322832 {
@@ -2871,7 +2871,7 @@ async def test_migrate_of_incompatible_config_entry(
28712871 id = "climate_high_low" ,
28722872 ),
28732873 pytest .param (
2874- MOCK_CLIMATE_NO_TARGET_TEMP_SUBENTRY_DATA_SINGLE ,
2874+ MOCK_CLIMATE_NO_TARGET_TEMP_SUBENTRY_DATA ,
28752875 {"name" : "Milk notifier" , "mqtt_settings" : {"qos" : 0 }},
28762876 {"name" : "Cooler" },
28772877 {
@@ -2900,7 +2900,7 @@ async def test_migrate_of_incompatible_config_entry(
29002900 id = "climate_no_target_temp" ,
29012901 ),
29022902 pytest .param (
2903- MOCK_CLIMATE_SUBENTRY_DATA_SINGLE ,
2903+ MOCK_CLIMATE_SUBENTRY_DATA ,
29042904 {"name" : "Milk notifier" , "mqtt_settings" : {"qos" : 0 }},
29052905 {"name" : "Cooler" },
29062906 {
@@ -3047,7 +3047,7 @@ async def test_migrate_of_incompatible_config_entry(
30473047 id = "climate_single" ,
30483048 ),
30493049 pytest .param (
3050- MOCK_COVER_SUBENTRY_DATA_SINGLE ,
3050+ MOCK_COVER_SUBENTRY_DATA ,
30513051 {"name" : "Milk notifier" , "mqtt_settings" : {"qos" : 0 }},
30523052 {"name" : "Blind" },
30533053 {"device_class" : "blind" },
@@ -3134,7 +3134,7 @@ async def test_migrate_of_incompatible_config_entry(
31343134 id = "cover" ,
31353135 ),
31363136 pytest .param (
3137- MOCK_FAN_SUBENTRY_DATA_SINGLE ,
3137+ MOCK_FAN_SUBENTRY_DATA ,
31383138 {"name" : "Milk notifier" , "mqtt_settings" : {"qos" : 0 }},
31393139 {"name" : "Breezer" },
31403140 {
@@ -3325,7 +3325,7 @@ async def test_migrate_of_incompatible_config_entry(
33253325 id = "notify_image_url" ,
33263326 ),
33273327 pytest .param (
3328- MOCK_LIGHT_BASIC_KELVIN_SUBENTRY_DATA_SINGLE ,
3328+ MOCK_LIGHT_BASIC_KELVIN_SUBENTRY_DATA ,
33293329 {"name" : "Milk notifier" , "mqtt_settings" : {"qos" : 1 }},
33303330 {"name" : "Basic light" },
33313331 {},
@@ -3371,7 +3371,7 @@ async def test_migrate_of_incompatible_config_entry(
33713371 id = "light_basic_kelvin" ,
33723372 ),
33733373 pytest .param (
3374- MOCK_LOCK_SUBENTRY_DATA_SINGLE ,
3374+ MOCK_LOCK_SUBENTRY_DATA ,
33753375 {"name" : "Milk notifier" , "mqtt_settings" : {"qos" : 0 }},
33763376 {"name" : "Lock" },
33773377 {},
@@ -3440,7 +3440,7 @@ async def test_migrate_of_incompatible_config_entry(
34403440 id = "notify_no_entity_name" ,
34413441 ),
34423442 pytest .param (
3443- MOCK_NOTIFY_SUBENTRY_DATA_SINGLE ,
3443+ MOCK_NOTIFY_SUBENTRY_DATA ,
34443444 {"name" : "Milk notifier" , "mqtt_settings" : {"qos" : 1 }},
34453445 {"name" : "Milkman alert" },
34463446 {},
@@ -3573,7 +3573,7 @@ async def test_migrate_of_incompatible_config_entry(
35733573 id = "select" ,
35743574 ),
35753575 pytest .param (
3576- MOCK_SENSOR_SUBENTRY_DATA_SINGLE ,
3576+ MOCK_SENSOR_SUBENTRY_DATA ,
35773577 {"name" : "Milk notifier" , "mqtt_settings" : {"qos" : 0 }},
35783578 {"name" : "Energy" },
35793579 {"device_class" : "enum" , "options" : ["low" , "medium" , "high" ]},
@@ -3630,7 +3630,7 @@ async def test_migrate_of_incompatible_config_entry(
36303630 id = "sensor_options" ,
36313631 ),
36323632 pytest .param (
3633- MOCK_SENSOR_SUBENTRY_DATA_SINGLE_STATE_CLASS ,
3633+ MOCK_SENSOR_SUBENTRY_DATA_STATE_CLASS ,
36343634 {"name" : "Milk notifier" , "mqtt_settings" : {"qos" : 0 }},
36353635 {"name" : "Energy" },
36363636 {
@@ -3653,7 +3653,7 @@ async def test_migrate_of_incompatible_config_entry(
36533653 id = "sensor_total" ,
36543654 ),
36553655 pytest .param (
3656- MOCK_SWITCH_SUBENTRY_DATA_SINGLE_STATE_CLASS ,
3656+ MOCK_SWITCH_SUBENTRY_DATA ,
36573657 {"name" : "Milk notifier" , "mqtt_settings" : {"qos" : 0 }},
36583658 {"name" : "Outlet" },
36593659 {"device_class" : "outlet" },
@@ -4101,7 +4101,7 @@ async def test_subentry_reconfigure_edit_entity_multi_entitites(
41014101 pytest .param (
41024102 (
41034103 ConfigSubentryData (
4104- data = MOCK_ALARM_CONTROL_PANEL_LOCAL_CODE_SUBENTRY_DATA_SINGLE ,
4104+ data = MOCK_ALARM_CONTROL_PANEL_LOCAL_CODE_SUBENTRY_DATA ,
41054105 subentry_type = "device" ,
41064106 title = "Mock subentry" ,
41074107 ),
@@ -4132,7 +4132,7 @@ async def test_subentry_reconfigure_edit_entity_multi_entitites(
41324132 pytest .param (
41334133 (
41344134 ConfigSubentryData (
4135- data = MOCK_ALARM_CONTROL_PANEL_REMOTE_CODE_SUBENTRY_DATA_SINGLE ,
4135+ data = MOCK_ALARM_CONTROL_PANEL_REMOTE_CODE_SUBENTRY_DATA ,
41364136 subentry_type = "device" ,
41374137 title = "Mock subentry" ,
41384138 ),
@@ -4164,7 +4164,7 @@ async def test_subentry_reconfigure_edit_entity_multi_entitites(
41644164 pytest .param (
41654165 (
41664166 ConfigSubentryData (
4167- data = MOCK_CLIMATE_HIGH_LOW_SUBENTRY_DATA_SINGLE ,
4167+ data = MOCK_CLIMATE_HIGH_LOW_SUBENTRY_DATA ,
41684168 subentry_type = "device" ,
41694169 title = "Mock subentry" ,
41704170 ),
@@ -4212,7 +4212,7 @@ async def test_subentry_reconfigure_edit_entity_multi_entitites(
42124212 pytest .param (
42134213 (
42144214 ConfigSubentryData (
4215- data = MOCK_CLIMATE_SUBENTRY_DATA_SINGLE ,
4215+ data = MOCK_CLIMATE_SUBENTRY_DATA ,
42164216 subentry_type = "device" ,
42174217 title = "Mock subentry" ,
42184218 ),
@@ -4302,7 +4302,7 @@ async def test_subentry_reconfigure_edit_entity_multi_entitites(
43024302 pytest .param (
43034303 (
43044304 ConfigSubentryData (
4305- data = MOCK_LIGHT_BASIC_KELVIN_SUBENTRY_DATA_SINGLE ,
4305+ data = MOCK_LIGHT_BASIC_KELVIN_SUBENTRY_DATA ,
43064306 subentry_type = "device" ,
43074307 title = "Mock subentry" ,
43084308 ),
@@ -4327,7 +4327,7 @@ async def test_subentry_reconfigure_edit_entity_multi_entitites(
43274327 pytest .param (
43284328 (
43294329 ConfigSubentryData (
4330- data = MOCK_NOTIFY_SUBENTRY_DATA_SINGLE ,
4330+ data = MOCK_NOTIFY_SUBENTRY_DATA ,
43314331 subentry_type = "device" ,
43324332 title = "Mock subentry" ,
43334333 ),
@@ -4350,7 +4350,7 @@ async def test_subentry_reconfigure_edit_entity_multi_entitites(
43504350 pytest .param (
43514351 (
43524352 ConfigSubentryData (
4353- data = MOCK_SENSOR_SUBENTRY_DATA_SINGLE ,
4353+ data = MOCK_SENSOR_SUBENTRY_DATA ,
43544354 subentry_type = "device" ,
43554355 title = "Mock subentry" ,
43564356 ),
@@ -4518,7 +4518,7 @@ async def test_subentry_reconfigure_edit_entity_single_entity(
45184518 (
45194519 (
45204520 ConfigSubentryData (
4521- data = MOCK_SENSOR_SUBENTRY_DATA_SINGLE_LAST_RESET_TEMPLATE ,
4521+ data = MOCK_SENSOR_SUBENTRY_DATA_LAST_RESET_TEMPLATE ,
45224522 subentry_type = "device" ,
45234523 title = "Mock subentry" ,
45244524 ),
@@ -4651,7 +4651,7 @@ async def test_subentry_reconfigure_edit_entity_reset_fields(
46514651 (
46524652 (
46534653 ConfigSubentryData (
4654- data = MOCK_NOTIFY_SUBENTRY_DATA_SINGLE ,
4654+ data = MOCK_NOTIFY_SUBENTRY_DATA ,
46554655 subentry_type = "device" ,
46564656 title = "Mock subentry" ,
46574657 ),
0 commit comments