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_environment(environment, &block)
  begin
    set_octopus_env(environment.to_s)
    clean_connection_proxy()
    yield
  ensure
    set_octopus_env('octopus')
    clean_connection_proxy()
  end
end

def set_octopus_env(env)
  Octopus.instance_variable_set(:@config, nil)
  Octopus.stub!(:env).and_return(env)
end

Version data entries

4 entries across 4 versions & 1 rubygems

Version	Path
ar-octopus-0.3.4	spec/octopus_helper.rb
ar-octopus-0.3.3	spec/octopus_helper.rb
ar-octopus-0.3.2	spec/octopus_helper.rb
ar-octopus-0.3.1	spec/octopus_helper.rb