156 lines
4.4 KiB
Ruby
156 lines
4.4 KiB
Ruby
require "set"
|
|
require "active_support/core_ext/module/delegation"
|
|
require "active_support/core_ext/module/attribute_accessors"
|
|
require "active_support/deprecation"
|
|
require "active_support/notifications"
|
|
|
|
require "factory_bot/definition_hierarchy"
|
|
require "factory_bot/configuration"
|
|
require "factory_bot/errors"
|
|
require "factory_bot/factory_runner"
|
|
require "factory_bot/strategy_syntax_method_registrar"
|
|
require "factory_bot/strategy_calculator"
|
|
require "factory_bot/strategy/build"
|
|
require "factory_bot/strategy/create"
|
|
require "factory_bot/strategy/attributes_for"
|
|
require "factory_bot/strategy/stub"
|
|
require "factory_bot/strategy/null"
|
|
require "factory_bot/registry"
|
|
require "factory_bot/null_factory"
|
|
require "factory_bot/null_object"
|
|
require "factory_bot/evaluation"
|
|
require "factory_bot/factory"
|
|
require "factory_bot/attribute_assigner"
|
|
require "factory_bot/evaluator"
|
|
require "factory_bot/evaluator_class_definer"
|
|
require "factory_bot/attribute"
|
|
require "factory_bot/callback"
|
|
require "factory_bot/callbacks_observer"
|
|
require "factory_bot/declaration_list"
|
|
require "factory_bot/declaration"
|
|
require "factory_bot/sequence"
|
|
require "factory_bot/attribute_list"
|
|
require "factory_bot/trait"
|
|
require "factory_bot/aliases"
|
|
require "factory_bot/definition"
|
|
require "factory_bot/definition_proxy"
|
|
require "factory_bot/syntax"
|
|
require "factory_bot/syntax_runner"
|
|
require "factory_bot/find_definitions"
|
|
require "factory_bot/reload"
|
|
require "factory_bot/decorator"
|
|
require "factory_bot/decorator/attribute_hash"
|
|
require "factory_bot/decorator/disallows_duplicates_registry"
|
|
require "factory_bot/decorator/invocation_tracker"
|
|
require "factory_bot/decorator/new_constructor"
|
|
require "factory_bot/linter"
|
|
require "factory_bot/version"
|
|
|
|
module FactoryBot
|
|
def self.configuration
|
|
@configuration ||= Configuration.new
|
|
end
|
|
|
|
def self.reset_configuration
|
|
@configuration = nil
|
|
end
|
|
|
|
mattr_accessor :use_parent_strategy, instance_accessor: false, default: true
|
|
|
|
# Look for errors in factories and (optionally) their traits.
|
|
# Parameters:
|
|
# factories - which factories to lint; omit for all factories
|
|
# options:
|
|
# traits: true - to lint traits as well as factories
|
|
# strategy: :create - to specify the strategy for linting
|
|
# verbose: true - to include full backtraces for each linting error
|
|
def self.lint(*args)
|
|
options = args.extract_options!
|
|
factories_to_lint = args[0] || FactoryBot.factories
|
|
Linter.new(factories_to_lint, options).lint!
|
|
end
|
|
|
|
class << self
|
|
delegate :factories,
|
|
:sequences,
|
|
:traits,
|
|
:callbacks,
|
|
:strategies,
|
|
:callback_names,
|
|
:to_create,
|
|
:skip_create,
|
|
:initialize_with,
|
|
:constructor,
|
|
to: :configuration
|
|
end
|
|
|
|
def self.register_factory(factory)
|
|
factory.names.each do |name|
|
|
factories.register(name, factory)
|
|
end
|
|
factory
|
|
end
|
|
|
|
def self.factory_by_name(name)
|
|
factories.find(name)
|
|
end
|
|
|
|
def self.register_sequence(sequence)
|
|
sequence.names.each do |name|
|
|
sequences.register(name, sequence)
|
|
end
|
|
sequence
|
|
end
|
|
|
|
def self.sequence_by_name(name)
|
|
sequences.find(name)
|
|
end
|
|
|
|
def self.rewind_sequences
|
|
sequences.each(&:rewind)
|
|
end
|
|
|
|
def self.register_trait(trait)
|
|
trait.names.each do |name|
|
|
traits.register(name, trait)
|
|
end
|
|
trait
|
|
end
|
|
|
|
def self.trait_by_name(name)
|
|
traits.find(name)
|
|
end
|
|
|
|
def self.register_strategy(strategy_name, strategy_class)
|
|
strategies.register(strategy_name, strategy_class)
|
|
StrategySyntaxMethodRegistrar.new(strategy_name).define_strategy_methods
|
|
end
|
|
|
|
def self.strategy_by_name(name)
|
|
strategies.find(name)
|
|
end
|
|
|
|
def self.register_default_strategies
|
|
register_strategy(:build, FactoryBot::Strategy::Build)
|
|
register_strategy(:create, FactoryBot::Strategy::Create)
|
|
register_strategy(:attributes_for, FactoryBot::Strategy::AttributesFor)
|
|
register_strategy(:build_stubbed, FactoryBot::Strategy::Stub)
|
|
register_strategy(:null, FactoryBot::Strategy::Null)
|
|
end
|
|
|
|
def self.register_default_callbacks
|
|
register_callback(:after_build)
|
|
register_callback(:after_create)
|
|
register_callback(:after_stub)
|
|
register_callback(:before_create)
|
|
end
|
|
|
|
def self.register_callback(name)
|
|
name = name.to_sym
|
|
callback_names << name
|
|
end
|
|
end
|
|
|
|
FactoryBot.register_default_strategies
|
|
FactoryBot.register_default_callbacks
|