[climate_ir] Only advertise HEAT_COOL when heat and cool are both supported (#19483)

tests/component_tests/climate_ir/config/cool_only.yaml

@@ -0,0 +1,15 @@
+esphome:
+  name: climateir-heatcool
+
+esp8266:
+  board: esp01_1m
+
+remote_transmitter:
+  pin: GPIO5
+  carrier_duty_percent: 50%
+
+climate:
+  - platform: coolix
+    id: test_coolix
+    name: Coolix
+    supports_heat: false

tests/component_tests/climate_ir/config/cool_only_override_on.yaml

@@ -0,0 +1,16 @@
+esphome:
+  name: climateir-heatcool
+
+esp8266:
+  board: esp01_1m
+
+remote_transmitter:
+  pin: GPIO5
+  carrier_duty_percent: 50%
+
+climate:
+  - platform: coolix
+    id: test_coolix
+    name: Coolix
+    supports_heat: false
+    supports_heat_cool: true

tests/component_tests/climate_ir/config/heat_and_cool.yaml

@@ -0,0 +1,14 @@
+esphome:
+  name: climateir-heatcool
+
+esp8266:
+  board: esp01_1m
+
+remote_transmitter:
+  pin: GPIO5
+  carrier_duty_percent: 50%
+
+climate:
+  - platform: coolix
+    id: test_coolix
+    name: Coolix

tests/component_tests/climate_ir/config/heat_and_cool_override_off.yaml

@@ -0,0 +1,15 @@
+esphome:
+  name: climateir-heatcool
+
+esp8266:
+  board: esp01_1m
+
+remote_transmitter:
+  pin: GPIO5
+  carrier_duty_percent: 50%
+
+climate:
+  - platform: coolix
+    id: test_coolix
+    name: Coolix
+    supports_heat_cool: false

tests/component_tests/climate_ir/config/heat_only.yaml

@@ -0,0 +1,15 @@
+esphome:
+  name: climateir-heatcool
+
+esp8266:
+  board: esp01_1m
+
+remote_transmitter:
+  pin: GPIO5
+  carrier_duty_percent: 50%
+
+climate:
+  - platform: coolix
+    id: test_coolix
+    name: Coolix
+    supports_cool: false

tests/component_tests/climate_ir/config/neither.yaml

@@ -0,0 +1,16 @@
+esphome:
+  name: climateir-heatcool
+
+esp8266:
+  board: esp01_1m
+
+remote_transmitter:
+  pin: GPIO5
+  carrier_duty_percent: 50%
+
+climate:
+  - platform: coolix
+    id: test_coolix
+    name: Coolix
+    supports_heat: false
+    supports_cool: false

tests/component_tests/climate_ir/test_supports_heat_cool.py

@@ -0,0 +1,53 @@
+"""Tests for the supports_heat_cool default resolved in climate_ir code generation."""
+
+from __future__ import annotations
+
+from collections.abc import Callable
+from pathlib import Path
+import re
+
+import pytest
+
+
+def _emitted_value(main_cpp: str) -> str | None:
+    """Return the argument of the generated set_supports_heat_cool() call, or None if absent."""
+    match = re.search(r"set_supports_heat_cool\((true|false)\)", main_cpp)
+    return match.group(1) if match else None
+
+
+@pytest.mark.parametrize(
+    ("config", "expected"),
+    [
+        ("heat_and_cool.yaml", None),
+        ("cool_only.yaml", "false"),
+        ("heat_only.yaml", "false"),
+        ("neither.yaml", "false"),
+    ],
+)
+def test_default_requires_heat_and_cool(
+    config: str,
+    expected: str | None,
+    generate_main: Callable[[str | Path], str],
+    component_config_path: Callable[[str], Path],
+) -> None:
+    """Without the key, HEAT_COOL follows supports_heat and supports_cool."""
+    main_cpp = generate_main(component_config_path(config))
+    assert _emitted_value(main_cpp) == expected
+
+
+@pytest.mark.parametrize(
+    ("config", "expected"),
+    [
+        ("cool_only_override_on.yaml", None),
+        ("heat_and_cool_override_off.yaml", "false"),
+    ],
+)
+def test_explicit_key_overrides_default(
+    config: str,
+    expected: str | None,
+    generate_main: Callable[[str | Path], str],
+    component_config_path: Callable[[str], Path],
+) -> None:
+    """A cool-only unit can still offer HEAT_COOL, and a heat+cool unit can drop it."""
+    main_cpp = generate_main(component_config_path(config))
+    assert _emitted_value(main_cpp) == expected

tests/components/climate_ir/__init__.py

@@ -0,0 +1,6 @@
+from tests.testing_helpers import ComponentManifestOverride
+
+
+def override_manifest(manifest: ComponentManifestOverride) -> None:
+    # ClimateIR derives from climate::Climate without declaring it as a dependency.
+    manifest.dependencies = manifest.dependencies + ["climate"]

tests/components/climate_ir/climate_ir_test.cpp

@@ -0,0 +1,56 @@
+#include <gtest/gtest.h>
+#include "esphome/components/climate_ir/climate_ir.h"
+
+namespace esphome::climate_ir::testing {
+
+class TestClimateIR : public ClimateIR {
+ public:
+  TestClimateIR() : ClimateIR(16.0f, 30.0f) {}
+
+  using ClimateIR::traits;
+
+ protected:
+  void transmit_state() override {}
+};
+
+// The HEAT_COOL default is covered in tests/component_tests/climate_ir.
+
+TEST(ClimateIRTest, HeatCoolAdvertisedWhenSupported) {
+  TestClimateIR climate;
+  climate.set_supports_heat(true);
+  climate.set_supports_cool(true);
+  climate.set_supports_heat_cool(true);
+  EXPECT_TRUE(climate.traits().supports_mode(climate::CLIMATE_MODE_HEAT_COOL));
+}
+
+TEST(ClimateIRTest, HeatCoolNotAdvertisedWhenUnsupported) {
+  TestClimateIR climate;
+  climate.set_supports_heat(true);
+  climate.set_supports_cool(true);
+  climate.set_supports_heat_cool(false);
+  EXPECT_FALSE(climate.traits().supports_mode(climate::CLIMATE_MODE_HEAT_COOL));
+}
+
+TEST(ClimateIRTest, HeatCoolAdvertisedForCoolOnlyDeviceThatSupportsIt) {
+  TestClimateIR climate;
+  climate.set_supports_heat(false);
+  climate.set_supports_cool(true);
+  climate.set_supports_heat_cool(true);
+  auto traits = climate.traits();
+  EXPECT_TRUE(traits.supports_mode(climate::CLIMATE_MODE_HEAT_COOL));
+  EXPECT_FALSE(traits.supports_mode(climate::CLIMATE_MODE_HEAT));
+}
+
+TEST(ClimateIRTest, HeatAndCoolModesFollowTheirOwnFlags) {
+  TestClimateIR climate;
+  climate.set_supports_heat(false);
+  climate.set_supports_cool(true);
+  climate.set_supports_heat_cool(false);
+  auto traits = climate.traits();
+  EXPECT_TRUE(traits.supports_mode(climate::CLIMATE_MODE_COOL));
+  EXPECT_FALSE(traits.supports_mode(climate::CLIMATE_MODE_HEAT));
+  EXPECT_FALSE(traits.supports_mode(climate::CLIMATE_MODE_HEAT_COOL));
+  EXPECT_TRUE(traits.supports_mode(climate::CLIMATE_MODE_OFF));
+}
+
+}  // namespace esphome::climate_ir::testing

tests/components/gree/validate-cool-only-heat-cool.esp32-idf.yaml

@@ -0,0 +1,10 @@
+packages:
+  remote_transmitter: !include ../../test_build_components/common/remote_transmitter/esp32-idf.yaml
+
+climate:
+  - platform: gree
+    name: GREE
+    transmitter_id: xmitr
+    model: YAN
+    supports_heat: false
+    supports_heat_cool: true

tests/components/gree/validate-no-cool.esp32-idf.yaml

@@ -0,0 +1,9 @@
+packages:
+  remote_transmitter: !include ../../test_build_components/common/remote_transmitter/esp32-idf.yaml
+
+climate:
+  - platform: gree
+    name: GREE
+    transmitter_id: xmitr
+    model: YAN
+    supports_cool: false

tests/components/gree/validate-no-heat-cool.esp32-idf.yaml

@@ -0,0 +1,9 @@
+packages:
+  remote_transmitter: !include ../../test_build_components/common/remote_transmitter/esp32-idf.yaml
+
+climate:
+  - platform: gree
+    name: GREE
+    transmitter_id: xmitr
+    model: YAN
+    supports_heat_cool: false

tests/components/gree/validate-no-heat.esp32-idf.yaml

@@ -0,0 +1,9 @@
+packages:
+  remote_transmitter: !include ../../test_build_components/common/remote_transmitter/esp32-idf.yaml
+
+climate:
+  - platform: gree
+    name: GREE
+    transmitter_id: xmitr
+    model: YAN
+    supports_heat: false