@@ -477,13 +477,13 @@ def trace_session_plpgsql_functions(node: PostgresNode):
477477with node .connect ("postgres" ,autocommit = True )as conn :
478478start_session_trace (conn )
479479
480- assert node .execute ("select calculate_order_total(2)" )== [(Decimal ('80000.00' ),)]
481- node .execute ("select update_order_status(2, 'SHIPPED')" )
482- node .execute ("select add_product_to_order(1, 2, 100)" )
483- assert node .execute ("select * from get_user_orders(1)" )== [(1 ,Decimal ('4050000.00' ),'NEW' ,2 ,)]
484- node .execute ("select create_user('aboba', 'boba', 'aboba_boba@example.ru', '+79132281337')" )
485- assert node .execute ("select * from get_low_stock_products(10000)" )== [(2 ,"Samsung Galaxy S10" ,50 ,40000.00 ),(1 ,"iPhone X" ,100 ,50000.00 ), (3 ,"Xiaomi Redmi Note 8 Pro" ,200 ,25000.00 )]
486- assert node .execute ("SELECT process_complete_order(1, ARRAY[2,3], ARRAY[10, 5])" )== [(4 ,)]
480+ assert conn .execute ("select calculate_order_total(2)" )== [(Decimal ('80000.00' ),)]
481+ conn .execute ("select update_order_status(2, 'SHIPPED')" )
482+ conn .execute ("select add_product_to_order(1, 2, 100)" )
483+ assert conn .execute ("select * from get_user_orders(1)" )== [(1 ,Decimal ('4050000.00' ),'NEW' ,2 ,)]
484+ conn .execute ("select create_user('aboba', 'boba', 'aboba_boba@example.ru', '+79132281337')" )
485+ assert conn .execute ("select * from get_low_stock_products(10000)" )== [(2 ,"Samsung Galaxy S10" ,50 ,40000.00 ),(1 ,"iPhone X" ,100 ,50000.00 ), (3 ,"Xiaomi Redmi Note 8 Pro" ,200 ,25000.00 )]
486+ assert conn .execute ("SELECT process_complete_order(1, ARRAY[2,3], ARRAY[10, 5])" )== [(4 ,)]
487487
488488stop_session_trace (conn )
489489
@@ -510,9 +510,9 @@ def trace_session_correct_executor_finish(node: PostgresNode):
510510with node .connect ("postgres" ,autocommit = True )as conn :
511511start_session_trace (conn )
512512
513- node .execute ("create table mlparted (a int, b int)" )
513+ conn .execute ("create table mlparted (a int, b int)" )
514514
515- node .execute ("with ins (a, b, c) as\
515+ conn .execute ("with ins (a, b, c) as\
516516\
517517
518518