chore: try to resolve pgjwt for pg 15

Author: samrose

ebssurrogate/files/unit-tests/unit-test-01.sql

@@ -5,37 +5,37 @@ DO $$
 DECLARE
     extension_array text[];
     orioledb_available boolean;
+    pg_version integer;
 BEGIN
+    -- Get PostgreSQL version (as integer, e.g., 15 for PostgreSQL 15.x)
+    SELECT current_setting('server_version_num')::integer / 10000 INTO pg_version;
+    
     -- Check if orioledb is available
     SELECT EXISTS (
         SELECT 1 FROM pg_available_extensions WHERE name = 'orioledb'
     ) INTO orioledb_available;
 
-    -- If available, create it and add to the expected extensions list
+    -- Base extensions list
+    extension_array := ARRAY[
+        'plpgsql',
+        'pg_stat_statements',
+        'pgsodium',
+        'pgtap',
+        'pg_graphql',
+        'pgcrypto',
+        'uuid-ossp',
+        'supabase_vault'
+    ];
+    
+    -- Add pgjwt if PostgreSQL version is 15 or higher
+    IF pg_version >= 15 THEN
+        extension_array := array_append(extension_array, 'pgjwt');
+    END IF;
+    
+    -- Add orioledb if available
     IF orioledb_available THEN
         CREATE EXTENSION IF NOT EXISTS orioledb;
-        extension_array := ARRAY[
-            'plpgsql',
-            'pg_stat_statements',
-            'pgsodium',
-            'pgtap',
-            'pg_graphql',
-            'pgcrypto',
-            'uuid-ossp',
-            'supabase_vault',
-            'orioledb'
-        ];
-    ELSE
-        extension_array := ARRAY[
-            'plpgsql',
-            'pg_stat_statements',
-            'pgsodium',
-            'pgtap',
-            'pg_graphql',
-            'pgcrypto',
-            'uuid-ossp',
-            'supabase_vault'
-        ];
+        extension_array := array_append(extension_array, 'orioledb');
     END IF;
 
     -- Set the array as a temporary variable to use in the test
@@ -48,7 +48,6 @@ SELECT extensions_are(
     string_to_array(current_setting('myapp.extensions'), ',')::text[]
 );
 
-
 SELECT has_schema('pg_toast');
 SELECT has_schema('pg_catalog');
 SELECT has_schema('information_schema');
@@ -59,4 +58,4 @@ SELECT function_privs_are('pgsodium', 'crypto_aead_det_encrypt', array['bytea',
 SELECT function_privs_are('pgsodium', 'crypto_aead_det_keygen', array[]::text[], 'service_role', array['EXECUTE']);
 
 SELECT * FROM finish();
-ROLLBACK;
+ROLLBACK;