pg_spec_helper 1.3.0 → 1.5.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c1285cbc5b11e282ca1e564530c3b031bf0d8455a48d7f7e2b6af112216e0287
-  data.tar.gz: 9c63189e6193d584bae3e84081c50a3dc151e4b717044eee56da6b18e78da48c
+  metadata.gz: '0448eb4ca85825f030ce96b4138d8bbf1ecce2dd51e6767a936591582034bd99'
+  data.tar.gz: bd24c7db5a5ad3c267dd4b26d7b753aaa547db79e662235d96bb027e5cc8e5e7
 SHA512:
-  metadata.gz: f150ef8957731ef55129548b3f8ccc3ff95bf211d57b2b01850df80cba4d080d72323095ee0f1648f8189fcf9487a6d9c418e53565047f8509942a05c71f4039
-  data.tar.gz: 77ad4f5fde90f10bf13eae8390a157adb281ee1e910992b4531604dd8cdef71e72286c38caf887814b051aab74e5d6cd81d4dc0b5e96070bc0c5a245cb3fba2f
+  metadata.gz: 8ec24b586e7b1550c638b9c1387ef3ac60b244ab35c857f95ab75e661f725f2c2eefeb65326ca946c2f0c97593db15bb8e3e156831cbbe72631d3afc46f29215
+  data.tar.gz: b4b3982f19db7d7fc2ee038167eb8fea189a368d1ac6cb98916da60008635ba2293998216a057e451523b45855b6c0d8d8992ecda80b86c030f929fa306d12f1

data/CHANGELOG.md

@@ -1,5 +1,19 @@
 # Changelog
 
