mirror of
https://github.com/ChuckBuilds/LEDMatrix.git
synced 2026-08-01 16:58:06 +00:00
ChuckandGitHub
2ffc57cf40
The safety harness's VisualTestDisplayManager (base of BoundsCheckingDisplayManager) doesn't implement process_deferred_updates(), which 5 first-party ledmatrix-plugins call unconditionally between set_scrolling_state() and their scroll-position update: news, odds-ticker, ledmatrix-leaderboard, stock-news, and ledmatrix-stocks. Any of them fails the harness with AttributeError the moment it's touched (surfaced when ledmatrix-plugins#177 had to add a local hasattr guard in ledmatrix-stocks just to pass CI). Add the method as a no-op, mirroring the existing "no-op for testing" pattern already used for set_scrolling_state, so these plugins render under the harness without every touching PR needing its own guard.
238 lines
9.1 KiB
Python
238 lines
9.1 KiB
Python
"""
|
|
Tests for VisualTestDisplayManager.
|
|
|
|
Verifies that the visual display manager actually renders pixels,
|
|
loads fonts, and can save snapshots.
|
|
"""
|
|
|
|
from PIL import Image
|
|
|
|
from src.plugin_system.testing import VisualTestDisplayManager
|
|
|
|
|
|
class TestVisualDisplayManager:
|
|
"""Test VisualTestDisplayManager pixel rendering."""
|
|
|
|
def test_creates_image_with_correct_dimensions(self):
|
|
vdm = VisualTestDisplayManager(width=128, height=32)
|
|
assert vdm.image.size == (128, 32)
|
|
|
|
def test_creates_image_custom_dimensions(self):
|
|
vdm = VisualTestDisplayManager(width=64, height=64)
|
|
assert vdm.image.size == (64, 64)
|
|
assert vdm.width == 64
|
|
assert vdm.height == 64
|
|
|
|
def test_draw_text_renders_pixels(self):
|
|
vdm = VisualTestDisplayManager(width=128, height=32)
|
|
vdm.draw_text("Hello", x=0, y=0, color=(255, 255, 255))
|
|
pixels = list(vdm.image.getdata())
|
|
non_black = [p for p in pixels if p != (0, 0, 0)]
|
|
assert len(non_black) > 0, "draw_text should render actual pixels"
|
|
|
|
def test_draw_text_centered(self):
|
|
vdm = VisualTestDisplayManager(width=128, height=32)
|
|
vdm.draw_text("Test", color=(255, 0, 0)) # x=None centers text
|
|
pixels = list(vdm.image.getdata())
|
|
non_black = [p for p in pixels if p != (0, 0, 0)]
|
|
assert len(non_black) > 0
|
|
|
|
def test_draw_text_with_centered_flag(self):
|
|
vdm = VisualTestDisplayManager(width=128, height=32)
|
|
vdm.draw_text("X", x=64, y=10, centered=True, color=(0, 255, 0))
|
|
pixels = list(vdm.image.getdata())
|
|
non_black = [p for p in pixels if p != (0, 0, 0)]
|
|
assert len(non_black) > 0
|
|
|
|
def test_draw_text_tracks_calls(self):
|
|
vdm = VisualTestDisplayManager(width=128, height=32)
|
|
vdm.draw_text("Hello", x=10, y=5, color=(255, 0, 0))
|
|
assert len(vdm.draw_calls) == 1
|
|
assert vdm.draw_calls[0]['type'] == 'text'
|
|
assert vdm.draw_calls[0]['text'] == 'Hello'
|
|
assert vdm.draw_calls[0]['x'] == 10
|
|
assert vdm.draw_calls[0]['y'] == 5
|
|
|
|
def test_clear_resets_canvas(self):
|
|
vdm = VisualTestDisplayManager(width=128, height=32)
|
|
vdm.draw_text("Hello", x=0, y=0, color=(255, 255, 255))
|
|
vdm.clear()
|
|
pixels = list(vdm.image.getdata())
|
|
non_black = [p for p in pixels if p != (0, 0, 0)]
|
|
assert len(non_black) == 0, "clear() should reset all pixels to black"
|
|
assert vdm.clear_called is True
|
|
|
|
def test_update_display_sets_flag(self):
|
|
vdm = VisualTestDisplayManager(width=128, height=32)
|
|
assert vdm.update_called is False
|
|
vdm.update_display()
|
|
assert vdm.update_called is True
|
|
|
|
def test_matrix_proxy(self):
|
|
vdm = VisualTestDisplayManager(width=128, height=32)
|
|
assert vdm.matrix.width == 128
|
|
assert vdm.matrix.height == 32
|
|
|
|
def test_width_height_properties(self):
|
|
vdm = VisualTestDisplayManager(width=64, height=32)
|
|
assert vdm.width == 64
|
|
assert vdm.height == 32
|
|
assert vdm.display_width == 64
|
|
assert vdm.display_height == 32
|
|
|
|
def test_save_snapshot(self, tmp_path):
|
|
vdm = VisualTestDisplayManager(width=128, height=32)
|
|
vdm.draw_text("Test", x=10, y=10, color=(255, 0, 0))
|
|
output = tmp_path / "test_render.png"
|
|
vdm.save_snapshot(str(output))
|
|
assert output.exists()
|
|
with Image.open(str(output)) as saved_img:
|
|
assert saved_img.size == (128, 32)
|
|
|
|
def test_get_image(self):
|
|
vdm = VisualTestDisplayManager(width=128, height=32)
|
|
img = vdm.get_image()
|
|
assert isinstance(img, Image.Image)
|
|
assert img.size == (128, 32)
|
|
|
|
def test_get_image_base64(self):
|
|
vdm = VisualTestDisplayManager(width=128, height=32)
|
|
vdm.draw_text("Hi", x=0, y=0, color=(255, 255, 255))
|
|
b64 = vdm.get_image_base64()
|
|
assert isinstance(b64, str)
|
|
assert len(b64) > 0
|
|
# Should be valid base64 PNG
|
|
import base64
|
|
decoded = base64.b64decode(b64)
|
|
assert decoded[:4] == b'\x89PNG'
|
|
|
|
def test_font_attributes_exist(self):
|
|
vdm = VisualTestDisplayManager(width=128, height=32)
|
|
assert hasattr(vdm, 'regular_font')
|
|
assert hasattr(vdm, 'small_font')
|
|
assert hasattr(vdm, 'extra_small_font')
|
|
assert hasattr(vdm, 'calendar_font')
|
|
assert hasattr(vdm, 'bdf_5x7_font')
|
|
assert hasattr(vdm, 'font')
|
|
|
|
def test_get_text_width(self):
|
|
vdm = VisualTestDisplayManager(width=128, height=32)
|
|
w = vdm.get_text_width("Hello", vdm.regular_font)
|
|
assert isinstance(w, int)
|
|
assert w > 0
|
|
|
|
def test_get_font_height(self):
|
|
vdm = VisualTestDisplayManager(width=128, height=32)
|
|
h = vdm.get_font_height(vdm.regular_font)
|
|
assert isinstance(h, int)
|
|
assert h > 0
|
|
|
|
def test_image_paste(self):
|
|
"""Verify plugins can paste images onto the display."""
|
|
vdm = VisualTestDisplayManager(width=128, height=32)
|
|
overlay = Image.new('RGB', (10, 10), (255, 0, 0))
|
|
vdm.image.paste(overlay, (0, 0))
|
|
pixel = vdm.image.getpixel((5, 5))
|
|
assert pixel == (255, 0, 0)
|
|
|
|
def test_image_assignment(self):
|
|
"""Verify plugins can assign a new image to display_manager.image."""
|
|
vdm = VisualTestDisplayManager(width=128, height=32)
|
|
new_img = Image.new('RGB', (128, 32), (0, 255, 0))
|
|
vdm.image = new_img
|
|
assert vdm.image.getpixel((0, 0)) == (0, 255, 0)
|
|
|
|
def test_draw_image(self):
|
|
vdm = VisualTestDisplayManager(width=128, height=32)
|
|
overlay = Image.new('RGB', (10, 10), (0, 0, 255))
|
|
vdm.draw_image(overlay, 5, 5)
|
|
assert len(vdm.draw_calls) == 1
|
|
assert vdm.draw_calls[0]['type'] == 'image'
|
|
# Verify pixels were actually pasted
|
|
pixel = vdm.image.getpixel((7, 7))
|
|
assert pixel == (0, 0, 255)
|
|
|
|
def test_reset(self):
|
|
vdm = VisualTestDisplayManager(width=128, height=32)
|
|
vdm.draw_text("Hi", x=0, y=0)
|
|
vdm.clear()
|
|
vdm.update_display()
|
|
vdm.reset()
|
|
assert vdm.clear_called is False
|
|
assert vdm.update_called is False
|
|
assert len(vdm.draw_calls) == 0
|
|
pixels = list(vdm.image.getdata())
|
|
assert all(p == (0, 0, 0) for p in pixels)
|
|
|
|
def test_scrolling_state(self):
|
|
vdm = VisualTestDisplayManager(width=128, height=32)
|
|
assert vdm.is_currently_scrolling() is False
|
|
vdm.set_scrolling_state(True)
|
|
assert vdm.is_currently_scrolling() is True
|
|
vdm.set_scrolling_state(False)
|
|
assert vdm.is_currently_scrolling() is False
|
|
|
|
def test_process_deferred_updates_is_noop(self):
|
|
# Ticker-style plugins (news, odds-ticker, leaderboard, stock-news,
|
|
# stocks) call this unconditionally alongside set_scrolling_state();
|
|
# it must exist and be harmless so those plugins render under the
|
|
# harness instead of raising AttributeError.
|
|
vdm = VisualTestDisplayManager(width=128, height=32)
|
|
vdm.set_scrolling_state(True)
|
|
vdm.process_deferred_updates() # should not raise
|
|
assert vdm.is_currently_scrolling() is True
|
|
|
|
def test_format_date_with_ordinal(self):
|
|
from datetime import datetime
|
|
vdm = VisualTestDisplayManager(width=128, height=32)
|
|
dt = datetime(2025, 8, 1)
|
|
result = vdm.format_date_with_ordinal(dt)
|
|
assert '1st' in result
|
|
dt = datetime(2025, 8, 3)
|
|
result = vdm.format_date_with_ordinal(dt)
|
|
assert '3rd' in result
|
|
dt = datetime(2025, 8, 11)
|
|
result = vdm.format_date_with_ordinal(dt)
|
|
assert '11th' in result
|
|
|
|
|
|
class TestWeatherDrawing:
|
|
"""Test weather icon rendering."""
|
|
|
|
def test_draw_sun(self):
|
|
vdm = VisualTestDisplayManager(width=128, height=32)
|
|
vdm.draw_sun(0, 0, 16)
|
|
pixels = list(vdm.image.getdata())
|
|
non_black = [p for p in pixels if p != (0, 0, 0)]
|
|
assert len(non_black) > 0
|
|
|
|
def test_draw_cloud(self):
|
|
vdm = VisualTestDisplayManager(width=128, height=32)
|
|
vdm.draw_cloud(0, 0, 16)
|
|
pixels = list(vdm.image.getdata())
|
|
non_black = [p for p in pixels if p != (0, 0, 0)]
|
|
assert len(non_black) > 0
|
|
|
|
def test_draw_rain(self):
|
|
vdm = VisualTestDisplayManager(width=128, height=32)
|
|
vdm.draw_rain(0, 0, 16)
|
|
pixels = list(vdm.image.getdata())
|
|
non_black = [p for p in pixels if p != (0, 0, 0)]
|
|
assert len(non_black) > 0
|
|
|
|
def test_draw_snow(self):
|
|
vdm = VisualTestDisplayManager(width=128, height=32)
|
|
vdm.draw_snow(0, 0, 16)
|
|
pixels = list(vdm.image.getdata())
|
|
non_black = [p for p in pixels if p != (0, 0, 0)]
|
|
assert len(non_black) > 0
|
|
|
|
def test_draw_weather_icon_dispatches(self):
|
|
vdm = VisualTestDisplayManager(width=128, height=32)
|
|
for condition in ['clear', 'cloudy', 'rain', 'snow', 'storm', 'unknown']:
|
|
vdm.clear()
|
|
vdm.draw_weather_icon(condition, 0, 0, 16)
|
|
pixels = list(vdm.image.getdata())
|
|
non_black = [p for p in pixels if p != (0, 0, 0)]
|
|
assert len(non_black) > 0, f"draw_weather_icon('{condition}') should render pixels"
|