"""The tests for mqtt number component."""
import json
from unittest.mock import patch

import pytest

from openpeerpower.components import number
from openpeerpower.components.mqtt.number import CONF_MAX, CONF_MIN
from openpeerpower.components.number import (
    ATTR_MAX,
    ATTR_MIN,
    ATTR_STEP,
    ATTR_VALUE,
    DOMAIN as NUMBER_DOMAIN,
    SERVICE_SET_VALUE,
)
from openpeerpower.const import ATTR_ASSUMED_STATE, ATTR_ENTITY_ID
import openpeerpower.core as ha
from openpeerpower.setup import async_setup_component

from .test_common import (
    help_test_availability_when_connection_lost,
    help_test_availability_without_topic,
    help_test_custom_availability_payload,
    help_test_default_availability_payload,
    help_test_discovery_broken,
    help_test_discovery_removal,
    help_test_discovery_update,
    help_test_discovery_update_attr,
    help_test_discovery_update_unchanged,
    help_test_entity_debug_info_message,
    help_test_entity_device_info_remove,
    help_test_entity_device_info_update,
    help_test_entity_device_info_with_connection,
    help_test_entity_device_info_with_identifier,
    help_test_entity_id_update_discovery_update,
    help_test_entity_id_update_subscriptions,
    help_test_setting_attribute_via_mqtt_json_message,
    help_test_setting_attribute_with_template,
    help_test_unique_id,
    help_test_update_with_json_attrs_bad_JSON,
    help_test_update_with_json_attrs_not_dict,
)

from tests.common import async_fire_mqtt_message

DEFAULT_CONFIG = {
    number.DOMAIN: {"platform": "mqtt", "name": "test", "command_topic": "test-topic"}
}


async def test_run_number_setup(opp, mqtt_mock):
    """Test that it fetches the given payload."""
    topic = "test/number"
    await async_setup_component(
        opp,
        "number",
        {
            "number": {
                "platform": "mqtt",
                "state_topic": topic,
                "command_topic": topic,
                "name": "<NAME>",
            }
        },
    )
    await opp.async_block_till_done()

    async_fire_mqtt_message(opp, topic, "10")

    await opp.async_block_till_done()

    state = opp.states.get("number.test_number")
    assert state.state == "10"

    async_fire_mqtt_message(opp, topic, "20.5")

    await opp.async_block_till_done()

    state = opp.states.get("number.test_number")
    assert state.state == "20.5"


async def test_run_number_service_optimistic(opp, mqtt_mock):
    """Test that set_value service works in optimistic mode."""
    topic = "test/number"

    fake_state = ha.State("switch.test", "3")

    with patch(
        "openpeerpower.helpers.restore_state.RestoreEntity.async_get_last_state",
        return_value=fake_state,
    ):
        assert await async_setup_component(
            opp,
            number.DOMAIN,
            {
                "number": {
                    "platform": "mqtt",
                    "command_topic": topic,
                    "name": "<NAME>",
                }
            },
        )
        await opp.async_block_till_done()

    state = opp.states.get("number.test_number")
    assert state.state == "3"
    assert state.attributes.get(ATTR_ASSUMED_STATE)

    # Integer
    await opp.services.async_call(
        NUMBER_DOMAIN,
        SERVICE_SET_VALUE,
        {ATTR_ENTITY_ID: "number.test_number", ATTR_VALUE: 30},
        blocking=True,
    )

    mqtt_mock.async_publish.assert_called_once_with(topic, "30", 0, False)
    mqtt_mock.async_publish.reset_mock()
    state = opp.states.get("number.test_number")
    assert state.state == "30"

    # Float with no decimal -> integer
    await opp.services.async_call(
        NUMBER_DOMAIN,
        SERVICE_SET_VALUE,
        {ATTR_ENTITY_ID: "number.test_number", ATTR_VALUE: 42.0},
        blocking=True,
    )

    mqtt_mock.async_publish.assert_called_once_with(topic, "42", 0, False)
    mqtt_mock.async_publish.reset_mock()
    state = opp.states.get("number.test_number")
    assert state.state == "42"

    # Float with decimal -> float
    await opp.services.async_call(
        NUMBER_DOMAIN,
        SERVICE_SET_VALUE,
        {ATTR_ENTITY_ID: "number.test_number", ATTR_VALUE: 42.1},
        blocking=True,
    )

    mqtt_mock.async_publish.assert_called_once_with(topic, "42.1", 0, False)
    mqtt_mock.async_publish.reset_mock()
    state = opp.states.get("number.test_number")
    assert state.state == "42.1"


