enhance mock methods