WIP: fixing regressions in tests

tests/test_log_management_ui.py

@@ -33,6 +33,7 @@ history = []
 @pytest.fixture
 def app_instance(mock_config: Path, mock_project: Path, monkeypatch: pytest.MonkeyPatch) -> App:
  monkeypatch.setattr("src.models.CONFIG_PATH", mock_config)
+ monkeypatch.setattr("src.gui_2.filedialog", MagicMock())
  with patch("src.project_manager.load_project") as mock_load, \
  patch("src.session_logger.open_session"), \
  patch("src.session_logger.reset_session"):
@@ -67,11 +68,10 @@ def test_render_log_management_logic(app_instance: App) -> None:
  app.show_windows["Log Management"] = True
  # Mock LogRegistry
  with patch("src.log_registry.LogRegistry") as MockRegistry, \
- patch("src.gui_2.imgui.begin") as mock_begin, \
+ patch("src.gui_2.imscope") as mock_imscope, \
  patch("src.gui_2.imgui.begin_table") as mock_begin_table, \
  patch("src.gui_2.imgui.text") as mock_text, \
  patch("src.gui_2.imgui.end_table"), \
- patch("src.gui_2.imgui.end"), \
  patch("src.gui_2.imgui.push_style_color"), \
  patch("src.gui_2.imgui.pop_style_color"), \
  patch("src.gui_2.imgui.table_setup_column"), \
@@ -81,7 +81,7 @@ def test_render_log_management_logic(app_instance: App) -> None:
  patch("src.gui_2.imgui.same_line"), \
  patch("src.gui_2.imgui.text_colored"), \
  patch("src.gui_2.imgui.separator"), \
- patch("src.gui_2.imgui.button"):
+ patch("src.gui_2.imgui.button", return_value=False):
 
   mock_reg = MockRegistry.return_value
   mock_reg.data = {
@@ -91,10 +91,15 @@ def test_render_log_management_logic(app_instance: App) -> None:
     "metadata": {"reason": "test", "size_kb": 10, "message_count": 5}
    }
   }
-  mock_begin.return_value = (True, True)
+  app._log_registry = mock_reg
+  
+  mock_window_cm = MagicMock()
+  mock_window_cm.__enter__.return_value = (True, True)
+  mock_imscope.window.return_value = mock_window_cm
+
   mock_begin_table.return_value = True
   app._render_log_management()
-  mock_begin.assert_called_with("Log Management", app.show_windows["Log Management"])
+  mock_imscope.window.assert_called_with("Log Management", app.show_windows["Log Management"])
   mock_begin_table.assert_called()
   mock_text.assert_any_call("session_1")