add: unit tests for controller classes