"""app.widgets.battery -- unit-level, no HTTP: constructs Widget/ BatteryWidgetConfig rows directly and sets Frame.battery_percent/ battery_as_of straight on the Frame row, same as routers/device.py's frame_battery would.""" from __future__ import annotations import time from app import widgets from app.models import BatteryWidgetConfig, Frame, Widget def _make_widget(db_session, mode="detailed") -> tuple[Frame, Widget]: frame = db_session.get(Frame, 1) widget = Widget(frame_id=frame.id, widget_type="battery", x=0, y=0, w=1, h=1, sort_order=0, created_at=time.time()) db_session.add(widget) db_session.flush() db_session.add(BatteryWidgetConfig(widget_id=widget.id, mode=mode)) db_session.commit() return frame, widget def test_render_shows_a_placeholder_when_frame_has_never_reported(db_session): frame, widget = _make_widget(db_session) assert frame.battery_percent == -1 img = widgets.battery.render(db_session, frame, widget, 300, 200) assert img.size == (300, 200) assert img.mode == "RGB" def test_render_shows_the_reported_percent(db_session): frame, widget = _make_widget(db_session) frame.battery_percent = 42 frame.battery_as_of = time.time() db_session.commit() img = widgets.battery.render(db_session, frame, widget, 300, 200) assert img.size == (300, 200) assert img.mode == "RGB" def test_render_compact_mode_omits_the_estimate_lines(db_session): frame, widget = _make_widget(db_session, mode="compact") frame.battery_percent = 80 frame.battery_as_of = time.time() db_session.commit() img = widgets.battery.render(db_session, frame, widget, 300, 200) assert img.size == (300, 200) def test_render_at_minimum_grid_footprint(db_session): """grid.MIN_FOOTPRINT["battery"] is (1, 1) cells -- on an 8x5 grid against a full 800x480 panel that's a 100x96 box, the smallest a battery widget can actually be placed at.""" frame, widget = _make_widget(db_session) frame.battery_percent = 15 db_session.commit() img = widgets.battery.render(db_session, frame, widget, 100, 96) assert img.size == (100, 96) def test_render_falls_back_to_detailed_mode_with_no_config_row(db_session): """widget_id has no BatteryWidgetConfig row at all -- render() must not raise, matching every other widget type's "never raises for a foreseeable failure" contract.""" frame = db_session.get(Frame, 1) widget = Widget(frame_id=frame.id, widget_type="battery", x=0, y=0, w=1, h=1, sort_order=0, created_at=time.time()) db_session.add(widget) db_session.flush() db_session.commit() frame.battery_percent = 60 db_session.commit() img = widgets.battery.render(db_session, frame, widget, 100, 96) assert img.size == (100, 96) def test_no_button_actions(): """A number the device itself pushes on every wake -- nothing to advance/back/check.""" assert widgets.battery.ACTIONS == {} assert widgets.battery.ACTION_LABELS == {}