Add controller's tests