declare_schema 2.0.0 → 2.1.0.pre.1

data/spec/lib/declare_schema/model/column_spec.rb

@@ -1,22 +1,14 @@
 # frozen_string_literal: true
 
-begin
-  require 'mysql2'
-rescue LoadError
-end
-
 require_relative '../../../../lib/declare_schema/model/column'
 
 RSpec.describe DeclareSchema::Model::Column do
-  before do
-    load File.expand_path('../prepare_testapp.rb', __dir__)
-  end
+  include_context 'prepare test app'
 
   describe 'class methods' do
     describe '.native_type?' do
       let(:native_types) { [:string, :text, :integer, :float, :decimal, :datetime, :time, :date, :binary, :boolean, :json] }
 
-
       it 'is falsey for :primary_key' do
         expect(described_class.native_type?(:primary_key)).to be_falsey
       end
@@ -40,60 +32,116 @@ RSpec.describe DeclareSchema::Model::Column do
     end
 
     describe '.native_types' do
-      subject { described_class.native_types }
-
-      it 'returns the native type for :primary_key' do
-        expect(subject[:primary_key]).to match(/auto_increment PRIMARY KEY|PRIMARY KEY AUTOINCREMENT NOT NULL/)
+      subject(:native_types) { described_class.native_types }
+
+      describe 'primary_key' do
+        subject { native_types[:primary_key] }
+        let(:expected_name) do
+          case current_adapter
+          when 'mysql2'
+            'bigint auto_increment PRIMARY KEY'
+          when 'postgresql'
+            'bigserial primary key'
+          when 'sqlite3'
+            'integer PRIMARY KEY AUTOINCREMENT NOT NULL'
+          end
+        end
+        it { is_expected.to eq(expected_name) }
       end
 
-      it 'returns the native type for :string' do
-        expect(subject.dig(:string, :name)).to eq('varchar')
+      describe 'string' do
+        subject { native_types[:string] }
+        let(:expected_name) { current_adapter == 'postgresql' ? 'character varying' : 'varchar' }
+        it { is_expected.to include(name: expected_name) }
       end
 
-      it 'returns the native type for :integer' do
-        expect(subject.dig(:integer, :name)).to match(/int/)
+      describe 'integer' do
+        subject { native_types[:integer] }
+        let(:expected_name) { /int/ }
+        it { is_expected.to include(name: expected_name) }
       end
 
-      it 'returns the native type for :datetime' do
-        expect(subject.dig(:datetime, :name)).to eq('datetime')
+      describe 'datetime' do
+        subject { native_types[:datetime] }
+        let(:expected_name) { current_adapter == 'postgresql' ? 'timestamp' : 'datetime' }
+        it { is_expected.to include(name: expected_name) }
       end
     end
 
     describe '.deserialize_default_value' do
-      it 'deserializes :boolean' do
-        expect(described_class.deserialize_default_value(nil, :boolean, 'true')).to eq(true)
-        expect(described_class.deserialize_default_value(nil, :boolean, 'false')).to eq(false)
+      subject { described_class.deserialize_default_value(col_spec, column_type, default) }
+
+      context 'when deserializing a boolean' do
+        let(:column_type) { :boolean }
+        let(:col_spec) do
+          if current_adapter == 'postgresql'
+            instance_double(ActiveRecord::ConnectionAdapters::PostgreSQL::Column, type: :boolean, sql_type: "boolean", oid: 16, fmod: -1)
+          else
+            instance_double(ActiveRecord::ConnectionAdapters::Column, type: :boolean, sql_type: "boolean")
+          end
+        end
+
+        context 'when the default is true' do
+          let(:default) { 'true' }
+          it { is_expected.to eq(true) }
+        end
+
+        context 'when the default is false' do
+          let(:default) { 'false' }
+          it { is_expected.to eq(false) }
+        end
       end
 
-      it 'deserializes :integer' do
-        expect(described_class.deserialize_default_value(nil, :integer, '12')).to eq(12)
+      context 'when deserializing an integer' do
+        let(:column_type) { :integer }
+        let(:default) { '12' }
+        let(:col_spec) do
+          if current_adapter == 'postgresql'
+            instance_double(ActiveRecord::ConnectionAdapters::PostgreSQL::Column, type: :integer, sql_type: "integer", limit: 4, oid: 23, fmod: -1)
+          else
+            instance_double(ActiveRecord::ConnectionAdapters::Column, type: :integer, sql_type: "integer", limit: 4)
+          end
+        end
+
+        it { is_expected.to eq(12) }
       end
 
