Add unit tests for the controller.
Add unit tests for the controller.