factory_girl 1.1

data/LICENSE

@@ -0,0 +1,19 @@
+Copyright (c) 2008 Joe Ferris and thoughtbot, inc.
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.

data/README.textile

@@ -0,0 +1,105 @@
+= factory_girl
+
+  written by Joe Ferris <jferris@thoughtbot.com>
+  thanks to Tammer Saleh, Dan Croak, and Jon Yurek of thoughtbot, inc.
+  Copyright 2008 Joe Ferris and thoughtbot, inc.
+
+== Defining factories
+
+  # This will guess the User class
+  Factory.define :user do |u|
+    u.first_name 'John'
+    u.last_name  'Doe'
+    u.admin false
+  end
+
+  # This will use the User class (Admin would have been guessed)
+  Factory.define :admin, :class => User do |u|
+    u.first_name 'Admin'
+    u.last_name  'User'
+    u.admin true
+  end
+
+It is recommended that you create a test/factories.rb file and define your
+factories there. This file can be included from test_helper or directly from
+your test files. Don't forget:
+  require 'factory_girl'
+
+== Lazy Attributes
+
+Most attributes can be added using static values that are evaluated when the
+factory is defined, but some attributes (such as associations and other
+attributes that must be dynamically generated) will need values assigned each
+time an instance is generated. These "lazy" attributes can be added by passing
+a block instead of a parameter:
+
+  Factory.define :user do |u|
+    # ...
+    u.activation_code { User.generate_activation_code }
+  end
+
+== Dependent Attributes
+
+Some attributes may need to be generated based on the values of other
+attributes. This can be done by calling the attribute name on
+Factory::AttributeProxy, which is yielded to lazy attribute blocks:
+
+  Factory.define :user do |u|
+    u.first_name 'Joe'
+    u.last_name  'Blow'
+    u.email {|a| "#{a.first_name}.#{a.last_name}@example.com".downcase }
+  end
+
+  Factory(:user, :last_name => 'Doe').email
+  # => "joe.doe@example.com"
+
+== Associations
+
+Associated instances can be generated by using the association method when
+defining a lazy attribute:
+
+  Factory.define :post do |p|
+    # ...
+    p.author {|author| author.association(:user, :last_name => 'Writely') }
+  end
+
+When using the association method, the same build strategy (build, create, or attributes_for) will be used for all generated instances:
+
+  # Builds and saves a User and a Post
+  post = Factory(:post)
+  post.new_record?       # => false
+  post.author.new_record # => false
+
+  # Builds but does not save a User and a Post
+  Factory.build(:post)
+  post.new_record?       # => true
+  post.author.new_record # => true
+
+== Sequences
+
+Unique values in a specific format (for example, e-mail addresses) can be
+generated using sequences. Sequences are defined by calling Factory.sequence,
+and values in a sequence are generated by calling Factory.next:
+
+  # Defines a new sequence
+  Factory.sequence :email do |n|
+    "person#{n}@example.com"
+  end
+
+  Factory.next :email
+  # => "person1@example.com"
+
+  Factory.next :email
+  # => "person2@example.com"
+
+== Using factories
+
+  # Build and save a User instance
+  Factory(:user)
+
+  # Build a User instance and override the first_name property
+  Factory.build(:user, :first_name => 'Joe')
+
+  # Return an attributes Hash that can be used to build a User instance
+  attrs = Factory.attributes_for(:user)
+

data/Rakefile

