BalajiN | #49 | Adding tests for the controller