async def test_run_number_service(opp, mqtt_mock):
    """Test that set_value service works in non optimistic mode."""
    cmd_topic = "test/number/set"
    state_topic = "test/number"

    assert await async_setup_component(
        opp,
        number.DOMAIN,
        {
            "number": {
                "platform": "mqtt",
                "command_topic": cmd_topic,
                "state_topic": state_topic,
                "name": "<NAME>",
            }
        },
    )
    await opp.async_block_till_done()

    async_fire_mqtt_message(opp, state_topic, "32")
    state = opp.states.get("number.test_number")
    assert state.state == "32"

    await opp.services.async_call(
        NUMBER_DOMAIN,
        SERVICE_SET_VALUE,
        {ATTR_ENTITY_ID: "number.test_number", ATTR_VALUE: 30},
        blocking=True,
    )
    mqtt_mock.async_publish.assert_called_once_with(cmd_topic, "30", 0, False)
    state = opp.states.get("number.test_number")
    assert state.state == "32"


async def test_availability_when_connection_lost(opp, mqtt_mock):
    """Test availability after MQTT disconnection."""
    await help_test_availability_when_connection_lost(
        opp, mqtt_mock, number.DOMAIN, DEFAULT_CONFIG
    )


async def test_availability_without_topic(opp, mqtt_mock):
    """Test availability without defined availability topic."""
    await help_test_availability_without_topic(
        opp, mqtt_mock, number.DOMAIN, DEFAULT_CONFIG
    )


async def test_default_availability_payload(opp, mqtt_mock):
    """Test availability by default payload with defined topic."""
    await help_test_default_availability_payload(
        opp, mqtt_mock, number.DOMAIN, DEFAULT_CONFIG
    )


async def test_custom_availability_payload(opp, mqtt_mock):
    """Test availability by custom payload with defined topic."""
    await help_test_custom_availability_payload(
        opp, mqtt_mock, number.DOMAIN, DEFAULT_CONFIG
    )


async def test_setting_attribute_via_mqtt_json_message(opp, mqtt_mock):
    """Test the setting of attribute via MQTT with JSON payload."""
    await help_test_setting_attribute_via_mqtt_json_message(
        opp, mqtt_mock, number.DOMAIN, DEFAULT_CONFIG
    )


async def test_setting_attribute_with_template(opp, mqtt_mock):
    """Test the setting of attribute via MQTT with JSON payload."""
    await help_test_setting_attribute_with_template(
        opp, mqtt_mock, number.DOMAIN, DEFAULT_CONFIG
    )


async def test_update_with_json_attrs_not_dict(opp, mqtt_mock, caplog):
    """Test attributes get extracted from a JSON result."""
    await help_test_update_with_json_attrs_not_dict(
        opp, mqtt_mock, caplog, number.DOMAIN, DEFAULT_CONFIG
    )


async def test_update_with_json_attrs_bad_JSON(opp, mqtt_mock, caplog):
    """Test attributes get extracted from a JSON result."""
    await help_test_update_with_json_attrs_bad_JSON(
        opp, mqtt_mock, caplog, number.DOMAIN, DEFAULT_CONFIG
    )


async def test_discovery_update_attr(opp, mqtt_mock, caplog):
    """Test update of discovered MQTTAttributes."""
    await help_test_discovery_update_attr(
        opp, mqtt_mock, caplog, number.DOMAIN, DEFAULT_CONFIG
    )


async def test_unique_id(opp, mqtt_mock):
    """Test unique id option only creates one number per unique_id."""
    config = {
        number.DOMAIN: [
            {
                "platform": "mqtt",
                "name": "Test 1",
                "state_topic": "test-topic",
                "command_topic": "test-topic",
                "unique_id": "TOTALLY_UNIQUE",
            },
            {
                "platform": "mqtt",
                "name": "Test 2",
                "state_topic": "test-topic",
                "command_topic": "test-topic",
                "unique_id": "TOTALLY_UNIQUE",
            },
        ]
    }
    await help_test_unique_id(opp, mqtt_mock, number.DOMAIN, config)