@@ -0,0 +1,69 @@
+require 'rubygems'
+require 'rake'
+require 'rake/testtask'
+require 'rake/rdoctask'
+require 'rake/gempackagetask'
+require 'date'
+
+desc 'Default: run unit tests.'
+task :default => :test
+
+desc 'Test the factory_girl plugin.'
+Rake::TestTask.new(:test) do |t|
+  t.libs << 'lib'
+  t.pattern = 'test/**/*_test.rb'
+  t.verbose = true
+end
+
+desc 'Generate documentation for the factory_girl plugin.'
+Rake::RDocTask.new(:rdoc) do |rdoc|
+  rdoc.rdoc_dir = 'rdoc'
+  rdoc.title    = 'Factory Girl'
+  rdoc.options << '--line-numbers' << '--inline-source' << "--main" << "README.textile"
+  rdoc.rdoc_files.include('README.textile')
+  rdoc.rdoc_files.include('lib/**/*.rb')
+end
+
+desc 'Update documentation on website'
+task :sync_docs => 'rdoc' do
+  `rsync -ave ssh rdoc/ dev@dev.thoughtbot.com:/home/dev/www/dev.thoughtbot.com/factory_girl`
+end
+
+spec = Gem::Specification.new do |s|
+  s.name        = %q{factory_girl}
+  s.version     = "1.1"
+  s.summary     = %q{factory_girl provides a framework and DSL for defining and
+                     using model instance factories.}
+  s.description = %q{factory_girl provides a framework and DSL for defining and
+                     using factories - less error-prone, more explicit, and 
+                     all-around easier to work with than fixtures.}
+
+  s.files        = FileList['[A-Z]*', 'lib/**/*.rb', 'test/**/*.rb']
+  s.require_path = 'lib'
+  s.test_files   = Dir[*['test/**/*_test.rb']]
+
+  s.has_rdoc         = true
+  s.extra_rdoc_files = ["README.textile"]
+  s.rdoc_options = ['--line-numbers', '--inline-source', "--main", "README.textile"]
+
+  s.authors = ["Joe Ferris"]
+  s.email   = %q{jferris@thoughtbot.com}
+
+  s.platform = Gem::Platform::RUBY
+  s.add_dependency(%q<activesupport>, [">= 1.0"])
+end
+
+Rake::GemPackageTask.new spec do |pkg|
+  pkg.need_tar = true
+  pkg.need_zip = true
+end
+
+desc "Clean files generated by rake tasks"
+task :clobber => [:clobber_rdoc, :clobber_package]
+
+desc "Generate a gemspec file"
+task :gemspec do
+  File.open("#{spec.name}.gemspec", 'w') do |f|
+    f.write spec.to_ruby
+  end
+end

data/lib/factory_girl.rb

@@ -0,0 +1,12 @@
+require 'activesupport'
+require 'factory_girl/factory'
+require 'factory_girl/attribute_proxy'
+require 'factory_girl/sequence'
+
+# Shortcut for Factory.create.
+#
+# Example:
+#   Factory(:user, :name => 'Joe')
+def Factory (name, attrs = {})
+  Factory.create(name, attrs)
+end

data/lib/factory_girl/attribute_proxy.rb

@@ -0,0 +1,92 @@
+class Factory
+
+  class AttributeProxy
+
+    attr_accessor :factory, :attribute_name, :strategy, :current_values #:nodoc:
+
+    def initialize (factory, attr, strategy, values) #:nodoc:
+      @factory        = factory
+      @attribute_name = attr
+      @strategy       = strategy
+      @current_values = values
+    end
+
+    # Generates an association using the current build strategy.
+    #
+    # Arguments:
+    #   name: (Symbol)
+    #     The name of the factory that should be used to generate this
+    #     association.
+    #   attributes: (Hash)
+    #     A hash of attributes that should be overridden for this association.
+    #
+    # Returns:
+    #   The generated association for the current build strategy. Note that
+    #   assocaitions are not generated for the attributes_for strategy. Returns
+    #   nil in this case.
+    #
+    # Example:
+    #
+    #   Factory.define :user do |f|
+    #     # ...
+    #   end
+    #
+    #   Factory.define :post do |f|
+    #     # ...
+    #     f.author {|a| a.association :user, :name => 'Joe' }
+    #   end
+    #
+    #   # Builds (but doesn't save) a Post and a User
+    #   Factory.build(:post)
+    #
+    #   # Builds and saves a User, builds a Post, assigns the User to the 
+    #   # author association, and saves the User.
+    #   Factory.create(:post)
+    #
+    def association (name, attributes = {})
+      if strategy == :attributes_for
+        nil
+      else
+        Factory.send(strategy, name, attributes)
+      end
+    end
+
+    # Returns the value for specified attribute. A value will only be available
+    # if it was overridden when calling the factory, or if a value is added
+    # earlier in the definition of the factory.
+    #
+    # Arguments:
+    #   attribute: (Symbol)
+    #     The attribute whose value should be returned.
+    #
+    # Returns:
+    #   The value of the requested attribute. (Object)
+    def value_for (attribute)
+      unless current_values.key?(attribute)
+        raise ArgumentError, "No such attribute: #{attribute.inspect}"
+      end
+      current_values[attribute]
+    end
+
+    # Undefined methods are delegated to value_for, which means that:
+    #
+    #   Factory.define :user do |f|
+    #     f.first_name 'Ben'
+    #     f.last_name {|a| a.value_for(:first_name) }
+    #   end
+    #
+    # and:
+    #
+    #   Factory.define :user do |f|
+    #     f.first_name 'Ben'
+    #     f.last_name {|a| a.first_name }
+    #   end
+    #
+    # are equivilent.
+    def method_missing (name, *args, &block)
+      current_values[name]
+    end
+
+  end
+
+end

