Merge branch 'main' into codex/add-integration-tests

Author: seanwevans

Makefile

@@ -1,6 +1,9 @@
 EXTENSION = pg_os
 DATA = pg_os--1.0.sql
 PG_CONFIG ?= pg_config
-REGRESS = pg_os_basic create_user lock_file create_process load_unload_module
+
+REGRESS = pg_os_basic create_user lock_file create_process load_unload_module modules
+REGRESS_OPTS = --outputdir=/tmp/pg_os_regress
+
 PGXS := $(shell $(PG_CONFIG) --pgxs)
 include $(PGXS)

expected/modules.out

@@ -0,0 +1,61 @@
+-- tests for load_module and unload_module
+\set ECHO none
+SELECT module_name, loaded FROM modules ORDER BY id;
+ module_name | loaded 
+-------------+--------
+ mod1        | f
+ mod2        | f
+(2 rows)
+
+SELECT load_module('mod1');
+ load_module 
+-------------
+ 
+(1 row)
+
+SELECT module_name, loaded FROM modules ORDER BY id;
+ module_name | loaded 
+-------------+--------
+ mod1        | t
+ mod2        | f
+(2 rows)
+
+SELECT unload_module('mod1');
+ unload_module 
+---------------
+ 
+(1 row)
+
+SELECT module_name, loaded FROM modules ORDER BY id;
+ module_name | loaded 
+-------------+--------
+ mod1        | f
+ mod2        | f
+(2 rows)
+
+SELECT load_module('mod3');
+ load_module 
+-------------
+ 
+(1 row)
+
+SELECT module_name, loaded FROM modules ORDER BY id;
+ module_name | loaded 
+-------------+--------
+ mod1        | f
+ mod2        | f
+(2 rows)
+
+SELECT load_module('mod2');
+ load_module 
+-------------
+ 
+(1 row)
+
+SELECT module_name, loaded FROM modules ORDER BY id;
+ module_name | loaded 
+-------------+--------
+ mod1        | f
+ mod2        | t
+(2 rows)
+

sql/modules.sql

@@ -0,0 +1,53 @@
+-- tests for load_module and unload_module
+\set ECHO none
+SET client_min_messages TO warning;
+
+DROP TABLE IF EXISTS modules CASCADE;
+
+CREATE TABLE modules (
+    id SERIAL PRIMARY KEY,
+    module_name TEXT UNIQUE NOT NULL,
+    loaded BOOLEAN DEFAULT FALSE,
+    code TEXT,
+    created_at TIMESTAMP DEFAULT now()
+);
+
+CREATE OR REPLACE FUNCTION load_module(module_name TEXT) RETURNS VOID AS $$
+BEGIN
+    UPDATE modules
+    SET loaded = TRUE
+    WHERE modules.module_name = load_module.module_name;
+END;
+$$ LANGUAGE plpgsql;
+
+CREATE OR REPLACE FUNCTION unload_module(module_name TEXT) RETURNS VOID AS $$
+BEGIN
+    UPDATE modules
+    SET loaded = FALSE
+    WHERE modules.module_name = unload_module.module_name;
+END;
+$$ LANGUAGE plpgsql;
+
+INSERT INTO modules (module_name) VALUES ('mod1'), ('mod2');
+
+\set ECHO queries
+\set VERBOSITY terse
+
+-- initial state
+SELECT module_name, loaded FROM modules ORDER BY id;
+
+-- load mod1
+SELECT load_module('mod1');
+SELECT module_name, loaded FROM modules ORDER BY id;
+
+-- unload mod1
+SELECT unload_module('mod1');
+SELECT module_name, loaded FROM modules ORDER BY id;
+
+-- load non-existent module
+SELECT load_module('mod3');
+SELECT module_name, loaded FROM modules ORDER BY id;
+
+-- load mod2
+SELECT load_module('mod2');
+SELECT module_name, loaded FROM modules ORDER BY id;