Module: ActiveRecord::TestFixtures::ClassMethods

Defined in:
lib/active_record/fixtures.rb

Instance Method Summary collapse

Instance Method Details

#fixtures(*table_names) ⇒ Object



854
855
856
857
858
859
860
861
862
863
864
865
# File 'lib/active_record/fixtures.rb', line 854

def fixtures(*table_names)
  if table_names.first == :all
    table_names = Dir["#{fixture_path}/**/*.{yml,csv}"]
    table_names.map! { |f| f[(fixture_path.size + 1)..-5] }
  else
    table_names = table_names.flatten.map { |n| n.to_s }
  end

  self.fixture_table_names |= table_names
  require_fixture_classes(table_names)
  setup_fixture_accessors(table_names)
end

#require_fixture_classes(table_names = nil) ⇒ Object



879
880
881
882
883
884
885
# File 'lib/active_record/fixtures.rb', line 879

def require_fixture_classes(table_names = nil)
  (table_names || fixture_table_names).each do |table_name|
    file_name = table_name.to_s
    file_name = file_name.singularize if ActiveRecord::Base.pluralize_table_names
    try_to_load_dependency(file_name)
  end
end

#set_fixture_class(class_names = {}) ⇒ Object



850
851
852
# File 'lib/active_record/fixtures.rb', line 850

def set_fixture_class(class_names = {})
  self.fixture_class_names = self.fixture_class_names.merge(class_names)
end

#setup_fixture_accessors(table_names = nil) ⇒ Object



887
888
889
890
891
892
893
894
895
896
897
898
899
900
901
902
903
904
905
906
907
908
909
910
911
# File 'lib/active_record/fixtures.rb', line 887

def setup_fixture_accessors(table_names = nil)
  table_names = Array.wrap(table_names || fixture_table_names)
  table_names.each do |table_name|
    table_name = table_name.to_s.tr('./', '_')

    redefine_method(table_name) do |*fixtures|
      force_reload = fixtures.pop if fixtures.last == true || fixtures.last == :reload

      @fixture_cache[table_name] ||= {}

      instances = fixtures.map do |fixture|
        @fixture_cache[table_name].delete(fixture) if force_reload

        if @loaded_fixtures[table_name][fixture.to_s]
          @fixture_cache[table_name][fixture] ||= @loaded_fixtures[table_name][fixture.to_s].find
        else
          raise StandardError, "No fixture with name '#{fixture}' found for table '#{table_name}'"
        end
      end

      instances.size == 1 ? instances.first : instances
    end
    private table_name
  end
end

#try_to_load_dependency(file_name) ⇒ Object



867
868
869
870
871
872
873
874
875
876
877
# File 'lib/active_record/fixtures.rb', line 867

def try_to_load_dependency(file_name)
  require_dependency file_name
rescue LoadError => e
  # Let's hope the developer has included it himself

  # Let's warn in case this is a subdependency, otherwise
  # subdependency error messages are totally cryptic
  if ActiveRecord::Base.logger
    ActiveRecord::Base.logger.warn("Unable to load #{file_name}, underlying cause #{e.message} \n\n #{e.backtrace.join("\n")}")
  end
end

#uses_transaction(*methods) ⇒ Object



913
914
915
916
# File 'lib/active_record/fixtures.rb', line 913

def uses_transaction(*methods)
  @uses_transaction = [] unless defined?(@uses_transaction)
  @uses_transaction.concat methods.map { |m| m.to_s }
end

#uses_transaction?(method) ⇒ Boolean

Returns:

  • (Boolean)


918
919
920
921
# File 'lib/active_record/fixtures.rb', line 918

def uses_transaction?(method)
  @uses_transaction = [] unless defined?(@uses_transaction)
  @uses_transaction.include?(method.to_s)
end