async def test_discovery_removal_number(opp, mqtt_mock, caplog):
    """Test removal of discovered number."""
    data = json.dumps(DEFAULT_CONFIG[number.DOMAIN])
    await help_test_discovery_removal(opp, mqtt_mock, caplog, number.DOMAIN, data)


async def test_discovery_update_number(opp, mqtt_mock, caplog):
    """Test update of discovered number."""
    data1 = (
        '{ "name": "Beer", "state_topic": "test-topic", "command_topic": "test-topic"}'
    )
    data2 = (
        '{ "name": "Milk", "state_topic": "test-topic", "command_topic": "test-topic"}'
    )

    await help_test_discovery_update(
        opp, mqtt_mock, caplog, number.DOMAIN, data1, data2
    )


async def test_discovery_update_unchanged_number(opp, mqtt_mock, caplog):
    """Test update of discovered number."""
    data1 = (
        '{ "name": "Beer", "state_topic": "test-topic", "command_topic": "test-topic"}'
    )
    with patch(
        "openpeerpower.components.mqtt.number.MqttNumber.discovery_update"
    ) as discovery_update:
        await help_test_discovery_update_unchanged(
            opp, mqtt_mock, caplog, number.DOMAIN, data1, discovery_update
        )


@pytest.mark.no_fail_on_log_exception
async def test_discovery_broken(opp, mqtt_mock, caplog):
    """Test handling of bad discovery message."""
    data1 = '{ "name": "Beer" }'
    data2 = (
        '{ "name": "Milk", "state_topic": "test-topic", "command_topic": "test-topic"}'
    )

    await help_test_discovery_broken(
        opp, mqtt_mock, caplog, number.DOMAIN, data1, data2
    )


async def test_entity_device_info_with_connection(opp, mqtt_mock):
    """Test MQTT number device registry integration."""
    await help_test_entity_device_info_with_connection(
        opp, mqtt_mock, number.DOMAIN, DEFAULT_CONFIG
    )


async def test_entity_device_info_with_identifier(opp, mqtt_mock):
    """Test MQTT number device registry integration."""
    await help_test_entity_device_info_with_identifier(
        opp, mqtt_mock, number.DOMAIN, DEFAULT_CONFIG
    )


async def test_entity_device_info_update(opp, mqtt_mock):
    """Test device registry update."""
    await help_test_entity_device_info_update(
        opp, mqtt_mock, number.DOMAIN, DEFAULT_CONFIG
    )


async def test_entity_device_info_remove(opp, mqtt_mock):
    """Test device registry remove."""
    await help_test_entity_device_info_remove(
        opp, mqtt_mock, number.DOMAIN, DEFAULT_CONFIG
    )


async def test_entity_id_update_subscriptions(opp, mqtt_mock):
    """Test MQTT subscriptions are managed when entity_id is updated."""
    await help_test_entity_id_update_subscriptions(
        opp, mqtt_mock, number.DOMAIN, DEFAULT_CONFIG
    )


async def test_entity_id_update_discovery_update(opp, mqtt_mock):
    """Test MQTT discovery update when entity_id is updated."""
    await help_test_entity_id_update_discovery_update(
        opp, mqtt_mock, number.DOMAIN, DEFAULT_CONFIG
    )


async def test_entity_debug_info_message(opp, mqtt_mock):
    """Test MQTT debug info."""
    await help_test_entity_debug_info_message(
        opp, mqtt_mock, number.DOMAIN, DEFAULT_CONFIG, payload=b"1"
    )


async def test_min_max_step_attributes(opp, mqtt_mock):
    """Test min/max/step attributes."""
    topic = "test/number"
    await async_setup_component(
        opp,
        "number",
        {
            "number": {
                "platform": "mqtt",
                "state_topic": topic,
                "command_topic": topic,
                "name": "<NAME>",
                "min": 5,
                "max": 110,
                "step": 20,
            }
        },
    )
    await opp.async_block_till_done()

    state = opp.states.get("number.test_number")
    assert state.attributes.get(ATTR_MIN) == 5
    assert state.attributes.get(ATTR_MAX) == 110
    assert state.attributes.get(ATTR_STEP) == 20


