Completed unit tests for entities