Add full test coverage for controllers (unit)