Refactoring TestableController to TestableControllerTrait