dima-exe-factory_girl 1.1.5.0

data/lib/factory_girl/factory.rb

@@ -0,0 +1,337 @@
+class Factory
+
+  class ScopeDefinitionError < RuntimeError
+  end
+
+  class ScopeNotFoundError < RuntimeError
+  end
+
+  class << self
+    attr_accessor :factories #:nodoc:
+
+    # An Array of strings specifying locations that should be searched for
+    # factory definitions. By default, factory_girl will attempt to require
+    # "factories," "test/factories," and "spec/factories." Only the first
+    # existing file will be loaded.
+    attr_accessor :definition_file_paths
+  end
+
+  self.factories = {}
+  self.definition_file_paths = %w(factories test/factories spec/factories)
+
+  attr_reader :factory_name
+
+  # Defines a new factory that can be used by the build strategies (create and
+  # build) to build new objects.
+  #
+  # Arguments:
+  #   name: (Symbol)
+  #     A unique name used to identify this factory.
+  #   options: (Hash)
+  #     class: the class that will be used when generating instances for this
+  #            factory. If not specified, the class will be guessed from the
+  #            factory name.
+  #
+  # Yields:
+  #    The newly created factory (Factory)
+  def self.define (name, options = {})
+    instance = Factory.new(name, options)
+    yield(instance)
+    self.factories[instance.factory_name] = instance
+  end
+
+  def build_class #:nodoc:
+    @build_class ||= class_for(@options[:class] || factory_name)
+  end
+
+  def initialize (name, options = {}) #:nodoc:
+    assert_valid_options(options)
+    @factory_name = factory_name_for(name)
+    @options      = options
+    @attributes   = []
+    @scopes       = {}
+  end
+
+  # Adds an attribute that should be assigned on generated instances for this
+  # factory.
+  #
+  # This method should be called with either a value or block, but not both. If
+  # called with a block, the attribute will be generated "lazily," whenever an
+  # instance is generated. Lazy attribute blocks will not be called if that
+  # attribute is overriden for a specific instance.
+  #
+  # When defining lazy attributes, an instance of Factory::AttributeProxy will
+  # be yielded, allowing associations to be built using the correct build
+  # strategy.
+  #
+  # Arguments:
+  #   name: (Symbol)
+  #     The name of this attribute. This will be assigned using :"#{name}=" for
+  #     generated instances.
+  #   value: (Object)
+  #     If no block is given, this value will be used for this attribute.
+  def add_attribute (name, value = nil, &block)
+    attribute = Attribute.new(name, value, block)
+
+    if attribute_defined?(attribute.name)
+      raise AttributeDefinitionError, "Attribute already defined: #{name}"
+    end
+
+    @attributes << attribute
+  end
+
+  def add_scope(name, attr = {})
+    if @scopes.has_key?(name.to_sym)
+      raise ScopeDefinitionError, "Scope already defined: #{name}"
+    end
+
+    attrs = []
+    attr.each_pair do |k,v|
+      attrs << Attribute.new(k, v, nil)
+    end
+
+    @scopes[name.to_sym] = attrs
+  end
+
+  # Calls add_attribute using the missing method name as the name of the
+  # attribute, so that:
+  #
+  #   Factory.define :user do |f|
+  #     f.name 'Billy Idol'
+  #   end
+  #
+  # and:
+  #
+  #   Factory.define :user do |f|
+  #     f.add_attribute :name, 'Billy Idol'
+  #   end
+  #
+  # are equivilent.
+  def method_missing (name, *args, &block)
+    add_attribute(name, *args, &block)
+  end
+
+  def self.method_missing(*args, &block)
+    chain   = args[0].to_s
+    name    = args[1]
+    factory = Factory.factories[name.to_sym]
+    super(*args, &block) unless factory
+
+    names = factory.scope_names.map(&:to_s)
+    super(*args, &block) if names.empty?
+
+    rs = true
+    assigned = {}
+    while rs
+      rs = names.any? do |n|
+        if chain.start_with?(n)
+          assigned.merge!(factory.scope_attributes_for(n.to_sym))
+          chain.gsub!(/^#{n}_?/, "")
+          !chain.empty?
+        end
+      end
+    end
+
+    raise ScopeNotFoundError, "Can't found scope with name '#{chain}'" unless chain.empty?
+    Factory(name.to_sym, assigned)
+  end
+
+  # Adds an attribute that builds an association. The associated instance will
+  # be built using the same build strategy as the parent instance.
+  #
+  # Example:
+  #   Factory.define :user do |f|
+  #     f.name 'Joey'
+  #   end
+  #
+  #   Factory.define :post do |f|
+  #     f.association :author, :factory => :user
+  #   end
+  #
+  # Arguments:
+  #   name: (Symbol)
+  #     The name of this attribute.
+  #   options: (Hash)
+  #     factory: (Symbol)
+  #       The name of the factory to use when building the associated instance.
+  #       If no name is given, the name of the attribute is assumed to be the
+  #       name of the factory. For example, a "user" association will by
+  #       default use the "user" factory.
+  def association (name, options = {})
+    name    = name.to_sym
+    options = symbolize_keys(options)
+    association_factory = options[:factory] || name
+
+    add_attribute(name) {|a| a.association(association_factory) }
+  end
+
+  def attributes_for (attrs = {}) #:nodoc:
+    build_attributes_hash(attrs, :attributes_for)
+  end
+
+  def scope_attributes_for(scope, attrs = {}) #:nodoc
+    build_attributes_hash(attrs, :attributes_for, @scopes[scope])
+  end
+
+  def scope_names
+    @scopes.keys
+  end
+
+  def build (attrs = {}) #:nodoc:
+    build_instance(attrs, :build)
+  end
+
+  def create (attrs = {}) #:nodoc:
+    instance = build_instance(attrs, :create)
+    instance.save!
+    instance
+  end
+
+  class << self
+
+    # Generates and returns a Hash of attributes from this factory. Attributes
+    # can be individually overridden by passing in a Hash of attribute => value
+    # pairs.
+    #
+    # Arguments:
+    #   attrs: (Hash)
+    #     Attributes to overwrite for this set.
+    #
+    # Returns:
+    #   A set of attributes that can be used to build an instance of the class
+    #   this factory generates. (Hash)
+    def attributes_for (name, attrs = {})
+      factory_by_name(name).attributes_for(attrs)
+    end
+
+    def scope_attributes_for(name, scope, attrs = {}) #:nodoc
+      factory_by_name(name).scope_attributes_for(scope, attrs)
+    end
+
+    # Generates and returns an instance from this factory. Attributes can be
+    # individually overridden by passing in a Hash of attribute => value pairs.
+    #
+    # Arguments:
+    #   attrs: (Hash)
+    #     See attributes_for
+    #
+    # Returns:
+    #   An instance of the class this factory generates, with generated
+    #   attributes assigned.
+    def build (name, attrs = {})
+      factory_by_name(name).build(attrs)
+    end
+
+    # Generates, saves, and returns an instance from this factory. Attributes can
+    # be individually overridden by passing in a Hash of attribute => value
+    # pairs.
+    #
+    # If the instance is not valid, an ActiveRecord::Invalid exception will be
+    # raised.
+    #
+    # Arguments:
+    #   attrs: (Hash)
+    #     See attributes_for
+    #
+    # Returns:
+    #   A saved instance of the class this factory generates, with generated
+    #   attributes assigned.
+    def create (name, attrs = {})
+      factory_by_name(name).create(attrs)
+    end
+
+    def find_definitions #:nodoc:
+      definition_file_paths.each do |path|
+        require("#{path}.rb") if File.exists?("#{path}.rb")
+
+        if File.directory? path
+          Dir[File.join(path, '*.rb')].each do |file|
+            require file
+          end
+        end
+      end
+    end
+
+    private
+
+    def factory_by_name (name)
+      factories[name.to_sym] or raise ArgumentError.new("No such factory: #{name.to_s}")
+    end
+
+  end
+
+  private
+
+  def build_attributes_hash (values, strategy, container = nil)
+    values = symbolize_keys(values)
+    passed_keys = values.keys.collect {|key| Factory.aliases_for(key) }.flatten
+    (container || @attributes).each do |attribute|
+      unless passed_keys.include?(attribute.name)
+        proxy = AttributeProxy.new(self, attribute.name, strategy, values)
+        values[attribute.name] = attribute.value(proxy)
+      end
+    end
+    values
+  end
+
+  def build_instance (override, strategy)
+    instance = build_class.new
+    attrs = build_attributes_hash(override, strategy)
+    attrs.each do |attr, value|
+      instance.send(:"#{attr}=", value)
+    end
+    instance
+  end
+
+  def class_for (class_or_to_s)
+    if class_or_to_s.respond_to?(:to_sym)
+      Object.const_get(variable_name_to_class_name(class_or_to_s))
+    else
+      class_or_to_s
+    end
+  end
+
+  def factory_name_for (class_or_to_s)
+    if class_or_to_s.respond_to?(:to_sym)
+      class_or_to_s.to_sym
+    else
+      class_name_to_variable_name(class_or_to_s).to_sym
+    end
+  end
+
+  def attribute_defined? (name)
+    !@attributes.detect {|attr| attr.name == name }.nil?
+  end
+
+  def assert_valid_options(options)
+    invalid_keys = options.keys - [:class]
+    unless invalid_keys == []
+      raise ArgumentError, "Unknown arguments: #{invalid_keys.inspect}"
+    end
+  end
+
+  # Based on ActiveSupport's underscore inflector
+  def class_name_to_variable_name(name)
+    name.to_s.gsub(/::/, '/').
+      gsub(/([A-Z]+)([A-Z][a-z])/,'\1_\2').
+      gsub(/([a-z\d])([A-Z])/,'\1_\2').
+      tr("-", "_").
+      downcase
+  end
+
+  # Based on ActiveSupport's camelize inflector
+  def variable_name_to_class_name(name)
+    name.to_s.
+      gsub(/\/(.?)/) { "::#{$1.upcase}" }.
+      gsub(/(?:^|_)(.)/) { $1.upcase }
+  end
+
+  # From ActiveSupport
+  def symbolize_keys(hash)
+    hash.inject({}) do |options, (key, value)|
+      options[(key.to_sym rescue key) || key] = value
+      options
+    end
+  end
+
+end

data/lib/factory_girl/sequence.rb

@@ -0,0 +1,58 @@
+class Factory
+
+  class Sequence
+
+    def initialize (&proc) #:nodoc:
+      @proc  = proc
+      @value = 0
+    end
+
+    # Returns the next value for this sequence
+    def next
+      @value += 1
+      @proc.call(@value)
+    end
+
+  end
+
+  class << self
+    attr_accessor :sequences #:nodoc:
+  end
+  self.sequences = {}
+
+  # Defines a new sequence that can be used to generate unique values in a specific format.
+  #
+  # Arguments:
+  #   name: (Symbol)
+  #     A unique name for this sequence. This name will be referenced when
+  #     calling next to generate new values from this sequence.
+  #   block: (Proc)
+  #     The code to generate each value in the sequence. This block will be
+  #     called with a unique number each time a value in the sequence is to be
+  #     generated. The block should return the generated value for the
+  #     sequence.
+  #
+  # Example:
+  #   
+  #   Factory.sequence(:email) {|n| "somebody_#{n}@example.com" }
+  def self.sequence (name, &block)
+    self.sequences[name] = Sequence.new(&block)
+  end
+
+  # Generates and returns the next value in a sequence.
+  #
+  # Arguments:
+  #   name: (Symbol)
+  #     The name of the sequence that a value should be generated for.
+  #
+  # Returns:
+  #   The next value in the sequence. (Object)
+  def self.next (sequence)
+    unless self.sequences.key?(sequence)
+      raise "No such sequence: #{sequence}"
+    end
+
+    self.sequences[sequence].next
+  end
+
+end

data/lib/factory_girl.rb

@@ -0,0 +1,27 @@
+require 'active_support'
+require 'factory_girl/factory'
+require 'factory_girl/attribute_proxy'
+require 'factory_girl/attribute'
+require 'factory_girl/sequence'
+require 'factory_girl/aliases'
+
+# Shortcut for Factory.create.
+#
+# Example:
+#   Factory(:user, :name => 'Joe')
+def Factory (name, attrs = {})
+  Factory.create(name, attrs)
+end
+
+if defined? Rails
+  Rails.configuration.after_initialize do
+    Factory.definition_file_paths = [
+      File.join(RAILS_ROOT, 'test', 'factories'),
+      File.join(RAILS_ROOT, 'spec', 'factories')
+    ]
+    Factory.find_definitions
+  end
+else
+  Factory.find_definitions
+end
+

data/test/aliases_test.rb

@@ -0,0 +1,29 @@
+require(File.join(File.dirname(__FILE__), 'test_helper'))
+
+class AliasesTest < Test::Unit::TestCase
+
+  should "include an attribute as an alias for itself by default" do
+    assert Factory.aliases_for(:test).include?(:test)
+  end
+
+  should "include the root of a foreign key as an alias by default" do
+    assert Factory.aliases_for(:test_id).include?(:test)
+  end
+
+  should "include an attribute's foreign key as an alias by default" do
+    assert Factory.aliases_for(:test).include?(:test_id)
+  end
+
+  context "after adding an alias" do
+
+    setup do
+      Factory.alias(/(.*)_suffix/, '\1')
+    end
+
+    should "return the alias in the aliases list" do
+      assert Factory.aliases_for(:test_suffix).include?(:test)
+    end
+
+  end
+
+end

data/test/attribute_proxy_test.rb

@@ -0,0 +1,121 @@
+require(File.join(File.dirname(__FILE__), 'test_helper'))
+
+class AttributeProxyTest < Test::Unit::TestCase
+
+  context "an association proxy" do
+
+    setup do
+      @factory  = mock('factory')
+      @attr     = :user
+      @attrs    = { :first_name => 'John' }
+      @strategy = :create
+      @proxy  = Factory::AttributeProxy.new(@factory, @attr, @strategy, @attrs)
+    end
+
+    should "have a factory" do
+      assert_equal @factory, @proxy.factory
+    end
+
+    should "have an attribute name" do
+      assert_equal @attr, @proxy.attribute_name
+    end
+
+    should "have a build strategy" do
+      assert_equal @strategy, @proxy.strategy
+    end
+
+    should "have attributes" do
+      assert_equal @attrs, @proxy.current_values
+    end
+
+    context "building an association" do
+
+      setup do
+        @association = mock('built-user')
+        @name        = :user
+        @attribs     = { :first_name => 'Billy' }
+
+        Factory.stubs(@strategy).returns(@association)
+      end
+
+      should "delegate to the appropriate method on Factory" do
+        Factory.expects(@strategy).with(@name, @attribs).returns(@association)
+        @proxy.association(@name, @attribs)
+      end
+
+      should "return the built association" do
+        assert_equal @association, @proxy.association(@name)
+      end
+
+    end
+
+    context "building an association using the attributes_for strategy" do
+
+      setup do
+        @strategy = :attributes_for
+        @proxy  = Factory::AttributeProxy.new(@factory, @attr, @strategy, @attrs)
+      end
+
+      should "not build the association" do
+        Factory.expects(@strategy).never
+        @proxy.association(:user)
+      end
+
+      should "return nil for the association" do
+        Factory.stubs(@strategy).returns(:user)
+        assert_nil @proxy.association(:user)
+      end
+
+    end
+
+    context "building an association using the build strategy" do
+
+      setup do
+        @strategy = :build
+        @built    = 'object'
+        @proxy    = Factory::AttributeProxy.new(@factory, @attr, @strategy, @attrs)
+        Factory.stubs(:create).returns(@built)
+      end
+
+      should "create the association" do
+        Factory.expects(:create).with(:user, {}).returns(@built)
+        @proxy.association(:user)
+      end
+
+      should "return the created object" do
+        assert_equal @built, @proxy.association(:user)
+      end
+
+    end
+
+    context "fetching the value of an attribute" do
+
+      setup do
+        @attr = :first_name
+      end
+
+      should "return the correct value" do
+        assert_equal @attrs[@attr], @proxy.value_for(@attr)
+      end
+
+      should "call value_for for undefined methods" do
+        assert_equal @attrs[@attr], @proxy.send(@attr)
+      end
+
+    end
+
+    context "fetching the value of an undefined attribute" do
+
+      setup do
+        @attr = :beachball
+      end
+
+      should "raise an ArgumentError" do
+        assert_raise(ArgumentError) { @proxy.value_for(@attr) }
+      end
+
+    end
+
+  end
+
+end

data/test/attribute_test.rb

@@ -0,0 +1,70 @@
+require(File.join(File.dirname(__FILE__), 'test_helper'))
+
+class AttributeTest < Test::Unit::TestCase
+
+  def setup
+    @proxy = mock('attribute-proxy')
+  end
+
+  context "an attribute" do
+
+    setup do
+      @name  = :user
+      @attr  = Factory::Attribute.new(@name, 'test', nil)
+    end
+
+    should "have a name" do
+      assert_equal @name, @attr.name
+    end
+
+  end
+
+  context "an attribute with a static value" do
+
+    setup do
+      @value = 'test'
+      @attr  = Factory::Attribute.new(:user, @value, nil)
+    end
+
+    should "return the value" do
+      assert_equal @value, @attr.value(@proxy)
+    end
+
+  end
+
+  context "an attribute with a lazy value" do
+
+    setup do
+      @block = lambda { 'value' }
+      @attr  = Factory::Attribute.new(:user, nil, @block)
+    end
+
+    should "call the block to return a value" do
+      assert_equal 'value', @attr.value(@proxy)
+    end
+
+    should "yield the attribute proxy to the block" do
+      @block = lambda {|a| a }
+      @attr  = Factory::Attribute.new(:user, nil, @block)
+      assert_equal @proxy, @attr.value(@proxy)
+    end
+
+  end
+
+  should "raise an error when defining an attribute writer" do
+    assert_raise Factory::AttributeDefinitionError do
+      Factory::Attribute.new('test=', nil, nil)
+    end
+  end
+
+  should "not allow attributes to be added with both a value parameter and a block" do
+    assert_raise(Factory::AttributeDefinitionError) do
+      Factory::Attribute.new(:name, 'value', lambda {})
+    end
+  end
+
+  should "convert names to symbols" do
+    assert_equal :name, Factory::Attribute.new('name', nil, nil).name
+  end
+
+end