fixture_seed 0.2.0 → 0.3.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 3017e602e40c59de792d03c63f0d126c06abb63e54476329c1c88e9d36a6f57a
-  data.tar.gz: c8c7267e55089bc38ae585e3d0329038a2416205da9ba9e4d9997674436ae891
+  metadata.gz: 74ef9a7ff52401cea08e20f1f6610eb509d339b8aabb7ae249a231f1818661e5
+  data.tar.gz: 3a7f4bbf8ea02c63979e01893dd7f03635fc164e046ec87035eb27b6525a9f80
 SHA512:
-  metadata.gz: 6c7af8d978f4b93ce5175106c08522923e4e1e287d9aee354c432e6526a0e677f64d2cde17bff6dc53b9122f5c3cbb69033f03beaa826fdfd4f81c80d2412d24
-  data.tar.gz: 57622614a7ed6514e1e9522ca62a11b2c69e3207f8caaa7e484f22757e994e3b44e4e3726d8ecd067134f7090f9c240a3f185e7f14f629170df23fe44bc2404f
+  metadata.gz: 88e226a51a806e9cf69f4e44067f8c892c5b48c80d700e86fc4bf371417de570f5221d4fbe6cea2f3010a4b58df2474cfc5fafd507dc15b93ac29b0b4abd8c66
+  data.tar.gz: dd409f1d8f5974b4bf1d169879dbdc078f065278d7313808bba22325543015b9d606b0ca660182465bd6ac9d2389f2a3fd9c875541f810623ff2f1b6bd26923d

data/README.md

@@ -52,6 +52,23 @@ user2:
 
 The labels (e.g., `user1`, `user2`) should follow the pattern of the table name in singular form followed by a number.
 
+#### ERB Support
+
+For dynamic fixture generation, you can use ERB templates in your YAML files:
+
+```erb
+# users.yml
+<% 10.times do |i| %>
+user<%= i + 1 %>:
+  id: <%= i + 1 %>
+  name: "User <%= i + 1 %>"
+  email: "user<%= i + 1 %>@example.com"
+  created_at: <%= Time.current.to_s(:db) %>
+<% end %>
+```
+
+ERB templates have access to the full Rails environment, allowing you to use helpers, constants, and other Ruby code to generate dynamic fixture data.
+
 ### Fixture Loading Order
 
 #### Order-independent Loading

data/lib/fixture_seed/loader.rb

@@ -0,0 +1,85 @@
+# frozen_string_literal: true
+
+require "active_record/fixtures"
+
+module FixtureSeed
+  class Loader
+    class << self
+      def load_fixtures(fixtures_path = nil)
+        fixtures_path ||= ENV["FIXTURES_PATH"] || "db/fixtures"
+        fixtures_dir = Rails.root.join(fixtures_path)
+
+        unless Dir.exist?(fixtures_dir)
+          logger = Rails.logger || Logger.new($stdout)
+          logger.info "[FixtureSeed] Fixtures directory not found: #{fixtures_path}"
+          return
+        end
+
+        logger = Rails.logger || Logger.new($stdout)
+        logger.info "[FixtureSeed] Loading fixtures from #{fixtures_path}"
+
+        ActiveRecord::Base.transaction do
+          with_foreign_keys_disabled do
+            load_fixture_files(fixtures_dir, logger)
+          end
+        end
+
+        logger.info "[FixtureSeed] Finished loading fixtures"
+      end
+
+      private
+
+      def load_fixture_files(fixtures_dir, logger)
+        fixture_files = Dir.glob("#{fixtures_dir}/*.yml")
+        return if fixture_files.empty?
+
+        table_names = fixture_files.map { |f| File.basename(f, ".yml") }
+        logger.info "[FixtureSeed] Found tables: #{table_names.join(', ')}"
+
+        ordered_tables = dependency_ordered_tables(table_names)
+
+        ordered_tables.each do |table_name|
+          ActiveRecord::FixtureSet.create_fixtures(fixtures_dir.to_s, [table_name])
+        end
+      end
+
+      def dependency_ordered_tables(table_names)
+        ordered = []
+        ordered << "users" if table_names.include?("users")
+        ordered += (table_names - ["users"])
+        ordered
+      end
+
+      def with_foreign_keys_disabled
+        adapter_name = ActiveRecord::Base.connection.adapter_name.downcase
+
+        disable_foreign_keys(adapter_name)
+        yield
+      ensure
+        enable_foreign_keys(adapter_name)
+      end
+
+      def disable_foreign_keys(adapter_name)
+        case adapter_name
+        when /postgresql/
+          ActiveRecord::Base.connection.execute("SET session_replication_role = replica;")
+        when /sqlite/
+          ActiveRecord::Base.connection.execute("PRAGMA foreign_keys = OFF;")
+        when /mysql/
+          ActiveRecord::Base.connection.execute("SET FOREIGN_KEY_CHECKS = 0;")
+        end
+      end
+
+      def enable_foreign_keys(adapter_name)
+        case adapter_name
+        when /postgresql/
+          ActiveRecord::Base.connection.execute("SET session_replication_role = DEFAULT;")
+        when /sqlite/
+          ActiveRecord::Base.connection.execute("PRAGMA foreign_keys = ON;")
+        when /mysql/
+          ActiveRecord::Base.connection.execute("SET FOREIGN_KEY_CHECKS = 1;")
+        end
+      end
+    end
+  end
+end

data/lib/fixture_seed/railtie.rb

