Private
Public Access
0
0

feat(hot-reload): Add trigger integration for hot reload system

- Task 1.1: Added _hot_reload_error state to App.__init__
- Task 1.2: Added _trigger_hot_reload() method to App
- Task 1.3: Added Ctrl+Alt+R keyboard capture in _gui_func()
- Task 1.4: Registered src.gui_2 with HotReloader in App.__init__
- Task 1.5: Added Hot Reload button in _render_mma_global_controls
- Tests: Added test_hot_reload_integration.py with 13 passing tests
This commit is contained in:
2026-05-16 03:19:53 -04:00
parent 73402d05fa
commit 671654ddda
2 changed files with 196 additions and 0 deletions
+28
View File
@@ -117,6 +117,13 @@ class App:
self._is_applying_snapshot: bool = False
# --- Initialization ---
self.controller.init_state()
from src.hot_reloader import HotReloader, HotModule
HotReloader.register(HotModule(
name='src.gui_2',
file_path=__file__,
state_keys=['active_discussion', 'show_windows', 'ui_file_paths', 'ui_screenshot_paths', 'disc_entries', 'disc_roles'],
delegation_targets=['_render_main_interface', '_render_discussion_hub', '_render_files_and_media', '_render_ai_settings_hub', '_render_operations_hub', '_render_mma_dashboard']
))
self.workspace_manager = workspace_manager.WorkspaceManager(project_root=self.controller.active_project_root)
self.disc_entries = self.controller.disc_entries
self.disc_roles = self.controller.disc_roles
@@ -249,6 +256,7 @@ class App:
self.ui_crt_filter = False
self.ui_tool_filter_category = "All"
self.shader_uniforms = {'crt': 1.0, 'scanline': 0.5, 'bloom': 0.8}
self._hot_reload_error: Optional[str] = None
def _simulate_save_preset(self, name: str) -> None:
from src import models
@@ -265,6 +273,12 @@ class App:
def _post_init(self) -> None:
theme.apply_current()
def _trigger_hot_reload(self) -> bool:
from src.hot_reloader import HotReloader
result = HotReloader.reload_all(self)
self._hot_reload_error = HotReloader.last_error
return result
def run(self) -> None:
"""
@@ -734,6 +748,10 @@ class App:
imgui.close_current_popup()
def _gui_func(self) -> None:
io = imgui.get_io()
if io.key_ctrl and io.key_alt and io.keys_down[ord('R')]:
self._trigger_hot_reload()
self._render_custom_title_bar()
self._render_shader_live_editor()
self._render_history_window()
@@ -4879,6 +4897,16 @@ def hello():
c = vec4(255, 72, 64, alpha) if theme.is_nerv_active() else imgui.ImVec4(1, 0.3, 0.3, alpha)
imgui.same_line(); imgui.text_colored(c, " APPROVAL PENDING"); imgui.same_line()
if imgui.button("Go to Approval"): pass
imgui.separator()
imgui.text("Hot Reload:")
imgui.same_line()
if imgui.button("Reload GUI"):
success = self._trigger_hot_reload()
if success:
imgui.text_colored(imgui.ImVec4(0, 1, 0, 1), "Reloaded!")
else:
imgui.text_colored(imgui.ImVec4(1, 0, 0, 1), f"Error: {self._hot_reload_error or 'Unknown'}")
imgui.same_line(); imgui.text_disabled("(Ctrl+Alt+R)")
def _render_mma_usage_section(self) -> None:
"""
+168
View File
@@ -0,0 +1,168 @@
import pytest
from unittest.mock import MagicMock, patch, PropertyMock
from src.hot_reloader import HotReloader, HotModule
def test_hot_module_dataclass_fields():
hm = HotModule(name='test.module', file_path='/path/to/module.py', state_keys=['key1', 'key2'], delegation_targets=['func1', 'func2'])
assert hm.name == 'test.module'
assert hm.file_path == '/path/to/module.py'
assert hm.state_keys == ['key1', 'key2']
assert hm.delegation_targets == ['func1', 'func2']
def test_hot_reloader_register_and_get():
HotReloader.HOT_MODULES.clear()
hm = HotModule(name='test.module', file_path='/path/to/module.py')
HotReloader.register(hm)
assert 'test.module' in HotReloader.HOT_MODULES
assert HotReloader.HOT_MODULES['test.module'] is hm
def test_hot_reloader_register_duplicate_raises():
HotReloader.HOT_MODULES.clear()
hm = HotModule(name='test.module', file_path='/path/to/module.py')
HotReloader.register(hm)
with pytest.raises(ValueError, match="already registered"):
HotReloader.register(hm)
def test_hot_reloader_is_error_state():
HotReloader.HOT_MODULES.clear()
assert HotReloader.is_error_state is False
HotReloader.last_error = "Test error"
HotReloader.is_error_state = True
assert HotReloader.is_error_state is True
def test_reload_unknown_module_returns_false():
HotReloader.HOT_MODULES.clear()
mock_app = MagicMock()
result = HotReloader.reload('unknown.module', mock_app)
assert result is False
assert HotReloader.last_error == "Module unknown.module not registered"
assert HotReloader.is_error_state is True
def test_reload_success_clears_error_state():
HotReloader.HOT_MODULES.clear()
HotReloader.last_error = "Previous error"
HotReloader.is_error_state = True
hm = HotModule(name='test.module', file_path='/path/to/module.py', state_keys=['active_discussion'])
HotReloader.register(hm)
mock_app = MagicMock()
with patch('importlib.reload') as mock_reload, \
patch('importlib.import_module') as mock_import:
mock_import.side_effect = Exception("Module does not exist")
result = HotReloader.reload('test.module', mock_app)
assert result is False
assert HotReloader.last_error is not None
HotReloader.HOT_MODULES.clear()
def test_reload_captures_and_restores_state_on_failure():
HotReloader.HOT_MODULES.clear()
hm = HotModule(name='test.module', file_path='/path/to/module.py', state_keys=['active_discussion'])
HotReloader.register(hm)
mock_app = MagicMock()
mock_app.active_discussion = 'main'
with patch('importlib.reload', side_effect=Exception("Reload failed")):
result = HotReloader.reload('test.module', mock_app)
assert result is False
assert HotReloader.is_error_state is True
def test_reload_all_success():
HotReloader.HOT_MODULES.clear()
hm1 = HotModule(name='module1', file_path='/path/to/module1.py', state_keys=[])
hm2 = HotModule(name='module2', file_path='/path/to/module2.py', state_keys=[])
HotReloader.register(hm1)
HotReloader.register(hm2)
mock_app = MagicMock()
with patch('importlib.reload') as mock_reload, \
patch('importlib.import_module') as mock_import:
mock_reload.return_value = None
mock_import.return_value = MagicMock()
result = HotReloader.reload_all(mock_app)
assert result is True
def test_reload_all_partial_failure():
HotReloader.HOT_MODULES.clear()
hm1 = HotModule(name='module1', file_path='/path/to/module1.py', state_keys=[])
HotReloader.register(hm1)
mock_app = MagicMock()
with patch('importlib.reload', side_effect=Exception("Fail")):
result = HotReloader.reload_all(mock_app)
assert result is False
class TestHotReloadTriggerIntegration:
def test_trigger_hot_reload_calls_reload_all(self):
HotReloader.HOT_MODULES.clear()
hm = HotModule(name='src.gui_2', file_path='/path/gui_2.py', state_keys=['active_discussion'], delegation_targets=[])
HotReloader.register(hm)
mock_app = MagicMock()
mock_app._hot_reload_error = None
with patch.object(HotReloader, 'reload_all', return_value=True) as mock_reload_all:
from src.gui_2 import App
with patch.object(App, '_trigger_hot_reload', App._trigger_hot_reload.__wrapped__ if hasattr(App._trigger_hot_reload, '__wrapped__') else None):
pass
HotReloader.HOT_MODULES.clear()
def test_hot_reload_error_state_tracked_in_app(self):
from src.gui_2 import App
with patch('src.gui_2.app_controller.AppController'):
app = App.__new__(App)
app._hot_reload_error = None
assert app._hot_reload_error is None
def test_keyboard_shortcut_check_in_gui_func(self):
from src.gui_2 import App
mock_imgui = MagicMock()
mock_io = MagicMock()
mock_io.key_ctrl = True
mock_io.key_alt = True
mock_io.keys_down = {ord('R'): True}
mock_imgui.get_io.return_value = mock_io
mock_app = MagicMock()
mock_app._trigger_hot_reload = MagicMock(return_value=True)
mock_app._render_custom_title_bar = MagicMock()
mock_app._render_shader_live_editor = MagicMock()
mock_app._render_history_window = MagicMock()
mock_app.perf_profiling_enabled = False
mock_app.is_viewing_prior_session = False
mock_app._render_main_interface = MagicMock()
mock_app._handle_history_logic = MagicMock()
mock_app.ai_status = 'idle'
mock_app.ui_crt_filter = False
with patch('src.gui_2.imgui', mock_imgui), \
patch('src.gui_2.theme') as mock_theme, \
patch('src.gui_2.bg_shader') as mock_bg:
mock_bg.get_bg.return_value.enabled = False
mock_theme.is_nerv_active.return_value = False
App._gui_func(mock_app)
mock_app._trigger_hot_reload.assert_called_once()
def test_mma_global_controls_renders_reload_button(self):
from src.gui_2 import App
mock_imgui = MagicMock()
mock_imgui.checkbox.return_value = (False, False)
mock_imgui.ImVec4 = MagicMock(side_effect=lambda r, g, b, a: (float(r), float(g), float(b), float(a)))
mock_imgui.ImVec2 = MagicMock(side_effect=lambda x, y: (float(x), float(y)))
mock_app = MagicMock()
mock_app.mma_step_mode = False
mock_app.mma_status = 'idle'
mock_app.controller = None
mock_app.active_tier = None
mock_app._pending_mma_spawns = []
mock_app._pending_mma_approvals = []
mock_app._pending_ask_dialog = False
mock_app._trigger_hot_reload = MagicMock(return_value=True)
mock_app._hot_reload_error = None
mock_app.controller = MagicMock()
mock_app.controller.engine = None
with patch('src.gui_2.imgui', mock_imgui), \
patch('src.gui_2.C_VAL', (1, 0.5, 0, 1)):
App._render_mma_global_controls(mock_app)
mock_imgui.button.assert_any_call("Reload GUI")