data/lib/factory_girl/factory.rb

@@ -0,0 +1,216 @@
+class Factory
+
+  cattr_accessor :factories, :sequences #:nodoc:
+  self.factories = {}
+  self.sequences = {}
+
+  attr_reader :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[name] = instance
+  end
+
+  # 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
+
+  def build_class #:nodoc:
+    @build_class ||= @options[:class] || name.to_s.classify.constantize
+  end
+
+  def initialize (name, options = {}) #:nodoc:
+    options.assert_valid_keys(:class)
+    @name    = name
+    @options = options
+
+    @static_attributes     = {}
+    @lazy_attribute_blocks = {}
+    @lazy_attribute_names  = []
+  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)
+    if block_given?
+      unless value.nil?
+        raise ArgumentError, "Both value and block given"
+      end
+      @lazy_attribute_blocks[name] = block
+      @lazy_attribute_names << name
+    else
+      @static_attributes[name] = value
+    end
+  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 :user, 'Billy Idol'
+  #   end
+  #
+  # are equivilent. 
+  def method_missing (name, *args, &block)
+    add_attribute(name, *args, &block)
+  end
+
+  def attributes_for (attrs = {}) #:nodoc:
+    build_attributes_hash(attrs, :attributes_for)
+  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
+
+    # 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
+
+    private
+
+    def factory_by_name (name)
+      factories[name] or raise ArgumentError.new("No such factory: #{name.inspect}")
+    end
+
+  end
+
+  private
+
+  def build_attributes_hash (override, strategy)
+    result = @static_attributes.merge(override)
+    @lazy_attribute_names.each do |name|
+      proxy = AttributeProxy.new(self, name, strategy, result)
+      result[name] = @lazy_attribute_blocks[name].call(proxy) unless override.key?(name)
+    end
+    result
+  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
+
+end

data/lib/factory_girl/sequence.rb

@@ -0,0 +1,17 @@
+class Factory
+
+  class Sequence
+
+    def initialize (&proc)
+      @proc  = proc
+      @value = 0
+    end
+
+    def next
+      @value += 1
+      @proc.call(@value)
+    end
+
+  end
+
+end

data/test/attribute_proxy_test.rb

@@ -0,0 +1,101 @@
+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 "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/factory_test.rb