async def test_invalid_min_max_attributes(opp, caplog, mqtt_mock):
    """Test invalid min/max attributes."""
    topic = "test/number"
    await async_setup_component(
        opp,
        "number",
        {
            "number": {
                "platform": "mqtt",
                "state_topic": topic,
                "command_topic": topic,
                "name": "<NAME>",
                "min": 35,
                "max": 10,
            }
        },
    )
    await opp.async_block_till_done()

    assert f"'{CONF_MAX}'' must be > '{CONF_MIN}'" in caplog.text


async def test_mqtt_payload_not_a_number_warning(opp, caplog, mqtt_mock):
    """Test warning for MQTT payload which is not a number."""
    topic = "test/number"
    await async_setup_component(
        opp,
        "number",
        {
            "number": {
                "platform": "mqtt",
                "state_topic": topic,
                "command_topic": topic,
                "name": "<NAME>",
            }
        },
    )
    await opp.async_block_till_done()

    async_fire_mqtt_message(opp, topic, "not_a_number")

    await opp.async_block_till_done()

    assert "Payload 'not_a_number' is not a Number" in caplog.text


async def test_mqtt_payload_out_of_range_error(opp, caplog, mqtt_mock):
    """Test error when MQTT payload is out of min/max range."""
    topic = "test/number"
    await async_setup_component(
        opp,
        "number",
        {
            "number": {
                "platform": "mqtt",
                "state_topic": topic,
                "command_topic": topic,
                "name": "<NAME>",
                "min": 5,
                "max": 110,
            }
        },
    )
    await opp.async_block_till_done()

    async_fire_mqtt_message(opp, topic, "115.5")

    await opp.async_block_till_done()

    assert (
        "Invalid value for number.test_number: 115.5 (range 5.0 - 110.0)" in caplog.text
    )

"""The tests for mqtt number component."""
import json
from unittest.mock import patch

import pytest

from openpeerpower.components import number
from openpeerpower.components.mqtt.number import CONF_MAX, CONF_MIN
from openpeerpower.components.number import (
    ATTR_MAX,
    ATTR_MIN,
    ATTR_STEP,
    ATTR_VALUE,
    DOMAIN as NUMBER_DOMAIN,
    SERVICE_SET_VALUE,
)
from openpeerpower.const import ATTR_ASSUMED_STATE, ATTR_ENTITY_ID
import openpeerpower.core as ha
from openpeerpower.setup import async_setup_component

from .test_common import (
    help_test_availability_when_connection_lost,
    help_test_availability_without_topic,
    help_test_custom_availability_payload,
    help_test_default_availability_payload,
    help_test_discovery_broken,
    help_test_discovery_removal,
    help_test_discovery_update,
    help_test_discovery_update_attr,
    help_test_discovery_update_unchanged,
    help_test_entity_debug_info_message,
    help_test_entity_device_info_remove,
    help_test_entity_device_info_update,
    help_test_entity_device_info_with_connection,
    help_test_entity_device_info_with_identifier,
    help_test_entity_id_update_discovery_update,
    help_test_entity_id_update_subscriptions,
    help_test_setting_attribute_via_mqtt_json_message,
    help_test_setting_attribute_with_template,
    help_test_unique_id,
    help_test_update_with_json_attrs_bad_JSON,
    help_test_update_with_json_attrs_not_dict,
)

from tests.common import async_fire_mqtt_message

DEFAULT_CONFIG = {
    number.DOMAIN: {"platform": "mqtt", "name": "test", "command_topic": "test-topic"}
}


async def test_run_number_setup(opp, mqtt_mock):
    """Test that it fetches the given payload."""
    topic = "test/number"
    await async_setup_component(
        opp,
        "number",
        {
            "number": {
                "platform": "mqtt",
                "state_topic": topic,
                "command_topic": topic,
                "name": "<NAME>",
            }
        },
    )
    await opp.async_block_till_done()

    async_fire_mqtt_message(opp, topic, "10")

    await opp.async_block_till_done()

    state = opp.states.get("number.test_number")
    assert state.state == "10"

    async_fire_mqtt_message(opp, topic, "20.5")

    await opp.async_block_till_done()

    state = opp.states.get("number.test_number")
    assert state.state == "20.5"


