vitalish-factory_girl 1.2.4

data/lib/factory_girl/proxy/create.rb

@@ -0,0 +1,12 @@
+module FactoryGirl
+  class Proxy #:nodoc:
+    class Create < Build #:nodoc:
+      def result
+        run_callbacks(:after_build)
+        @instance.save!
+        run_callbacks(:after_create)
+        @instance
+      end
+    end
+  end
+end

data/lib/factory_girl/proxy/stub.rb

@@ -0,0 +1,64 @@
+module FactoryGirl
+  class Proxy
+    class Stub < Proxy #:nodoc:
+      @@next_id = 1000
+
+      def initialize(klass)
+        @instance = klass.new
+        @instance.id = next_id
+        @instance.instance_eval do
+          def new_record?
+            id.nil?
+          end
+
+          def save(*args)
+            raise "stubbed models are not allowed to access the database"
+          end
+
+          def destroy(*args)
+            raise "stubbed models are not allowed to access the database"
+          end
+
+          def connection
+            raise "stubbed models are not allowed to access the database"
+          end
+
+          def reload
+            raise "stubbed models are not allowed to access the database"
+          end
+
+          def update_attribute(*args)
+            raise "stubbed models are not allowed to access the database"
+          end
+        end
+      end
+
+      def next_id
+        @@next_id += 1
+      end
+
+      def get(attribute)
+        @instance.send(attribute)
+      end
+
+      def set(attribute, value)
+        @instance.send(:"#{attribute}=", value)
+      end
+
+      def associate(name, factory_name, overrides)
+        factory = FactoryGirl.factory_by_name(factory_name)
+        set(name, factory.run(Proxy::Stub, overrides))
+      end
+
+      def association(factory_name, overrides = {})
+        factory = FactoryGirl.factory_by_name(factory_name)
+        factory.run(Proxy::Stub, overrides)
+      end
+
+      def result
+        run_callbacks(:after_stub)
+        @instance
+      end
+    end
+  end
+end

data/lib/factory_girl/sequence.rb

@@ -0,0 +1,28 @@
+module FactoryGirl
+
+  # Raised when calling Factory.sequence from a dynamic attribute block
+  class SequenceAbuseError < StandardError; end
+
+  # Sequences are defined using Factory.sequence. Sequence values are generated
+  # using next.
+  class Sequence
+
+    def initialize(value = 1, &proc) #:nodoc:
+      @proc  = proc
+      @value = value || 1
+    end
+
+    # Returns the next value for this sequence
+    def next
+      @proc.call(@value)
+    ensure
+      @value = @value.next
+    end
+
+  end
+
+  class << self
+    attr_accessor :sequences #:nodoc:
+  end
+  self.sequences = {}
+end

data/lib/factory_girl/step_definitions.rb

