pg_spec_helper 1.8.5 → 1.9.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 80425037527e477dc7bbb76dbeba05d0a9339f6cf829ee1f9d17338c99bb3e95
-  data.tar.gz: b7a29ce7a34962211ce7100330754b9d10dc53c0910efd5d987c776b9ca9e819
+  metadata.gz: 111948cb669c62fe5e30ff56765ec20856dc53493a87a16304087068dfae8e2e
+  data.tar.gz: 56df8e6b579bb74df75be7e5b8e5a62b2db35ef737d084f92706982e5e39a032
 SHA512:
-  metadata.gz: a47cc3dc6c0744df722e420af888ffb552b6df4aeb44caeef0e8d5c6a8f25e4ec2218ac6f537caca7d23d891e09631aa8951b57934b5bea070a8535b21d22807
-  data.tar.gz: d255fd2adaffd140017ff8c8a8961e1251782b6bee1773ebaee283e86be75a5acbd70c5df5aa87a9f236266b34f9cffc331e07e0581958c62d4918e1f47d83f0
+  metadata.gz: a03eb171d281e9689c2e4bc1928845a3014e4a1f411f2c6419150bf617ce3fab4d4a6c9e9de8daa1ce885a65daa9e82310bcc92b9b61152eea1b8d1afa6f260a
+  data.tar.gz: e2749f90e7d1dab240f19548de538095d378a62b1b98ae6c69be64866a07fa267175ead068be77fbd15f1f70af514a4a7fa63974e9c8aec1bf8b3962dd8c06cd

data/CHANGELOG.md

@@ -1,5 +1,12 @@
 # Changelog
 
+## [1.9.0](https://github.com/craigulliott/pg_spec_helper/compare/v1.8.5...v1.9.0) (2023-08-21)
+
+
+### Features
+
+* added support for enums and extensions ([b22f662](https://github.com/craigulliott/pg_spec_helper/commit/b22f662cad22c1220ac6f4bc029ca534348a4bea))
+
 ## [1.8.5](https://github.com/craigulliott/pg_spec_helper/compare/v1.8.4...v1.8.5) (2023-08-19)
 
 

data/lib/pg_spec_helper/enums.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+class PGSpecHelper
+  module Enums
+    # Create an enum
+    def create_enum schema_name, enum_name, values
+      connection.exec(<<~SQL)
+        CREATE TYPE #{schema_name}.#{enum_name} as ENUM ('#{values.join("','")}')
+      SQL
+    end
+
+    # Drop an enum
+    def drop_enum schema_name, enum_name
+      connection.exec(<<~SQL)
+        DROP TYPE #{schema_name}.#{enum_name}
+      SQL
+    end
+
+    # get a list of enum names for the provided schema
+    def get_enum_names schema_name
+      rows = connection.exec_params(<<~SQL, [schema_name.to_s])
+        SELECT
+          t.typname AS enum_name
+        FROM pg_type t
+          JOIN pg_catalog.pg_namespace n ON n.oid = t.typnamespace
+        WHERE n.nspname = $1 AND t.typname[0] != '_'
+        GROUP BY t.typname
+      SQL
+      rows.map { |row| row["enum_name"].to_sym }
+    end
+  end
+end

data/lib/pg_spec_helper/extensions.rb

@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+
+class PGSpecHelper
+  module Extensions
+    # Create an extension
+    def create_extension extension_name
+      connection.exec(<<~SQL)
+        CREATE EXTENSION "#{extension_name}"
+      SQL
+    end
+
+    # Drop an extension
+    def drop_extension extension_name
+      connection.exec(<<~SQL)
+        DROP EXTENSION "#{extension_name}"
+      SQL
+    end
+
+    # get a list of extension names for the provided table
+    def get_extension_names
+      rows = connection.exec_params(<<~SQL)
+        SELECT
+          extname AS name
+        FROM pg_extension
+      SQL
+      rows.map { |row| row["name"].to_sym }
+    end
+  end
+end

data/lib/pg_spec_helper/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 class PGSpecHelper
-  VERSION = "1.8.5"
+  VERSION = "1.9.0"
 end

data/lib/pg_spec_helper.rb

@@ -17,6 +17,8 @@ require "pg_spec_helper/primary_keys"
 require "pg_spec_helper/indexes"
 require "pg_spec_helper/triggers"
 require "pg_spec_helper/functions"
+require "pg_spec_helper/enums"
+require "pg_spec_helper/extensions"
 require "pg_spec_helper/models"
 require "pg_spec_helper/materialized_views"
 require "pg_spec_helper/reset"
@@ -41,6 +43,8 @@ class PGSpecHelper
   include Indexes
   include Triggers
   include Functions
+  include Extensions
+  include Enums
   include Models
   include MaterializedViews
   include Reset

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: pg_spec_helper
 version: !ruby/object:Gem::Version
-  version: 1.8.5
+  version: 1.9.0
 platform: ruby
 authors:
 - Craig Ulliott
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2023-08-19 00:00:00.000000000 Z
+date: 2023-08-21 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: pg
@@ -54,6 +54,8 @@ files:
 - lib/pg_spec_helper/columns.rb
 - lib/pg_spec_helper/connection.rb
 - lib/pg_spec_helper/empty_database.rb
+- lib/pg_spec_helper/enums.rb
+- lib/pg_spec_helper/extensions.rb
 - lib/pg_spec_helper/foreign_keys.rb
 - lib/pg_spec_helper/functions.rb
 - lib/pg_spec_helper/ignored_schemas.rb