More unit tests for UnitOfWork