fabrication 0.0.3 → 0.1.0

data/README.markdown

@@ -0,0 +1,3 @@
+### Fabrication ###
+
+

data/lib/fabrication.rb

@@ -15,9 +15,28 @@ module Fabrication
     end
 
     def generate(name, options)
+      find_definitions if fabricators.empty?
       fabricators[name].fabricate(options)
     end
 
+    def find_definitions
+      fabricator_file_paths = [
+        File.join('test', 'fabricators'),
+        File.join('spec', 'fabricators')
+      ]
+      fabricator_file_paths.each do |path|
+        if File.exists?("#{path}.rb")
+          require("#{path}.rb") 
+        end
+
+        if File.directory? path
+          Dir[File.join(path, '*.rb')].each do |file|
+            require file
+          end
+        end
+      end
+    end
+
     private
 
     def fabricators

data/lib/fabrication/generator/active_record.rb

@@ -1,10 +1,10 @@
 class Fabrication::Generator::ActiveRecord < Fabrication::Generator::Base
 
+  attr_accessor :instance, :options
+
   def generate(options)
-    @options = options
-    @instance = super
-    @instance.save
-    @instance
+    self.options = options
+    self.instance = super.tap { |t| t.save }
   end
 
   def self.supports?(klass)
@@ -13,21 +13,26 @@ class Fabrication::Generator::ActiveRecord < Fabrication::Generator::Base
 
   def method_missing(method_name, *args, &block)
     method_name = method_name.to_s
-    unless @options.include?(method_name.to_sym)
+    count = (args && args.first && args.first[:count]) || 1
+    unless options.include?(method_name.to_sym)
       if block_given?
-        unless args.include?(:force) || @instance.class.columns.map(&:name).include?(method_name)
+        unless (args.first && args.first[:force]) || instance.class.columns.map(&:name).include?(method_name)
           # copy the original getter
-          @instance.instance_variable_set("@__#{method_name}_original", @instance.method(method_name).clone)
+          instance.instance_variable_set("@__#{method_name}_original", instance.method(method_name).clone)
 
           # store the block for lazy generation
-          @instance.instance_variable_set("@__#{method_name}_block", block)
+          instance.instance_variable_set("@__#{method_name}_block", block)
 
           # redefine the getter
-          @instance.instance_eval %<
+          instance.instance_eval %<
             def #{method_name}
               original_value = @__#{method_name}_original.call
               if @__#{method_name}_block.present?
-                original_value = #{method_name}= @__#{method_name}_block.call(self)
+                if #{count} \> 1
+                  original_value = #{method_name}= (1..#{count}).map { |i| @__#{method_name}_block.call(self, i) }
+                else
+                  original_value = #{method_name}= @__#{method_name}_block.call(self)
+                end
                 @__#{method_name}_block = nil
                 @__#{method_name}_original.call
               end
@@ -35,10 +40,10 @@ class Fabrication::Generator::ActiveRecord < Fabrication::Generator::Base
             end
           >
         else
-          assign(@instance, method_name, yield)
+          assign(instance, method_name, args.first, &block)
         end
       else
-        assign(@instance, method_name, args.first)
+        assign(instance, method_name, args)
       end
     end
   end

data/lib/fabrication/generator/base.rb

@@ -1,26 +1,36 @@
 class Fabrication::Generator::Base
 
+  attr_accessor :klass, :block, :instance
+
   def initialize(klass, &block)
-    @klass = klass
-    @block = block
+    self.klass = klass
+    self.block = block
   end
 
   def generate(options)
-    @instance = @klass.new
-    instance_eval &@block
-    options.each { |k,v| assign(@instance, k, v) }
-    @instance
+    self.instance = klass.new
+    instance_eval &block
+    options.each { |k,v| assign(instance, k, v) }
+    instance
   end
 
   def method_missing(method_name, *args, &block)
-    assign(@instance, method_name.to_s, block_given? ? yield : args.first)
+    assign(instance, method_name.to_s, args.first, &block)
   end
 
   def self.supports?(klass); true end
 
   private
 
-  def assign(instance, method_name, value)
+  def assign(instance, method_name, param)
+    value = nil
+    if param.is_a?(Hash) && param[:count] && param[:count] > 1
+      value = (1..param[:count]).map do |i|
+        block_given? ? yield(i) : param
+      end
+    else
+      value = block_given? ? yield : param
+    end
     instance.send("#{method_name.to_s}=", value)
   end
 