async def test_run_number_service_optimistic(opp, mqtt_mock):
    """Test that set_value service works in optimistic mode."""
    topic = "test/number"

    fake_state = ha.State("switch.test", "3")

    with patch(
        "openpeerpower.helpers.restore_state.RestoreEntity.async_get_last_state",
        return_value=fake_state,
    ):
        assert await async_setup_component(
            opp,
            number.DOMAIN,
            {
                "number": {
                    "platform": "mqtt",
                    "command_topic": topic,
                    "name": "<NAME>",
                }
            },
        )
        await opp.async_block_till_done()

    state = opp.states.get("number.test_number")
    assert state.state == "3"
    assert state.attributes.get(ATTR_ASSUMED_STATE)

    # Integer
    await opp.services.async_call(
        NUMBER_DOMAIN,
        SERVICE_SET_VALUE,
        {ATTR_ENTITY_ID: "number.test_number", ATTR_VALUE: 30},
        blocking=True,
    )

    mqtt_mock.async_publish.assert_called_once_with(topic, "30", 0, False)
    mqtt_mock.async_publish.reset_mock()
    state = opp.states.get("number.test_number")
    assert state.state == "30"

    # Float with no decimal -> integer
    await opp.services.async_call(
        NUMBER_DOMAIN,
        SERVICE_SET_VALUE,
        {ATTR_ENTITY_ID: "number.test_number", ATTR_VALUE: 42.0},
        blocking=True,
    )

    mqtt_mock.async_publish.assert_called_once_with(topic, "42", 0, False)
    mqtt_mock.async_publish.reset_mock()
    state = opp.states.get("number.test_number")
    assert state.state == "42"

    # Float with decimal -> float
    await opp.services.async_call(
        NUMBER_DOMAIN,
        SERVICE_SET_VALUE,
        {ATTR_ENTITY_ID: "number.test_number", ATTR_VALUE: 42.1},
        blocking=True,
    )

    mqtt_mock.async_publish.assert_called_once_with(topic, "42.1", 0, False)
    mqtt_mock.async_publish.reset_mock()
    state = opp.states.get("number.test_number")
    assert state.state == "42.1"


async def test_run_number_service(opp, mqtt_mock):
    """Test that set_value service works in non optimistic mode."""
    cmd_topic = "test/number/set"
    state_topic = "test/number"

    assert await async_setup_component(
        opp,
        number.DOMAIN,
        {
            "number": {
                "platform": "mqtt",
                "command_topic": cmd_topic,
                "state_topic": state_topic,
                "name": "<NAME>",
            }
        },
    )
    await opp.async_block_till_done()

    async_fire_mqtt_message(opp, state_topic, "32")
    state = opp.states.get("number.test_number")
    assert state.state == "32"

    await opp.services.async_call(
        NUMBER_DOMAIN,
        SERVICE_SET_VALUE,
        {ATTR_ENTITY_ID: "number.test_number", ATTR_VALUE: 30},
        blocking=True,
    )
    mqtt_mock.async_publish.assert_called_once_with(cmd_topic, "30", 0, False)
    state = opp.states.get("number.test_number")
    assert state.state == "32"


async def test_availability_when_connection_lost(opp, mqtt_mock):
    """Test availability after MQTT disconnection."""
    await help_test_availability_when_connection_lost(
        opp, mqtt_mock, number.DOMAIN, DEFAULT_CONFIG
    )


async def test_availability_without_topic(opp, mqtt_mock):
    """Test availability without defined availability topic."""
    await help_test_availability_without_topic(
        opp, mqtt_mock, number.DOMAIN, DEFAULT_CONFIG
    )


async def test_default_availability_payload(opp, mqtt_mock):
    """Test availability by default payload with defined topic."""
    await help_test_default_availability_payload(
        opp, mqtt_mock, number.DOMAIN, DEFAULT_CONFIG
    )


async def test_custom_availability_payload(opp, mqtt_mock):
    """Test availability by custom payload with defined topic."""
    await help_test_custom_availability_payload(
        opp, mqtt_mock, number.DOMAIN, DEFAULT_CONFIG
    )


async def test_setting_attribute_via_mqtt_json_message(opp, mqtt_mock):
    """Test the setting of attribute via MQTT with JSON payload."""
    await help_test_setting_attribute_via_mqtt_json_message(
        opp, mqtt_mock, number.DOMAIN, DEFAULT_CONFIG
    )


async def test_setting_attribute_with_template(opp, mqtt_mock):
    """Test the setting of attribute via MQTT with JSON payload."""
    await help_test_setting_attribute_with_template(
        opp, mqtt_mock, number.DOMAIN, DEFAULT_CONFIG
    )


