|
| 1 | +-- Fixtures to test deparser fixes from constructive-db PR #229 |
| 2 | +-- These exercise: PERFORM, INTO clause placement, record field qualification, RETURN handling |
| 3 | + |
| 4 | +-- Test 1: PERFORM statement (parser stores as SELECT, deparser must strip SELECT) |
| 5 | +CREATE FUNCTION test_perform_basic() RETURNS trigger |
| 6 | +LANGUAGE plpgsql AS $$ |
| 7 | +BEGIN |
| 8 | + PERFORM pg_notify('test_channel', 'message'); |
| 9 | + RETURN NEW; |
| 10 | +END$$; |
| 11 | + |
| 12 | +-- Test 2: PERFORM with function call and arguments |
| 13 | +CREATE FUNCTION test_perform_with_args() RETURNS trigger |
| 14 | +LANGUAGE plpgsql AS $$ |
| 15 | +BEGIN |
| 16 | + IF (TG_OP = 'INSERT' OR TG_OP = 'UPDATE') THEN |
| 17 | + PERFORM pg_notify(TG_ARGV[0], to_json(NEW)::text); |
| 18 | + RETURN NEW; |
| 19 | + END IF; |
| 20 | + IF (TG_OP = 'DELETE') THEN |
| 21 | + PERFORM pg_notify(TG_ARGV[0], to_json(OLD)::text); |
| 22 | + RETURN OLD; |
| 23 | + END IF; |
| 24 | + RETURN NULL; |
| 25 | +END$$; |
| 26 | + |
| 27 | +-- Test 3: INTO clause with record field target (recfield qualification) |
| 28 | +CREATE FUNCTION test_into_record_field() RETURNS trigger |
| 29 | +LANGUAGE plpgsql AS $$ |
| 30 | +BEGIN |
| 31 | + SELECT |
| 32 | + NEW.is_approved IS TRUE |
| 33 | + AND NEW.is_verified IS TRUE |
| 34 | + AND NEW.is_disabled IS FALSE INTO NEW.is_active; |
| 35 | + RETURN NEW; |
| 36 | +END$$; |
| 37 | + |
| 38 | +-- Test 4: INTO clause with subquery (depth-aware scanner must skip nested FROM) |
| 39 | +CREATE FUNCTION test_into_with_subquery() RETURNS trigger |
| 40 | +LANGUAGE plpgsql AS $$ |
| 41 | +DECLARE |
| 42 | + result_value int; |
| 43 | +BEGIN |
| 44 | + SELECT count(*) INTO result_value |
| 45 | + FROM (SELECT id FROM users WHERE id = NEW.user_id) sub; |
| 46 | + RETURN NEW; |
| 47 | +END$$; |
| 48 | + |
| 49 | +-- Test 5: INTO clause with multiple record fields |
| 50 | +CREATE FUNCTION test_into_multiple_fields() RETURNS trigger |
| 51 | +LANGUAGE plpgsql AS $$ |
| 52 | +BEGIN |
| 53 | + SELECT is_active, is_verified INTO NEW.is_active, NEW.is_verified |
| 54 | + FROM users WHERE id = NEW.user_id; |
| 55 | + RETURN NEW; |
| 56 | +END$$; |
| 57 | + |
| 58 | +-- Test 6: SETOF function with RETURN QUERY and bare RETURN |
| 59 | +CREATE FUNCTION test_setof_return_query(p_limit int) |
| 60 | +RETURNS SETOF int |
| 61 | +LANGUAGE plpgsql AS $$ |
| 62 | +BEGIN |
| 63 | + RETURN QUERY SELECT generate_series(1, p_limit); |
| 64 | + RETURN; |
| 65 | +END$$; |
| 66 | + |
| 67 | +-- Test 7: SETOF function with RETURN NEXT |
| 68 | +CREATE FUNCTION test_setof_return_next(p_count int) |
| 69 | +RETURNS SETOF text |
| 70 | +LANGUAGE plpgsql AS $$ |
| 71 | +DECLARE |
| 72 | + i int; |
| 73 | +BEGIN |
| 74 | + FOR i IN 1..p_count LOOP |
| 75 | + RETURN NEXT 'item_' || i::text; |
| 76 | + END LOOP; |
| 77 | + RETURN; |
| 78 | +END$$; |
| 79 | + |
| 80 | +-- Test 8: Void function with bare RETURN |
| 81 | +CREATE FUNCTION test_void_function(p_value text) |
| 82 | +RETURNS void |
| 83 | +LANGUAGE plpgsql AS $$ |
| 84 | +BEGIN |
| 85 | + RAISE NOTICE 'Value: %', p_value; |
| 86 | + RETURN; |
| 87 | +END$$; |
| 88 | + |
| 89 | +-- Test 9: Scalar function with RETURN NULL |
| 90 | +CREATE FUNCTION test_scalar_return_null() |
| 91 | +RETURNS int |
| 92 | +LANGUAGE plpgsql AS $$ |
| 93 | +BEGIN |
| 94 | + RETURN NULL; |
| 95 | +END$$; |
| 96 | + |
| 97 | +-- Test 10: Scalar function with conditional RETURN |
| 98 | +CREATE FUNCTION test_scalar_conditional(p_value int) |
| 99 | +RETURNS int |
| 100 | +LANGUAGE plpgsql AS $$ |
| 101 | +BEGIN |
| 102 | + IF p_value > 0 THEN |
| 103 | + RETURN p_value * 2; |
| 104 | + END IF; |
| 105 | + RETURN NULL; |
| 106 | +END$$; |
| 107 | + |
| 108 | +-- Test 11: OUT parameter function with bare RETURN |
| 109 | +CREATE FUNCTION test_out_params(OUT ok boolean, OUT message text) |
| 110 | +LANGUAGE plpgsql AS $$ |
| 111 | +BEGIN |
| 112 | + ok := true; |
| 113 | + message := 'success'; |
| 114 | + RETURN; |
| 115 | +END$$; |
| 116 | + |
| 117 | +-- Test 12: RETURNS TABLE function with RETURN QUERY |
| 118 | +CREATE FUNCTION test_returns_table(p_prefix text) |
| 119 | +RETURNS TABLE(id int, name text) |
| 120 | +LANGUAGE plpgsql AS $$ |
| 121 | +BEGIN |
| 122 | + RETURN QUERY SELECT 1, p_prefix || '_one'; |
| 123 | + RETURN QUERY SELECT 2, p_prefix || '_two'; |
| 124 | + RETURN; |
| 125 | +END$$; |
| 126 | + |
| 127 | +-- Test 13: Trigger function with complex logic |
| 128 | +CREATE FUNCTION test_trigger_complex() RETURNS trigger |
| 129 | +LANGUAGE plpgsql AS $$ |
| 130 | +DECLARE |
| 131 | + defaults_record record; |
| 132 | + bit_len int; |
| 133 | +BEGIN |
| 134 | + bit_len := bit_length(NEW.permissions); |
| 135 | + |
| 136 | + SELECT * INTO defaults_record |
| 137 | + FROM permission_defaults AS t |
| 138 | + LIMIT 1; |
| 139 | + |
| 140 | + IF found THEN |
| 141 | + NEW.is_approved := defaults_record.is_approved; |
| 142 | + NEW.is_verified := defaults_record.is_verified; |
| 143 | + END IF; |
| 144 | + |
| 145 | + IF NEW.is_owner IS TRUE THEN |
| 146 | + NEW.is_admin := true; |
| 147 | + NEW.is_approved := true; |
| 148 | + NEW.is_verified := true; |
| 149 | + END IF; |
| 150 | + |
| 151 | + SELECT |
| 152 | + NEW.is_approved IS TRUE |
| 153 | + AND NEW.is_verified IS TRUE |
| 154 | + AND NEW.is_disabled IS FALSE INTO NEW.is_active; |
| 155 | + |
| 156 | + RETURN NEW; |
| 157 | +END$$; |
| 158 | + |
| 159 | +-- Test 14: Procedure (implicit void return) |
| 160 | +CREATE PROCEDURE test_procedure(p_message text) |
| 161 | +LANGUAGE plpgsql AS $$ |
| 162 | +BEGIN |
| 163 | + RAISE NOTICE '%', p_message; |
| 164 | +END$$; |
0 commit comments