gemini quota exhausted: fixing regressions in test suite
This commit is contained in:
@@ -1,6 +1,7 @@
|
||||
from __future__ import annotations
|
||||
from unittest.mock import patch, MagicMock
|
||||
|
||||
from src import gui_2
|
||||
from src.gui_2 import App
|
||||
|
||||
|
||||
@@ -64,7 +65,9 @@ class TestMMADashboardStreams:
|
||||
app = _make_app(mma_streams={"Tier 1": "hello"})
|
||||
imgui_mock = _make_imgui_mock()
|
||||
imgui_mock.begin_child.return_value = True
|
||||
with patch("src.gui_2.imgui", imgui_mock), patch("src.gui_2.imscope") as mock_imscope:
|
||||
with patch("src.gui_2.imgui", imgui_mock), \
|
||||
patch("src.gui_2.imscope") as mock_imscope, \
|
||||
patch("src.gui_2.render_selectable_label") as mock_render:
|
||||
# Setup imscope mocks
|
||||
mock_imscope.window.return_value.__enter__.return_value = (True, True)
|
||||
mock_imscope.child.return_value.__enter__.return_value = True
|
||||
@@ -74,9 +77,9 @@ class TestMMADashboardStreams:
|
||||
mock_imscope.style_color.return_value.__enter__.return_value = None
|
||||
mock_imscope.style_var.return_value.__enter__.return_value = None
|
||||
|
||||
App._render_tier_stream_panel(app, "Tier 1", "Tier 1")
|
||||
gui_2.render_tier_stream_panel(app, "Tier 1", "Tier 1")
|
||||
|
||||
app._render_selectable_label.assert_called_with('stream_Tier 1', 'hello', width=-1, multiline=True, height=0)
|
||||
mock_render.assert_called_with(app, 'stream_Tier 1', 'hello', width=-1, multiline=True, height=0)
|
||||
|
||||
def test_tier3_renders_worker_subheaders(self):
|
||||
"""_render_tier_stream_panel for Tier 3 must render a sub-header for each worker stream key."""
|
||||
@@ -86,7 +89,9 @@ class TestMMADashboardStreams:
|
||||
})
|
||||
imgui_mock = _make_imgui_mock()
|
||||
imgui_mock.begin_child.return_value = True
|
||||
with patch("src.gui_2.imgui", imgui_mock), patch("src.gui_2.imscope") as mock_imscope:
|
||||
with patch("src.gui_2.imgui", imgui_mock), \
|
||||
patch("src.gui_2.imscope") as mock_imscope, \
|
||||
patch("src.gui_2.render_selectable_label"):
|
||||
# Setup imscope mocks
|
||||
mock_imscope.window.return_value.__enter__.return_value = (True, True)
|
||||
mock_imscope.child.return_value.__enter__.return_value = True
|
||||
@@ -96,7 +101,7 @@ class TestMMADashboardStreams:
|
||||
mock_imscope.style_color.return_value.__enter__.return_value = None
|
||||
mock_imscope.style_var.return_value.__enter__.return_value = None
|
||||
|
||||
App._render_tier_stream_panel(app, "Tier 3", None)
|
||||
gui_2.render_tier_stream_panel(app, "Tier 3", None)
|
||||
text_args = " ".join(str(c) for c in imgui_mock.text.call_args_list)
|
||||
assert "T-001" in text_args, "imgui.text not called with 'T-001' worker sub-header"
|
||||
assert "T-002" in text_args, "imgui.text not called with 'T-002' worker sub-header"
|
||||
Reference in New Issue
Block a user