-      it 'deserializes :json' do
-        expect(described_class.deserialize_default_value(nil, :json, '{}')).to eq({})
+      context 'when deserializing json' do
+        let(:column_type) { :json }
+        let(:default) { '{}' }
+        let(:col_spec) do
+          if current_adapter == 'postgresql'
+            instance_double(ActiveRecord::ConnectionAdapters::PostgreSQL::Column, type: :json, sql_type: "json", oid: 114, fmod: -1)
+          else
+            instance_double(ActiveRecord::ConnectionAdapters::Column, type: :json, sql_type: "json")
+          end
+        end
+
+        it { is_expected.to eq({}) }
       end
     end
   end
 
   describe 'instance methods' do
+    subject { described_class.new(model, current_table_name, column) }
+
     let(:model) { ColumnTestModel }
     let(:type) { :integer }
     let(:current_table_name) { model.table_name }
-    let(:column) { double("ActiveRecord Column",
-                          name: 'count',
-                          type: type,
-                          limit: nil,
-                          precision: nil,
-                          scale: nil,
-                          type_cast_from_database: nil,
-                          null: false,
-                          default: nil,
-                          sql_type_metadata: {}) }
-    subject { described_class.new(model, current_table_name, column) }
+    let(:column) do
+      if current_adapter == 'postgresql'
+        instance_double(ActiveRecord::ConnectionAdapters::PostgreSQL::Column,
+                        name: 'count', type: type, sql_type: type, limit: 4, precision: nil, scale: nil, null: false, default: nil,
+                        oid: 23, fmod: -1)
+      else
+        instance_double(ActiveRecord::ConnectionAdapters::Column, name: 'count', type: type, sql_type: type, limit: nil,
+                        precision: nil, scale: nil, null: false, default: nil)
+      end
+    end
 
     context 'Using declare_schema' do
       before do
-        class ColumnTestModel < ActiveRecord::Base
+        class ColumnTestModel < ActiveRecord::Base # rubocop:disable Lint/ConstantDefinitionInBlock
           declare_schema do
             string :title, limit: 127, null: false
             integer :count, null: false
@@ -109,10 +157,10 @@ RSpec.describe DeclareSchema::Model::Column do
 
       describe '#schema_attributes' do
         it 'returns a hash with relevant key/values' do
-          if defined?(Mysql2)
-            expect(subject.schema_attributes).to eq(type: :integer, null: false, limit: 4)
-          else
+          if current_adapter == 'sqlite3'
             expect(subject.schema_attributes).to eq(type: :integer, null: false)
+          else
+            expect(subject.schema_attributes).to eq(type: :integer, null: false, limit: 4)
           end
         end
       end

data/spec/lib/declare_schema/model/foreign_key_definition_spec.rb

@@ -3,13 +3,13 @@
 require_relative '../../../../lib/declare_schema/model/foreign_key_definition'
 
 RSpec.describe DeclareSchema::Model::ForeignKeyDefinition do
+  include_context 'prepare test app'
+
   let(:model_class) { Network }
 
   context 'Using declare_schema' do
     before do
-      load File.expand_path('../prepare_testapp.rb', __dir__)
-
-      class Network < ActiveRecord::Base
+      class Network < ActiveRecord::Base # rubocop:disable Lint/ConstantDefinitionInBlock
         declare_schema do
           string :name, limit: 127, index: true
 
@@ -95,13 +95,19 @@ RSpec.describe DeclareSchema::Model::ForeignKeyDefinition do
     end
 
     describe 'class << self' do
-      let(:connection) { instance_double(ActiveRecord::Base.connection.class) }
-      let(:model) { instance_double('Model', table_name: 'models', connection: connection) }
+      let(:connection)     { instance_double(ActiveRecord::Base.connection.class) }
+      let(:model)          { instance_double('Model', table_name: 'models', connection: connection) }
       let(:old_table_name) { 'networks' }
