pg_spec_helper 1.8.5 → 1.9.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 80425037527e477dc7bbb76dbeba05d0a9339f6cf829ee1f9d17338c99bb3e95
-  data.tar.gz: b7a29ce7a34962211ce7100330754b9d10dc53c0910efd5d987c776b9ca9e819
+  metadata.gz: 5c916286f724e81e2f7e1dd400127a4146f0d9153e378ed57ae89ee48530d545
+  data.tar.gz: 5a027b821ffbbd5cdec38f2d7b9116a10f7f4ba26190ff0d63db9d021eb3ce39
 SHA512:
-  metadata.gz: a47cc3dc6c0744df722e420af888ffb552b6df4aeb44caeef0e8d5c6a8f25e4ec2218ac6f537caca7d23d891e09631aa8951b57934b5bea070a8535b21d22807
-  data.tar.gz: d255fd2adaffd140017ff8c8a8961e1251782b6bee1773ebaee283e86be75a5acbd70c5df5aa87a9f236266b34f9cffc331e07e0581958c62d4918e1f47d83f0
+  metadata.gz: f19e1f15f8779f2d9fc34af866f632bd3e6999712ba08d025498955a1f36af0a4b0606e42f03dea66de15de838354b8b5b0f44ea4b1fc941b4de43ee7aac96af
+  data.tar.gz: a25b821f11ef368720ab26ab3a85c31bfa5930402841f4f8a8030ae11071d71276dece9ded790e73925e70a03bf08121a2395cefd670a02530fe69602e9dd140

data/CHANGELOG.md

@@ -1,5 +1,19 @@
 # Changelog
 
+## [1.9.1](https://github.com/craigulliott/pg_spec_helper/compare/v1.9.0...v1.9.1) (2023-08-23)
+
+
+### Bug Fixes
+
+* track use of functions, triggers and enums so we erase the database correctly between tests ([c705a0d](https://github.com/craigulliott/pg_spec_helper/commit/c705a0defe011c715de18c10f16861a02308122b))
+
+## [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/track_changes.rb

@@ -18,7 +18,10 @@ class PGSpecHelper
       :create_index,
       :create_primary_key,
       :create_unique_constraint,
-      :create_validation
+      :create_validation,
+      :create_function,
+      :create_trigger,
+      :create_enum
     ]
 
     # returns true if any changes have been made to the database structure
@@ -63,11 +66,11 @@ class PGSpecHelper
         # keep a pointer to the original method
         original_method = self.class.instance_method(method_name)
         # ovveride the original method
-        self.class.define_method(method_name) do |*args|
+        self.class.define_method(method_name) do |*args, **kwargs|
           # note that this method was called
           track_change method_name
           # call the original method
-          original_method.bind_call(self, *args)
+          original_method.bind_call(self, *args, **kwargs)
           # do any materialized views need to be refreshed?
           refresh_materialized_views_by_method method_name
         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.1"
 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.1
 platform: ruby
 authors:
 - Craig Ulliott
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2023-08-19 00:00:00.000000000 Z
+date: 2023-08-23 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