Module: ActiveRecord::TestFixtures::ClassMethods

Defined in:
lib/active_record/fixtures.rb

Instance Method Summary collapse

Instance Method Details

#fixtures(*fixture_set_names) ⇒ Object 



850
851
852
853
854
855
856
857
858
859
860
861
 
# File 'lib/active_record/fixtures.rb', line 850

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

  self.fixture_table_names |= fixture_set_names
  require_fixture_classes(fixture_set_names, self.config)
  setup_fixture_accessors(fixture_set_names)
end

#require_fixture_classes(fixture_set_names = nil, config = ActiveRecord::Base) ⇒ Object 



874
875
876
877
878
879
880
881
882
883
884
885
 
# File 'lib/active_record/fixtures.rb', line 874

def require_fixture_classes(fixture_set_names = nil, config = ActiveRecord::Base)
  if fixture_set_names
    fixture_set_names = fixture_set_names.map { |n| n.to_s }
  else
    fixture_set_names = fixture_table_names
  end

  fixture_set_names.each do |file_name|
    file_name = file_name.singularize if config.pluralize_table_names
    try_to_load_dependency(file_name)
  end
end

#set_fixture_class(class_names = {}) ⇒ Object

Sets the model class for a fixture when the class name cannot be inferred from the fixture name.

Examples:

set_fixture_class some_fixture:        SomeModel,
                  'namespaced/fixture' => Another::Model

The keys must be the fixture names, that coincide with the short paths to the fixture files.



846
847
848
 
# File 'lib/active_record/fixtures.rb', line 846

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

#setup_fixture_accessors(fixture_set_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
912
913
914
915
916
 
# File 'lib/active_record/fixtures.rb', line 887

def setup_fixture_accessors(fixture_set_names = nil)
  fixture_set_names = Array(fixture_set_names || fixture_table_names)
  methods = Module.new do
    fixture_set_names.each do |fs_name|
      fs_name = fs_name.to_s
      accessor_name = fs_name.tr('/', '_').to_sym

      define_method(accessor_name) do |*fixture_names|
        force_reload = fixture_names.pop if fixture_names.last == true || fixture_names.last == :reload

        @fixture_cache[fs_name] ||= {}

        instances = fixture_names.map do |f_name|
          f_name = f_name.to_s
          @fixture_cache[fs_name].delete(f_name) if force_reload

          if @loaded_fixtures[fs_name][f_name]
            @fixture_cache[fs_name][f_name] ||= @loaded_fixtures[fs_name][f_name].find
          else
            raise StandardError, "No fixture named '#{f_name}' found for fixture set '#{fs_name}'"
          end
        end

        instances.size == 1 ? instances.first : instances
      end
      private accessor_name
    end
  end
  include methods
end

#try_to_load_dependency(file_name) ⇒ Object 



863
864
865
866
867
868
869
870
871
872
 
# File 'lib/active_record/fixtures.rb', line 863

def try_to_load_dependency(file_name)
  require_dependency file_name
rescue LoadError => e
  unless fixture_class_names.key?(file_name.pluralize)
    if ActiveRecord::Base.logger
      ActiveRecord::Base.logger.warn("Unable to load #{file_name}, make sure you added it to ActiveSupport::TestCase.set_fixture_class")
      ActiveRecord::Base.logger.warn("underlying cause #{e.message} \n\n #{e.backtrace.join("\n")}")
    end
  end
end

#uses_transaction(*methods) ⇒ Object 



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

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) 


923
924
925
926
 
# File 'lib/active_record/fixtures.rb', line 923

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