Replace should_receive with expect...to receive
Replace stub with allow...to receive