Add some tests for AIController