Added unit test for my controller code.