data/lib/fabrication/generator/mongoid.rb

@@ -1,10 +1,9 @@
 class Fabrication::Generator::Mongoid < Fabrication::Generator::Base
 
+  attr_accessor :instance
+
   def generate(options)
-    @options = options
-    @instance = super
-    @instance.save
-    @instance
+    self.instance = super.tap { |t| t.save }
   end
 
   def self.supports?(klass)

data/lib/fabrication/version.rb

@@ -1,3 +1,3 @@
 module Fabrication
-  VERSION = '0.0.3'
+  VERSION = '0.1.0'
 end

data/spec/fabrication_spec.rb

@@ -2,6 +2,98 @@ require 'spec_helper'
 
 describe Fabrication do
 
+  context 'static fields' do
+
+    let(:person) do
+      Fabricate(:person, :last_name => 'Awesome')
+    end
+
+    before(:all) do
+      Fabricator(:person) do
+        first_name 'Joe'
+        last_name 'Schmoe'
+        age 78
+      end
+    end
+
+    it 'has the default first name' do
+      person.first_name.should == 'Joe'
+    end
+
+    it 'has an overridden last name' do
+      person.last_name.should == 'Awesome'
+    end
+
+    it 'has the default age' do
+      person.age.should == 78
+    end
+
+    it 'generates a fresh object every time' do
+      Fabricate(:person).should_not == person
+    end
+
+  end
+
+  context 'block generated fields' do
+
+    let(:person) do
+      Fabricate(:person)
+    end
+
+    before(:all) do
+      Fabricator(:person) do
+        first_name { Faker::Name.first_name }
+        last_name { Faker::Name.last_name }
+        age { rand(100) }
+        shoes(:count => 10) { |i| "shoe #{i}" }
+      end
+    end
+
+    it 'has a first name' do
+      person.first_name.should be
+    end
+
+    it 'has a last name' do
+      person.last_name.should be
+    end
+
+    it 'has an age' do
+      person.age.should be
+    end
+
+    it 'has 10 shoes' do
+      person.shoes.should == (1..10).map { |i| "shoe #{i}" }
+    end
+
+  end
+
+  context 'multiple instance' do
+
+    let(:person1) { Fabricate(:person, :first_name => 'Jane') }
+    let(:person2) { Fabricate(:person, :first_name => 'John') }
+
+    before(:all) do
+      Fabricator(:person) do
+        first_name { Faker::Name.first_name }
+        last_name { Faker::Name.last_name }
+        age { rand(100) }
+      end
+    end
+
+    it 'person1 is named Jane' do
+      person1.first_name.should == 'Jane'
+    end
+
+    it 'person2 is named John' do
+      person2.first_name.should == 'John'
+    end
+
+    it 'they have different last names' do
+      person1.last_name.should_not == person2.last_name
+    end
+
+  end
+
   context 'with an active record object' do
 
     before(:all) { TestMigration.up }
@@ -10,7 +102,8 @@ describe Fabrication do
     before do
       Fabricator(:company) do
         name { Faker::Company.name }
-        divisions(:force) { [Fabricate(:division)] }
+        divisions(:force => true, :count => 4) { Fabricate(:division) }
+        after_create { |o| o.update_attribute(:city, "Jacksonville Beach") }
       end
 
       Fabricator(:division) do
@@ -25,12 +118,35 @@ describe Fabrication do
     end
 
     it 'generates associations immediately when forced' do
-      Division.find_all_by_company_id(company.id).count.should == 1
+      Division.find_all_by_company_id(company.id).count.should == 4
     end
 
     it 'overrides associations' do
       Fabricate(:company, :divisions => []).divisions.should == []
     end
+
+    it 'executes after create blocks' do
+      company.city.should == 'Jacksonville Beach'
+    end
+
+  end
+
+  context 'with a mongoid document' do
+
+    let(:author) do
+      Fabricator(:author) do
+        name "George Orwell"
+        books(:count => 4) { |i| "book title #{i}" }
+      end.fabricate
+    end
+
+    it "sets the author name" do
+      author.name.should == "George Orwell"
+    end
+
+    it 'generates four books' do
+      author.books.should == (1..4).map { |i| "book title #{i}" }
+    end
   end
 
 end

