diff --git a/tests/ast/decoder.py b/tests/ast/decoder.py index c7e95ff8e3a..a797c04ff55 100644 --- a/tests/ast/decoder.py +++ b/tests/ast/decoder.py @@ -987,6 +987,9 @@ def decode_expr(self, expr: proto.Expr, **kwargs) -> Any: return call_table_function( fn_name, *pos_args, **named_args ) + + case "stored_procedure": + return self.session.call(fn_name, *pos_args, **named_args) case _: raise ValueError( "Unknown function reference type: %s" @@ -2813,8 +2816,13 @@ def decode_expr(self, expr: proto.Expr, **kwargs) -> Any: return_type = self.decode_data_type_expr( expr.stored_procedure.return_type ) + name = None + if expr.stored_procedure.HasField("name"): + name = self.decode_name_expr(expr.stored_procedure.name) + ret_sproc = sproc( - lambda *args: None, + self.session.sproc._registry[registered_object_name], + name=name, return_type=return_type, input_types=input_types, execute_as=execute_as,