fabrication 0.7.1 → 0.8.0

data/README.markdown

@@ -34,6 +34,15 @@ Now you can define fabricators in any of the following locations.
 
 They are automatically loaded, so no additional requires are necessary.
 
+### Rails 3 Generators ###
+
+They are really easy to configure! Just add this to your `config/application.rb`:
+
+    config.generators do |g|
+      g.test_framework      :rspec, :fixture => true
+      g.fixture_replacement :fabrication, :dir => "spec/fabricators"
+    end
+
 ### Usage ###
 
 Define your fabricators.
@@ -70,6 +79,18 @@ You can also explicitly specify the class being fabricated with the :class_name
       type "LLC"
     end
 
+The callbacks will be stacked when inheriting from other fabricators. For example, when you define something like this:
+
+    Fabricator(:user) do
+      after_create { |user| user.confirm! }
+    end
+
+    Fabricator(:admin, :from => :user) do
+      after_create { |user| user.make_admin! }
+    end
+
+When calling `Fabricate(:admin)`, the user callback will be executed first and then the admin callback.
+
 ### Fabricating ###
 
 Now that your Fabricators are defined, you can start generating some objects! To generate the LLC from the previous example, just do this:
@@ -135,3 +156,4 @@ To run rake successfully:
 * Matt (winescout)
 * Lar Van Der Jagt (supaspoida)
 * Justin Smestad (jsmestad)
+* chanks

data/lib/fabrication/generator/base.rb

@@ -2,23 +2,12 @@ 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)
-    callbacks[:after_create] = block if block_given?
-  end
-
-  def callbacks
-    @callbacks ||= {}
-  end
-
-  def generate(options={:save => true}, attributes=[])
+  def generate(options={:save => true}, attributes=[], callbacks={})
     process_attributes(attributes)
-    callbacks[:after_build].call(instance) if callbacks.include?(:after_build)
+
+    callbacks[:after_build].each { |callback| callback.call(instance) } if callbacks[:after_build]
     after_generation(options)
-    callbacks[:after_create].call(instance) if callbacks.include?(:after_create)
+    callbacks[:after_create].each { |callback| callback.call(instance) } if callbacks[:after_create]
     instance
   end
 
@@ -49,15 +38,10 @@ class Fabrication::Generator::Base
 
   def process_attributes(attributes)
     attributes.each do |attribute|
-      case attribute.name
-      when :after_create; after_create(&attribute.value)
-      when :after_build; after_build(&attribute.value)
+      if Proc === attribute.value
+        method_missing(attribute.name, attribute.params, &attribute.value)
       else
-        if Proc === attribute.value
-          method_missing(attribute.name, attribute.params, &attribute.value)
-        else
-          method_missing(attribute.name, attribute.value)
-        end
+        method_missing(attribute.name, attribute.value)
       end
     end
   end

data/lib/fabrication/schematic.rb

@@ -13,6 +13,16 @@ class Fabrication::Schematic
     instance_eval(&block) if block_given?
   end
 
+  def after_build(&block)
+    callbacks[:after_build] ||= []
+    callbacks[:after_build] << block
+  end
+
+  def after_create(&block)
+    callbacks[:after_create] ||= []
+    callbacks[:after_create] << block
+  end
+
   def attribute(name)
     attributes.select { |a| a.name == name }.first
   end
@@ -22,16 +32,26 @@ class Fabrication::Schematic
     @attributes ||= []
   end
 
+  attr_writer :callbacks
+  def callbacks
+    @callbacks ||= {}
+  end
+
   def generate(options={}, overrides={}, &block)
     attributes = merge(overrides, &block).attributes
     if options[:attributes]
       to_hash(attributes, overrides)
     else
-      generator.new(klass).generate(options, attributes)
+      generator.new(klass).generate(options, attributes, callbacks)
     end
   end
 
   def initialize_copy(original)
+    self.callbacks = {}
+    original.callbacks.each do |type, callbacks|
+      self.callbacks[type] = callbacks.clone
+    end
+
     self.attributes = original.attributes.map do |a|
       Fabrication::Attribute.new(a.name, a.params, a.value)
     end

