Rakefile

require "knapsack"
Dir.glob('lib/tasks/*.rake').each { |r| import r }

namespace :db do
  env = ENV['ENV'] || 'test'
  file  = 'db/structure.sql'

  desc "Fetch DB schema dump"
  task :fetch_schema do
    if branch = ENV['TRAVIS_MIGRATIONS_BRANCH'] and !branch.empty?
      $stderr.puts 'Warning: travis-migrations branch overridden by environment variable.'
    else
      branch = 'master'
    end

    url   = "https://raw.githubusercontent.com/travis-ci/travis-migrations/#{branch}/db/main/structure.sql"
    puts url
    system "curl -fs #{url} -o #{file} --create-dirs"
    abort "failed to download #{url}" unless File.exist?(file)
  end

  desc "Create and migrate the #{env} database"
  task create: :fetch_schema do
    abort 'Cannot run rake db:create in production.' if env == 'production'

    sh "createdb travis_#{env}" rescue nil
    sh "psql -q travis_#{env} < #{file}"
  end
end

namespace :spec do
  desc 'Run all specs'
  task :all do
    sh 'bundle exec rspec -r spec_helper spec'
  end
end

task :default => :'spec:all'
Knapsack.load_tasks if defined?(Knapsack)