@@ -0,0 +1,333 @@
+require(File.join(File.dirname(__FILE__), 'test_helper'))
+
+class FactoryTest < Test::Unit::TestCase
+
+  def self.should_instantiate_class
+
+    should "instantiate the build class" do
+      assert_kind_of @class, @instance
+    end
+
+    should "assign attributes on the instance" do
+      assert_equal @first_name, @instance.first_name
+      assert_equal @last_name,  @instance.last_name
+    end
+
+    should "override attributes using the passed hash" do
+      @value = 'Davis'
+      @instance = @factory.build(:first_name => @value)
+      assert_equal @value, @instance.first_name
+    end
+
+  end
+
+  context "defining a factory" do
+
+    setup do
+      @name    = :user
+      @factory = mock('factory')
+      @options = { :class => 'magic' }
+      Factory.stubs(:new).returns(@factory)
+    end
+
+    should "create a new factory using the specified name and options" do
+      Factory.expects(:new).with(@name, @options)
+      Factory.define(@name, @options) {|f| }
+    end
+
+    should "pass the factory do the block" do
+      yielded = nil
+      Factory.define(@name) do |y|
+        yielded = y
+      end
+      assert_equal @factory, yielded
+    end
+
+    should "add the factory to the list of factories" do
+      Factory.define(@name) {|f| }
+      assert_equal Factory.factories[@name], 
+                   @factory,
+                   "Factories: #{Factory.factories.inspect}"
+    end
+
+  end
+
+  context "defining a sequence" do
+
+    setup do
+      @sequence = mock('sequence')
+      @name     = :count
+      Factory::Sequence.stubs(:new).returns(@sequence)
+    end
+
+    should "create a new sequence" do
+      Factory::Sequence.expects(:new).with().returns(@sequence)
+      Factory.sequence(@name)
+    end
+
+    should "use the supplied block as the sequence generator" do
+      Factory::Sequence.stubs(:new).yields(1)
+      yielded = false
+      Factory.sequence(@name) {|n| yielded = true }
+      assert yielded
+    end
+
+  end
+
+  context "a factory" do
+
+    setup do
+      @name    = :user
+      @class   = User
+      @factory = Factory.new(@name)
+    end
+
+    should "have a name" do
+      assert_equal @name, @factory.name
+    end
+
+    should "have a build class" do
+      assert_equal @class, @factory.build_class
+    end
+
+    context "when adding an attribute with a value parameter" do
+
+      setup do
+        @attr  = :name
+        @value = 'Elvis lives!'
+        @factory.add_attribute(@attr, @value)
+      end
+
+      should "include that value in the generated attributes hash" do
+        assert_equal @value, @factory.attributes_for[@attr]
+      end
+
+    end
+
+    context "when adding an attribute with a block" do
+
+      setup do
+        @attr  = :name
+        @attrs = {}
+        @proxy = mock('attr-proxy')
+        Factory::AttributeProxy.stubs(:new).returns(@proxy)
+      end
+
+      should "not evaluate the block when the attribute is added" do
+        @factory.add_attribute(@attr) { flunk }
+      end
+
+      should "evaluate the block when attributes are generated" do
+        called = false
+        @factory.add_attribute(@attr) do
+          called = true
+        end
+        @factory.attributes_for
+        assert called
+      end
+
+      should "use the result of the block as the value of the attribute" do
+        value = "Watch out for snakes!"
+        @factory.add_attribute(@attr) { value }
+        assert_equal value, @factory.attributes_for[@attr]
+      end
+
+      should "build an attribute proxy" do
+        Factory::AttributeProxy.expects(:new).with(@factory, @attr, :attributes_for, @attrs)
+        @factory.add_attribute(@attr) {}
+        @factory.attributes_for
+      end
+
+      should "yield an attribute proxy to the block" do
+        yielded = nil
+        @factory.add_attribute(@attr) {|y| yielded = y }
+        @factory.attributes_for
+        assert_equal @proxy, yielded
+      end
+
+      context "when other attributes have previously been defined" do
+        
+        setup do
+          @attr  = :unimportant
+          @attrs = {
+            :one     => 'whatever',
+            :another => 'soup'
+          }
+          @factory.add_attribute(:one, 'whatever')
+          @factory.add_attribute(:another) { 'soup' }
+          @factory.add_attribute(@attr) {}
+        end
+
+        should "provide previously set attributes" do
+          Factory::AttributeProxy.expects(:new).with(@factory, @attr, :attributes_for, @attrs)
+          @factory.attributes_for
+        end
+
+      end
+
+    end
+
+    should "add an attribute using the method name when passed an undefined method" do
+      @attr  = :first_name
+      @value = 'Sugar'
+      @factory.send(@attr, @value)
+      assert_equal @value, @factory.attributes_for[@attr]
+    end
+
+    should "not allow attributes to be added with both a value parameter and a block" do
+      assert_raise(ArgumentError) do
+        @factory.add_attribute(:name, 'value') {}
+      end
+    end
+
+    context "when overriding generated attributes with a hash" do
+
+      setup do
+        @attr  = :name
+        @value = 'The price is right!'
+        @hash  = { @attr => @value }
+      end
+
+      should "return the overridden value in the generated attributes" do
+        @factory.add_attribute(@attr, 'The price is wrong, Bob!')
+        assert_equal @value, @factory.attributes_for(@hash)[@attr]
+      end
+
+      should "not call a lazy attribute block for an overridden attribute" do
+        @factory.add_attribute(@attr) { flunk }
+        @factory.attributes_for(@hash)
+      end
+
+    end
+
+    should "guess the build class from the factory name" do
+      assert_equal User, @factory.build_class
+    end
+
+    context "when defined with a custom class" do
+
+      setup do
+        @class   = User
+        @factory = Factory.new(:author, :class => @class)
+      end
+
+      should "use the specified class as the build class" do
+        assert_equal @class, @factory.build_class
+      end
+
+    end
+
+    context "with some attributes added" do
+
+      setup do
+        @first_name = 'Billy'
+        @last_name  = 'Idol'
+        @email      = 'test@something.com'
+
+        @factory.add_attribute(:first_name, @first_name)
+        @factory.add_attribute(:last_name,  @last_name)
+        @factory.add_attribute(:email,      @email)
+      end
+
+      context "when building an instance" do
+
+        setup do
+          @instance = @factory.build
+        end
+
+        should_instantiate_class
+
+        should "not save the instance" do
+          assert @instance.new_record?
+        end
+
+      end
+      
+      context "when creating an instance" do
+
+        setup do
+          @instance = @factory.create
+        end
+
+        should_instantiate_class
+
+        should "save the instance" do
+          assert !@instance.new_record?
+        end
+
+      end
+
+      should "raise an ActiveRecord::RecordInvalid error for invalid instances" do
+        assert_raise(ActiveRecord::RecordInvalid) do
+          @factory.create(:first_name => nil)
+        end
+      end
+
+    end
+
+  end
+
+  context "Factory class" do
+
+    setup do
+      @name       = :user
+      @attrs      = { :last_name => 'Override' }
+      @first_name = 'Johnny'
+      @last_name  = 'Winter'
+      @class      = User
+
+      Factory.define(@name) do |u|
+        u.first_name @first_name
+        u.last_name  { @last_name }
+        u.email      'jwinter@guitar.org'
+      end
+
+      @factory = Factory.factories[@name]
+    end
+
+    [:build, :create, :attributes_for].each do |method|
+
+      should "delegate the #{method} method to the factory instance" do
+        @factory.expects(method).with(@attrs)
+        Factory.send(method, @name, @attrs)
+      end
+
+      should "raise an ArgumentError when #{method} is called with a nonexistant factory" do
+        assert_raise(ArgumentError) { Factory.send(method, :bogus) }
+      end
+
+    end
+
+    should "call the create method from the top-level Factory() method" do
+      @factory.expects(:create).with(@attrs)
+      Factory(@name, @attrs)
+    end
+
+  end
+
+  context "after defining a sequence" do
+
+    setup do
+      @sequence = mock('sequence')
+      @name     = :test
+      @value    = '1 2 5'
+
+      @sequence.        stubs(:next).returns(@value)
+      Factory::Sequence.stubs(:new). returns(@sequence)
+
+      Factory.sequence(@name) {}
+    end
+
+    should "call next on the sequence when sent next" do
+      @sequence.expects(:next)
+
+      Factory.next(@name)
+    end
+
+    should "return the value from the sequence" do
+      assert_equal @value, Factory.next(@name)
+    end
+
+  end
+
+end