+      let(:foreign_keys) do
+        [
+          instance_double(ActiveRecord::ConnectionAdapters::ForeignKeyDefinition,
+                          column: 'network_id', name: 'constraint',
+                          from_table: 'networks', to_table: 'models', on_delete: nil)
+        ]
+      end
+
       before do
-        allow(connection).to receive(:quote_table_name).with('networks') { 'networks' }
-        allow(connection).to receive(:select_rows) { [['CONSTRAINT `constraint` FOREIGN KEY (`network_id`) REFERENCES `networks` (`id`)']] }
-        allow(connection).to receive(:index_name).with('models', column: 'network_id') { }
+        allow(connection).to receive(:foreign_keys).with(old_table_name) { foreign_keys }
       end
 
       describe '.for_table' do

data/spec/lib/declare_schema/model/habtm_model_shim_spec.rb

@@ -1,26 +1,21 @@
 # frozen_string_literal: true
 
-begin
-  require 'mysql2'
-rescue LoadError
-end
-
 require_relative '../../../../lib/declare_schema/model/habtm_model_shim'
 
 RSpec.describe DeclareSchema::Model::HabtmModelShim do
+  include_context 'prepare test app'
+
   let(:join_table) { "customers_users" }
   let(:foreign_keys) { ["user_id", "customer_id"] }
   let(:parent_table_names) { ["users", "customers"] }
   let(:connection) { instance_double(ActiveRecord::Base.connection.class, "connection") }
 
   before do
-    load File.expand_path('../prepare_testapp.rb', __dir__)
-
-    class User < ActiveRecord::Base
+    class User < ActiveRecord::Base # rubocop:disable Lint/ConstantDefinitionInBlock
       self.table_name = "users"
     end
 
-    class Customer < ActiveRecord::Base
+    class Customer < ActiveRecord::Base # rubocop:disable Lint/ConstantDefinitionInBlock
       self.table_name = "customers"
     end
   end

data/spec/lib/declare_schema/model/index_definition_spec.rb

@@ -14,10 +14,10 @@ RSpec.describe DeclareSchema::Model::IndexDefinition do
   let(:table_name) { model_class.table_name }
 
   context 'Using declare_schema' do
-    before do
-      load File.expand_path('../prepare_testapp.rb', __dir__)
+    include_context 'prepare test app'
 
-      class IndexDefinitionTestModel < ActiveRecord::Base
+    before do
+      class IndexDefinitionTestModel < ActiveRecord::Base # rubocop:disable Lint/ConstantDefinitionInBlock
         declare_schema do
           string :name, limit: 127, index: true
 
@@ -25,7 +25,7 @@ RSpec.describe DeclareSchema::Model::IndexDefinition do
         end
       end
 
-      class IndexDefinitionCompoundIndexModel < ActiveRecord::Base
+      class IndexDefinitionCompoundIndexModel < ActiveRecord::Base # rubocop:disable Lint/ConstantDefinitionInBlock
         declare_schema do
           integer :fk1_id
           integer :fk2_id
@@ -135,13 +135,13 @@ RSpec.describe DeclareSchema::Model::IndexDefinition do
         ActiveRecord::Base.connection.execute <<~EOS
           CREATE TABLE index_definition_test_models (
             id INTEGER NOT NULL PRIMARY KEY,
-            name #{if defined?(SQLite3) then 'TEXT' else 'VARCHAR(255)' end} NOT NULL
+            name #{if current_adapter == 'sqlite3' then 'TEXT' else 'VARCHAR(255)' end} NOT NULL
           )
         EOS
         ActiveRecord::Base.connection.execute <<~EOS
           CREATE UNIQUE INDEX index_definition_test_models_on_name ON index_definition_test_models(name)
         EOS
-        if defined?(Mysql2)
+        if current_adapter == 'mysql2'
           ActiveRecord::Base.connection.execute <<~EOS
             CREATE INDEX index_definition_test_models_on_name_partial ON index_definition_test_models(name(10))
           EOS
