--
-- SQL_PROCEDURE
--
-- Tests basic PROCEDURE functionality with SQL and PLPGSQL procedures.
--
-- print whether we're using version > 10 to make version-specific tests clear
SHOW server_version \gset
SELECT substring(:'server_version', '\d+')::int > 10 AS version_above_ten;
 version_above_ten 
-------------------
 f
(1 row)

SET citus.next_shard_id TO 100500;
CREATE SCHEMA procedure_schema;
SET SEARCH_PATH = procedure_schema;
CREATE TABLE test_table(id integer , org_id integer);
CREATE UNIQUE INDEX idx_table ON test_table(id, org_id);
SELECT create_distributed_table('test_table','id');
 create_distributed_table 
--------------------------
 
(1 row)

INSERT INTO test_table VALUES(1, 1);
-- test CREATE PROCEDURE
CREATE PROCEDURE test_procedure_delete_insert(id int, org_id int) LANGUAGE SQL AS $$
	DELETE FROM test_table;
	INSERT INTO test_table VALUES(id, org_id);
$$;
ERROR:  syntax error at or near "PROCEDURE"
LINE 1: CREATE PROCEDURE test_procedure_delete_insert(id int, org_id...
               ^
CALL test_procedure_delete_insert(2,3);
ERROR:  syntax error at or near "CALL"
LINE 1: CALL test_procedure_delete_insert(2,3);
        ^
SELECT * FROM test_table ORDER BY 1, 2;
 id | org_id 
----+--------
  1 |      1
(1 row)

-- commit/rollback is not allowed in procedures in SQL
-- following calls should fail
CREATE PROCEDURE test_procedure_commit(tt_id int, tt_org_id int) LANGUAGE SQL AS $$
	DELETE FROM test_table;
	COMMIT;
	INSERT INTO test_table VALUES(tt_id, -1);
	UPDATE test_table SET org_id = tt_org_id WHERE id = tt_id;
	COMMIT;
$$;
ERROR:  syntax error at or near "PROCEDURE"
LINE 1: CREATE PROCEDURE test_procedure_commit(tt_id int, tt_org_id ...
               ^
CALL test_procedure_commit(2,5);
ERROR:  syntax error at or near "CALL"
LINE 1: CALL test_procedure_commit(2,5);
        ^
SELECT * FROM test_table ORDER BY 1, 2;
 id | org_id 
----+--------
  1 |      1
(1 row)

CREATE PROCEDURE test_procedure_rollback(tt_id int, tt_org_id int) LANGUAGE SQL AS $$
	DELETE FROM test_table;
	ROLLBACK;
	UPDATE test_table SET org_id = tt_org_id WHERE id = tt_id;
    COMMIT;
$$;
ERROR:  syntax error at or near "PROCEDURE"
LINE 1: CREATE PROCEDURE test_procedure_rollback(tt_id int, tt_org_i...
               ^
CALL test_procedure_rollback(2,15);
ERROR:  syntax error at or near "CALL"
LINE 1: CALL test_procedure_rollback(2,15);
        ^
SELECT * FROM test_table ORDER BY 1, 2;
 id | org_id 
----+--------
  1 |      1
(1 row)

DROP PROCEDURE test_procedure_delete_insert(int, int);
ERROR:  syntax error at or near "PROCEDURE"
LINE 1: DROP PROCEDURE test_procedure_delete_insert(int, int);
             ^
DROP PROCEDURE test_procedure_commit(int, int);
ERROR:  syntax error at or near "PROCEDURE"
LINE 1: DROP PROCEDURE test_procedure_commit(int, int);
             ^
DROP PROCEDURE test_procedure_rollback(int, int);
ERROR:  syntax error at or near "PROCEDURE"
LINE 1: DROP PROCEDURE test_procedure_rollback(int, int);
             ^
-- same tests with plpgsql
-- test CREATE PROCEDURE
CREATE PROCEDURE test_procedure_delete_insert(id int, org_id int) LANGUAGE PLPGSQL AS $$
BEGIN
	DELETE FROM test_table;
	INSERT INTO test_table VALUES(id, org_id);
END;
$$;
ERROR:  syntax error at or near "PROCEDURE"
LINE 1: CREATE PROCEDURE test_procedure_delete_insert(id int, org_id...
               ^
CALL test_procedure_delete_insert(2,3);
ERROR:  syntax error at or near "CALL"
LINE 1: CALL test_procedure_delete_insert(2,3);
        ^
SELECT * FROM test_table ORDER BY 1, 2;
 id | org_id 
----+--------
  1 |      1
(1 row)

-- notice that the update succeed and committed
CREATE PROCEDURE test_procedure_modify_insert(tt_id int, tt_org_id int) LANGUAGE PLPGSQL AS $$
BEGIN
	UPDATE test_table SET org_id = tt_org_id WHERE id = tt_id;
	COMMIT;
	INSERT INTO test_table VALUES (tt_id, tt_org_id);
	ROLLBACK;
END;
$$;
ERROR:  syntax error at or near "PROCEDURE"
LINE 1: CREATE PROCEDURE test_procedure_modify_insert(tt_id int, tt_...
               ^
CALL test_procedure_modify_insert(2,12);
ERROR:  syntax error at or near "CALL"
LINE 1: CALL test_procedure_modify_insert(2,12);
        ^
SELECT * FROM test_table ORDER BY 1, 2;
 id | org_id 
----+--------
  1 |      1
(1 row)

CREATE PROCEDURE test_procedure_modify_insert_commit(tt_id int, tt_org_id int) LANGUAGE PLPGSQL AS $$
BEGIN
	UPDATE test_table SET org_id = tt_org_id WHERE id = tt_id;
	COMMIT;
	INSERT INTO test_table VALUES (tt_id, tt_org_id);
	COMMIT;
END;
$$;
ERROR:  syntax error at or near "PROCEDURE"
LINE 1: CREATE PROCEDURE test_procedure_modify_insert_commit(tt_id i...
               ^
CALL test_procedure_modify_insert_commit(2,30);
ERROR:  syntax error at or near "CALL"
LINE 1: CALL test_procedure_modify_insert_commit(2,30);
        ^
SELECT * FROM test_table ORDER BY 1, 2;
 id | org_id 
----+--------
  1 |      1
(1 row)

-- delete is commited but insert is rolled back
-- there is a bug #2371 on that preventing rollback
CREATE PROCEDURE test_procedure_rollback(tt_id int, tt_org_id int) LANGUAGE PLPGSQL AS $$
BEGIN
	DELETE FROM test_table;
	COMMIT;
	INSERT INTO test_table VALUES (tt_id, tt_org_id);
	ROLLBACK;
END;
$$;
ERROR:  syntax error at or near "PROCEDURE"
LINE 1: CREATE PROCEDURE test_procedure_rollback(tt_id int, tt_org_i...
               ^
CALL test_procedure_rollback(2,5);
ERROR:  syntax error at or near "CALL"
LINE 1: CALL test_procedure_rollback(2,5);
        ^
SELECT * FROM test_table ORDER BY 1, 2;
 id | org_id 
----+--------
  1 |      1
(1 row)

-- rollback is successfull when insert is on multiple rows
CREATE PROCEDURE test_procedure_rollback_2(tt_id int, tt_org_id int) LANGUAGE PLPGSQL AS $$
BEGIN
	DELETE FROM test_table;
	COMMIT;
	INSERT INTO test_table VALUES (tt_id, tt_org_id), (tt_id+1, tt_org_id+1);
	ROLLBACK;
END;
$$;
ERROR:  syntax error at or near "PROCEDURE"
LINE 1: CREATE PROCEDURE test_procedure_rollback_2(tt_id int, tt_org...
               ^
CALL test_procedure_rollback_2(12, 15);
ERROR:  syntax error at or near "CALL"
LINE 1: CALL test_procedure_rollback_2(12, 15);
        ^
SELECT * FROM test_table ORDER BY 1, 2;
 id | org_id 
----+--------
  1 |      1
(1 row)

-- delete is rolled back, update is committed
CREATE PROCEDURE test_procedure_rollback_3(tt_id int, tt_org_id int) LANGUAGE PLPGSQL AS $$
BEGIN
	DELETE FROM test_table;
	ROLLBACK;
	UPDATE test_table SET org_id = tt_org_id WHERE id = tt_id;
    COMMIT;
END;
$$;
ERROR:  syntax error at or near "PROCEDURE"
LINE 1: CREATE PROCEDURE test_procedure_rollback_3(tt_id int, tt_org...
               ^
INSERT INTO test_table VALUES (1, 1), (2, 2);
ERROR:  duplicate key value violates unique constraint "idx_table_100500"
DETAIL:  Key (id, org_id)=(1, 1) already exists.
CONTEXT:  while executing command on localhost:57638
CALL test_procedure_rollback_3(2,15);
ERROR:  syntax error at or near "CALL"
LINE 1: CALL test_procedure_rollback_3(2,15);
        ^
SELECT * FROM test_table ORDER BY 1, 2;
 id | org_id 
----+--------
  1 |      1
(1 row)

DROP SCHEMA procedure_schema CASCADE;
NOTICE:  drop cascades to table test_table
RESET SEARCH_PATH;