From 8bfb08ca0dcff3e023ea79efd1bb8c78ead22deb Mon Sep 17 00:00:00 2001 From: Patrick Surry Date: Sat, 9 Dec 2023 08:37:19 -0500 Subject: [PATCH] reorg --- tests/mock_board.py | 19 +++++-------------- tests/mocks.py | 13 +++++++++++++ 2 files changed, 18 insertions(+), 14 deletions(-) diff --git a/tests/mock_board.py b/tests/mock_board.py index c379e0761..cbb259282 100644 --- a/tests/mock_board.py +++ b/tests/mock_board.py @@ -2,26 +2,17 @@ # python -m tests.mock_board boards/nullbitsco/tidbit import sys -from unittest.mock import Mock -from .mocks import init_circuit_python_modules_mocks +from .mocks import init_board_module_mocks -init_circuit_python_modules_mocks() -sys.modules['rp2pio'] = Mock() -sys.modules['pwmio'] = Mock() -sys.modules['rotaryio'] = Mock() -sys.modules['displayio'] = Mock() -sys.modules['terminalio'] = Mock() -sys.modules['adafruit_pixelbuf'] = Mock() -sys.modules['adafruit_pixelbuf'].PixelBuf = Mock() -sys.modules['adafruit_displayio_ssd1306'] = Mock() -sys.modules['adafruit_display_text'] = Mock() +init_board_module_mocks() -sys.path.insert(0, sys.argv[1]) +board_dir = sys.argv[1] +sys.path.insert(0, board_dir) # TODO pass tests for dactyl keebs which seem to map a mystery pin B7 -if 'dactyl' in sys.argv[1]: +if 'dactyl' in board_dir: from kmk.quickpin.pro_micro.avr_promicro import translate as avr avr['B7'] = -1 diff --git a/tests/mocks.py b/tests/mocks.py index 9b6c92fb9..b1991dbe0 100644 --- a/tests/mocks.py +++ b/tests/mocks.py @@ -36,3 +36,16 @@ def init_circuit_python_modules_mocks(): from . import task sys.modules['_asyncio'] = task + + +def init_board_module_mocks(): + init_circuit_python_modules_mocks() + sys.modules['rp2pio'] = Mock() + sys.modules['pwmio'] = Mock() + sys.modules['rotaryio'] = Mock() + sys.modules['displayio'] = Mock() + sys.modules['terminalio'] = Mock() + sys.modules['adafruit_pixelbuf'] = Mock() + sys.modules['adafruit_pixelbuf'].PixelBuf = Mock() + sys.modules['adafruit_displayio_ssd1306'] = Mock() + sys.modules['adafruit_display_text'] = Mock()