|
| 1 | +# ----------------------------------------------------------------------------- |
| 2 | +# Predbat Home Battery System |
| 3 | +# Copyright Trefor Southwell 2026 - All Rights Reserved |
| 4 | +# This application maybe used for personal use only and not for commercial use |
| 5 | +# ----------------------------------------------------------------------------- |
| 6 | +"""Exercise the shared YAML through real native inverter and executor paths, without hardware.""" |
| 7 | + |
| 8 | +import copy |
| 9 | +from datetime import timedelta |
| 10 | +from pathlib import Path |
| 11 | +from unittest.mock import patch |
| 12 | + |
| 13 | +import yaml |
| 14 | + |
| 15 | +from config import INVERTER_DEF |
| 16 | +from inverter import Inverter |
| 17 | +from tests.test_infra import reset_inverter |
| 18 | + |
| 19 | + |
| 20 | +def run_deye_solarman_template_tests(my_predbat): |
| 21 | + """Validate six-row charge/demand, failure barriers and the absence of unrelated writes.""" |
| 22 | + template_path = Path(__file__).resolve().parents[3] / "templates" / "deye_solarman.yaml" |
| 23 | + with template_path.open() as stream: |
| 24 | + template = yaml.safe_load(stream)["pred_bat"] |
| 25 | + saved_definition = copy.deepcopy(INVERTER_DEF.get("DeyeSolarman")) |
| 26 | + saved_call = my_predbat.call_service_wrapper |
| 27 | + writes = [] |
| 28 | + rejected = set() |
| 29 | + failed = False |
| 30 | + ha = my_predbat.ha_interface |
| 31 | + sources = ["select.test_deye_program_{}_charging".format(row) for row in range(1, 7)] |
| 32 | + targets = ["number.test_deye_program_{}_soc".format(row) for row in range(1, 7)] |
| 33 | + protected = {"time.test_deye_program_{}_time".format(row): "{:02d}:00:00".format((row - 1) * 4) for row in range(1, 7)} |
| 34 | + protected.update({"number.test_deye_program_{}_power".format(row): 3000 for row in range(1, 7)}) |
| 35 | + protected.update( |
| 36 | + { |
| 37 | + "select.test_deye_work_mode": "Zero Export To CT", |
| 38 | + "select.test_deye_time_of_use": "Week", |
| 39 | + "number.test_deye_battery_max_charging_current": 59, |
| 40 | + "number.test_deye_battery_max_discharging_current": 160, |
| 41 | + "number.test_deye_battery_shutdown_soc": 5, |
| 42 | + } |
| 43 | + ) |
| 44 | + |
| 45 | + def check(condition, message): |
| 46 | + """Collect assertions so one failing case does not hide the remaining evidence.""" |
| 47 | + nonlocal failed |
| 48 | + if not condition: |
| 49 | + print("ERROR: Deye/Solarman template: " + message) |
| 50 | + failed = True |
| 51 | + |
| 52 | + def service_call(service, **data): |
| 53 | + """Model HA's grouped state-setting services, retaining entity attributes.""" |
| 54 | + entities = data.get("entity_id", "").split(",") |
| 55 | + writes.append((service, copy.deepcopy(data))) |
| 56 | + if rejected.intersection(entities): |
| 57 | + return False |
| 58 | + for entity in entities: |
| 59 | + check(entity in sources + targets, "unexpected physical write to " + entity) |
| 60 | + if entity in sources + targets: |
| 61 | + ha.dummy_items[entity]["state"] = data.get("option", data.get("value")) |
| 62 | + return True |
| 63 | + |
| 64 | + def rows(source, target): |
| 65 | + """Check all six observations, never just the first row.""" |
| 66 | + return all(ha.get_state(entity) == source for entity in sources) and all(float(ha.get_state(entity)) == target for entity in targets) |
| 67 | + |
| 68 | + try: |
| 69 | + reset_inverter(my_predbat) |
| 70 | + my_predbat.args.clear() |
| 71 | + my_predbat.args.update(template) |
| 72 | + my_predbat.args["deye_entity_prefix"] = "test_deye" |
| 73 | + my_predbat.args_from_apps_yaml = copy.deepcopy(my_predbat.args) |
| 74 | + my_predbat.last_service_hash = {} |
| 75 | + ha.dummy_items.update(protected) |
| 76 | + for entity in sources: |
| 77 | + ha.dummy_items[entity] = {"state": "Grid", "options": ["Disabled", "Grid", "Generator", "Both"]} |
| 78 | + for entity in targets: |
| 79 | + ha.dummy_items[entity] = {"state": 40, "min": 0, "max": 100, "step": 1, "unit_of_measurement": "%"} |
| 80 | + for suffix, value in (("battery", 30), ("battery_voltage", 51.2), ("battery_temperature", 20), ("battery_power", 0), ("pv_power", 0), ("load_power", 500), ("grid_power", 500)): |
| 81 | + ha.dummy_items["sensor.test_deye_" + suffix] = value |
| 82 | + my_predbat.call_service_wrapper = service_call |
| 83 | + inv = Inverter(my_predbat, 0, quiet=True) |
| 84 | + inv.sleep = lambda seconds: None |
| 85 | + inv.inv_write_and_poll_sleep = 0 |
| 86 | + inv.update_status(my_predbat.minutes_now) |
| 87 | + my_predbat.inverters = [inv] |
| 88 | + check(not writes, "construction must not send hardware services") |
| 89 | + check(template["template"] is True, "shipped example must stay in template mode") |
| 90 | + check(not inv.inv_support_charge_freeze and not inv.inv_support_discharge_freeze and not inv.inv_has_timed_pause, "unsupported freeze/pause must remain disabled") |
| 91 | + check(inv.inv_output_charge_control == "none", "global rate controls must remain unmapped") |
| 92 | + check("discharge_start_service" not in template, "forced export must be absent") |
| 93 | + |
| 94 | + # Both window bookkeeping and rate/target updates are private sensors, not ToU rows. |
| 95 | + inv.adjust_charge_window(my_predbat.midnight_utc + timedelta(hours=23), my_predbat.midnight_utc + timedelta(hours=25), 23 * 60) |
| 96 | + inv.adjust_charge_rate(1000) |
| 97 | + inv.adjust_discharge_rate(0) |
| 98 | + inv.adjust_battery_target(80, isCharging=True) |
| 99 | + check(not writes, "native bookkeeping must not write physical time, current or SOC controls") |
| 100 | + for key in ("charge_start_time", "charge_end_time", "discharge_start_time", "discharge_end_time"): |
| 101 | + check(my_predbat.get_arg(key, indirect=False, index=0) in inv.created_attributes, "time bookkeeping must use created sensors") |
| 102 | + |
| 103 | + inv.adjust_charge_immediate(80) |
| 104 | + check(rows("Grid", 80), "charge must set all six targets and sources") |
| 105 | + check([data.get("option", data.get("value")) for _, data in writes] == ["Disabled", 80.0, "Grid"], "charge must observe disable → SOC → enable") |
| 106 | + writes.clear() |
| 107 | + inv.adjust_charge_immediate(80) |
| 108 | + inv.adjust_export_immediate(100) |
| 109 | + check(not writes and rows("Grid", 80), "repeated charge and idle export hook must not clobber a charge") |
| 110 | + |
| 111 | + inv.adjust_charge_immediate(0) |
| 112 | + check(rows("Disabled", 10), "demand must disable charging and restore the domestic floor, not shutdown") |
| 113 | + my_predbat.expose_config("set_reserve_min", 25) |
| 114 | + inv.refresh_config() |
| 115 | + inv.adjust_charge_immediate(0) |
| 116 | + check(rows("Disabled", 25), "demand must follow the current manual domestic minimum") |
| 117 | + |
| 118 | + writes.clear() |
| 119 | + rejected.add(targets[-1]) |
| 120 | + inv.adjust_charge_immediate(90) |
| 121 | + check(not any(data.get("option") == "Grid" for _, data in writes), "failed last target must prevent source enable") |
| 122 | + rejected.clear() |
| 123 | + inv.adjust_charge_immediate(90) |
| 124 | + check(rows("Grid", 90), "next cycle must retry a failed sequence") |
| 125 | + |
| 126 | + # Native reset writes 100 to its private target, never to the physical ToU targets. |
| 127 | + my_predbat.set_read_only = False |
| 128 | + my_predbat.set_charge_window = True |
| 129 | + my_predbat.set_export_window = False |
| 130 | + my_predbat.reset_inverter() |
| 131 | + check(rows("Disabled", 25), "native reset must leave demand, not grid-charge to 100") |
| 132 | + |
| 133 | + # Drive an active native plan, including the trailing discharge-stop hook and rates. |
| 134 | + my_predbat.args["template"] = False |
| 135 | + my_predbat.inverter_hybrid = True |
| 136 | + my_predbat.set_soc_enable = True |
| 137 | + my_predbat.set_reserve_enable = False |
| 138 | + my_predbat.set_charge_freeze = False |
| 139 | + my_predbat.set_charge_low_power = False |
| 140 | + my_predbat.set_discharge_during_charge = True |
| 141 | + my_predbat.inverter_needs_reset = False |
| 142 | + my_predbat.fetch_inverter_data() |
| 143 | + inv = my_predbat.inverters[0] |
| 144 | + inv.sleep = lambda seconds: None |
| 145 | + my_predbat.charge_window_best = [{"start": my_predbat.minutes_now, "end": my_predbat.minutes_now + 60, "average": 1}] |
| 146 | + my_predbat.charge_limit_best = [8] |
| 147 | + my_predbat.charge_limit_percent_best = [80] |
| 148 | + my_predbat.execute_plan() |
| 149 | + check(rows("Grid", 80), "native active charge plan must reach all six rows") |
| 150 | + my_predbat.charge_window_best = [] |
| 151 | + my_predbat.charge_limit_best = [] |
| 152 | + my_predbat.execute_plan() |
| 153 | + check(rows("Disabled", 25), "native demand plan must stop grid charging") |
| 154 | + writes.clear() |
| 155 | + my_predbat.set_read_only = True |
| 156 | + my_predbat.execute_plan() |
| 157 | + check(not writes, "read-only native plan must not dispatch services") |
| 158 | + my_predbat.set_read_only = False |
| 159 | + my_predbat.set_charge_window = False |
| 160 | + my_predbat.set_export_window = False |
| 161 | + my_predbat.set_soc_enable = False |
| 162 | + my_predbat.execute_plan() |
| 163 | + check(not writes, "Monitor must not dispatch services") |
| 164 | + check(all(ha.get_state(entity) == value for entity, value in protected.items()), "all unmapped installation settings must stay untouched") |
| 165 | + |
| 166 | + # A transport ValueError is not one of our local diagnostics. Never echo its body. |
| 167 | + with patch.object(my_predbat, "call_service_wrapper", side_effect=ValueError("private-response-marker")), patch.object(inv, "log") as log, patch.object(my_predbat, "record_status") as status: |
| 168 | + check(inv.call_service_template("charge_start_service", {"target_soc": 95}) is True, "a transport exception must remain handled") |
| 169 | + check(status.called, "transport exception must record an error") |
| 170 | + check("private-response-marker" not in str(log.call_args_list) + str(status.call_args_list), "transport response must not leak into diagnostics") |
| 171 | + finally: |
| 172 | + my_predbat.call_service_wrapper = saved_call |
| 173 | + if saved_definition is None: |
| 174 | + INVERTER_DEF.pop("DeyeSolarman", None) |
| 175 | + else: |
| 176 | + INVERTER_DEF["DeyeSolarman"] = saved_definition |
| 177 | + print("Deye/Solarman template tests: {}".format("FAILED" if failed else "PASSED")) |
| 178 | + return failed |
0 commit comments