@@ -0,0 +1,60 @@
+module FactoryGirlStepHelpers
+
+  def convert_association_string_to_instance(factory_name, assignment)
+    attribute, value = assignment.split(':', 2)
+    return if value.blank?
+    factory = FactoryGirl.factory_by_name(factory_name)
+    attributes = convert_human_hash_to_attribute_hash({attribute => value.strip}, factory.associations)
+    attributes_find = {}
+    attributes.each do |k, v|
+      k = "#{k}_id" if v.is_a? ActiveRecord::Base
+      attributes_find[k] = v
+    end
+    model_class = factory.build_class
+    model_class.find(:first, :conditions => attributes_find) or
+      Factory(factory_name, attributes)
+  end
+
+  def convert_human_hash_to_attribute_hash(human_hash, associations = [])
+    human_hash.inject({}) do |attribute_hash, (human_key, value)|
+      key = human_key.downcase.gsub(' ', '_').to_sym
+      if association = associations.detect {|association| association.name == key }
+        value = convert_association_string_to_instance(association.factory, value)
+      end
+      attribute_hash.merge(key => value)
+    end
+  end
+end
+
+World(FactoryGirlStepHelpers)
+
+FactoryGirl.factories.values.each do |factory|
+  Given /^the following (?:#{factory.human_name}|#{factory.human_name.pluralize}) exists?:$/ do |table|
+    table.hashes.each do |human_hash|
+      attributes = convert_human_hash_to_attribute_hash(human_hash, factory.associations)
+      factory.run(FactoryGirl::Proxy::Create, attributes)
+    end
+  end
+
+  Given /^an? #{factory.human_name} exists$/ do
+    Factory(factory.name)
+  end
+
+  Given /^(\d+) #{factory.human_name.pluralize} exist$/ do |count|
+    count.to_i.times { Factory(factory.name) }
+  end
+
+  if factory.build_class.respond_to?(:columns)
+    factory.build_class.columns.each do |column|
+      human_column_name = column.name.downcase.gsub('_', ' ')
+      Given /^an? #{factory.human_name} exists with an? #{human_column_name} of "([^"]*)"$/i do |value|
+        Factory(factory.name, column.name => value)
+      end
+
+      Given /^(\d+) #{factory.human_name.pluralize} exist with an? #{human_column_name} of "([^"]*)"$/i do |count, value|
+        count.to_i.times { Factory(factory.name, column.name => value) }
+      end
+    end
+  end
+end
+

data/lib/factory_girl/syntax.rb

@@ -0,0 +1,12 @@
+module FactoryGirl
+  # Provides alternate syntaxes for factory_girl. If you don't like the default
+  # syntax for defining or using factories, look at one of the
+  # FactoryGirl::Syntax modules:
+  #
+  # * FactoryGirl::Syntax::Blueprint: definition syntax similar to Machinist
+  # * FactoryGirl::Syntax::Generate: usage syntax similar to Object Daddy
+  # * FactoryGirl::Syntax::Make: usage syntax similar to Machinist
+  # * FactoryGirl::Syntax::Sham: sequence syntax similar to Machinist
+  module Syntax
+  end
+end

data/lib/factory_girl/syntax/blueprint.rb

@@ -0,0 +1,42 @@
+module FactoryGirl
+  module Syntax
+
+    # Extends ActiveRecord::Base to provide a make class method, which is an
+    # alternate syntax for defining factories.
+    #
+    # Usage:
+    #
+    #   require 'factory_girl/syntax/blueprint'
+    #
+    #   User.blueprint do
+    #     name  { 'Billy Bob'             }
+    #     email { 'billy@bob.example.com' }
+    #   end
+    #
+    #   Factory(:user, :name => 'Johnny')
+    #
+    # This syntax was derived from Pete Yandell's machinist.
+    module Blueprint
+      module ActiveRecord #:nodoc:
+
+        def self.included(base) # :nodoc:
+          base.extend ClassMethods
+        end
+
+        module ClassMethods #:nodoc:
+
+          def blueprint(&block)
+            instance = Factory.new(name.underscore, :class => self)
+            proxy = FactoryGirl::DefinitionProxy.new(instance)
+            proxy.instance_eval(&block)
+            FactoryGirl.register_factory(instance)
+          end
+
+        end
+
+      end
+    end
+  end
+end
+
+ActiveRecord::Base.send(:include, FactoryGirl::Syntax::Blueprint::ActiveRecord)

data/lib/factory_girl/syntax/generate.rb

@@ -0,0 +1,73 @@
+module FactoryGirl
+  module Syntax
+
+    # Extends ActiveRecord::Base to provide generation methods for factories.
+    #
+    # Usage:
+    #
+    #   require 'factory_girl/syntax/generate'
+    #
+    #   FactoryGirl.define do
+    #     factory :user do
+    #       name 'Billy Bob'
+    #       email 'billy@bob.example.com'
+    #     end
+    #   end
+    #
+    #   # Creates a saved instance without raising (same as saving the result
+    #   # of Factory.build)
+    #   User.generate(:name => 'Johnny')
+    #
+    #   # Creates a saved instance and raises when invalid (same as
+    #   # Factory.create)
+    #   User.generate!
+    #
+    #   # Creates an unsaved instance (same as Factory.build)
+    #   User.spawn
+    #
+    #   # Creates an instance and yields it to the passed block
+    #   User.generate do |user|
+    #     # ...do something with user...
+    #   end
+    #
+    # This syntax was derived from Rick Bradley and Yossef Mendelssohn's
+    # object_daddy.
+    module Generate
+      module ActiveRecord #:nodoc:
+
+        def self.included(base) # :nodoc:
+          base.extend ClassMethods
+        end
+
+        module ClassMethods #:nodoc:
+
+          def generate(overrides = {}, &block)
+            factory = FactoryGirl.factory_by_name(name.underscore)
+            instance = factory.run(Proxy::Build, overrides)
+            instance.save
+            yield(instance) if block_given?
+            instance
+          end
+
+          def generate!(overrides = {}, &block)
+            factory = FactoryGirl.factory_by_name(name.underscore)
+            instance = factory.run(Proxy::Create, overrides)
+            yield(instance) if block_given?
+            instance
+          end
+
+          def spawn(overrides = {}, &block)
+            factory = FactoryGirl.factory_by_name(name.underscore)
+            instance = factory.run(Proxy::Build, overrides)
+            yield(instance) if block_given?
+            instance
+          end
+
+        end
+
+      end
+    end
+  end
+end
+
+ActiveRecord::Base.send(:include, FactoryGirl::Syntax::Generate::ActiveRecord)

data/lib/factory_girl/syntax/make.rb

@@ -0,0 +1,41 @@
+module FactoryGirl
+  module Syntax
+
+    # Extends ActiveRecord::Base to provide a make class method, which is a
+    # shortcut for Factory.create.
+    #
+    # Usage:
+    #
+    #   require 'factory_girl/syntax/make'
+    #
+    #   FactoryGirl.define do
+    #     factory :user do
+    #       name 'Billy Bob'
+    #       email 'billy@bob.example.com'
+    #     end
+    #   end
+    #
+    #   User.make(:name => 'Johnny')
+    #
+    # This syntax was derived from Pete Yandell's machinist.
+    module Make
+      module ActiveRecord #:nodoc:
+
+        def self.included(base) # :nodoc:
+          base.extend ClassMethods
+        end
+
+        module ClassMethods #:nodoc:
+
+          def make(overrides = {})
+            FactoryGirl.factory_by_name(name.underscore).run(Proxy::Create, overrides)
+          end
+
+        end
+
+      end
+    end
+  end
+end
+
+ActiveRecord::Base.send(:include, FactoryGirl::Syntax::Make::ActiveRecord)

data/lib/factory_girl/syntax/sham.rb

@@ -0,0 +1,45 @@
+module FactoryGirl
+  module Syntax
+
+    # Adds a Sham module, which provides an alternate interface to
+    # FactoryGirl::Sequence.
+    #
+    # Usage:
+    #
+    #   require 'factory_girl/syntax/sham'
+    #
+    #   Sham.email {|n| "somebody#{n}@example.com" }
+    #
+    #   FactoryGirl.define do
+    #     factory :user do
+    #       email
+    #     end
+    #   end
+    #
+    # Note that you can also use Faker, but it is recommended that you simply
+    # use a sequence as in the above example, as factory_girl does not provide
+    # protection against duplication in randomized sequences, and a randomized
+    # value does not provide any tangible benefits over an ascending sequence.
+    #
+    # This syntax was derived from Pete Yandell's machinist.
+    module Sham
+      module Sham #:nodoc:
+        def self.method_missing(name, *args, &block)
+          if block_given?
+            start_value = args.first
+            FactoryGirl.sequences[name] = Sequence.new(start_value || 1, &block)
+          else
+            FactoryGirl.sequences[name].next
+          end
+        end
+
+        # overrides name on Module
+        def self.name(&block)
+          method_missing('name', &block)
+        end
+      end
+    end
+  end
+end
+
+include FactoryGirl::Syntax::Sham

data/spec/factory_girl/aliases_spec.rb

@@ -0,0 +1,33 @@
+require 'spec_helper'
+
+describe Factory, "aliases" do
+
+  it "should include an attribute as an alias for itself by default" do
+    FactoryGirl.aliases_for(:test).should include(:test)
+  end
+
+  it "should include the root of a foreign key as an alias by default" do
+    FactoryGirl.aliases_for(:test_id).should include(:test)
+  end
+
+  it "should include an attribute's foreign key as an alias by default" do
+    FactoryGirl.aliases_for(:test).should include(:test_id)
+  end
+
+  it "should NOT include an attribute as an alias when it starts with underscore" do
+    FactoryGirl.aliases_for(:_id).should_not include(:id)
+  end
+
+  describe "after adding an alias" do
+
+    before do
+      Factory.alias(/(.*)_suffix/, '\1')
+    end
+
+    it "should return the alias in the aliases list" do
+      FactoryGirl.aliases_for(:test_suffix).should include(:test)
+    end
+
+  end
+
+end

data/spec/factory_girl/attribute/association_spec.rb

@@ -0,0 +1,29 @@
+require 'spec_helper'
+
+describe FactoryGirl::Attribute::Association do
+  before do
+    @name      = :author
+    @factory   = :user
+    @overrides = { :first_name => 'John' }
+    @attr      = FactoryGirl::Attribute::Association.new(@name, @factory, @overrides)
+  end
+
+  it "should have a name" do
+    @attr.name.should == @name
+  end
+
+  it "should have a factory" do
+    @attr.factory.should == @factory
+  end
+
+  it "should tell the proxy to associate when being added to a proxy" do
+    proxy = "proxy"
+    stub(proxy).associate
+    @attr.add_to(proxy)
+    proxy.should have_received.associate(@name, @factory, @overrides)
+  end
+
+  it "should convert names to symbols" do
+    FactoryGirl::Attribute::Association.new('name', :user, {}).name.should == :name
+  end
+end