-- Copyright (c) Microsoft Corporation.
-- Licensed under the PostgreSQL License.
-- Test: Variable substitution using |=> operator and df.as() function
SET ROLE df_regress_user;
DROP TABLE IF EXISTS test_vars_log;
NOTICE:  table "test_vars_log" does not exist, skipping
CREATE TABLE test_vars_log (id SERIAL, val TEXT, variant TEXT);
-- Test A: Using |=> operator
SELECT df.start(
    'SELECT 100 as num' |=> 'x'
    ~> 'INSERT INTO test_vars_log (val, variant) VALUES ($x::text, ''op'')',
    'test-variables-op'
) AS instance_id \gset
SELECT df.await_instance(:'instance_id') AS status;
  status   
-----------
 completed
(1 row)

-- Test B: Using df.as() function
SELECT df.start(
    df.seq(
        df.as('SELECT 200 as num', 'y'),
        'INSERT INTO test_vars_log (val, variant) VALUES ($y::text, ''fn'')'
    ),
    'test-variables-fn'
) AS instance_id \gset
SELECT df.await_instance(:'instance_id') AS status;
  status   
-----------
 completed
(1 row)

-- Verify results (deterministic output, ordered by id)
SELECT val, variant FROM test_vars_log ORDER BY id;
 val | variant 
-----+---------
 100 | op
 200 | fn
(2 rows)

-- Cleanup
DROP TABLE test_vars_log;