Module: PGSpecHelper::Models

Included in:
PGSpecHelper
Defined in:
lib/pg_spec_helper/models.rb

Instance Method Summary collapse

Instance Method Details

#create_model(schema_name, table_name, &block) ⇒ Object

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



12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
 
# File 'lib/pg_spec_helper/models.rb', line 12

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
  # required for auto increment
  connection.exec(<<~SQL)
    -- temporarily set the client_min_messages to WARNING to
    -- suppress the NOTICE messages about extension already existing
    SET client_min_messages TO WARNING;
    CREATE EXTENSION IF NOT EXISTS "uuid-ossp";
    SET client_min_messages TO NOTICE;
  SQL
  # create the standard columns
  create_column schema_name, table_name, :id, :uuid, false, "uuid_generate_v4()"
  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], :"#{table_name}_pkey"
  # 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