diff --git a/Gemfile.lock b/Gemfile.lock index f7d69e42..393698d9 100644 --- a/Gemfile.lock +++ b/Gemfile.lock @@ -21,6 +21,11 @@ GEM erubi (~> 1.4) rails-dom-testing (~> 2.0) rails-html-sanitizer (~> 1.1, >= 1.2.0) + activemodel (7.0.4.3) + activesupport (= 7.0.4.3) + activerecord (7.0.4.3) + activemodel (= 7.0.4.3) + activesupport (= 7.0.4.3) activesupport (7.0.4.3) concurrent-ruby (~> 1.0, >= 1.0.2) i18n (>= 1.6, < 2) @@ -150,6 +155,8 @@ GEM rubocop (>= 1.7.0, < 2.0) rubocop-ast (>= 0.4.0) ruby-progressbar (1.13.0) + sqlite3 (1.6.2) + mini_portile2 (~> 2.8.0) standard (1.27.0) language_server-protocol (~> 3.17.0.2) rubocop (~> 1.50.2) @@ -167,12 +174,14 @@ PLATFORMS ruby DEPENDENCIES + activerecord (>= 5.0.0) appraisal aruba cucumber factory_bot_rails! rake rspec-rails + sqlite3 standard BUNDLED WITH diff --git a/factory_bot_rails.gemspec b/factory_bot_rails.gemspec index b4c518f9..1d83c52e 100644 --- a/factory_bot_rails.gemspec +++ b/factory_bot_rails.gemspec @@ -16,4 +16,7 @@ Gem::Specification.new do |s| s.add_runtime_dependency("factory_bot", "~> 6.2.0") s.add_runtime_dependency("railties", ">= 5.0.0") + + s.add_development_dependency("sqlite3") + s.add_development_dependency("activerecord", ">= 5.0.0") end diff --git a/lib/factory_bot_rails/factory_validator.rb b/lib/factory_bot_rails/factory_validator.rb new file mode 100644 index 00000000..8603c90c --- /dev/null +++ b/lib/factory_bot_rails/factory_validator.rb @@ -0,0 +1,17 @@ +module FactoryBotRails + class FactoryValidator + def initialize(validators = []) + @validators = Array(validators) + end + + def add_validator(validator) + @validators << validator + end + + def run + ActiveSupport::Notifications.subscribe("factory_bot.compile_factory") do |event| + @validators.each { |validator| validator.validate!(event.payload) } + end + end + end +end diff --git a/lib/factory_bot_rails/factory_validator/active_record_validator.rb b/lib/factory_bot_rails/factory_validator/active_record_validator.rb new file mode 100644 index 00000000..17eaec5e --- /dev/null +++ b/lib/factory_bot_rails/factory_validator/active_record_validator.rb @@ -0,0 +1,16 @@ +module FactoryBotRails + class FactoryValidator + class ActiveRecordValidator + def validate!(payload) + attributes, for_class = payload.values_at(:attributes, :class) + attributes.each do |attribute| + if for_class < ActiveRecord::Base && for_class.primary_key == attribute.name.to_s + raise FactoryBot::AttributeDefinitionError, <<~ERROR + Attribute generates #{for_class.primary_key.inspect} primary key for #{for_class.name}" + ERROR + end + end + end + end + end +end diff --git a/lib/factory_bot_rails/railtie.rb b/lib/factory_bot_rails/railtie.rb index 8dca2f4f..912b18b9 100644 --- a/lib/factory_bot_rails/railtie.rb +++ b/lib/factory_bot_rails/railtie.rb @@ -3,12 +3,15 @@ require "factory_bot" require "factory_bot_rails/generator" require "factory_bot_rails/reloader" +require "factory_bot_rails/factory_validator" require "rails" module FactoryBotRails class Railtie < Rails::Railtie config.factory_bot = ActiveSupport::OrderedOptions.new config.factory_bot.definition_file_paths = FactoryBot.definition_file_paths + config.factory_bot.validator = FactoryBotRails::FactoryValidator.new + config.factory_bot.reject_primary_key_attributes = true initializer "factory_bot.set_fixture_replacement" do Generator.new(config).run @@ -18,9 +21,22 @@ class Railtie < Rails::Railtie FactoryBot.definition_file_paths = definition_file_paths end + initializer "factory_bot.active_record_integration" do + ActiveSupport.on_load :active_record do + config = Rails.configuration.factory_bot + + if config.reject_primary_key_attributes + require "factory_bot_rails/factory_validator/active_record_validator" + + Rails.configuration.factory_bot.validator.add_validator FactoryValidator::ActiveRecordValidator.new + end + end + end + config.after_initialize do |app| FactoryBot.find_definitions Reloader.new(app).run + app.config.factory_bot.validator.run end private diff --git a/spec/factory_bot_rails/factory_validator/active_record_validator_spec.rb b/spec/factory_bot_rails/factory_validator/active_record_validator_spec.rb new file mode 100644 index 00000000..000476cc --- /dev/null +++ b/spec/factory_bot_rails/factory_validator/active_record_validator_spec.rb @@ -0,0 +1,24 @@ +# frozen_string_literal: true + +describe FactoryBotRails::FactoryValidator do + before { FactoryBot.reload } + + describe "ActiveRecordValidator" do + context "when defined with a class that descends from ActiveRecord::Base" do + it "raises an error for a sequence generating its primary key" do + define_model "Article", an_id: :integer do + self.primary_key = :an_id + end + + FactoryBot.define do + factory :article do + sequence(:an_id) + end + end + + expect { FactoryBot.create(:article) } + .to raise_error(FactoryBot::AttributeDefinitionError) + end + end + end +end diff --git a/spec/spec_helper.rb b/spec/spec_helper.rb index 5cbcde8a..1c77212d 100644 --- a/spec/spec_helper.rb +++ b/spec/spec_helper.rb @@ -5,6 +5,8 @@ require "factory_bot_rails" require "fake_app" +Dir["spec/support/**/*.rb"].each { |f| require File.expand_path(f) } + RSpec.configure do |config| config.run_all_when_everything_filtered = true config.filter_run :focus diff --git a/spec/support/macros/define_constant.rb b/spec/support/macros/define_constant.rb new file mode 100644 index 00000000..2621dd1e --- /dev/null +++ b/spec/support/macros/define_constant.rb @@ -0,0 +1,67 @@ +require "active_record" + +module DefineConstantMacros + def define_class(path, base = Object, &block) + const = stub_const(path, Class.new(base)) + const.class_eval(&block) if block + const + end + + def define_model(name, columns = {}, &block) + model = define_class(name, ActiveRecord::Base, &block) + create_table(model.table_name) do |table| + columns.each do |column_name, type| + table.column column_name, type + end + end + model + end + + def create_table(table_name, &block) + connection = ActiveRecord::Base.connection + + begin + connection.execute("DROP TABLE IF EXISTS #{table_name}") + connection.create_table(table_name, &block) + created_tables << table_name + connection + rescue Exception => e # rubocop:disable Lint/RescueException + connection.execute("DROP TABLE IF EXISTS #{table_name}") + raise e + end + end + + def clear_generated_tables + created_tables.each do |table_name| + clear_generated_table(table_name) + end + created_tables.clear + end + + def clear_generated_table(table_name) + ActiveRecord::Base + .connection + .execute("DROP TABLE IF EXISTS #{table_name}") + end + + private + + def created_tables + @created_tables ||= [] + end +end + +RSpec.configure do |config| + config.include DefineConstantMacros + + config.before(:all) do + ActiveRecord::Base.establish_connection( + adapter: "sqlite3", + database: ":memory:" + ) + end + + config.after do + clear_generated_tables + end +end