Added test for UnitOfWork