async def test_update_with_json_attrs_not_dict(opp, mqtt_mock, caplog):
    """Test attributes get extracted from a JSON result."""
    await help_test_update_with_json_attrs_not_dict(
        opp, mqtt_mock, caplog, number.DOMAIN, DEFAULT_CONFIG
    )


async def test_update_with_json_attrs_bad_JSON(opp, mqtt_mock, caplog):
    """Test attributes get extracted from a JSON result."""
    await help_test_update_with_json_attrs_bad_JSON(
        opp, mqtt_mock, caplog, number.DOMAIN, DEFAULT_CONFIG
    )


async def test_discovery_update_attr(opp, mqtt_mock, caplog):
    """Test update of discovered MQTTAttributes."""
    await help_test_discovery_update_attr(
        opp, mqtt_mock, caplog, number.DOMAIN, DEFAULT_CONFIG
    )


async def test_unique_id(opp, mqtt_mock):
    """Test unique id option only creates one number per unique_id."""
    config = {
        number.DOMAIN: [
            {
                "platform": "mqtt",
                "name": "Test 1",
                "state_topic": "test-topic",
                "command_topic": "test-topic",
                "unique_id": "TOTALLY_UNIQUE",
            },
            {
                "platform": "mqtt",
                "name": "Test 2",
                "state_topic": "test-topic",
                "command_topic": "test-topic",
                "unique_id": "TOTALLY_UNIQUE",
            },
        ]
    }
    await help_test_unique_id(opp, mqtt_mock, number.DOMAIN, config)


async def test_discovery_removal_number(opp, mqtt_mock, caplog):
    """Test removal of discovered number."""
    data = json.dumps(DEFAULT_CONFIG[number.DOMAIN])
    await help_test_discovery_removal(opp, mqtt_mock, caplog, number.DOMAIN, data)


async def test_discovery_update_number(opp, mqtt_mock, caplog):
    """Test update of discovered number."""
    data1 = (
        '{ "name": "Beer", "state_topic": "test-topic", "command_topic": "test-topic"}'
    )
    data2 = (
        '{ "name": "Milk", "state_topic": "test-topic", "command_topic": "test-topic"}'
    )

    await help_test_discovery_update(
        opp, mqtt_mock, caplog, number.DOMAIN, data1, data2
    )


async def test_discovery_update_unchanged_number(opp, mqtt_mock, caplog):
    """Test update of discovered number."""
    data1 = (
        '{ "name": "Beer", "state_topic": "test-topic", "command_topic": "test-topic"}'
    )
    with patch(
        "openpeerpower.components.mqtt.number.MqttNumber.discovery_update"
    ) as discovery_update:
        await help_test_discovery_update_unchanged(
            opp, mqtt_mock, caplog, number.DOMAIN, data1, discovery_update
        )


@pytest.mark.no_fail_on_log_exception
async def test_discovery_broken(opp, mqtt_mock, caplog):
    """Test handling of bad discovery message."""
    data1 = '{ "name": "Beer" }'
    data2 = (
        '{ "name": "Milk", "state_topic": "test-topic", "command_topic": "test-topic"}'
    )

    await help_test_discovery_broken(
        opp, mqtt_mock, caplog, number.DOMAIN, data1, data2
    )


async def test_entity_device_info_with_connection(opp, mqtt_mock):
    """Test MQTT number device registry integration."""
    await help_test_entity_device_info_with_connection(
        opp, mqtt_mock, number.DOMAIN, DEFAULT_CONFIG
    )


async def test_entity_device_info_with_identifier(opp, mqtt_mock):
    """Test MQTT number device registry integration."""
    await help_test_entity_device_info_with_identifier(
        opp, mqtt_mock, number.DOMAIN, DEFAULT_CONFIG
    )


async def test_entity_device_info_update(opp, mqtt_mock):
    """Test device registry update."""
    await help_test_entity_device_info_update(
        opp, mqtt_mock, number.DOMAIN, DEFAULT_CONFIG
    )


async def test_entity_device_info_remove(opp, mqtt_mock):
    """Test device registry remove."""
    await help_test_entity_device_info_remove(
        opp, mqtt_mock, number.DOMAIN, DEFAULT_CONFIG
    )


