Module: ActiveRecord::TestFixtures

Extended by:

ActiveSupport::Concern

Included in:

ActiveSupport::TestCase

Defined in:

activerecord/lib/active_record/fixtures.rb

Defined Under Namespace

Modules: ClassMethods

Instance Method Summary

• #enlist_fixture_connections ⇒ Object
• #run_in_transaction? ⇒ Boolean
• #setup_fixtures ⇒ Object
• #teardown_fixtures ⇒ Object

Methods included from ActiveSupport::Concern

append_features, extended, included

Instance Method Details

#enlist_fixture_connections ⇒ Object

# File 'activerecord/lib/active_record/fixtures.rb', line 889

def enlist_fixture_connections
  ActiveRecord::Base.connection_handler.connection_pools.values.map(&:connection)
end

#run_in_transaction? ⇒ Boolean

Returns:

• (Boolean)

# File 'activerecord/lib/active_record/fixtures.rb', line 828

def run_in_transaction?
  use_transactional_fixtures &&
    !self.class.uses_transaction?(method_name)
end

#setup_fixtures ⇒ Object

# File 'activerecord/lib/active_record/fixtures.rb', line 833

def setup_fixtures
  return unless !ActiveRecord::Base.configurations.blank?

  if pre_loaded_fixtures && !use_transactional_fixtures
    raise RuntimeError, 'pre_loaded_fixtures requires use_transactional_fixtures'
  end

  @fixture_cache = {}
  @fixture_connections = []
  @@already_loaded_fixtures ||= {}

  # Load fixtures once and begin transaction.
  if run_in_transaction?
    if @@already_loaded_fixtures[self.class]
      @loaded_fixtures = @@already_loaded_fixtures[self.class]
    else
      @loaded_fixtures = load_fixtures
      @@already_loaded_fixtures[self.class] = @loaded_fixtures
    end
    @fixture_connections = enlist_fixture_connections
    @fixture_connections.each do |connection|
      connection.increment_open_transactions
      connection.transaction_joinable = false
      connection.begin_db_transaction
    end
  # Load fixtures for every test.
  else
    ActiveRecord::Fixtures.reset_cache
    @@already_loaded_fixtures[self.class] = nil
    @loaded_fixtures = load_fixtures
  end

  # Instantiate fixtures for every test if requested.
  instantiate_fixtures if use_instantiated_fixtures
end

#teardown_fixtures ⇒ Object

# File 'activerecord/lib/active_record/fixtures.rb', line 869

def teardown_fixtures
  return unless defined?(ActiveRecord) && !ActiveRecord::Base.configurations.blank?

  unless run_in_transaction?
    ActiveRecord::Fixtures.reset_cache
  end

  # Rollback changes if a transaction is active.
  if run_in_transaction?
    @fixture_connections.each do |connection|
      if connection.open_transactions != 0
        connection.rollback_db_transaction
        connection.decrement_open_transactions
      end
    end
    @fixture_connections.clear
  end
  ActiveRecord::Base.clear_active_connections!
end