Added unit test for the repository