Added tests to repository classes