-- Citus upgrades are finished by calling a procedure
-- Note that pg_catalog.citus_finish_citus_upgrade() behaves differently
-- when last upgrade citus version is less than 11
-- so we have two alternative outputs for this test
\set upgrade_test_old_citus_version `echo "$CITUS_OLD_VERSION"`
SELECT substring(:'upgrade_test_old_citus_version', 'v(\d+)\.\d+\.\d+')::int < 11
AS upgrade_test_old_citus_version_lt_11_0;
 upgrade_test_old_citus_version_lt_11_0
---------------------------------------------------------------------
 t
(1 row)

-- this is a transactional procedure, so rollback should be fine
BEGIN;
	CALL citus_finish_citus_upgrade();
NOTICE:  Preparing all the existing partitioned table indexes
NOTICE:  Preparing to sync the metadata to all nodes
ROLLBACK;
-- do the actual job
CALL citus_finish_citus_upgrade();
NOTICE:  Preparing all the existing partitioned table indexes
NOTICE:  Preparing to sync the metadata to all nodes
-- show that the upgrade is successfull
SELECT metadata->>'last_upgrade_version' = extversion
FROM pg_dist_node_metadata, pg_extension WHERE extname = 'citus';
 ?column?
---------------------------------------------------------------------
 t
(1 row)

-- idempotent, should be called multiple times
-- still, do not NOTICE the version as it changes per release
SET client_min_messages TO WARNING;
CALL citus_finish_citus_upgrade();
-- we should be able to sync metadata in nontransactional way as well
SET citus.metadata_sync_mode TO 'nontransactional';
SELECT start_metadata_sync_to_all_nodes();
 start_metadata_sync_to_all_nodes
---------------------------------------------------------------------
 t
(1 row)

RESET citus.metadata_sync_mode;
