Adding integration test for the controller