SET citus.next_shard_id TO 20030000;
SET client_min_messages TO ERROR;
CREATE USER procedureuser;
SELECT 1 FROM run_command_on_workers($$CREATE USER procedureuser;$$);
 ?column?
---------------------------------------------------------------------
        1
        1
(2 rows)

RESET client_min_messages;
CREATE SCHEMA procedure_tests AUTHORIZATION procedureuser;
CREATE SCHEMA procedure_tests2 AUTHORIZATION procedureuser;
SET search_path TO procedure_tests;
SET citus.shard_count TO 4;
-- Create and distribute a simple function
CREATE OR REPLACE PROCEDURE raise_info(text)
    LANGUAGE PLPGSQL AS $proc$
BEGIN
    RAISE INFO 'information message %', $1;
END;
$proc$;
-- procedures are distributed by text arguments, when run in isolation it is not guaranteed a table actually exists.
CREATE TABLE colocation_table(id text);
SET citus.shard_replication_factor TO 1;
SELECT create_distributed_table('colocation_table','id');
 create_distributed_table
---------------------------------------------------------------------

(1 row)

SELECT create_distributed_function('raise_info(text)', '$1', colocate_with := 'colocation_table');
 create_distributed_function
---------------------------------------------------------------------

(1 row)

SELECT * FROM run_command_on_workers($$CALL procedure_tests.raise_info('hello');$$) ORDER BY 1,2;
 nodename  | nodeport | success | result
---------------------------------------------------------------------
 localhost |    57637 | t       | CALL
 localhost |    57638 | t       | CALL
(2 rows)

SELECT public.verify_function_is_same_on_workers('procedure_tests.raise_info(text)');
 verify_function_is_same_on_workers
---------------------------------------------------------------------
 t
(1 row)

-- testing alter statements for a distributed function
-- ROWS 5, untested because;
-- ERROR:  ROWS is not applicable when function does not return a set
ALTER PROCEDURE raise_info(text) SECURITY INVOKER;
SELECT public.verify_function_is_same_on_workers('procedure_tests.raise_info(text)');
 verify_function_is_same_on_workers
---------------------------------------------------------------------
 t
(1 row)

ALTER PROCEDURE raise_info(text) SECURITY DEFINER;
SELECT public.verify_function_is_same_on_workers('procedure_tests.raise_info(text)');
 verify_function_is_same_on_workers
---------------------------------------------------------------------
 t
(1 row)

-- Test SET/RESET for alter procedure
ALTER PROCEDURE raise_info(text) SET client_min_messages TO warning;
SELECT public.verify_function_is_same_on_workers('procedure_tests.raise_info(text)');
 verify_function_is_same_on_workers
---------------------------------------------------------------------
 t
(1 row)

ALTER PROCEDURE raise_info(text) SET client_min_messages TO error;
SELECT public.verify_function_is_same_on_workers('procedure_tests.raise_info(text)');
 verify_function_is_same_on_workers
---------------------------------------------------------------------
 t
(1 row)

ALTER PROCEDURE raise_info(text) SET client_min_messages TO debug;
SELECT public.verify_function_is_same_on_workers('procedure_tests.raise_info(text)');
 verify_function_is_same_on_workers
---------------------------------------------------------------------
 t
(1 row)

ALTER PROCEDURE raise_info(text) RESET client_min_messages;
SELECT public.verify_function_is_same_on_workers('procedure_tests.raise_info(text)');
 verify_function_is_same_on_workers
---------------------------------------------------------------------
 t
(1 row)

-- rename function and make sure the new name can be used on the workers while the old name can't
ALTER PROCEDURE raise_info(text) RENAME TO raise_info2;
SELECT public.verify_function_is_same_on_workers('procedure_tests.raise_info2(text)');
 verify_function_is_same_on_workers
---------------------------------------------------------------------
 t
(1 row)

SELECT * FROM run_command_on_workers($$CALL procedure_tests.raise_info('hello');$$) ORDER BY 1,2;
 nodename  | nodeport | success |                                result
---------------------------------------------------------------------
 localhost |    57637 | f       | ERROR:  procedure procedure_tests.raise_info(unknown) does not exist
 localhost |    57638 | f       | ERROR:  procedure procedure_tests.raise_info(unknown) does not exist
(2 rows)

SELECT * FROM run_command_on_workers($$CALL procedure_tests.raise_info2('hello');$$) ORDER BY 1,2;
 nodename  | nodeport | success | result
---------------------------------------------------------------------
 localhost |    57637 | t       | CALL
 localhost |    57638 | t       | CALL
(2 rows)

ALTER PROCEDURE raise_info2(text) RENAME TO raise_info;
-- change the owner of the function and verify the owner has been changed on the workers
ALTER PROCEDURE raise_info(text) OWNER TO procedureuser;
SELECT public.verify_function_is_same_on_workers('procedure_tests.raise_info(text)');
 verify_function_is_same_on_workers
---------------------------------------------------------------------
 t
(1 row)

SELECT run_command_on_workers($$
SELECT row(usename, nspname, proname)
FROM pg_proc
JOIN pg_user ON (usesysid = proowner)
JOIN pg_namespace ON (pg_namespace.oid = pronamespace)
WHERE proname = 'raise_info';
$$);
                      run_command_on_workers
---------------------------------------------------------------------
 (localhost,57637,t,"(procedureuser,procedure_tests,raise_info)")
 (localhost,57638,t,"(procedureuser,procedure_tests,raise_info)")
(2 rows)

-- change the schema of the procedure and verify the old schema doesn't exist anymore while
-- the new schema has the function.
ALTER PROCEDURE raise_info(text) SET SCHEMA procedure_tests2;
SELECT public.verify_function_is_same_on_workers('procedure_tests2.raise_info(text)');
 verify_function_is_same_on_workers
---------------------------------------------------------------------
 t
(1 row)

SELECT * FROM run_command_on_workers($$CALL procedure_tests.raise_info('hello');$$) ORDER BY 1,2;
 nodename  | nodeport | success |                                result
---------------------------------------------------------------------
 localhost |    57637 | f       | ERROR:  procedure procedure_tests.raise_info(unknown) does not exist
 localhost |    57638 | f       | ERROR:  procedure procedure_tests.raise_info(unknown) does not exist
(2 rows)

SELECT * FROM run_command_on_workers($$CALL procedure_tests2.raise_info('hello');$$) ORDER BY 1,2;
 nodename  | nodeport | success | result
---------------------------------------------------------------------
 localhost |    57637 | t       | CALL
 localhost |    57638 | t       | CALL
(2 rows)

ALTER PROCEDURE procedure_tests2.raise_info(text) SET SCHEMA procedure_tests;
DROP PROCEDURE raise_info(text);
-- call should fail as procedure should have been dropped
SELECT * FROM run_command_on_workers($$CALL procedure_tests.raise_info('hello');$$) ORDER BY 1,2;
 nodename  | nodeport | success |                                result
---------------------------------------------------------------------
 localhost |    57637 | f       | ERROR:  procedure procedure_tests.raise_info(unknown) does not exist
 localhost |    57638 | f       | ERROR:  procedure procedure_tests.raise_info(unknown) does not exist
(2 rows)

SET client_min_messages TO error; -- suppress cascading objects dropping
DROP SCHEMA procedure_tests CASCADE;
DROP SCHEMA procedure_tests2 CASCADE;
DROP USER procedureuser;
SELECT 1 FROM run_command_on_workers($$DROP USER procedureuser;$$);
 ?column?
---------------------------------------------------------------------
        1
        1
(2 rows)