Add unit test for entity
Merge pull request #42 from cotal/fixUT

Add unit test for entity