Unit tests for agent_controller.py