+## [1.5.0](https://github.com/craigulliott/pg_spec_helper/compare/v1.4.0...v1.5.0) (2023-08-01)
+
+
+### Features
+
+* adding a new create_model method which yields a block to expose a DSL for easily working with the table structure ([dc28d5e](https://github.com/craigulliott/pg_spec_helper/commit/dc28d5ef599d8306564aa7c29d2220fb22ee6dd6))
+
+## [1.4.0](https://github.com/craigulliott/pg_spec_helper/compare/v1.3.0...v1.4.0) (2023-07-10)
+
+
+### Features
+
+* made `refresh_all_materialized_views` available as a public method ([dd88ea0](https://github.com/craigulliott/pg_spec_helper/commit/dd88ea0877ba75f5c78ce5083421dd20090be6cb))
+
 ## [1.3.0](https://github.com/craigulliott/pg_spec_helper/compare/v1.2.0...v1.3.0) (2023-07-10)
 
 

data/lib/pg_spec_helper/columns.rb

@@ -2,12 +2,15 @@
 
 class PGSpecHelper
   module Columns
+    class ColumnDoesNotExistError < StandardError
+    end
+
     # create a column for the provided table
-    def create_column schema_name, table_name, column_name, type
+    def create_column schema_name, table_name, column_name, type, null = true
       # note the `type` is safe from sql_injection due to the validation above
       connection.exec(<<-SQL)
         ALTER TABLE #{connection.quote_ident schema_name.to_s}.#{connection.quote_ident table_name.to_s}
-          ADD COLUMN #{connection.quote_ident column_name.to_s} #{sanitize_name type}
+          ADD COLUMN #{connection.quote_ident column_name.to_s} #{sanitize_name type} #{null ? "" : "NOT NULL"}
       SQL
     end
 
@@ -22,5 +25,18 @@ class PGSpecHelper
       SQL
       rows.map { |row| row["column_name"].to_sym }
     end
+
+    # return an array of column names for the provided table
+    def is_column_nullable schema_name, table_name, column_name
+      rows = connection.exec_params(<<-SQL, [schema_name.to_s, table_name.to_s, column_name.to_s])
+        SELECT is_nullable
+        FROM information_schema.columns
+        WHERE table_schema = $1
+          AND table_name = $2
+          AND column_name = $3;
+      SQL
+      raise ColumnDoesNotExistError if rows.first.nil?
+      rows.first["is_nullable"] == "YES"
+    end
   end
 end

data/lib/pg_spec_helper/materialized_views.rb

@@ -30,6 +30,17 @@ class PGSpecHelper
       }
     end
 
+    # refresh all materialized views that have been tracked
+    def refresh_all_materialized_views
+      @materialized_views&.each do |schema_name, views|
+        views.each do |materialized_view_name, view|
+          if materialized_view_exists? schema_name, materialized_view_name
+            refresh_materialized_view schema_name, materialized_view_name
+          end
+        end
+      end
+    end
+
     private
 
     # given a trackable method name, refreshes any materialized
@@ -64,17 +75,6 @@ class PGSpecHelper
       SQL
     end
 
-    # refresh all materialized views that have been tracked
-    def refresh_all_materialized_views
-      @materialized_views&.each do |schema_name, views|
-        views.each do |materialized_view_name, view|
-          if materialized_view_exists? schema_name, materialized_view_name
-            refresh_materialized_view schema_name, materialized_view_name
-          end
-        end
-      end
-    end
-
     # whenever schema changes are made from within the test suite we need to
     # rebuild the materialized views that hold a cached representation of the
     # database structure

data/lib/pg_spec_helper/models.rb

@@ -0,0 +1,35 @@
+# frozen_string_literal: true
+
+class PGSpecHelper
+  module Models
+    # create a new table in the provided schema with an auto incrementing id
+    # created_at and updated_at, optionally provide a block which will be
+    # yeilded This allows for syntax such as:
+    #
+    # create_model :users, :optional_schema_name do |t|
+    #   t.add_colmn string :name
+    # end
+    def create_model schema_name, table_name, &block
+      unless schema_exists? schema_name
+        create_schema schema_name
+      end
+      # create the table
+      create_table schema_name, table_name
+      # create the standard columns
+      create_column schema_name, table_name, :id, :serial
+      create_column schema_name, table_name, :created_at, :timestamp
+      create_column schema_name, table_name, :updated_at, :timestamp
+      # add the primary key
+      create_primary_key schema_name, table_name, [:id], :pk
+      # execute the optional block
+      if block
+        this = self
+        if this.is_a? PGSpecHelper
+          TableExecuter.new(this, schema_name, table_name, &block)
+        else
+          raise "Module should be added to PGSpecHelper"
+        end
+      end
+    end
+  end
+end

data/lib/pg_spec_helper/table_executer.rb

@@ -0,0 +1,36 @@
+# frozen_string_literal: true
+
+class PGSpecHelper
+  class TableExecuter
+    def initialize pg_spec_helper, schema_name, table_name, &block
+      @pg_spec_helper = pg_spec_helper
+      @schema_name = schema_name
+      @table_name = table_name
+      instance_exec(&block)
+    end
+
+    def add_column column_name, type, null = true
+      @pg_spec_helper.create_column @schema_name, @table_name, column_name, type, null
+    end
+
+    def add_foreign_key column_names, foreign_schema_name, foreign_table_name, foreign_column_names, foreign_key_name
+      @pg_spec_helper.create_foreign_key @schema_name, @table_name, column_names, foreign_schema_name, foreign_table_name, foreign_column_names, foreign_key_name
+    end
+
+    def add_index column_names, index_name, unique = false
+      @pg_spec_helper.create_index @schema_name, @table_name, column_names, index_name
+    end
+
+    def add_primary_key column_names, primary_key_name
+      @pg_spec_helper.create_primary_key @schema_name, @table_name, column_names, primary_key_name
+    end
+
+    def add_unique_constraint column_names, constraint_key_name
+      @pg_spec_helper.create_unique_constraint @schema_name, @table_name, column_names, constraint_key_name
+    end
+
+    def add_validation validation_name, check_clause
+      @pg_spec_helper.create_validation @schema_name, @table_name, validation_name, check_clause
+    end
+  end
+end

data/lib/pg_spec_helper/version.rb

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

data/lib/pg_spec_helper.rb

@@ -16,11 +16,14 @@ require "pg_spec_helper/foreign_keys"
 require "pg_spec_helper/unique_constraints"
 require "pg_spec_helper/primary_keys"
 require "pg_spec_helper/indexes"
+require "pg_spec_helper/models"
 require "pg_spec_helper/materialized_views"
 require "pg_spec_helper/reset"
 require "pg_spec_helper/empty_database"
 require "pg_spec_helper/track_changes"
 
+require "pg_spec_helper/table_executer"
+
 class PGSpecHelper
   class MissingRequiredOptionError < StandardError
   end
@@ -36,6 +39,7 @@ class PGSpecHelper
   include UniqueConstraints
   include PrimaryKeys
   include Indexes
+  include Models
   include MaterializedViews
   include Reset
   include EmptyDatabase

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: pg_spec_helper
 version: !ruby/object:Gem::Version
-  version: 1.3.0
+  version: 1.5.0
 platform: ruby
 authors:
 - Craig Ulliott
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2023-07-10 00:00:00.000000000 Z
+date: 2023-08-01 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: pg
@@ -58,10 +58,12 @@ files:
 - lib/pg_spec_helper/ignored_schemas.rb
 - lib/pg_spec_helper/indexes.rb
 - lib/pg_spec_helper/materialized_views.rb
+- lib/pg_spec_helper/models.rb
 - lib/pg_spec_helper/primary_keys.rb
 - lib/pg_spec_helper/reset.rb
 - lib/pg_spec_helper/sanitize.rb
 - lib/pg_spec_helper/schemas.rb
+- lib/pg_spec_helper/table_executer.rb
 - lib/pg_spec_helper/tables.rb
 - lib/pg_spec_helper/track_changes.rb
 - lib/pg_spec_helper/unique_constraints.rb