@@ -164,7 +164,7 @@ RSpec.describe DeclareSchema::Model::IndexDefinition do
           it 'returns the indexes for the model' do
             expect(subject.map(&:to_key)).to eq([
               ["index_definition_test_models_on_name", ["name"], { unique: true, where: nil, length: nil }],
-              (["index_definition_test_models_on_name_partial", ["name"], { unique: false, where: nil, length: { name: 10 } }] if defined?(Mysql2)),
+              (["index_definition_test_models_on_name_partial", ["name"], { unique: false, where: nil, length: { name: 10 } }] if current_adapter == 'mysql2'),
               ["PRIMARY", ["id"], { unique: true, where: nil, length: nil }]
             ].compact)
           end
@@ -185,7 +185,7 @@ RSpec.describe DeclareSchema::Model::IndexDefinition do
 
           it 'skips the ignored index' do
             expect(subject.map(&:to_key)).to eq([
-              (["index_definition_test_models_on_name_partial", ["name"], { unique: false, where: nil, length: { name: 10 } }] if defined?(Mysql2)),
+              (["index_definition_test_models_on_name_partial", ["name"], { unique: false, where: nil, length: { name: 10 } }] if current_adapter == 'mysql2'),
               ["PRIMARY", ["id"], { length: nil, unique: true, where: nil }]
             ].compact)
           end
@@ -214,12 +214,20 @@ RSpec.describe DeclareSchema::Model::IndexDefinition do
       end
 
       context 'with short max_index_and_constraint_name_length' do
+        include_context 'skip if' do
+          let(:adapter) { 'postgresql' }
+        end
+
         let(:max_index_and_constraint_name_length) { 40 }
 
         it { is_expected.to eq("users__last_name_first_name_middle_name") }
       end
 
       context 'with long table name' do
