Contents

def clean_all_shards()
  ActiveRecord::Base.using(:master).connection.instance_variable_get(:@shards).keys.each do |shard_symbol|
    ['schema_migrations', 'users', 'clients', 'cats', 'items', 'keyboards', 'computers', 'permissions_roles', 'roles', 'permissions', 'assignments', 'projects', 'programmers'].each do |tables|
      ActiveRecord::Base.using(shard_symbol).connection.execute("DELETE FROM #{tables};") 
    end
  end
end

def clean_connection_proxy()
  Thread.current[:connection_proxy] = nil  
end

def migrating_to_version(version, &block)
  begin
    ActiveRecord::Migrator.run(:up, MIGRATIONS_ROOT, version)
    yield
  ensure
    ActiveRecord::Migrator.run(:down, MIGRATIONS_ROOT, version)
  end
end

def using_enviroment(enviroment, &block)
  begin
    Octopus.instance_variable_set(:@env, enviroment.to_s)
    clean_connection_proxy()
    yield
  ensure
    Octopus.instance_variable_set(:@env, 'production')
    clean_connection_proxy()
  end
end

Version data entries

7 entries across 7 versions & 1 rubygems

Version	Path
ar-octopus-0.0.12	spec/octopus_helper.rb
ar-octopus-0.0.11	spec/octopus_helper.rb
ar-octopus-0.0.10	spec/octopus_helper.rb
ar-octopus-0.0.9	spec/octopus_helper.rb
ar-octopus-0.0.8	spec/octopus_helper.rb
ar-octopus-0.0.7	spec/octopus_helper.rb
ar-octopus-0.0.6	spec/octopus_helper.rb