Adding upstream version 14.5.upstream/14.5upstream

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>

1 files changed, 287 insertions, 0 deletions

diff --git a/src/test/regress/sql/object_address.sql b/src/test/regress/sql/object_address.sql
new file mode 100644
index 0000000..2f4f66e
--- /dev/null
+++ b/src/test/regress/sql/object_address.sql
@@ -0,0 +1,287 @@
+--
+-- Test for pg_get_object_address
+--
+
+-- Clean up in case a prior regression run failed
+SET client_min_messages TO 'warning';
+DROP ROLE IF EXISTS regress_addr_user;
+RESET client_min_messages;
+
+CREATE USER regress_addr_user;
+
+-- Test generic object addressing/identification functions
+CREATE SCHEMA addr_nsp;
+SET search_path TO 'addr_nsp';
+CREATE FOREIGN DATA WRAPPER addr_fdw;
+CREATE SERVER addr_fserv FOREIGN DATA WRAPPER addr_fdw;
+CREATE TEXT SEARCH DICTIONARY addr_ts_dict (template=simple);
+CREATE TEXT SEARCH CONFIGURATION addr_ts_conf (copy=english);
+CREATE TEXT SEARCH TEMPLATE addr_ts_temp (lexize=dsimple_lexize);
+CREATE TEXT SEARCH PARSER addr_ts_prs
+    (start = prsd_start, gettoken = prsd_nexttoken, end = prsd_end, lextypes = prsd_lextype);
+CREATE TABLE addr_nsp.gentable (
+	a serial primary key CONSTRAINT a_chk CHECK (a > 0),
+	b text DEFAULT 'hello');
+CREATE TABLE addr_nsp.parttable (
+	a int PRIMARY KEY
+) PARTITION BY RANGE (a);
+CREATE VIEW addr_nsp.genview AS SELECT * from addr_nsp.gentable;
+CREATE MATERIALIZED VIEW addr_nsp.genmatview AS SELECT * FROM addr_nsp.gentable;
+CREATE TYPE addr_nsp.gencomptype AS (a int);
+CREATE TYPE addr_nsp.genenum AS ENUM ('one', 'two');
+CREATE FOREIGN TABLE addr_nsp.genftable (a int) SERVER addr_fserv;
+CREATE AGGREGATE addr_nsp.genaggr(int4) (sfunc = int4pl, stype = int4);
+CREATE DOMAIN addr_nsp.gendomain AS int4 CONSTRAINT domconstr CHECK (value > 0);
+CREATE FUNCTION addr_nsp.trig() RETURNS TRIGGER LANGUAGE plpgsql AS $$ BEGIN END; $$;
+CREATE TRIGGER t BEFORE INSERT ON addr_nsp.gentable FOR EACH ROW EXECUTE PROCEDURE addr_nsp.trig();
+CREATE POLICY genpol ON addr_nsp.gentable;
+CREATE PROCEDURE addr_nsp.proc(int4) LANGUAGE SQL AS $$ $$;
+CREATE SERVER "integer" FOREIGN DATA WRAPPER addr_fdw;
+CREATE USER MAPPING FOR regress_addr_user SERVER "integer";
+ALTER DEFAULT PRIVILEGES FOR ROLE regress_addr_user IN SCHEMA public GRANT ALL ON TABLES TO regress_addr_user;
+ALTER DEFAULT PRIVILEGES FOR ROLE regress_addr_user REVOKE DELETE ON TABLES FROM regress_addr_user;
+-- this transform would be quite unsafe to leave lying around,
+-- except that the SQL language pays no attention to transforms:
+CREATE TRANSFORM FOR int LANGUAGE SQL (
+	FROM SQL WITH FUNCTION prsd_lextype(internal),
+	TO SQL WITH FUNCTION int4recv(internal));
+-- suppress warning that depends on wal_level
+SET client_min_messages = 'ERROR';
+CREATE PUBLICATION addr_pub FOR TABLE addr_nsp.gentable;
+RESET client_min_messages;
+CREATE SUBSCRIPTION regress_addr_sub CONNECTION '' PUBLICATION bar WITH (connect = false, slot_name = NONE);
+CREATE STATISTICS addr_nsp.gentable_stat ON a, b FROM addr_nsp.gentable;
+
+-- test some error cases
+SELECT pg_get_object_address('stone', '{}', '{}');
+SELECT pg_get_object_address('table', '{}', '{}');
+SELECT pg_get_object_address('table', '{NULL}', '{}');
+
+-- unrecognized object types
+DO $$
+DECLARE
+	objtype text;
+BEGIN
+	FOR objtype IN VALUES ('toast table'), ('index column'), ('sequence column'),
+		('toast table column'), ('view column'), ('materialized view column')
+	LOOP
+		BEGIN
+			PERFORM pg_get_object_address(objtype, '{one}', '{}');
+		EXCEPTION WHEN invalid_parameter_value THEN
+			RAISE WARNING 'error for %: %', objtype, sqlerrm;
+		END;
+	END LOOP;
+END;
+$$;
+
+-- miscellaneous other errors
+select * from pg_get_object_address('operator of access method', '{btree,integer_ops,1}', '{int4,bool}');
+select * from pg_get_object_address('operator of access method', '{btree,integer_ops,99}', '{int4,int4}');
+select * from pg_get_object_address('function of access method', '{btree,integer_ops,1}', '{int4,bool}');
+select * from pg_get_object_address('function of access method', '{btree,integer_ops,99}', '{int4,int4}');
+
+DO $$
+DECLARE
+	objtype text;
+	names	text[];
+	args	text[];
+BEGIN
+	FOR objtype IN VALUES
+		('table'), ('index'), ('sequence'), ('view'),
+		('materialized view'), ('foreign table'),
+		('table column'), ('foreign table column'),
+		('aggregate'), ('function'), ('procedure'), ('type'), ('cast'),
+		('table constraint'), ('domain constraint'), ('conversion'), ('default value'),
+		('operator'), ('operator class'), ('operator family'), ('rule'), ('trigger'),
+		('text search parser'), ('text search dictionary'),
+		('text search template'), ('text search configuration'),
+		('policy'), ('user mapping'), ('default acl'), ('transform'),
+		('operator of access method'), ('function of access method'),
+		('publication relation')
+	LOOP
+		FOR names IN VALUES ('{eins}'), ('{addr_nsp, zwei}'), ('{eins, zwei, drei}')
+		LOOP
+			FOR args IN VALUES ('{}'), ('{integer}')
+			LOOP
+				BEGIN
+					PERFORM pg_get_object_address(objtype, names, args);
+				EXCEPTION WHEN OTHERS THEN
+						RAISE WARNING 'error for %,%,%: %', objtype, names, args, sqlerrm;
+				END;
+			END LOOP;
+		END LOOP;
+	END LOOP;
+END;
+$$;
+
+-- these object types cannot be qualified names
+SELECT pg_get_object_address('language', '{one}', '{}');
+SELECT pg_get_object_address('language', '{one,two}', '{}');
+SELECT pg_get_object_address('large object', '{123}', '{}');
+SELECT pg_get_object_address('large object', '{123,456}', '{}');
+SELECT pg_get_object_address('large object', '{blargh}', '{}');
+SELECT pg_get_object_address('schema', '{one}', '{}');
+SELECT pg_get_object_address('schema', '{one,two}', '{}');
+SELECT pg_get_object_address('role', '{one}', '{}');
+SELECT pg_get_object_address('role', '{one,two}', '{}');
+SELECT pg_get_object_address('database', '{one}', '{}');
+SELECT pg_get_object_address('database', '{one,two}', '{}');
+SELECT pg_get_object_address('tablespace', '{one}', '{}');
+SELECT pg_get_object_address('tablespace', '{one,two}', '{}');
+SELECT pg_get_object_address('foreign-data wrapper', '{one}', '{}');
+SELECT pg_get_object_address('foreign-data wrapper', '{one,two}', '{}');
+SELECT pg_get_object_address('server', '{one}', '{}');
+SELECT pg_get_object_address('server', '{one,two}', '{}');
+SELECT pg_get_object_address('extension', '{one}', '{}');
+SELECT pg_get_object_address('extension', '{one,two}', '{}');
+SELECT pg_get_object_address('event trigger', '{one}', '{}');
+SELECT pg_get_object_address('event trigger', '{one,two}', '{}');
+SELECT pg_get_object_address('access method', '{one}', '{}');
+SELECT pg_get_object_address('access method', '{one,two}', '{}');
+SELECT pg_get_object_address('publication', '{one}', '{}');
+SELECT pg_get_object_address('publication', '{one,two}', '{}');
+SELECT pg_get_object_address('subscription', '{one}', '{}');
+SELECT pg_get_object_address('subscription', '{one,two}', '{}');
+
+-- test successful cases
+WITH objects (type, name, args) AS (VALUES
+				('table', '{addr_nsp, gentable}'::text[], '{}'::text[]),
+				('table', '{addr_nsp, parttable}'::text[], '{}'::text[]),
+				('index', '{addr_nsp, gentable_pkey}', '{}'),
+				('index', '{addr_nsp, parttable_pkey}', '{}'),
+				('sequence', '{addr_nsp, gentable_a_seq}', '{}'),
+				-- toast table
+				('view', '{addr_nsp, genview}', '{}'),
+				('materialized view', '{addr_nsp, genmatview}', '{}'),
+				('foreign table', '{addr_nsp, genftable}', '{}'),
+				('table column', '{addr_nsp, gentable, b}', '{}'),
+				('foreign table column', '{addr_nsp, genftable, a}', '{}'),
+				('aggregate', '{addr_nsp, genaggr}', '{int4}'),
+				('function', '{pg_catalog, pg_identify_object}', '{pg_catalog.oid, pg_catalog.oid, int4}'),
+				('procedure', '{addr_nsp, proc}', '{int4}'),
+				('type', '{pg_catalog._int4}', '{}'),
+				('type', '{addr_nsp.gendomain}', '{}'),
+				('type', '{addr_nsp.gencomptype}', '{}'),
+				('type', '{addr_nsp.genenum}', '{}'),
+				('cast', '{int8}', '{int4}'),
+				('collation', '{default}', '{}'),
+				('table constraint', '{addr_nsp, gentable, a_chk}', '{}'),
+				('domain constraint', '{addr_nsp.gendomain}', '{domconstr}'),
+				('conversion', '{pg_catalog, koi8_r_to_mic}', '{}'),
+				('default value', '{addr_nsp, gentable, b}', '{}'),
+				('language', '{plpgsql}', '{}'),
+				-- large object
+				('operator', '{+}', '{int4, int4}'),
+				('operator class', '{btree, int4_ops}', '{}'),
+				('operator family', '{btree, integer_ops}', '{}'),
+				('operator of access method', '{btree,integer_ops,1}', '{integer,integer}'),
+				('function of access method', '{btree,integer_ops,2}', '{integer,integer}'),
+				('rule', '{addr_nsp, genview, _RETURN}', '{}'),
+				('trigger', '{addr_nsp, gentable, t}', '{}'),
+				('schema', '{addr_nsp}', '{}'),
+				('text search parser', '{addr_ts_prs}', '{}'),
+				('text search dictionary', '{addr_ts_dict}', '{}'),
+				('text search template', '{addr_ts_temp}', '{}'),
+				('text search configuration', '{addr_ts_conf}', '{}'),
+				('role', '{regress_addr_user}', '{}'),
+				-- database
+				-- tablespace
+				('foreign-data wrapper', '{addr_fdw}', '{}'),
+				('server', '{addr_fserv}', '{}'),
+				('user mapping', '{regress_addr_user}', '{integer}'),
+				('default acl', '{regress_addr_user,public}', '{r}'),
+				('default acl', '{regress_addr_user}', '{r}'),
+				-- extension
+				-- event trigger
+				('policy', '{addr_nsp, gentable, genpol}', '{}'),
+				('transform', '{int}', '{sql}'),
+				('access method', '{btree}', '{}'),
+				('publication', '{addr_pub}', '{}'),
+				('publication relation', '{addr_nsp, gentable}', '{addr_pub}'),
+				('subscription', '{regress_addr_sub}', '{}'),
+				('statistics object', '{addr_nsp, gentable_stat}', '{}')
+        )
+SELECT (pg_identify_object(addr1.classid, addr1.objid, addr1.objsubid)).*,
+	-- test roundtrip through pg_identify_object_as_address
+	ROW(pg_identify_object(addr1.classid, addr1.objid, addr1.objsubid)) =
+	ROW(pg_identify_object(addr2.classid, addr2.objid, addr2.objsubid))
+	  FROM objects, pg_get_object_address(type, name, args) addr1,
+			pg_identify_object_as_address(classid, objid, objsubid) ioa(typ,nms,args),
+			pg_get_object_address(typ, nms, ioa.args) as addr2
+	ORDER BY addr1.classid, addr1.objid, addr1.objsubid;
+
+---
+--- Cleanup resources
+---
+DROP FOREIGN DATA WRAPPER addr_fdw CASCADE;
+DROP PUBLICATION addr_pub;
+DROP SUBSCRIPTION regress_addr_sub;
+
+DROP SCHEMA addr_nsp CASCADE;
+
+DROP OWNED BY regress_addr_user;
+DROP USER regress_addr_user;
+
+--
+-- Checks for invalid objects
+--
+-- Make sure that NULL handling is correct.
+\pset null 'NULL'
+-- Temporarily disable fancy output, so as future additions never create
+-- a large amount of diffs.
+\a\t
+
+-- Keep this list in the same order as getObjectIdentityParts()
+-- in objectaddress.c.
+WITH objects (classid, objid, objsubid) AS (VALUES
+    ('pg_class'::regclass, 0, 0), -- no relation
+    ('pg_class'::regclass, 'pg_class'::regclass, 100), -- no column for relation
+    ('pg_proc'::regclass, 0, 0), -- no function
+    ('pg_type'::regclass, 0, 0), -- no type
+    ('pg_cast'::regclass, 0, 0), -- no cast
+    ('pg_collation'::regclass, 0, 0), -- no collation
+    ('pg_constraint'::regclass, 0, 0), -- no constraint
+    ('pg_conversion'::regclass, 0, 0), -- no conversion
+    ('pg_attrdef'::regclass, 0, 0), -- no default attribute
+    ('pg_language'::regclass, 0, 0), -- no language
+    ('pg_largeobject'::regclass, 0, 0), -- no large object, no error
+    ('pg_operator'::regclass, 0, 0), -- no operator
+    ('pg_opclass'::regclass, 0, 0), -- no opclass, no need to check for no access method
+    ('pg_opfamily'::regclass, 0, 0), -- no opfamily
+    ('pg_am'::regclass, 0, 0), -- no access method
+    ('pg_amop'::regclass, 0, 0), -- no AM operator
+    ('pg_amproc'::regclass, 0, 0), -- no AM proc
+    ('pg_rewrite'::regclass, 0, 0), -- no rewrite
+    ('pg_trigger'::regclass, 0, 0), -- no trigger
+    ('pg_namespace'::regclass, 0, 0), -- no schema
+    ('pg_statistic_ext'::regclass, 0, 0), -- no statistics
+    ('pg_ts_parser'::regclass, 0, 0), -- no TS parser
+    ('pg_ts_dict'::regclass, 0, 0), -- no TS dictionnary
+    ('pg_ts_template'::regclass, 0, 0), -- no TS template
+    ('pg_ts_config'::regclass, 0, 0), -- no TS configuration
+    ('pg_authid'::regclass, 0, 0), -- no role
+    ('pg_database'::regclass, 0, 0), -- no database
+    ('pg_tablespace'::regclass, 0, 0), -- no tablespace
+    ('pg_foreign_data_wrapper'::regclass, 0, 0), -- no FDW
+    ('pg_foreign_server'::regclass, 0, 0), -- no server
+    ('pg_user_mapping'::regclass, 0, 0), -- no user mapping
+    ('pg_default_acl'::regclass, 0, 0), -- no default ACL
+    ('pg_extension'::regclass, 0, 0), -- no extension
+    ('pg_event_trigger'::regclass, 0, 0), -- no event trigger
+    ('pg_policy'::regclass, 0, 0), -- no policy
+    ('pg_publication'::regclass, 0, 0), -- no publication
+    ('pg_publication_rel'::regclass, 0, 0), -- no publication relation
+    ('pg_subscription'::regclass, 0, 0), -- no subscription
+    ('pg_transform'::regclass, 0, 0) -- no transformation
+  )
+SELECT ROW(pg_identify_object(objects.classid, objects.objid, objects.objsubid))
+         AS ident,
+       ROW(pg_identify_object_as_address(objects.classid, objects.objid, objects.objsubid))
+         AS addr,
+       pg_describe_object(objects.classid, objects.objid, objects.objsubid)
+         AS descr
+FROM objects
+ORDER BY objects.classid, objects.objid, objects.objsubid;
+
+-- restore normal output mode
+\a\t