+        include_context 'skip if' do
+          let(:adapter) { 'postgresql' }
+        end
+
         let(:table_name2) { 'user_domains_extra' }
         {
           34 => '__last_name_first_name_middle_name',
@@ -279,7 +287,7 @@ RSpec.describe DeclareSchema::Model::IndexDefinition do
         context 'with multiple columns' do
           let(:columns) { ["last_name", "first_name"] }
 
-          it { expect { subject }.to raise_exception(ArgumentError, /Index length of Integer only allowed when exactly one column; got 10 for \["last_name", "first_name"]/i) }
+          it { expect { subject }.to raise_exception(ArgumentError, /Index length of Integer only allowed when exactly one column; got 10 for \["last_name", "first_name"\]/i) }
         end
       end
 
@@ -320,7 +328,7 @@ RSpec.describe DeclareSchema::Model::IndexDefinition do
       context 'with an invalid length' do
         let(:length) { 10.5 }
 
-        it { expect { subject }.to raise_exception(ArgumentError, /length must be nil or Integer or a Hash of column names to lengths; got 10\.5 for \[:last_name]/i) }
+        it { expect { subject }.to raise_exception(ArgumentError, /length must be nil or Integer or a Hash of column names to lengths; got 10\.5 for \[:last_name\]/i) }
       end
     end
   end

data/spec/lib/declare_schema/model/table_options_definition_spec.rb

@@ -1,22 +1,17 @@
 # frozen_string_literal: true
 
-begin
-  require 'mysql2'
-  require 'active_record/connection_adapters/mysql2_adapter'
-rescue LoadError
-end
 require_relative '../../../../lib/declare_schema/model/table_options_definition'
 
 RSpec.describe DeclareSchema::Model::TableOptionsDefinition do
+  include_context 'prepare test app'
+
   let(:model_class) { TableOptionsDefinitionTestModel }
   let(:charset) { DeclareSchema.normalize_charset('utf8') }
   let(:collation) { DeclareSchema.normalize_collation('utf8_general') } # adapt so that tests will pass on MySQL 5.7 or 8+
 
   context 'Using declare_schema' do
     before do
-      load File.expand_path('../prepare_testapp.rb', __dir__)
-
-      class TableOptionsDefinitionTestModel < ActiveRecord::Base
+      class TableOptionsDefinitionTestModel < ActiveRecord::Base # rubocop:disable Lint/ConstantDefinitionInBlock
         declare_schema do
           string :name, limit: 127, index: true
         end
@@ -36,7 +31,11 @@ RSpec.describe DeclareSchema::Model::TableOptionsDefinition do
         subject { model.settings }
         it { is_expected.to eq("CHARACTER SET #{charset} COLLATE #{collation}") }
 
-        if defined?(Mysql2)
+        context 'MySQL only' do
+          include_context 'skip unless' do
+            let(:adapter) { 'mysql2' }
+          end
+
           context 'when running in MySQL 8' do
             around do |spec|
               DeclareSchema.mysql_version = Gem::Version.new('8.0.21')
@@ -75,10 +74,11 @@ RSpec.describe DeclareSchema::Model::TableOptionsDefinition do
       describe '#for_model' do
         context 'when database migrated' do
           let(:options) do
-            if defined?(Mysql2)
+            case current_adapter(model_class)
+            when 'mysql2'
               { charset: "utf8mb4", collation: "utf8mb4_bin" }
             else
-              { }
+              {}
             end
           end
           subject { described_class.for_model(model_class) }

data/spec/lib/declare_schema/schema_change/base_spec.rb

@@ -3,24 +3,22 @@
 require_relative '../../../../lib/declare_schema/schema_change/base'
 
 RSpec.describe DeclareSchema::SchemaChange::Base do
-  before do
-    load File.expand_path('../prepare_testapp.rb', __dir__)
-  end
+  include_context 'prepare test app'
 
   before :all do
-    class ChangeDefault < described_class
+    class ChangeDefault < described_class # rubocop:disable Lint/ConstantDefinitionInBlock
       attr_reader :up_command, :down_command
 
-      def initialize(up:, down:)
+      def initialize(up:, down:) # rubocop:disable Naming/MethodParameterName
         @up_command = up
         @down_command = down
       end
     end
 
-    class ChangeOverride < described_class
+    class ChangeOverride < described_class # rubocop:disable Lint/ConstantDefinitionInBlock
       attr_reader :up_command, :down_command
 
-      def initialize(up:, down:)
+      def initialize(up:, down:) # rubocop:disable Naming/MethodParameterName
         @up_command = up
         @down_command = down
       end

data/spec/lib/declare_schema/schema_change/column_add_spec.rb

@@ -3,9 +3,7 @@
 require_relative '../../../../lib/declare_schema/schema_change/column_add'
 
 RSpec.describe DeclareSchema::SchemaChange::ColumnAdd do
-  before do
-    load File.expand_path('../prepare_testapp.rb', __dir__)
-  end
+  include_context 'prepare test app'
 
   let(:table_name) { 'networks' }
   let(:column_name) { 'title' }

data/spec/lib/declare_schema/schema_change/column_change_spec.rb

@@ -3,9 +3,7 @@
 require_relative '../../../../lib/declare_schema/schema_change/column_change'
 
 RSpec.describe DeclareSchema::SchemaChange::ColumnChange do
-  before do
-    load File.expand_path('../prepare_testapp.rb', __dir__)
-  end
+  include_context 'prepare test app'
 
   let(:table_name) { 'networks' }
   let(:column_name) { 'title' }

data/spec/lib/declare_schema/schema_change/column_remove_spec.rb

@@ -3,9 +3,7 @@
 require_relative '../../../../lib/declare_schema/schema_change/column_remove'
 
 RSpec.describe DeclareSchema::SchemaChange::ColumnRemove do
-  before do
-    load File.expand_path('../prepare_testapp.rb', __dir__)
-  end
+  include_context 'prepare test app'
 
   let(:table_name) { 'networks' }
   let(:column_name) { 'title' }

data/spec/lib/declare_schema/schema_change/column_rename_spec.rb

@@ -3,9 +3,7 @@
 require_relative '../../../../lib/declare_schema/schema_change/column_rename'
 
 RSpec.describe DeclareSchema::SchemaChange::ColumnRename do
-  before do
-    load File.expand_path('../prepare_testapp.rb', __dir__)
-  end
+  include_context 'prepare test app'
 
   let(:table_name) { 'networks' }
   let(:old_name) { 'title' }

data/spec/lib/declare_schema/schema_change/foreign_key_add_spec.rb

@@ -3,9 +3,7 @@
 require_relative '../../../../lib/declare_schema/schema_change/foreign_key_add'
 
 RSpec.describe DeclareSchema::SchemaChange::ForeignKeyAdd do
-  before do
-    load File.expand_path('../prepare_testapp.rb', __dir__)
-  end
+  include_context 'prepare test app'
 
   let(:table_name) { 'users' }
   let(:parent_table_name) { 'organization' }

data/spec/lib/declare_schema/schema_change/foreign_key_remove_spec.rb

@@ -3,9 +3,7 @@
 require_relative '../../../../lib/declare_schema/schema_change/foreign_key_remove'
 
 RSpec.describe DeclareSchema::SchemaChange::ForeignKeyRemove do
-  before do
-    load File.expand_path('../prepare_testapp.rb', __dir__)
-  end
+  include_context 'prepare test app'
 
   let(:table_name) { 'users' }
   let(:parent_table_name) { 'organization' }

data/spec/lib/declare_schema/schema_change/index_add_spec.rb

@@ -3,9 +3,7 @@
 require_relative '../../../../lib/declare_schema/schema_change/index_add'
 
 RSpec.describe DeclareSchema::SchemaChange::IndexAdd do
-  before do
-    load File.expand_path('../prepare_testapp.rb', __dir__)
-  end
+  include_context 'prepare test app'
 
   let(:table_name) { 'users' }
   let(:column_names) { [:last_name, 'first_name'] }

data/spec/lib/declare_schema/schema_change/index_remove_spec.rb

@@ -3,9 +3,7 @@
 require_relative '../../../../lib/declare_schema/schema_change/index_remove'
 
 RSpec.describe DeclareSchema::SchemaChange::IndexRemove do
-  before do
-    load File.expand_path('../prepare_testapp.rb', __dir__)
-  end
+  include_context 'prepare test app'
 
   let(:table_name) { 'users' }
   let(:column_names) { [:last_name, 'first_name'] }

data/spec/lib/declare_schema/schema_change/primary_key_change_spec.rb

@@ -3,9 +3,7 @@
 require_relative '../../../../lib/declare_schema/schema_change/primary_key_change'
 
 RSpec.describe DeclareSchema::SchemaChange::PrimaryKeyChange do
-  before do
-    load File.expand_path('../prepare_testapp.rb', __dir__)
-  end
+  include_context 'prepare test app'
 
   let(:table_name) { 'users' }
   let(:old_column_names) { ['id'] }
@@ -17,15 +15,25 @@ RSpec.describe DeclareSchema::SchemaChange::PrimaryKeyChange do
     context 'when PRIMARY KEY set -> set' do
       describe '#up' do
         it 'responds with command' do
-          command = "ALTER TABLE #{ActiveRecord::Base.connection.quote_table_name(table_name)} DROP PRIMARY KEY, ADD PRIMARY KEY (#{new_column_names.join(', ')})"
-          expect(subject.up).to eq("execute #{command.inspect}\n")
+          if current_adapter == 'postgresql'
+            expect(subject.up.split("\n")).to include('execute "ALTER TABLE \"users\" DROP CONSTRAINT users_pkey;"')
+            expect(subject.up.split("\n")).to include('execute "ALTER TABLE \"users\" ADD PRIMARY KEY (last_name, first_name);"')
+          else
+            command = "ALTER TABLE #{ActiveRecord::Base.connection.quote_table_name(table_name)} DROP PRIMARY KEY, ADD PRIMARY KEY (last_name, first_name)"
+            expect(subject.up).to eq("execute #{command.inspect}\n")
+          end
         end
       end
 
       describe '#down' do
         it 'responds with command' do
-          command = "ALTER TABLE #{ActiveRecord::Base.connection.quote_table_name(table_name)} DROP PRIMARY KEY, ADD PRIMARY KEY (#{old_column_names.join(', ')})"
-          expect(subject.down).to eq("execute #{command.inspect}\n")
+          if current_adapter == 'postgresql'
+            expect(subject.down.split("\n")).to include('execute "ALTER TABLE \"users\" DROP CONSTRAINT users_pkey;"')
+            expect(subject.down.split("\n")).to include('execute "ALTER TABLE \"users\" ADD PRIMARY KEY (id);"')
+          else
+            command = "ALTER TABLE #{ActiveRecord::Base.connection.quote_table_name(table_name)} DROP PRIMARY KEY, ADD PRIMARY KEY (id)"
+            expect(subject.down).to eq("execute #{command.inspect}\n")
+          end
         end
       end
     end
@@ -35,15 +43,23 @@ RSpec.describe DeclareSchema::SchemaChange::PrimaryKeyChange do
 
       describe '#up' do
         it 'responds with command' do
-          command = "ALTER TABLE #{ActiveRecord::Base.connection.quote_table_name(table_name)} ADD PRIMARY KEY (#{new_column_names.join(', ')})"
-          expect(subject.up).to eq("execute #{command.inspect}\n")
+          if current_adapter == 'postgresql'
+            expect(subject.up.split("\n")).to include('execute "ALTER TABLE \"users\" ADD PRIMARY KEY (last_name, first_name);"')
+          else
+            command = "ALTER TABLE #{ActiveRecord::Base.connection.quote_table_name(table_name)} ADD PRIMARY KEY (#{new_column_names.join(', ')})"
+            expect(subject.up).to eq("execute #{command.inspect}\n")
+          end
         end
       end
 
       describe '#down' do
         it 'responds with command' do
-          command = "ALTER TABLE #{ActiveRecord::Base.connection.quote_table_name(table_name)} DROP PRIMARY KEY"
-          expect(subject.down).to eq("execute #{command.inspect}\n")
+          if current_adapter == 'postgresql'
+            expect(subject.down.split("\n")).to include('execute "ALTER TABLE \"users\" DROP CONSTRAINT users_pkey;"')
+          else
+            command = "ALTER TABLE #{ActiveRecord::Base.connection.quote_table_name(table_name)} DROP PRIMARY KEY"
+            expect(subject.down).to eq("execute #{command.inspect}\n")
+          end
         end
       end
     end
@@ -53,15 +69,23 @@ RSpec.describe DeclareSchema::SchemaChange::PrimaryKeyChange do
 
       describe '#up' do
         it 'responds with command' do
-          command = "ALTER TABLE #{ActiveRecord::Base.connection.quote_table_name(table_name)} DROP PRIMARY KEY"
-          expect(subject.up).to eq("execute #{command.inspect}\n")
+          if current_adapter == 'postgresql'
+            expect(subject.up.split("\n")).to include('execute "ALTER TABLE \"users\" DROP CONSTRAINT users_pkey;"')
+          else
+            command = "ALTER TABLE #{ActiveRecord::Base.connection.quote_table_name(table_name)} DROP PRIMARY KEY"
+            expect(subject.up).to eq("execute #{command.inspect}\n")
+          end
         end
       end
 
       describe '#down' do
         it 'responds with command' do
-          command = "ALTER TABLE #{ActiveRecord::Base.connection.quote_table_name(table_name)} ADD PRIMARY KEY (#{old_column_names.join(', ')})"
-          expect(subject.down).to eq("execute #{command.inspect}\n")
+          if current_adapter == 'postgresql'
+            expect(subject.down.split("\n")).to include('execute "ALTER TABLE \"users\" ADD PRIMARY KEY (id);"')
+          else
+            command = "ALTER TABLE #{ActiveRecord::Base.connection.quote_table_name(table_name)} ADD PRIMARY KEY (#{old_column_names.join(', ')})"
+            expect(subject.down).to eq("execute #{command.inspect}\n")
+          end
         end
       end
     end

data/spec/lib/declare_schema/schema_change/table_add_spec.rb

@@ -3,9 +3,7 @@
 require_relative '../../../../lib/declare_schema/schema_change/table_add'
 
 RSpec.describe DeclareSchema::SchemaChange::TableAdd do
-  before do
-    load File.expand_path('../prepare_testapp.rb', __dir__)
-  end
+  include_context 'prepare test app'
 
   let(:table_name) { 'networks' }
   let(:fields) { [[:string, :title, limit: 255, null: false ], [:boolean, :admin, null: false]] }

data/spec/lib/declare_schema/schema_change/table_change_spec.rb

@@ -3,9 +3,7 @@
 require_relative '../../../../lib/declare_schema/schema_change/table_change'
 
 RSpec.describe DeclareSchema::SchemaChange::TableChange do
-  before do
-    load File.expand_path('../prepare_testapp.rb', __dir__)
-  end
+  include_context 'prepare test app'
 
   let(:table_name)     { 'networks' }
   let(:old_options)    { { charset: 'utf8', collation: 'utf8_ci' } }