add get_entity mock support