Conductor
44 lines
2.0 KiB
Python
44 lines
2.0 KiB
Python
from unittest.mock import patch, MagicMock
|
|
from src.markdown_table import render_table, TableBlock
|
|
|
|
def test_render_table_uses_text_wrapped_for_cells():
|
|
block = TableBlock(headers=["A"], rows=[["hello"]], span=(0, 2))
|
|
with patch("src.markdown_table.imgui") as mock_imgui:
|
|
mock_imgui.TableFlags_ = type("T", (), {"borders": 1, "row_bg": 2, "resizable": 4})()
|
|
mock_imgui.begin_table.return_value = True
|
|
mock_imgui.table_next_column = MagicMock()
|
|
mock_imgui.table_next_row = MagicMock()
|
|
mock_imgui.table_headers_row = MagicMock()
|
|
mock_imgui.text_wrapped = MagicMock()
|
|
mock_imgui.text = MagicMock()
|
|
mock_imgui.end_table = MagicMock()
|
|
render_table(block)
|
|
mock_imgui.text_wrapped.assert_any_call("hello")
|
|
|
|
def test_render_table_uses_text_wrapped_for_headers():
|
|
block = TableBlock(headers=["A"], rows=[["hello"]], span=(0, 2))
|
|
with patch("src.markdown_table.imgui") as mock_imgui:
|
|
mock_imgui.TableFlags_ = type("T", (), {"borders": 1, "row_bg": 2, "resizable": 4})()
|
|
mock_imgui.begin_table.return_value = True
|
|
mock_imgui.table_next_column = MagicMock()
|
|
mock_imgui.table_next_row = MagicMock()
|
|
mock_imgui.table_headers_row = MagicMock()
|
|
mock_imgui.text_wrapped = MagicMock()
|
|
mock_imgui.text = MagicMock()
|
|
mock_imgui.end_table = MagicMock()
|
|
render_table(block)
|
|
mock_imgui.text_wrapped.assert_any_call("A")
|
|
|
|
def test_render_table_does_not_use_text_for_cells():
|
|
block = TableBlock(headers=["A"], rows=[["hello"]], span=(0, 2))
|
|
with patch("src.markdown_table.imgui") as mock_imgui:
|
|
mock_imgui.TableFlags_ = type("T", (), {"borders": 1, "row_bg": 2, "resizable": 4})()
|
|
mock_imgui.begin_table.return_value = True
|
|
mock_imgui.table_next_column = MagicMock()
|
|
mock_imgui.table_next_row = MagicMock()
|
|
mock_imgui.table_headers_row = MagicMock()
|
|
mock_imgui.text_wrapped = MagicMock()
|
|
mock_imgui.text = MagicMock()
|
|
mock_imgui.end_table = MagicMock()
|
|
render_table(block)
|
|
assert not mock_imgui.text.called, "imgui.text should not be called for table cells" |