chore: unit test

Author: samrose

Diff for: ebssurrogate/files/unit-tests/unit-test-01.sql

@@ -5,37 +5,39 @@ DO $$
 DECLARE
     extension_array text[];
     orioledb_available boolean;
+    pgcrypto_available boolean;
 BEGIN
     -- 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
+    
+    -- Check if pgcrypto is available
+    SELECT EXISTS (
+        SELECT 1 FROM pg_available_extensions WHERE name = 'pgcrypto'
+    ) INTO pgcrypto_available;
+    
+    -- Initialize base extension array
+    extension_array := ARRAY[
+        'plpgsql',
+        'pg_stat_statements',
+        'pgtap',
+        'pg_graphql',
+        'pgjwt',
+        'uuid-ossp',
+        'supabase_vault'
+    ];
+
+    -- If orioledb is available, add it
     IF orioledb_available THEN
         CREATE EXTENSION IF NOT EXISTS orioledb;
-        extension_array := ARRAY[
-            'plpgsql',
-            'pg_stat_statements',
-            'pgtap',
-            'pg_graphql',
-            'pgcrypto',
-            'pgjwt',
-            'uuid-ossp',
-            'supabase_vault',
-            'orioledb'
-        ];
-    ELSE
-        extension_array := ARRAY[
-            'plpgsql',
-            'pg_stat_statements',
-            'pgtap',
-            'pg_graphql',
-            'pgcrypto',
-            'pgjwt',
-            'uuid-ossp',
-            'supabase_vault'
-        ];
+        extension_array := array_append(extension_array, 'orioledb');
+    END IF;
+
+    -- If pgcrypto is available, add it
+    IF pgcrypto_available THEN
+        CREATE EXTENSION IF NOT EXISTS pgcrypto;
+        extension_array := array_append(extension_array, 'pgcrypto');
     END IF;
 
     -- Set the array as a temporary variable to use in the test