data/spec/support/active_record_setup.rb

@@ -11,6 +11,7 @@ class TestMigration < ActiveRecord::Migration
   def self.up
     create_table :companies, :force => true do |t|
       t.column :name, :string
+      t.column :city, :string
     end
 
     create_table :divisions, :force => true do |t|

data/spec/support/helper_objects.rb

@@ -1 +1,3 @@
-class Person; attr_accessor :age, :first_name, :last_name end
+class Person
+  attr_accessor :age, :first_name, :last_name, :shoes
+end

data/spec/support/mongoid_setup.rb

@@ -1,9 +1,10 @@
 Mongoid.configure do |config|
-  config.allow_dynamic_fields = false
+  config.allow_dynamic_fields = true
   config.master = Mongo::Connection.new.db("fabrication_test")
 end
 
 class Author
   include Mongoid::Document
   field :name
+  field :books, :type => Array
 end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: fabrication
 version: !ruby/object:Gem::Version 
-  hash: 25
+  hash: 27
   prerelease: false
   segments: 
   - 0
+  - 1
   - 0
-  - 3
-  version: 0.0.3
+  version: 0.1.0
 platform: ruby
 authors: 
 - Paul Elliott
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-06-17 00:00:00 -04:00
+date: 2010-06-19 00:00:00 -04:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -114,14 +114,11 @@ files:
 - lib/fabrication/generator/mongoid.rb
 - lib/fabrication/version.rb
 - lib/fabrication.rb
-- spec/block_generated_fields_spec.rb
 - spec/fabrication_spec.rb
 - spec/fabricator_spec.rb
 - spec/generator/active_record_spec.rb
 - spec/generator/base_spec.rb
 - spec/generator/mongoid_spec.rb
-- spec/multiple_instance_spec.rb
-- spec/simple_object_spec.rb
 - spec/spec.opts
 - spec/spec_helper.rb
 - spec/support/active_record_setup.rb

data/spec/block_generated_fields_spec.rb

@@ -1,33 +0,0 @@
-require 'spec_helper'
-
-describe Fabrication do
-
-  context 'block generated fields' do
-
-    let(:person) do
-      Fabricate(:person)
-    end
-
-    before(:all) do
-      Fabricator(:person) do
-        first_name { Faker::Name.first_name }
-        last_name { Faker::Name.last_name }
-        age { rand(100) }
-      end
-    end
-
-    it 'has a first name' do
-      person.first_name.should be
-    end
-
-    it 'has a last name' do
-      person.last_name.should be
-    end
-
-    it 'has an age' do
-      person.age.should be
-    end
-
-  end
-
-end

data/spec/multiple_instance_spec.rb

@@ -1,32 +0,0 @@
-require 'spec_helper'
-
-describe Fabrication do
-
-  context 'multiple instance' do
-
-    let(:person1) { Fabricate(:person, :first_name => 'Jane') }
-    let(:person2) { Fabricate(:person, :first_name => 'John') }
-
-    before(:all) do
-      Fabricator(:person) do
-        first_name { Faker::Name.first_name }
-        last_name { Faker::Name.last_name }
-        age { rand(100) }
-      end
-    end
-
-    it 'person1 is named Jane' do
-      person1.first_name.should == 'Jane'
-    end
-
-    it 'person2 is named John' do
-      person2.first_name.should == 'John'
-    end
-
-    it 'they have different last names' do
-      person1.last_name.should_not == person2.last_name
-    end
-
-  end
-
-end

data/spec/simple_object_spec.rb

@@ -1,37 +0,0 @@
-require 'spec_helper'
-
-describe Fabrication do
-
-  context 'static fields' do
-
-    let(:person) do
-      Fabricate(:person, :last_name => 'Awesome')
-    end
-
-    before(:all) do
-      Fabricator(:person) do
-        first_name 'Joe'
-        last_name 'Schmoe'
-        age 78
-      end
-    end
-
-    it 'has the default first name' do
-      person.first_name.should == 'Joe'
-    end
-
-    it 'has an overridden last name' do
-      person.last_name.should == 'Awesome'
-    end
-
-    it 'has the default age' do
-      person.age.should == 78
-    end
-
-    it 'generates a fresh object every time' do
-      Fabricate(:person).should_not == person
-    end
-
-  end
-
-end