feat(controller): Add test coverage