codeship.rake

codeship.rake

namespace :db do
  
  desc 'Drops all tables in the database'
  task drop_all_tables: :environment do
    fail 'Only allowed in test environment' unless Rails.env.test?

    connection = ActiveRecord::Base.connection
    case connection.adapter_name
    # SQL Server via TinyTDS
    when 'SQLServer'
      last_error_message = nil
      begin
        command = "EXEC sp_msforeachtable 'DROP TABLE ?'"
        connection.raw_connection.execute(command).do 
      rescue TinyTds::Error => e
        if last_error_message != e.message
          last_error_message = e.message
          retry
        end
      end
    else
      fail "Don't know how to drop tables for #{connection.adapter_name}"
    end
    
  end

end