Add test for controller