fabrication 0.6.2 → 0.6.3

data/README.markdown

@@ -34,6 +34,7 @@ Define your fabricators.
       name "Fun Factory"
       employees(:count => 20) { |company, i| Fabricate(:drone, :company => company, :name => "Drone #{i}") }
       location! { Fabricate(:location) }
+      after_build { |company| company.name = "Another #{company.name}" }
       after_create { |company| company.update_attribute(:ceo, Fabricate(:drone, :name => 'Lead Drone') }
     end
 
@@ -42,6 +43,7 @@ Breaking down the above, we are defining a "company" fabricator, which will gene
 * The object has a name field, which is statically filled with "Fun Factory".
 * It has a has_many association to employees and will generate an array of 20 records as indicated by the :count => 20. The block is passed the company object being fabricated and index of the array being created.
 * It has a belongs_to association to location and this will be generated immediately with the company object. This is because of the "!" after the association name.
+* After the object is built but before it is saved, it will update the name to "Another Fun Factory".
 * After the object is created, it will update the "ceo" association with a new "drone" record.
 
 Alternatively, you can Fabricate(:company) without first defining the Fabricator. Doing so will create an empty Fabricator called ":company" and prevent you from defining the Fabricator explicitly later.
@@ -118,3 +120,11 @@ To run rake successfully:
 3. Install bundler
 4. Run `bundle install` from the project root
 5. Run `rake` and the test suite should be all green!
+
+### Contributors ###
+
+* Les Hill (leshill)
+* Jim Remsik (bigtiger)
+* Dave Ott (daveott)
+* Matt (winescout)
+* Lar Van Der Jagt (supaspoida)

data/lib/fabrication/fabricator.rb

@@ -46,3 +46,10 @@ class Fabrication::Fabricator
   end
 
 end
+
+# Adds support for reload! in the Rails 2.3.x console
+if defined? ActionController and ActionController::Dispatcher.respond_to? :reload_application
+  ActionController::Dispatcher.to_prepare :fabrication do
+    Fabrication.clear_definitions
+  end
+end

data/lib/fabrication/generator/base.rb

@@ -2,14 +2,23 @@ class Fabrication::Generator::Base
 
   def self.supports?(klass); true end
 
+  def after_build(&block)
+    callbacks[:after_build] = block if block_given?
+  end
+
   def after_create(&block)
-    self.after_create_block = block if block_given?
+    callbacks[:after_create] = block if block_given?
+  end
+
+  def callbacks
+    @callbacks ||= {}
   end
 
   def generate(options={:save => true}, attributes=[])
     process_attributes(attributes)
+    callbacks[:after_build].call(instance) if callbacks.include?(:after_build)
     after_generation(options)
-    after_create_block.call(instance) if after_create_block
+    callbacks[:after_create].call(instance) if callbacks.include?(:after_create)
     instance
   end
 
@@ -23,11 +32,9 @@ class Fabrication::Generator::Base
 
   protected
 
-  def after_generation(options); end
-
-  private
+  attr_accessor :instance
 
-  attr_accessor :after_create_block, :instance
+  def after_generation(options); end
 
   def assign(method_name, param)
     value = nil
@@ -43,8 +50,9 @@ class Fabrication::Generator::Base
 
   def process_attributes(attributes)
     attributes.each do |attribute|
-      if attribute.name == :after_create
-        after_create(&attribute.value)
+      case attribute.name
+      when :after_create; after_create(&attribute.value)
+      when :after_build; after_build(&attribute.value)
       else
         if Proc === attribute.value
           method_missing(attribute.name, attribute.params, &attribute.value)

data/lib/fabrication/support.rb

@@ -26,10 +26,10 @@ class Fabrication::Support
       ['test', 'spec'].map do |folder|
         path = File.expand_path(File.join(folder, 'fabricators'))
 
-        require("#{path}.rb") if File.exists?("#{path}.rb")
+        load("#{path}.rb") if File.exists?("#{path}.rb")
 
         File.directory? path and Dir[File.join(path, '*.rb')].each do |file|
-          require file
+          load file
         end
       end
     end

data/lib/fabrication/version.rb

@@ -1,3 +1,3 @@
 module Fabrication
-  VERSION = '0.6.2'
+  VERSION = '0.6.3'
 end

data/spec/fabrication_spec.rb

@@ -102,10 +102,12 @@ describe Fabrication do
 
       Fabricator(:other_company, :from => :company) do
         divisions(:count => 2) { Fabricate(:division) }
+        after_build { |o| o.name = "Crazysauce" }
       end
     end
 
     let(:company) { Fabricate(:company, :name => "Hashrocket") }
+    let(:other_company) { Fabricate(:other_company) }
 
     it 'generates field blocks immediately' do
       company.name.should == "Hashrocket"
@@ -115,6 +117,10 @@ describe Fabrication do
       Division.find_all_by_company_id(company.id).count.should == 4
     end
 
+    it 'executes after build blocks' do
+      other_company.name.should == 'Crazysauce'
+    end
+
     it 'executes after create blocks' do
       company.city.should == 'Jacksonville Beach'
     end
@@ -124,7 +130,7 @@ describe Fabrication do
     end
 
     it 'overrides inherited associations' do
-      Fabricate(:other_company).divisions.count.should == 2
+      other_company.divisions.count.should == 2
       Division.count.should == 2
     end
 

data/spec/fabricators.rb

@@ -0,0 +1 @@
+Fabricator(:awesome_object, :from => :object)

data/spec/fabricators/cool_object_fabricator.rb

@@ -0,0 +1 @@
+Fabricator(:cool_object, :from => :object)

data/spec/generator/base_spec.rb

@@ -9,11 +9,19 @@ describe Fabrication::Generator::Base do
     end
   end
 
+  describe "#after_build" do
+    subject { Fabrication::Generator::Base.new(Object) }
+    before { subject.after_build { "something" } }
+    it "stores the after create block" do
+      Proc.should === subject.callbacks[:after_build]
+    end
+  end
+
   describe "#after_create" do
     subject { Fabrication::Generator::Base.new(Object) }
     before { subject.after_create { "something" } }
     it "stores the after create block" do
-      Proc.should === subject.send(:after_create_block)
+      Proc.should === subject.callbacks[:after_create]
     end
   end
 

data/spec/support_spec.rb

@@ -36,17 +36,7 @@ describe Fabrication::Support do
 
   describe ".find_definitions" do
 
-    before(:all) do
-      File.open("spec/fabricators.rb", "w") do |f|
-        f.write("Fabricator(:awesome_object, :from => :object)")
-      end
-      File.open("spec/fabricators/cool_object.rb", "w") do |f|
-        f.write("Fabricator(:cool_object, :from => :object)")
-      end
-      Fabrication::Support.find_definitions
-      File.delete("spec/fabricators.rb")
-      File.delete("spec/fabricators/cool_object.rb")
-    end
+    before(:all) { Fabrication::Support.find_definitions }
 
     it "has an awesome object" do
       Fabrication::Fabricator.schematics[:awesome_object].should be

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: fabrication
 version: !ruby/object:Gem::Version 
-  hash: 3
+  hash: 1
   prerelease: false
   segments: 
   - 0
   - 6
-  - 2
-  version: 0.6.2
+  - 3
+  version: 0.6.3
 platform: ruby
 authors: 
 - Paul Elliott
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-08-16 00:00:00 -04:00
+date: 2010-08-18 00:00:00 -04:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -136,6 +136,8 @@ files:
 - lib/fabrication.rb
 - spec/fabrication_spec.rb
 - spec/fabricator_spec.rb
+- spec/fabricators/cool_object_fabricator.rb
+- spec/fabricators.rb
 - spec/generator/active_record_spec.rb
 - spec/generator/base_spec.rb
 - spec/generator/mongoid_spec.rb