feat: expose the data migrations code as a rake task

Author: bethesque

lib/pact_broker/db.rb

@@ -24,7 +24,7 @@ def self.run_migrations database_connection, options = {}
     end
 
     def self.run_data_migrations database_connection
-      PactBroker::DB::MigrateData.(connection)
+      PactBroker::DB::MigrateData.(database_connection)
     end
 
     def self.validate_connection_config

lib/pact_broker/tasks.rb

@@ -1,3 +1,4 @@
 require 'pact_broker/tasks/migration_task'
+require 'pact_broker/tasks/data_migration_task'
 require 'pact_broker/tasks/version_task'
 require 'pact_broker/tasks/clean_task'

lib/pact_broker/tasks/data_migration_task.rb

@@ -0,0 +1,38 @@
+require 'rake/tasklib'
+
+=begin
+
+require 'pact_broker/tasks'
+
+PactBroker::DB::DataMigrationTask.new do | task |
+  require 'my_app/db'
+  task.database_connection = MyApp::DB
+end
+
+=end
+
+module PactBroker
+  module DB
+    class DataMigrationTask < ::Rake::TaskLib
+
+      attr_accessor :database_connection
+
+      def initialize &block
+        rake_task &block
+      end
+
+      def rake_task &block
+        namespace :pact_broker do
+          namespace :db do
+            desc "Run data migrations for pact broker database"
+            task :migrate_data do | t, args |
+              require 'pact_broker/db'
+              instance_eval(&block)
+              PactBroker::DB.run_data_migrations database_connection
+            end
+          end
+        end
+      end
+    end
+  end
+end

tasks/test_db.rake

@@ -6,6 +6,12 @@ PactBroker::DB::MigrationTask.new do | task |
   task.database_connection = DB::PACT_BROKER_DB
 end
 
+PactBroker::DB::DataMigrationTask.new do | task |
+  ENV['RACK_ENV'] ||= 'test'
+  require 'db'
+  task.database_connection = DB::PACT_BROKER_DB
+end
+
 PactBroker::DB::VersionTask.new do | task |
   ENV['RACK_ENV'] ||= 'test'
   require 'db'