93 lines
4.6 KiB
Python
93 lines
4.6 KiB
Python
"""Run with: python -m unittest discover -s esphome/bambu-display -v."""
|
|
|
|
import copy
|
|
import unittest
|
|
|
|
import yaml
|
|
|
|
from generate import HERE, generate, validate_printers
|
|
|
|
|
|
class FleetTests(unittest.TestCase):
|
|
def setUp(self):
|
|
self.printers = yaml.safe_load((HERE / 'printers.yaml').read_text())['printers']
|
|
|
|
def test_all_subscriptions_match_repository_entities(self):
|
|
known = {line.split(' | ')[0] for line in
|
|
(HERE.parent / 'ha-entities.txt').read_text().splitlines()}
|
|
package = generate(self.printers)
|
|
subscribed = {entry['entity_id'] for domain in ('sensor', 'text_sensor')
|
|
for entry in package[domain]}
|
|
self.assertFalse(subscribed - known)
|
|
|
|
def test_seven_printers_have_independent_pages_and_metrics(self):
|
|
for index in (6, 7):
|
|
extra = copy.deepcopy(self.printers[0])
|
|
extra.update(id=f'printer_{index}', entity_prefix=f'new_printer_{index}')
|
|
self.printers.append(extra)
|
|
package = generate(self.printers)
|
|
self.assertEqual(len([page for page in package['lvgl']['pages'] if not page.get('skip')]), 7)
|
|
self.assertEqual(len(package['script']), 7)
|
|
subscriptions = {entry['id']: entry['entity_id'] for entry in package['sensor']}
|
|
self.assertEqual(subscriptions['printer_7_progress'], 'sensor.new_printer_7_print_progress')
|
|
for page in package['lvgl']['pages']:
|
|
self.assertIn('lvgl.page.next', page['on_swipe_left'][-1])
|
|
self.assertIn('lvgl.page.previous', page['on_swipe_right'][-1])
|
|
|
|
def test_overrides_apply_to_values_and_unit_subscriptions(self):
|
|
self.printers[0]['entities'] = {'time_left': 'sensor.custom_remaining'}
|
|
package = generate(self.printers)
|
|
for domain, key in [('sensor', 'time_left'), ('text_sensor', 'remaining_unit')]:
|
|
entry = next(row for row in package[domain] if row['id'] == f'prints_charming_{key}')
|
|
self.assertEqual(entry['entity_id'], 'sensor.custom_remaining')
|
|
|
|
def test_reject_empty_duplicate_and_malformed_printer_ids(self):
|
|
for invalid in ([], self.printers + [self.printers[0]],
|
|
[{**self.printers[0], 'id': 'invalid-id'}]):
|
|
with self.subTest(invalid=invalid), self.assertRaises(ValueError):
|
|
validate_printers(invalid)
|
|
|
|
def test_reject_overflowing_filament_strip(self):
|
|
self.printers[0]['external_spools'] = [{'label': 'EXT', 'entity': 'sensor.external_spool'}]
|
|
with self.assertRaisesRegex(ValueError, 'eight slots'):
|
|
generate(self.printers)
|
|
|
|
def test_progress_and_dual_nozzle_widgets_fit_the_panel(self):
|
|
package = generate(self.printers)
|
|
for page in package['lvgl']['pages']:
|
|
for widget in page['widgets']:
|
|
props = next(iter(widget.values()))
|
|
self.assertGreaterEqual(props.get('x', 0), 0)
|
|
self.assertGreaterEqual(props.get('y', 0), 0)
|
|
self.assertLessEqual(props.get('x', 0) + props.get('width', 0), 800)
|
|
self.assertLessEqual(props.get('y', 0) + props.get('height', 0), 480)
|
|
|
|
def test_rgb565_primaries_reach_the_correct_panel_bits(self):
|
|
# mipi_rgb emits named RGB data pins with bytes exchanged, then copies
|
|
# LVGL RGB565 bytes straight into the ESP32 little-endian framebuffer.
|
|
# Exercise red, green, blue and combined primaries through that path.
|
|
class Loader(yaml.SafeLoader):
|
|
pass
|
|
for tag in ('!secret', '!include'):
|
|
Loader.add_constructor(tag, lambda loader, node: loader.construct_scalar(node))
|
|
config = yaml.load((HERE.parent / 'bambu-printer-display.yaml').read_text(), Loader=Loader)
|
|
display = config['display'][0]
|
|
pins = display['data_pins']
|
|
natural = pins['blue'] + pins['green'] + pins['red']
|
|
driver = natural[8:] + natural[:8]
|
|
order = {'big_endian': 'big', 'little_endian': 'little'}[config['lvgl']['byte_order']]
|
|
for color in (0xF800, 0x07E0, 0x001F, 0xFFFF, 0x07FF, 0xF81F):
|
|
memory_word = int.from_bytes(color.to_bytes(2, order), 'little')
|
|
panel_word = sum(((memory_word >> i) & 1) << natural.index(pin)
|
|
for i, pin in enumerate(driver))
|
|
self.assertEqual(panel_word, color, f'RGB565 color {color:#06x} is wired incorrectly')
|
|
|
|
def test_diagnostic_color_page_is_skipped_during_normal_navigation(self):
|
|
package = generate(self.printers)
|
|
page = next(page for page in package['lvgl']['pages'] if page['id'] == 'color_test_page')
|
|
self.assertTrue(page['skip'])
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|