Contents

# frozen_string_literal: true

module FixtureBot
  module FixtureCreator
    extend ::FactoryBot::Syntax::Methods
    extend ::FixtureBot::Helpers

    class << self
      attr_accessor :tables, :record_ids, :fixtures, :fixtures_with_id
    end

    self.tables = {}
    self.record_ids = {}

    module_function

    def load_to_db
      tmp_fixtures = []
      tmp_fixtures_with_id = []

      tables.each do |table, blocks|
        table_preload = TableLoader.new(table)

        blocks.each do |block|
          table_preload.instance_eval(&block)
        end

        tmp_fixtures.concat(table_preload.fixtures)
        tmp_fixtures_with_id.concat(table_preload.fixtures_with_id)
      end

      self.fixtures = tmp_fixtures.to_h
      self.fixtures_with_id = tmp_fixtures_with_id.to_h

      ::ActiveRecord::Base.connection.transaction requires_new: true do
        fixtures_with_id.each do |key, block|
          table, fixture_name = key.split("/")
          fixture_with_id(table.to_sym, fixture_name.to_sym, &block)
        end

        fixtures.each do |key, block|
          table, fixture_name = key.split("/")
          fixture(table.to_sym, fixture_name.to_sym, &block)
        end
      end
    end

    private def force_load_fixture(table, name)
      if (block = fixtures["#{table}/#{name}"] || fixtures_with_id["#{table}/#{name}"])
        fixture(table, name, &block)
      end
    end

    private def fixture(table, name, &)
      record_ids[table] ||= {}
      record_ids[table][name] ||=
        begin
          model = instance_eval(&)
          [model.class.name, model.id]
        end
    end

    private def fixture_with_id(table, name, &)
      _, record_id = fixture(table, name, &)

      if ::ActiveRecord::Base.connection.respond_to?(:postgresql_version)
        ::ActiveRecord::Base.connection.execute <<~SQL
          SELECT setval(pg_get_serial_sequence('#{table}', 'id'), GREATEST(#{record_id}, nextval(pg_get_serial_sequence('#{table}', 'id'))))
        SQL
      end
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
fixture_bot-0.3.0	lib/fixture_bot/fixture_creator.rb