data/lib/fabrication/version.rb

@@ -1,3 +1,3 @@
 module Fabrication
-  VERSION = '0.7.1'
+  VERSION = '0.8.0'
 end

data/lib/rails/generators/fabrication/model/model_generator.rb

@@ -0,0 +1,15 @@
+require 'rails/generators/fabrication_generator'
+
+module Fabrication
+  module Generators
+    class ModelGenerator < Base
+      argument :attributes, :type => :array, :default => [], :banner => "field:type field:type"
+      class_option :dir, :type => :string, :default => "test/fabricators", :desc => "The directory where the fabricators should go"
+      class_option :extension, :type => :string, :default => "rb", :desc => "file extension name"
+        
+      def create_fabrication_file
+        template 'fabricator.rb', File.join(options[:dir], "#{table_name}.#{options[:extension].to_s}")
+      end
+    end
+  end
+end

data/lib/rails/generators/fabrication/model/templates/fabricator.rb

@@ -0,0 +1,5 @@
+Fabricator :<%= singular_name %> do
+<% attributes.each do |attribute| -%>
+  <%= attribute.name %> <%= attribute.default.inspect %>
+<% end -%>
+end

data/lib/rails/generators/fabrication_generator.rb

@@ -0,0 +1,11 @@
+require 'rails/generators/named_base'
+
+module Fabrication
+  module Generators
+    class Base < Rails::Generators::NamedBase #:nodoc:
+      def self.source_root
+        @_fabrication_source_root ||= File.expand_path(File.join(File.dirname(__FILE__), 'fabrication', generator_name, 'templates'))
+      end
+    end
+  end
+end

data/spec/fabrication/generator/base_spec.rb

@@ -9,22 +9,6 @@ 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.callbacks[:after_create]
-    end
-  end
-
   describe "#generate" do
 
     let(:generator) { Fabrication::Generator::Base.new(Person) }

data/spec/fabrication_spec.rb

@@ -219,6 +219,26 @@ describe Fabrication do
     end
   end
 
+  context 'with multiple callbacks' do
+    let(:child) { Fabricate(:child) }
+
+    it "runs the first callback" do
+      child.first_name.should == "Johnny"
+    end
+
+    it "runs the second callback" do
+      child.age.should == 10
+    end
+  end
+
+  context 'with multiple, inherited callbacks' do
+    let(:senior) { Fabricate(:senior) }
+
+    it "runs the parent callbacks first" do
+      senior.age.should == 70
+    end
+  end
+
   context 'with a parent fabricator' do
 
     context 'and a previously defined parent' do

data/spec/fabricators.rb

@@ -20,6 +20,15 @@ Fabricator(:person) do
   location
 end
 
+Fabricator(:child, :from => :person) do
+  after_build { |child| child.first_name = 'Johnny' }
+  after_build { |child| child.age = 10 }
+end
+
+Fabricator(:senior, :from => :child) do
+  after_build { |senior| senior.age *= 7 }
+end
+
 # ActiveRecord Objects
 Fabricator(:division) do
   name "Division Name"

metadata

@@ -4,9 +4,9 @@ version: !ruby/object:Gem::Version
   prerelease: false
   segments: 
   - 0
-  - 7
-  - 1
-  version: 0.7.1
+  - 8
+  - 0
+  version: 0.8.0
 platform: ruby
 authors: 
 - Paul Elliott
@@ -14,7 +14,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-09-12 00:00:00 -04:00
+date: 2010-09-13 00:00:00 -04:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -132,6 +132,9 @@ files:
 - lib/fabrication/support.rb
 - lib/fabrication/version.rb
 - lib/fabrication.rb
+- lib/rails/generators/fabrication/model/model_generator.rb
+- lib/rails/generators/fabrication/model/templates/fabricator.rb
+- lib/rails/generators/fabrication_generator.rb
 - spec/fabrication/attribute_spec.rb
 - spec/fabrication/fabricator_spec.rb
 - spec/fabrication/generator/active_record_spec.rb