def METHOD_NAME(self):
    self.execute_times = 0
    @retry_on_rpc_failure()
    def test_api(self, code):
        self.execute_times += 1
        self.mock_milvus_exception(code)
    with pytest.raises(MilvusException) as e:
        test_api(self, common_pb2.ForceDeny)
    print(e)
    assert "force deny" in e.value.message
    # the first execute + 0 retry times
    assert self.execute_times == 1