pg_spec_helper 1.4.0 → 1.5.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 68e7d2fca3250ae565d38d0709b1eb73d212b5e6af6c642702e6345123a55e3a
-  data.tar.gz: 9d2340c8d9f285e97ffce51942c359843067c663f0383e0e4a9f752382b91eff
+  metadata.gz: '0448eb4ca85825f030ce96b4138d8bbf1ecce2dd51e6767a936591582034bd99'
+  data.tar.gz: bd24c7db5a5ad3c267dd4b26d7b753aaa547db79e662235d96bb027e5cc8e5e7
 SHA512:
-  metadata.gz: d124e944257073e9a37a1c39ef1a50ba64a86adc4f436312af1b69554a6b8ebdf36d0fedb375e89a9a521fc7e23f2c08f23bf80b15cdbe7c90c27e04542fc656
-  data.tar.gz: 3cb9e58e2bd59553f94cee19af7f991f14174e8ee51fc8160eb923ebe467e54543a03b9d0895a6f09dd58489763af2b0299dbee3c6e04e8063af28502c17f2f1
+  metadata.gz: 8ec24b586e7b1550c638b9c1387ef3ac60b244ab35c857f95ab75e661f725f2c2eefeb65326ca946c2f0c97593db15bb8e3e156831cbbe72631d3afc46f29215
+  data.tar.gz: b4b3982f19db7d7fc2ee038167eb8fea189a368d1ac6cb98916da60008635ba2293998216a057e451523b45855b6c0d8d8992ecda80b86c030f929fa306d12f1

data/CHANGELOG.md

@@ -1,5 +1,12 @@
 # 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)
 
 

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/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.4.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.4.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