data/test/integration_test.rb

@@ -0,0 +1,143 @@
+require(File.join(File.dirname(__FILE__), 'test_helper'))
+
+class IntegrationTest < Test::Unit::TestCase
+
+  def setup
+    Factory.define :user do |f|
+      f.first_name 'Jimi'
+      f.last_name  'Hendrix'
+      f.admin       false
+      f.email {|a| "#{a.first_name}.#{a.last_name}@example.com".downcase }
+    end
+
+    Factory.define :post do |f|
+      f.title 'Test Post'
+      f.author {|a| a.association(:user) }
+    end
+
+    Factory.define :admin, :class => User do |f|
+      f.first_name 'Ben'
+      f.last_name  'Stein'
+      f.admin       true
+      f.email { Factory.next(:email) }
+    end
+
+    Factory.sequence :email do |n|
+      "somebody#{n}@example.com"
+    end
+  end
+
+  def teardown
+    Factory.send(:class_variable_get, "@@factories").clear
+  end
+
+  context "a generated attributes hash" do
+
+    setup do
+      @attrs = Factory.attributes_for(:user, :first_name => 'Bill')
+    end
+
+    should "assign all attributes" do
+      assert_equal [:admin, :email, :first_name, :last_name],
+                   @attrs.keys.sort {|a, b| a.to_s <=> b.to_s }
+    end
+
+    should "correctly assign lazy, dependent attributes" do
+      assert_equal "bill.hendrix@example.com", @attrs[:email]
+    end
+
+    should "override attrbutes" do
+      assert_equal 'Bill', @attrs[:first_name]
+    end
+
+    should "not assign associations" do
+      assert_nil Factory.attributes_for(:post)[:author]
+    end
+
+  end
+
+  context "a built instance" do
+
+    setup do
+      @instance = Factory.build(:post)
+    end
+
+    should "not be saved" do
+      assert @instance.new_record?
+    end
+
+    should "assign associations" do
+      assert_kind_of User, @instance.author
+    end
+
+    should "not save associations" do
+      assert @instance.author.new_record?
+    end
+
+  end
+
+  context "a created instance" do
+
+    setup do
+      @instance = Factory.create(:post)
+    end
+
+    should "be saved" do
+      assert !@instance.new_record?
+    end
+
+    should "assign associations" do
+      assert_kind_of User, @instance.author
+    end
+
+    should "save associations" do
+      assert !@instance.author.new_record?
+    end
+
+  end
+
+  context "an instance generated by a factory with a custom class name" do
+
+    setup do
+      @instance = Factory.create(:admin)
+    end
+
+    should "use the correct class name" do
+      assert_kind_of User, @instance
+    end
+
+    should "use the correct factory definition" do
+      assert @instance.admin?
+    end
+
+  end
+
+  context "an attribute generated by a sequence" do
+
+    setup do
+      @email = Factory.attributes_for(:admin)[:email]
+    end
+
+    should "match the correct format" do
+      assert_match /^somebody\d+@example\.com$/, @email
+    end
+
+    context "after the attribute has already been generated once" do
+
+      setup do
+        @another_email = Factory.attributes_for(:admin)[:email]
+      end
+
+      should "match the correct format" do
+        assert_match /^somebody\d+@example\.com$/, @email
+      end
+
+      should "not be the same as the first generated value" do
+        assert_not_equal @email, @another_email
+      end
+
+    end
+
+  end
+
+end

