Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Extra unit tests #1842

Merged
merged 2 commits into from
Jan 5, 2025
Merged

Extra unit tests #1842

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
147fb32
Extra unit tests
springfall2008 Jan 4, 2025
5a42f30
[pre-commit.ci lite] apply automatic fixes
pre-commit-ci-lite[bot] Jan 4, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
192 changes: 191 additions & 1 deletion apps/predbat/unit_test.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -885,7 +885,6 @@ def test_inverter_update(

dummy_rest = DummyRestAPI()
my_predbat.args["givtcp_rest"] = "dummy"
inv.sleep = dummy_sleep

dummy_rest.rest_data = {}
dummy_rest.rest_data["Control"] = {}
Expand All @@ -910,6 +909,7 @@ def test_inverter_update(
dummy_items["sensor.soc_kw"] = -1

inv = Inverter(my_predbat, 0, rest_postCommand=dummy_rest.dummy_rest_postCommand, rest_getData=dummy_rest.dummy_rest_getData)
inv.sleep = dummy_sleep

print("Test: Update Inverter - REST")
inv.update_status(my_predbat.minutes_now)
Expand Down Expand Up @@ -948,6 +948,31 @@ def test_inverter_update(
return failed


def test_auto_restart(test_name, my_predbat, ha, inv, dummy_items, service, expected, active=False):
print("**** Running Test: {} ****".format(test_name))
failed = 0
ha.service_store_enable = True
ha.service_store = []
my_predbat.restart_active = active

my_predbat.args["auto_restart"] = service

failed = 1 if not active else 0
try:
inv.auto_restart("Crashed")
except Exception as e:
failed = 0
if str(e) != "Auto-restart triggered":
print("ERROR: Auto-restart should be triggered got {}".format(e))
failed = 1

result = ha.get_service_store()
if json.dumps(expected) != json.dumps(result):
print("ERROR: Auto-restart service should be {} got {}".format(expected, result))
failed = 1
return failed


def test_call_adjust_charge_immediate(test_name, my_predbat, ha, inv, dummy_items, soc, repeat=False, freeze=False, clear=False, stop_discharge=False, charge_start_time="00:00:00", charge_end_time="23:55:00", no_freeze=False):
"""
Tests;
Expand Down Expand Up @@ -1160,6 +1185,107 @@ def run_car_charging_smart_tests(my_predbat):
return failed


def test_inverter_self_test(test_name, my_predbat):
failed = 0

print("**** Running Test: {} ****".format(test_name))
# Call self test - doesn't really check much as such except the code isn't dead
dummy_rest = DummyRestAPI()
my_predbat.args["givtcp_rest"] = "dummy"

dummy_rest.rest_data = {}
dummy_rest.rest_data["Control"] = {}
dummy_rest.rest_data["Control"]["Target_SOC"] = 99
dummy_rest.rest_data["Control"]["Mode"] = "Eco"
dummy_rest.rest_data["Control"]["Battery_Power_Reserve"] = 4.0
dummy_rest.rest_data["Control"]["Battery_Charge_Rate"] = 1100
dummy_rest.rest_data["Control"]["Battery_Discharge_Rate"] = 1500
dummy_rest.rest_data["Control"]["Enable_Charge_Schedule"] = "enable"
dummy_rest.rest_data["Control"]["Enable_Discharge_Schedule"] = "enable"
dummy_rest.rest_data["Timeslots"] = {}
dummy_rest.rest_data["Timeslots"]["Charge_start_time_slot_1"] = "00:30:00"
dummy_rest.rest_data["Timeslots"]["Charge_end_time_slot_1"] = "22:00:00"
dummy_rest.rest_data["Timeslots"]["Discharge_start_time_slot_1"] = "01:00:00"
dummy_rest.rest_data["Timeslots"]["Discharge_end_time_slot_1"] = "02:30:00"
dummy_rest.rest_data["Power"] = {}
dummy_rest.rest_data["Power"]["Power"] = {}
dummy_rest.rest_data["Power"]["Power"]["SOC_kWh"] = 1.0
dummy_rest.rest_data["Power"]["Power"]["Battery_Power"] = 100
dummy_rest.rest_data["Power"]["Power"]["PV_Power"] = 200
dummy_rest.rest_data["Power"]["Power"]["Load_Power"] = 300

inv = Inverter(my_predbat, 0, rest_postCommand=dummy_rest.dummy_rest_postCommand, rest_getData=dummy_rest.dummy_rest_getData)
inv.sleep = dummy_sleep
inv.self_test(my_predbat.minutes_now)
rest = dummy_rest.get_commands()
expected = [
["dummy/setChargeTarget", {"chargeToPercent": 100}],
["dummy/setChargeTarget", {"chargeToPercent": 100}],
["dummy/setChargeTarget", {"chargeToPercent": 100}],
["dummy/setChargeTarget", {"chargeToPercent": 100}],
["dummy/setChargeTarget", {"chargeToPercent": 100}],
["dummy/setChargeRate", {"chargeRate": 215}],
["dummy/setChargeRate", {"chargeRate": 215}],
["dummy/setChargeRate", {"chargeRate": 215}],
["dummy/setChargeRate", {"chargeRate": 215}],
["dummy/setChargeRate", {"chargeRate": 215}],
["dummy/setChargeRate", {"chargeRate": 0}],
["dummy/setChargeRate", {"chargeRate": 0}],
["dummy/setChargeRate", {"chargeRate": 0}],
["dummy/setChargeRate", {"chargeRate": 0}],
["dummy/setChargeRate", {"chargeRate": 0}],
["dummy/setDischargeRate", {"dischargeRate": 220}],
["dummy/setDischargeRate", {"dischargeRate": 220}],
["dummy/setDischargeRate", {"dischargeRate": 220}],
["dummy/setDischargeRate", {"dischargeRate": 220}],
["dummy/setDischargeRate", {"dischargeRate": 220}],
["dummy/setDischargeRate", {"dischargeRate": 0}],
["dummy/setDischargeRate", {"dischargeRate": 0}],
["dummy/setDischargeRate", {"dischargeRate": 0}],
["dummy/setDischargeRate", {"dischargeRate": 0}],
["dummy/setDischargeRate", {"dischargeRate": 0}],
["dummy/setBatteryReserve", {"reservePercent": 100}],
["dummy/setBatteryReserve", {"reservePercent": 100}],
["dummy/setBatteryReserve", {"reservePercent": 100}],
["dummy/setBatteryReserve", {"reservePercent": 100}],
["dummy/setBatteryReserve", {"reservePercent": 100}],
["dummy/setBatteryReserve", {"reservePercent": 6}],
["dummy/setBatteryReserve", {"reservePercent": 6}],
["dummy/setBatteryReserve", {"reservePercent": 6}],
["dummy/setBatteryReserve", {"reservePercent": 6}],
["dummy/setBatteryReserve", {"reservePercent": 6}],
["dummy/enableChargeSchedule", {"state": "disable"}],
["dummy/enableChargeSchedule", {"state": "disable"}],
["dummy/enableChargeSchedule", {"state": "disable"}],
["dummy/enableChargeSchedule", {"state": "disable"}],
["dummy/enableChargeSchedule", {"state": "disable"}],
["dummy/setChargeSlot1", {"start": "23:01", "finish": "05:01"}],
["dummy/setChargeSlot1", {"start": "23:01", "finish": "05:01"}],
["dummy/setChargeSlot1", {"start": "23:01", "finish": "05:01"}],
["dummy/setChargeSlot1", {"start": "23:01", "finish": "05:01"}],
["dummy/setChargeSlot1", {"start": "23:01", "finish": "05:01"}],
["dummy/setChargeSlot1", {"start": "23:00", "finish": "05:00"}],
["dummy/setChargeSlot1", {"start": "23:00", "finish": "05:00"}],
["dummy/setChargeSlot1", {"start": "23:00", "finish": "05:00"}],
["dummy/setChargeSlot1", {"start": "23:00", "finish": "05:00"}],
["dummy/setChargeSlot1", {"start": "23:00", "finish": "05:00"}],
["dummy/setDischargeSlot1", {"start": "23:00", "finish": "23:01"}],
["dummy/setDischargeSlot1", {"start": "23:00", "finish": "23:01"}],
["dummy/setDischargeSlot1", {"start": "23:00", "finish": "23:01"}],
["dummy/setDischargeSlot1", {"start": "23:00", "finish": "23:01"}],
["dummy/setDischargeSlot1", {"start": "23:00", "finish": "23:01"}],
["dummy/setBatteryMode", {"mode": "Timed Export"}],
["dummy/setBatteryMode", {"mode": "Timed Export"}],
["dummy/setBatteryMode", {"mode": "Timed Export"}],
["dummy/setBatteryMode", {"mode": "Timed Export"}],
["dummy/setBatteryMode", {"mode": "Timed Export"}],
]
if json.dumps(expected) != json.dumps(rest):
print("ERROR: Self test should be {} got {}".format(expected, rest))
failed = True
return failed


def run_inverter_tests():
"""
Test the inverter functions
Expand Down Expand Up @@ -1374,6 +1500,8 @@ def run_inverter_tests():
failed |= test_call_adjust_charge_immediate("charge_immediate6", my_predbat, ha, inv, dummy_items, 49, charge_start_time="00:00:00", charge_end_time="11:00:00")
failed |= test_call_adjust_charge_immediate("charge_immediate7", my_predbat, ha, inv, dummy_items, 50, freeze=True)
failed |= test_call_adjust_charge_immediate("charge_immediate8", my_predbat, ha, inv, dummy_items, 50, freeze=False, no_freeze=True)
if failed:
return failed

failed |= test_call_adjust_export_immediate("export_immediate1", my_predbat, ha, inv, dummy_items, 100, repeat=True)
failed |= test_call_adjust_export_immediate("export_immediate3", my_predbat, ha, inv, dummy_items, 0, repeat=True)
Expand All @@ -1383,7 +1511,69 @@ def run_inverter_tests():
failed |= test_call_adjust_export_immediate("export_immediate6", my_predbat, ha, inv, dummy_items, 49, discharge_start_time="00:00:00", discharge_end_time="09:00:00")
failed |= test_call_adjust_export_immediate("export_immediate7", my_predbat, ha, inv, dummy_items, 50, freeze=True)
failed |= test_call_adjust_export_immediate("export_immediate8", my_predbat, ha, inv, dummy_items, 50, freeze=False, no_freeze=True)
if failed:
return failed

failed |= test_auto_restart(
"auto_restart1",
my_predbat,
ha,
inv,
dummy_items,
service={"command": "service", "service": "restart_service", "addon": "adds"},
expected=[["restart_service", {"addon": "adds"}], ["notify/notify", {"message": "Auto-restart service restart_service called due to: Crashed"}]],
)
if failed:
return failed

failed |= test_auto_restart(
"auto_restart2", my_predbat, ha, inv, dummy_items, service={"command": "service", "service": "restart_service"}, expected=[["restart_service", {}], ["notify/notify", {"message": "Auto-restart service restart_service called due to: Crashed"}]]
)
if failed:
return failed

failed |= test_auto_restart(
"auto_restart3",
my_predbat,
ha,
inv,
dummy_items,
service={"command": "service", "service": "restart_service"},
expected=[],
active=True,
)
if failed:
return failed

failed |= test_auto_restart(
"auto_restart4",
my_predbat,
ha,
inv,
dummy_items,
service={"command": "service", "service": "restart_service", "entity_id": "switch.restart"},
expected=[["restart_service", {"entity_id": "switch.restart"}], ["notify/notify", {"message": "Auto-restart service restart_service called due to: Crashed"}]],
)
if failed:
return failed

os.system("touch tmp1234")
failed |= test_auto_restart(
"auto_restart5",
my_predbat,
ha,
inv,
dummy_items,
service={"command": "service", "shell": "rm tmp1234"},
expected=[],
)
if failed:
return failed
if os.path.exists("tmp1234"):
print("ERROR: File should be deleted")
failed = True

failed |= test_inverter_self_test("self_test1", my_predbat)
return failed


Expand Down
Loading