ridgepole-rails 1.0.2 → 1.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 20921cf112d6479993b6c11c6678fbb9814ab992
-  data.tar.gz: 8270126ebd78803bfd36da5b2ecb9a7265038d0c
+  metadata.gz: 69049e4874316519f312e8f5903e7b293fdaaf57
+  data.tar.gz: 7cd7ea48280f524858b1227f3c5ff6583314502c
 SHA512:
-  metadata.gz: 8934a116c5038c175c648f0ed460144bbbefa49dddedadeca997a062edaa320ed59152db44fdd5b56f0f80b2b2ef941c0078e29cb008d13d85e7c0740b0ddd4b
-  data.tar.gz: 1491d15980d74404a9dfe8e492b47712c873f2bd4caa987b1e3d0e0fe2b050150edadd03b664a7e147be680c47d5fd81babe99099cfe5b9b9f25f1ca5820bf1c
+  metadata.gz: b09690dba1a5d24d3555d340140aa9e0131a25d1f81e03e8a92a80adb16fc43b2d2b4d33e6a07faef00ad70bdfdd7302c170bd3f2deb68f03912da09ad3a8ff9
+  data.tar.gz: eb987e84629819549303714bdf70f73b0029e96cbf175e7db1d4a46b1d2949946c3fc57730e0c89c8841c4d368884bb91bc96aa08f051a34f8e39ef3f7828feb

data/lib/ridgepole/rails/rake_task.rb

@@ -1,6 +1,7 @@
 require 'rake'
 require 'rake/tasklib'
 require 'shellwords'
+require 'ridgepole/rails/restore_extensions_on_purge'
 
 module Ridgepole
   module Rails

data/lib/ridgepole/rails/restore_extensions_on_purge.rb

@@ -0,0 +1,37 @@
+require 'active_record/base'
+require 'active_record/tasks/database_tasks'
+require 'active_support/core_ext/module/delegation'
+
+module Ridgepole
+  module Rails
+    module RestoreExtensionsOnPurge
+      delegate :establish_connection, :connection, to: ActiveRecord::Base
+
+      # Restore extensions to recreated database if any
+      def purge(configuration)
+        establish_connection(configuration)
+        saved_extensions = connection.extensions
+
+        super
+
+        # Re-establish connection since the last one is closed by super#purge
+        establish_connection(configuration)
+        saved_extensions.each do |extension|
+          connection.enable_extension(extension)
+        end
+      end
+
+      def self.prepended(mod)
+        mod.extend(self)
+      end
+    end
+  end
+end
+
+module ActiveRecord
+  module Tasks
+    module DatabaseTasks
+      prepend Ridgepole::Rails::RestoreExtensionsOnPurge
+    end
+  end
+end

data/lib/ridgepole/rails/version.rb

@@ -1,5 +1,5 @@
 module Ridgepole
   module Rails
-    VERSION = '1.0.2'
+    VERSION = '1.0.3'
   end
 end

data/lib/tasks/ridgepole.rake

@@ -21,7 +21,7 @@ desc 'Apply Schemafile to the database'
 task 'db:schema:load' => 'ridgepole:apply'
 
 Rake.application.lookup('db:test:load').clear
-task 'db:test:load' do
+task 'db:test:load' => 'db:test:purge' do
   Rake::Task['ridgepole:apply'].invoke('test')
 end
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ridgepole-rails
 version: !ruby/object:Gem::Version
-  version: 1.0.2
+  version: 1.0.3
 platform: ruby
 authors:
 - OZAWA Sakuro
@@ -67,6 +67,7 @@ files:
 - lib/ridgepole/rails.rb
 - lib/ridgepole/rails/engine.rb
 - lib/ridgepole/rails/rake_task.rb
+- lib/ridgepole/rails/restore_extensions_on_purge.rb
 - lib/ridgepole/rails/version.rb
 - lib/tasks/ridgepole.rake
 homepage: https://github.com/sakuro/ridgepole-rails