@@ -1,23 +1,18 @@
 # frozen_string_literal: true
 
+require_relative "loader"
+
 module FixtureSeed
   class Railtie < Rails::Railtie
-    initializer "fixture_seed.initialize" do
-      Rails.logger.info "[FixtureSeed] Railtie initialized"
-    end
-
     rake_tasks do
-      if Rake::Task.task_defined?("db:seed")
-        Rake::Task["db:seed"].enhance([:load_fixtures])
-
-        task :load_fixtures do
-          Rails.logger.info "[FixtureSeed] Starting to load fixtures before db:seed"
-          FixtureSeed.load_fixtures
-          Rails.logger.info "[FixtureSeed] Finished loading fixtures"
+      namespace :fixture_seed do
+        desc "Load fixtures from specified directory"
+        task load_fixtures: :environment do
+          FixtureSeed::Loader.load_fixtures
         end
-      else
-        Rails.logger.warn "[FixtureSeed] db:seed task not found, fixture loading enhancement not applied"
       end
+
+      Rake::Task["db:seed"].enhance(["fixture_seed:load_fixtures"]) if Rake::Task.task_defined?("db:seed")
     end
   end
 end

data/lib/fixture_seed/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module FixtureSeed
-  VERSION = "0.2.0"
+  VERSION = "0.3.1"
 end

data/lib/fixture_seed.rb

@@ -1,105 +1,13 @@
 # frozen_string_literal: true
 
 require_relative "fixture_seed/version"
+require_relative "fixture_seed/loader"
 require_relative "fixture_seed/railtie" if defined?(Rails::Railtie)
 
 module FixtureSeed
   class Error < StandardError; end
 
-  class << self
-    def load_fixtures
-      fixture_dir = Rails.root.join("db/fixtures").to_s
-      files_to_load = Dir.glob(File.join(fixture_dir, "*.yml")).sort
-
-      file_names = files_to_load.map { |f| File.basename(f) }.join(", ")
-      Rails.logger.info "[FixtureSeed] Found #{files_to_load.size} fixture files in #{fixture_dir}: #{file_names}"
-
-      # Load all fixtures with retry logic
-      unloaded_files = process_fixtures(fixture_dir, files_to_load)
-
-      # Report any fixtures that still failed
-      if unloaded_files.empty?
-        Rails.logger.info "[FixtureSeed] All fixtures loaded successfully."
-      else
-        message = "The following fixtures could not be loaded: #{unloaded_files.join(', ')}"
-        Rails.logger.warn "[FixtureSeed] #{message}"
-      end
-    end
-
-    private
-
-    def process_fixtures(fixture_dir, files)
-      remaining_files = files.dup
-      previous_count = remaining_files.size + 1
-
-      # Continue until all fixtures are loaded or no progress is made
-      while !remaining_files.empty? && remaining_files.size < previous_count
-        previous_count = remaining_files.size
-        remaining_files = process_batch(fixture_dir, files, remaining_files)
-      end
-
-      remaining_files
-    end
-
-    def process_batch(fixture_dir, original_files, current_files)
-      still_failed = []
-
-      current_files.each do |file|
-        log_loading(file)
-        load_single_fixture(fixture_dir, file)
-        log_success(file, original_files, current_files)
-      rescue ActiveRecord::InvalidForeignKey, ActiveRecord::StatementInvalid => e
-        log_failure(file, e, original_files, current_files)
-        still_failed << file
-      end
-
-      still_failed
-    end
-
-    def load_single_fixture(_fixture_dir, file)
-      table_name = File.basename(file, ".yml")
-      yaml_data = YAML.load_file(file)
-
-      model_class = begin
-        table_name.classify.constantize
-      rescue StandardError
-        nil
-      end
-
-      if model_class
-        yaml_data.each do |_fixture_name, attributes|
-          model_class.create!(attributes)
-        rescue StandardError => e
-          raise ActiveRecord::InvalidForeignKey, e.message
-        end
-      else
-        Rails.logger.warn "[FixtureSeed] Model for table #{table_name} not found"
-      end
-    end
-
-    def log_loading(file)
-      Rails.logger.info "[FixtureSeed] Loading fixture #{File.basename(file)}..."
-    end
-
-    def log_success(file, original_files, current_files)
-      is_retry = original_files.size != current_files.size
-      if is_retry
-        Rails.logger.info "[FixtureSeed] Successfully loaded fixture #{File.basename(file)} on retry."
-      else
-        Rails.logger.info "[FixtureSeed] Loaded fixture #{File.basename(file)}"
-      end
-    end
-
-    def log_failure(file, error, original_files, current_files)
-      is_retry = original_files.size != current_files.size
-      filename = File.basename(file)
-      if is_retry
-        Rails.logger.warn "[FixtureSeed] Still failed to load fixture #{filename}: " \
-                          "#{error.message}. Will retry later."
-      else
-        Rails.logger.warn "[FixtureSeed] Failed to load fixture #{filename}: " \
-                          "#{error.message}. Will retry later."
-      end
-    end
+  def self.load_fixtures(fixtures_path = nil)
+    Loader.load_fixtures(fixtures_path)
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: fixture_seed
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.3.1
 platform: ruby
 authors:
 - Masaki Komagata
@@ -36,6 +36,7 @@ files:
 - LICENSE.txt
 - README.md
 - lib/fixture_seed.rb
+- lib/fixture_seed/loader.rb
 - lib/fixture_seed/railtie.rb
 - lib/fixture_seed/version.rb
 - sig/fixture_seed.rbs