Use should_receive instead of stub.