def test_get_desired_value(): """Test that sensor returns correct desired value in different states.""" const = get_const("1.4") sensor_id = 1 child_id = 0 value_type = const.SetReq.V_LIGHT_LEVEL wrong_child_id = 100 wrong_value_type = 9999 sensor = Sensor(sensor_id) sensor.add_child_sensor(child_id, const.Presentation.S_LIGHT_LEVEL) assert sensor.get_desired_value(child_id, value_type) is None sensor.update_child_value(child_id, value_type, "50") assert sensor.get_desired_value(child_id, value_type) == "50" sensor.init_smart_sleep_mode() sensor.set_child_desired_state(child_id, value_type, "90") assert sensor.get_desired_value(child_id, value_type) == "90" sensor.update_child_value(child_id, value_type, "70") assert sensor.get_desired_value(child_id, value_type) == "70" assert sensor.get_desired_value(wrong_child_id, value_type) is None assert sensor.get_desired_value(child_id, wrong_value_type) is None
def test_get_desired_value_empty_string(): """Test that sensor returns correct value if the desired state is empty string.""" const = get_const("1.4") sensor_id = 1 child_id = 0 value_type = const.SetReq.V_VAR1 sensor = Sensor(sensor_id) sensor.add_child_sensor(child_id, const.Presentation.S_CUSTOM) assert sensor.get_desired_value(child_id, value_type) is None sensor.update_child_value(child_id, value_type, "Some string") assert sensor.get_desired_value(child_id, value_type) == "Some string" sensor.init_smart_sleep_mode() sensor.set_child_desired_state(child_id, value_type, "") assert sensor.get_desired_value(child_id, value_type) == ""