fix ActiveRecord::Fixtures deprecation warning in AR 4.2

rdy · Jun 4, 2015 · 6999fc3 · 6999fc3
1 parent 6b58c49
commit 6999fc3
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/lib/fixture_builder/builder.rb b/lib/fixture_builder/builder.rb
@@ -27,13 +27,22 @@ def create_fixture_objects
 
     def load_legacy_fixtures
       legacy_fixtures.each do |fixture_file|
-        # Rails 3.0 and 3.1+ support
-        fixtures_class = defined?(ActiveRecord::Fixtures) ? ActiveRecord::Fixtures : ::Fixtures
         fixtures = fixtures_class.create_fixtures(File.dirname(fixture_file), File.basename(fixture_file, '.*'))
         populate_custom_names(fixtures)
       end
     end
 
+    # Rails 3.0 and 3.1+ support
+    def fixtures_class
+      if defined?(ActiveRecord::FixtureSet)
+        ActiveRecord::FixtureSet
+      elsif defined?(ActiveRecord::Fixtures)
+        ActiveRecord::Fixtures
+      else
+        ::Fixtures
+      end
+    end
+
     def surface_errors
       yield
     rescue Object => error