Added tests for the main controller.