data/test/models.rb

@@ -0,0 +1,32 @@
+ActiveRecord::Base.establish_connection(
+  :adapter  => 'sqlite3',
+  :database => File.join(File.dirname(__FILE__), 'test.db')
+)
+
+class CreateSchema < ActiveRecord::Migration
+  def self.up
+    create_table :users, :force => true do |t|
+      t.string  :first_name
+      t.string  :last_name
+      t.string  :email
+      t.boolean :admin, :default => false
+    end
+
+    create_table :posts, :force => true do |t|
+      t.string  :title
+      t.integer :author_id
+    end
+  end
+end
+
+CreateSchema.suppress_messages { CreateSchema.migrate(:up) }
+
+class User < ActiveRecord::Base
+  validates_presence_of :first_name, :last_name, :email
+  has_many :posts, :foreign_key => 'author_id'
+end
+
+class Post < ActiveRecord::Base
+  validates_presence_of :title, :author_id
+  belongs_to :author, :class_name => 'User'
+end

data/test/sequence_test.rb

@@ -0,0 +1,29 @@
+require(File.join(File.dirname(__FILE__), 'test_helper'))
+
+class SequenceTest < Test::Unit::TestCase
+
+  context "a sequence" do
+
+    setup do
+      @sequence = Factory::Sequence.new {|n| "=#{n}" }
+    end
+
+    should "start with a value of 1" do
+      assert_equal "=1", @sequence.next
+    end
+
+    context "after being called" do
+
+      setup do
+        @sequence.next
+      end
+
+      should "use the next value" do
+        assert_equal "=2", @sequence.next
+      end
+
+    end
+
+  end
+
+end

data/test/test_helper.rb

@@ -0,0 +1,10 @@
+$: << File.join(File.dirname(__FILE__), '..', 'lib')
+$: << File.join(File.dirname(__FILE__))
+
+require 'rubygems'
+require 'test/unit'
+require 'activerecord'
+require 'factory_girl'
+require 'shoulda'
+require 'mocha'
+require 'models'

metadata

@@ -0,0 +1,82 @@
+--- !ruby/object:Gem::Specification 
+name: factory_girl
+version: !ruby/object:Gem::Version 
+  version: "1.1"
+platform: ruby
+authors: 
+- Joe Ferris
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2008-06-06 00:00:00 -04:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: activesupport
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "1.0"
+    version: 
+description: factory_girl provides a framework and DSL for defining and using factories - less error-prone, more explicit, and  all-around easier to work with than fixtures.
+email: jferris@thoughtbot.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- README.textile
+files: 
+- LICENSE
+- Rakefile
+- README.textile
+- lib/factory_girl/attribute_proxy.rb
+- lib/factory_girl/factory.rb
+- lib/factory_girl/sequence.rb
+- lib/factory_girl.rb
+- test/attribute_proxy_test.rb
+- test/factory_test.rb
+- test/integration_test.rb
+- test/models.rb
+- test/sequence_test.rb
+- test/test_helper.rb
+has_rdoc: true
+homepage: 
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --line-numbers
+- --inline-source
+- --main
+- README.textile
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.5
+signing_key: 
+specification_version: 2
+summary: factory_girl provides a framework and DSL for defining and using model instance factories.
+test_files: 
+- test/attribute_proxy_test.rb
+- test/factory_test.rb
+- test/integration_test.rb
+- test/sequence_test.rb