Add identity map
Add test for UnitOfWork#createEntity