Ticket #1911: more_flexible_fixture_architecture-3b.patch

lib/active_record/fixtures.rb

@@ -2 +2 @@
-require 'yaml'
-require 'csv'
-
+# A FixtureGroup is a set of fixtures identified by a name.  Normally, this is the name of the
+# corresponding fixture filename.  For example, when you declare the use of fixtures in a
+# TestUnit class, like so:
+#   fixtures :users
+# you are creating a FixtureGroup whose name is 'users', and whose defaults are set such that the
+# +class_name+, +file_name+ and +table_name+ are guessed from the FixtureGroup's name.
+class FixtureGroup
+  attr_accessor :table_name, :class_name, :connection
+  attr_reader :group_name, :file_name
+
+  def initialize(file_name, optional_names = {})
+    self.file_name = file_name
+    self.group_name = optional_names[:group_name] || file_name
+    if optional_names[:table_name]
+      self.table_name = optional_names[:table_name]
+      self.class_name = optional_names[:class_name] || Inflector.classify(@table_name.to_s.gsub('.','_'))
+    elsif optional_names[:class_name]
+      self.class_name = optional_names[:class_name]
+      if Object.const_defined?(@class_name)
+        model_class = Object.const_get(@class_name)
+        self.table_name = ActiveRecord::Base.table_name_prefix + model_class.table_name + ActiveRecord::Base.table_name_suffix
+      end
+    end
+
+    # In case either :table_name or :class_name was not set:
+    self.table_name ||= ActiveRecord::Base.table_name_prefix + @group_name.to_s + ActiveRecord::Base.table_name_suffix
+    self.class_name ||= Inflector.classify(@table_name.to_s.gsub('.','_'))
+  end
+
+  def file_name=(name)
+    @file_name = name.to_s
+  end
+  
+  def group_name=(name)
+    @group_name = name.to_sym
+  end
+
+  def class_file_name
+    Inflector.underscore(@class_name)
+  end
+  
+  # Instantiate an array of FixtureGroup objects from an array of strings (table_names)
+  def self.array_from_names(names)
+    names.collect { |n| FixtureGroup.new(n) }
+  end
+  
+  def hash
+    @group_name.hash
+  end
+  
+  def eql?(other)
+    @group_name.eql? other.group_name
+  end
+end
+
-# Fixtures are a way of organizing data that you want to test against; in short, sample data. They come in 3 flavours:
-#
-#   1.  YAML fixtures
@@ -193 +248 @@
-#      Use InnoDB, MaxDB, or NDB instead.
-class Fixtures < Hash
-  DEFAULT_FILTER_RE = /\.ya?ml$/
+  
+  cattr_accessor :all_loaded_fixtures
+  self.all_loaded_fixtures = {}
-
-
-
-
+
+
+
+
-
-
-
-
-
-
+
+
+
+
+
+
+
-        end
-      end
+
+      ActiveRecord::Base.logger.level = old_logger_level
-    end
-
-
-
-
-
-
-
+
+
+
+
-    end
-  end
-  
-  cattr_accessor :all_loaded_fixtures
-  self.all_loaded_fixtures = {}
-
-
-
-
+
+
+
+
+
+
+
+
+
+
-
-
-
+  
+  
-
-
-
-
-
-
-
-
-
-
-
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
-      end
-      
-      reset_sequences(connection, table_names) if connection.is_a?(ActiveRecord::ConnectionAdapters::PostgreSQLAdapter)
-
-      return fixtures.size > 1 ? fixtures : fixtures.first
-    ensure
-      ActiveRecord::Base.logger.level = old_logger_level
-    end
-  end
-
-
-
-
-
-
-
-
-
-
-
-
+
+
+
+
+
+
+
+
+
+
+
+
-        end
-      end
-    end
-  end
-
-
+
+
-
-
-
-
-
-
+
+
+
+
-    read_fixture_files
-  end
-
-  def delete_existing_fixtures
-
+fixture_group.
-  end
-
-  def insert_fixtures
-    values.each do |fixture|
-
+fixture_group.
-    end
-  end
-
-  private
-    def read_fixture_files
-      if File.file?(yaml_file_path)
-
-
-
-
-
-
-
+
-      elsif File.file?(csv_file_path)
-
-
-
-
-
-
-
-
-
+
-      elsif File.file?(deprecated_yaml_file_path)
-        raise Fixture::FormatError, ".yml extension required: rename #{deprecated_yaml_file_path} to #{yaml_file_path}"
+      elsif File.directory?(single_file_fixtures_path)
+        read_standard_fixture_files
-      else
-
-
-
-
-
-
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
-        end
-      end
-    end
-
-    def yaml_file_path
-"#{@fixture_path}
+fixture_path_with_extension "
-    end
-
-    def deprecated_yaml_file_path
-"#{@fixture_path}
+fixture_path_with_extension "
-    end
-
-    def csv_file_path
-@fixture_path +
+fixture_path_with_extension
-    end
-
-yaml_fixtures_key(path)
-File.basename(@fixture_path).split(".").first
+
+single_file_fixtures_path
+fixture_path_with_extension ""
-    end
-
+    def fixture_path_with_extension(ext)
+      File.join(@fixtures_directory, @fixture_group.file_name + ext)
+    end      
+
-    def erb_render(fixture_content)
-      ERB.new(fixture_content).result
-    end
@@ -399 +483 @@
-module Test #:nodoc:
-  module Unit #:nodoc:
-    class TestCase #:nodoc:
-
+
+
-      class_inheritable_accessor :fixture_table_names
-      class_inheritable_accessor :use_transactional_fixtures
-      class_inheritable_accessor :use_instantiated_fixtures   # true, false, or :no_instances
-      class_inheritable_accessor :pre_loaded_fixtures
-
-table_name
+group
-      self.use_transactional_fixtures = false
-      self.use_instantiated_fixtures = true
-      self.pre_loaded_fixtures = false
-
-      @@already_loaded_fixtures = {}
-
-
-
-
-
-
+
+
+
+
+
+
+
+
+
+
-      end
-
-
-
+
+
+
+
+
+
+
+
-          begin
-Inflector.singularize(table_name.to_s)
+group.class_file_name
-          rescue LoadError
-            # Let's hope the developer has included it himself
-          end
-        end
-      end
-
-
-
-
-
+
+
+
-            force_reload = optionals.shift
-table
-table
-table_name][fixture] ||= @loaded_fixtures[table
+group.group
+group.group
+group.group_name][fixture] ||= @loaded_fixtures[group.group
-          end
-        end
-      end
@@ -519 +614 @@
-      private
-        def load_fixtures
-          @loaded_fixtures = {}
-_path, fixture_table_name
+s_directory, fixture_group
-          unless fixtures.nil?
-            if fixtures.instance_of?(Fixtures)
-table
+fixture_group.group
-            else
-table
+fixture_group.group
-            end
-          end
-        end
@@ -536 +631 @@
-          if pre_loaded_fixtures
-            raise RuntimeError, 'Load fixtures before instantiating them.' if Fixtures.all_loaded_fixtures.empty?
-            unless @@required_fixture_classes
-
+
+
-              @@required_fixture_classes = true
-            end
-            Fixtures.instantiate_all_loaded_fixtures(self, load_instances?)
-          else
-            raise RuntimeError, 'Load fixtures before instantiating them.' if @loaded_fixtures.nil?
-table
-table
+fixture_group
+fixture_group
-            end
-          end
-        end

test/fixtures/topics2.yml

@@ +1 @@
+third:
+  id: 3
+  title: The Third Topic
+  author_name: Duane
+  author_email_address: david@loudthinking.com
+  written_on: 2003-07-16t15:28:00.00+01:00
+  last_read: 2004-04-15
+  bonus_time: 2005-01-30t15:28:00.00+01:00
+  content: Have a nice day
+  approved: 0
+  replies_count: 0
+
+fourth:
+  id: 4
+  title: The Fourth topic
+  author_name: John
+  written_on: 2003-07-15t15:28:00.00+01:00
+  content: Have a nice day
+  approved: 1
+  replies_count: 2
+  parent_id: 1

test/fixtures_test.rb

@@ -106 +106 @@
-    end
-  end
-
+  def test_fixtures_not_found
+    assert_raise(Fixture::FixtureError) {
+      Fixtures.new(nil, File.join(File.dirname(__FILE__), 'fixtures'), FixtureGroup.new("bad_extension"))
+    }
+  end
+
-  def test_deprecated_yaml_extension
-    assert_raise(Fixture::FormatError) {
-'bad_extension', File.join(File.dirname(__FILE__), 'fixtures'
+File.join(File.dirname(__FILE__), 'fixtures', 'bad_fixtures'), FixtureGroup.new("deprecated"
-    }
-  end
-
@@ -139 +145 @@
-  end
-
-  def test_empty_yaml_fixture
-"accounts", File.dirname(__FILE__) + "/fixtures/naked/yml/accounts"
+File.dirname(__FILE__) + "/fixtures/naked/yml/", FixtureGroup.new("accounts")
-  end
-
-  def test_empty_yaml_fixture_with_a_comment_in_it
-"companies", File.dirname(__FILE__) + "/fixtures/naked/yml/companies"
+File.dirname(__FILE__) + "/fixtures/naked/yml/", FixtureGroup.new("companies")
-  end
-
-  def test_dirty_dirty_yaml_file
-    assert_raises(Fixture::FormatError) do
-"courses", File.dirname(__FILE__) + "/fixtures/naked/yml/courses"
+File.dirname(__FILE__) + "/fixtures/naked/yml/", FixtureGroup.new("courses")
-    end
-  end
-
-  def test_empty_csv_fixtures
-"accounts", File.dirname(__FILE__) + "/fixtures/naked/csv/accounts"
+File.dirname(__FILE__) + "/fixtures/naked/csv/", FixtureGroup.new("accounts")
-  end
-end
-
@@ -250 +256 @@
-
-end
-
+class FixturesGroupTest < Test::Unit::TestCase
+  def test_get_table_name_from_model_class
+    fg = FixtureGroup.new(:filedoesnotexist, :class_name => "DeveloperWithAggregate")
+    assert_equal fg.table_name, 'developers'
+    assert_equal fg.class_name, 'DeveloperWithAggregate'
+  end
+  
+  def test_assume_correct_class_name_from_table_name
+    fg = FixtureGroup.new(:developers, :table_name => "tests")
+    assert_equal fg.table_name, 'tests'
+    assert_equal fg.class_name, 'Test'
+  end
-
+  def test_file_name_is_a_string
+    fg = FixtureGroup.new(:developers)
+    assert String === fg.file_name
+  end
+  
+  def test_fixture_group_name
+    fg = FixtureGroup.new(:developers)
+    assert Symbol === fg.group_name
+    assert_equal fg.group_name, :developers
+    
+    fg = FixtureGroup.new(:developers, :group_name => "other")
+    assert Symbol === fg.group_name
+    assert_equal fg.group_name, :other
+  end
+end
-
+class MultipleFixturesForTheSameTableTest < Test::Unit::TestCase
+  self.use_instantiated_fixtures = true
+  fixture :topics, :class_name => "Topic"
+  fixture :topics2, :class_name => "Topic"
+  
+  def test_loaded_fixtures
+    assert_equal @topics.size, 2
+    assert_equal @topics2.size, 2
+    union = Topic.find(:all)
+    assert_equal union.size, 4
+  end
+end
-
+class MixSingularAndPluralFixturesTest < Test::Unit::TestCase
+  self.use_instantiated_fixtures = true
+  fixtures :topics, :developers, :accounts
+  fixture :topics2, :table_name => "topics"
+  
+  def test_loaded_fixtures
+    assert_equal @topics.size, 2
+    assert_equal @topics2.size, 2
+    union = Topic.find(:all)
+    assert_equal union.size, 4
+  end
+end
-
-
+
+
+
+
+
+
+
+
+
+
+
+
+
+