Add AIController unit test