module DataMapper::TestFixtures

Public Instance Methods

enlist_fixture_connections() click to toggle source
# File lib/dm-fixtures/fixtures.rb, line 962
def enlist_fixture_connections
  #ActiveRecord::Base.connection_handler.connection_pools.values.map(&:connection)
  DataMapper::Repository.adapters.values 
  #TODO think I actually want DataObjects stuff: https://groups.google.com/d/msg/datamapper/16DJGo9DLSI/oqjWvrnJp3oJ
  # or http://stackoverflow.com/a/11594219/1749924
end
run_in_transaction?() click to toggle source
# File lib/dm-fixtures/fixtures.rb, line 889
def run_in_transaction?
  use_transactional_fixtures && defined? DataMapper::Transaction && #make sure we have dm-transactions
    !self.class.uses_transaction?(method_name)
end
setup_fixtures() click to toggle source
# File lib/dm-fixtures/fixtures.rb, line 894
def setup_fixtures
  # Make sure we're connected to a DB
  # return unless !ActiveRecord::Base.configurations.blank?
  return if DataMapper::Repository.adapters.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 |repository|
      # http://git.io/2QmaGg
      # connection.increment_open_transactions
      # connection.transaction_joinable = false
      # connection.begin_db_transaction
      transaction = DataMapper::Transaction.new(repository)
      transaction.begin
      repository.push_transaction(transaction)
    end
  # Load fixtures for every test.
  else
    DataMapper::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() click to toggle source
# File lib/dm-fixtures/fixtures.rb, line 936
def teardown_fixtures
  #                                     ActiveRecord::Base.configurations.blank?
  return unless defined?(DataMapper) && !DataMapper::Repository.adapters.blank?

  # Rollback changes if a transaction is active.
  if run_in_transaction?
    @fixture_connections.each do |repository|
      # if connection.open_transactions != 0
      #   connection.rollback_db_transaction
      #   connection.decrement_open_transactions
      # end
      # adapter = repository.adapter
      while repository.current_transaction
        repository.current_transaction.rollback
        repository.pop_transaction
      end
    end
    @fixture_connections.clear
  else
    DataMapper::Fixtures.reset_cache
  end
  
  #
  # ActiveRecord::Base.clear_active_connections!
end

Private Instance Methods

instantiate_fixtures() click to toggle source
# File lib/dm-fixtures/fixtures.rb, line 978
def instantiate_fixtures
  if pre_loaded_fixtures
    raise RuntimeError, 'Load fixtures before instantiating them.' if DataMapper::Fixtures.all_loaded_fixtures.empty?
    unless @@required_fixture_classes
      self.class.require_fixture_classes DataMapper::Fixtures.all_loaded_fixtures.keys
      @@required_fixture_classes = true
    end
    DataMapper::Fixtures.instantiate_all_loaded_fixtures(self, load_instances?)
  else
    raise RuntimeError, 'Load fixtures before instantiating them.' if @loaded_fixtures.nil?
    @loaded_fixtures.each_value do |fixture_set|
      DataMapper::Fixtures.instantiate_fixtures(self, fixture_set, load_instances?)
    end
  end
end
load_fixtures() click to toggle source
# File lib/dm-fixtures/fixtures.rb, line 970
def load_fixtures
  fixtures = DataMapper::Fixtures.create_fixtures(fixture_path, fixture_table_names, fixture_class_names)
  Hash[fixtures.map { |f| [f.name, f] }]
end
load_instances?() click to toggle source
# File lib/dm-fixtures/fixtures.rb, line 994
def load_instances?
  use_instantiated_fixtures != :no_instances
end