async def test_entity_id_update_subscriptions(opp, mqtt_mock):
    """Test MQTT subscriptions are managed when entity_id is updated."""
    await help_test_entity_id_update_subscriptions(
        opp, mqtt_mock, number.DOMAIN, DEFAULT_CONFIG
    )


async def test_entity_id_update_discovery_update(opp, mqtt_mock):
    """Test MQTT discovery update when entity_id is updated."""
    await help_test_entity_id_update_discovery_update(
        opp, mqtt_mock, number.DOMAIN, DEFAULT_CONFIG
    )


async def test_entity_debug_info_message(opp, mqtt_mock):
    """Test MQTT debug info."""
    await help_test_entity_debug_info_message(
        opp, mqtt_mock, number.DOMAIN, DEFAULT_CONFIG, payload=b"1"
    )


async def test_min_max_step_attributes(opp, mqtt_mock):
    """Test min/max/step attributes."""
    topic = "test/number"
    await async_setup_component(
        opp,
        "number",
        {
            "number": {
                "platform": "mqtt",
                "state_topic": topic,
                "command_topic": topic,
                "name": "<NAME>",
                "min": 5,
                "max": 110,
                "step": 20,
            }
        },
    )
    await opp.async_block_till_done()

    state = opp.states.get("number.test_number")
    assert state.attributes.get(ATTR_MIN) == 5
    assert state.attributes.get(ATTR_MAX) == 110
    assert state.attributes.get(ATTR_STEP) == 20


async def test_invalid_min_max_attributes(opp, caplog, mqtt_mock):
    """Test invalid min/max attributes."""
    topic = "test/number"
    await async_setup_component(
        opp,
        "number",
        {
            "number": {
                "platform": "mqtt",
                "state_topic": topic,
                "command_topic": topic,
                "name": "<NAME>",
                "min": 35,
                "max": 10,
            }
        },
    )
    await opp.async_block_till_done()

    assert f"'{CONF_MAX}'' must be > '{CONF_MIN}'" in caplog.text


async def test_mqtt_payload_not_a_number_warning(opp, caplog, mqtt_mock):
    """Test warning for MQTT payload which is not a number."""
    topic = "test/number"
    await async_setup_component(
        opp,
        "number",
        {
            "number": {
                "platform": "mqtt",
                "state_topic": topic,
                "command_topic": topic,
                "name": "<NAME>",
            }
        },
    )
    await opp.async_block_till_done()

    async_fire_mqtt_message(opp, topic, "not_a_number")

    await opp.async_block_till_done()

    assert "Payload 'not_a_number' is not a Number" in caplog.text


async def test_mqtt_payload_out_of_range_error(opp, caplog, mqtt_mock):
    """Test error when MQTT payload is out of min/max range."""
    topic = "test/number"
    await async_setup_component(
        opp,
        "number",
        {
            "number": {
                "platform": "mqtt",
                "state_topic": topic,
                "command_topic": topic,
                "name": "<NAME>",
                "min": 5,
                "max": 110,
            }
        },
    )
    await opp.async_block_till_done()

    async_fire_mqtt_message(opp, topic, "115.5")

    await opp.async_block_till_done()

    assert (
        "Invalid value for number.test_number: 115.5 (range 5.0 - 110.0)" in caplog.text
    )

The tests for mqtt number component. Test that it fetches the given payload. Test that set_value service works in optimistic mode. # Integer # Float with no decimal -> integer # Float with decimal -> float Test that set_value service works in non optimistic mode. Test availability after MQTT disconnection. Test availability without defined availability topic. Test availability by default payload with defined topic. Test availability by custom payload with defined topic. Test the setting of attribute via MQTT with JSON payload. Test the setting of attribute via MQTT with JSON payload. Test attributes get extracted from a JSON result. Test attributes get extracted from a JSON result. Test update of discovered MQTTAttributes. Test unique id option only creates one number per unique_id. Test removal of discovered number. Test update of discovered number. Test update of discovered number. Test handling of bad discovery message. Test MQTT number device registry integration. Test MQTT number device registry integration. Test device registry update. Test device registry remove. Test MQTT subscriptions are managed when entity_id is updated. Test MQTT discovery update when entity_id is updated. Test MQTT debug info. Test min/max/step attributes. Test invalid min/max attributes. Test warning for MQTT payload which is not a number. Test error when MQTT payload is out of min/max range.