Remotion Video Generator

# chuk-motion/src/chuk_motion/components/charts/BarChart/test_bar_chart.py """ Tests for BarChart template generation. """ from tests.components.conftest import ( assert_has_interface, assert_valid_typescript, ) class TestBarChartBasic: """Basic BarChart generation tests.""" def test_basic_generation(self, component_builder, theme_name): """Test basic BarChart generation with all props.""" tsx = component_builder.build_component( "BarChart", { "data": [[0, 10], [1, 20], [2, 30]], "title": "Test Chart", "xlabel": "X", "ylabel": "Y", }, theme_name, ) assert tsx is not None assert "BarChart" in tsx assert_valid_typescript(tsx) assert_has_interface(tsx, "BarChart") def test_minimal_props(self, component_builder, theme_name): """Test BarChart with minimal props.""" tsx = component_builder.build_component("BarChart", {}, theme_name) assert tsx is not None class TestBarChartRendering: """Tests for BarChart rendering.""" def test_svg_element(self, component_builder, theme_name): """Test SVG element is created.""" tsx = component_builder.build_component( "BarChart", { "data": [[0, 10], [1, 20], [2, 30]], "title": "Test Chart", "xlabel": "X", "ylabel": "Y", }, theme_name, ) assert "<svg" in tsx assert "width=" in tsx assert "height=" in tsx def test_data_handling(self, component_builder, theme_name): """Test data prop is used.""" tsx = component_builder.build_component( "BarChart", { "data": [[0, 10], [1, 20], [2, 30]], "title": "Test Chart", "xlabel": "X", "ylabel": "Y", }, theme_name, ) assert "data" in tsx class TestBarChartBuilderMethod: """Tests for BarChart builder method.""" def test_add_to_composition_basic(self): """Test add_to_composition creates ComponentInstance.""" from chuk_motion.components.charts.BarChart.builder import add_to_composition from chuk_motion.generator.composition_builder import CompositionBuilder builder = CompositionBuilder() data = [{"label": "A", "value": 10}, {"label": "B", "value": 20}] result = add_to_composition(builder, data=data, start_time=0.0) assert result is builder assert len(builder.components) == 1 assert builder.components[0].component_type == "BarChart" assert builder.components[0].props["data"] == data def test_add_to_composition_all_props(self): """Test all props are set correctly.""" from chuk_motion.components.charts.BarChart.builder import add_to_composition from chuk_motion.generator.composition_builder import CompositionBuilder builder = CompositionBuilder() data = [{"label": "A", "value": 10}, {"label": "B", "value": 20}] add_to_composition( builder, data=data, title="Test Chart", xlabel="X Axis", ylabel="Y Axis", start_time=1.0, duration=5.0, ) props = builder.components[0].props assert props["data"] == data assert props["title"] == "Test Chart" assert props["xlabel"] == "X Axis" assert props["ylabel"] == "Y Axis" def test_add_to_composition_timing(self): """Test add_to_composition handles timing correctly.""" from chuk_motion.components.charts.BarChart.builder import add_to_composition from chuk_motion.generator.composition_builder import CompositionBuilder builder = CompositionBuilder(fps=30) add_to_composition(builder, data=[], start_time=2.0, duration=4.0) component = builder.components[0] assert component.start_frame == 60 assert component.duration_frames == 120 class TestBarChartToolRegistration: """Tests for BarChart MCP tool.""" def test_register_tool(self): """Test tool registration.""" from unittest.mock import Mock from chuk_motion.components.charts.BarChart.tool import register_tool mcp = Mock() project_manager = Mock() register_tool(mcp, project_manager) assert mcp.tool.called or hasattr(mcp, "tool") def test_tool_execution(self): """Test tool execution creates component.""" import asyncio import json from unittest.mock import Mock from chuk_motion.components.charts.BarChart.tool import register_tool from chuk_motion.generator.timeline import Timeline mcp = Mock() project_manager = Mock() timeline = Timeline(fps=30) project_manager.current_timeline = timeline register_tool(mcp, project_manager) tool_func = mcp.tool.call_args[0][0] result = asyncio.run(tool_func(data='[{"label": "A", "value": 10}]', duration=4.0)) # Check component was added assert len(timeline.get_all_components()) >= 1 result_data = json.loads(result) assert result_data["component"] == "BarChart" def test_tool_execution_no_project(self): """Test tool execution when no project exists.""" import asyncio import json from unittest.mock import Mock from chuk_motion.components.charts.BarChart.tool import register_tool mcp = Mock() project_manager = Mock() project_manager.current_timeline = None # No project register_tool(mcp, project_manager) tool_func = mcp.tool.call_args[0][0] result = asyncio.run(tool_func(data='[{"label": "A", "value": 10}]', duration=4.0)) result_data = json.loads(result) assert "error" in result_data assert "No active project" in result_data["error"] def test_tool_execution_error_handling(self): """Test tool execution handles exceptions.""" import asyncio import json from unittest.mock import Mock, patch from chuk_motion.components.charts.BarChart.tool import register_tool from chuk_motion.generator.timeline import Timeline mcp = Mock() project_manager = Mock() timeline = Timeline(fps=30) project_manager.current_timeline = timeline register_tool(mcp, project_manager) tool_func = mcp.tool.call_args[0][0] # Mock add_component to raise exception with patch.object(timeline, "add_component", side_effect=Exception("Test error")): result = asyncio.run(tool_func(data='[{"label": "A", "value": 10}]', duration=4.0)) result_data = json.loads(result) assert "error" in result_data assert "Test error" in result_data["error"] def test_tool_execution_invalid_json(self): """Test tool execution handles invalid JSON data.""" import asyncio import json from unittest.mock import Mock from chuk_motion.components.charts.BarChart.tool import register_tool from chuk_motion.generator.timeline import Timeline mcp = Mock() project_manager = Mock() timeline = Timeline(fps=30) project_manager.current_timeline = timeline register_tool(mcp, project_manager) tool_func = mcp.tool.call_args[0][0] result = asyncio.run(tool_func(data="invalid json {[}", title="Test", duration=4.0)) result_data = json.loads(result) assert "error" in result_data assert "Invalid data JSON" in result_data["error"]