Add some entity tests