Added integration test for controller