From c0853547063f8bf1adc9be006849722b7242bae3 Mon Sep 17 00:00:00 2001
From: carlad <carla@travis-ci.com>
Date: Thu, 4 Feb 2016 21:27:20 +0100
Subject: [PATCH] new rakefile

---
 .travis.yml |   5 +-
 Rakefile    | 132 ++++++++++++++++++++++++++++++++++++----------------
 2 files changed, 94 insertions(+), 43 deletions(-)

diff --git a/.travis.yml b/.travis.yml
index c0470291..58812bc5 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -18,10 +18,7 @@ services:
   - redis
 
 before_script:
-  - RAILS_ENV=test
-  - createdb travis_test
-  - createdb travis_pro_test
-  - bundle exec rake db:migrate --trace
+  - 'RAILS_ENV=test bundle exec rake db:create db:migrate --trace'
 
 script:
   - bundle exec rspec spec
diff --git a/Rakefile b/Rakefile
index c4af9f66..095cccec 100644
--- a/Rakefile
+++ b/Rakefile
@@ -1,47 +1,101 @@
+require 'rspec/core/rake_task'
 require 'bundler/setup'
+require 'micro_migrations'
 require 'travis'
-require 'travis/engine'
 
-begin
-  # ENV['SCHEMA'] = File.expand_path('../db/schema.rb', $:.detect { |p| p.include?('travis-core') })
-  require 'micro_migrations'
-rescue LoadError
-  # we can't load micro migrations on production
+ActiveRecord::Base.schema_format = :sql
+Rails.application.config.paths["db/migrate"] = ["db/migrate", "#{Gem.loaded_specs['travis-core'].full_gem_path}/db/migrate"]
+
+desc 'Run specs'
+RSpec::Core::RakeTask.new do |t|
+  t.pattern = './spec/**/*_spec.rb'
 end
 
-begin
-  require 'rspec/core/rake_task'
-  RSpec::Core::RakeTask.new
-  task default: :spec
-rescue LoadError
-  warn "could not load rspec"
-end
-
-desc "generate gemspec"
-task 'travis-api.gemspec' do
-  content = File.read 'travis-api.gemspec'
-
-  fields = {
-    authors: `git shortlog -sn`.scan(/[^\d\s].*/),
-    email:   `git shortlog -sne`.scan(/[^<]+@[^>]+/),
-    files:   `git ls-files`.split("\n").reject { |f| f =~ /^(\.|Gemfile)/ }
-  }
-
-  fields.each do |field, values|
-    updated = "  s.#{field} = ["
-    updated << values.map { |v| "\n    %p" % v }.join(',')
-    updated << "\n  ]"
-    content.sub!(/  s\.#{field} = \[\n(    .*\n)*  \]/, updated)
-  end
-
-  File.open('travis-api.gemspec', 'w') { |f| f << content }
-end
-
-task default: 'travis-api.gemspec'
-
-tasks_path = File.expand_path('../lib/tasks/*.rake', __FILE__)
-Dir.glob(tasks_path).each { |r| import r }
-
+module ActiveRecord
+  class Migration
+    class << self
+      attr_accessor :disable_ddl_transaction
+    end
+
+    # Disable DDL transactions for this migration.
+    def self.disable_ddl_transaction!
+      @disable_ddl_transaction = true
+    end
+
+    def disable_ddl_transaction # :nodoc:
+      self.class.disable_ddl_transaction
+    end
+  end
+
+  class Migrator
+    def use_transaction?(migration)
+      !migration.disable_ddl_transaction && Base.connection.supports_ddl_transactions?
+    end
+
+    def ddl_transaction(migration, &block)
+      if use_transaction?(migration)
+        Base.transaction { block.call }
+      else
+        block.call
+      end
+    end
+
+    def migrate(&block)
+      current = migrations.detect { |m| m.version == current_version }
+      target = migrations.detect { |m| m.version == @target_version }
+
+      if target.nil? && @target_version && @target_version > 0
+        raise UnknownMigrationVersionError.new(@target_version)
+      end
+
+      start = up? ? 0 : (migrations.index(current) || 0)
+      finish = migrations.index(target) || migrations.size - 1
+      runnable = migrations[start..finish]
+
+      # skip the last migration if we're headed down, but not ALL the way down
+      runnable.pop if down? && target
+
+      ran = []
+      runnable.each do |migration|
+        if block && !block.call(migration)
+          next
+        end
+
+        Base.logger.info "Migrating to #{migration.name} (#{migration.version})" if Base.logger
+
+        seen = migrated.include?(migration.version.to_i)
+
+        # On our way up, we skip migrating the ones we've already migrated
+        next if up? && seen
+
+        # On our way down, we skip reverting the ones we've never migrated
+        if down? && !seen
+          migration.announce 'never migrated, skipping'; migration.write
+          next
+        end
+
+        begin
+          ddl_transaction(migration) do
+            migration.migrate(@direction)
+            record_version_state_after_migrating(migration.version)
+          end
+          ran << migration
+        rescue => e
+          canceled_msg = Base.connection.supports_ddl_transactions? ? "this and " : ""
+          raise StandardError, "An error has occurred, #{canceled_msg}all later migrations canceled:\n\n#{e}", e.backtrace
+        end
+      end
+      ran
+    end
+  end
+
+  class MigrationProxy
+    delegate :disable_ddl_transaction, to: :migration
+  end
+end
+
+task :default => :spec
+
 module ActiveRecord
   class Migration
     class << self