fabrication 1.0.1 → 1.1.0

data/lib/fabrication.rb

@@ -53,7 +53,7 @@ class Fabricate
     }, options, &block)
   end
 
-  def self.sequence(name, start=0, &block)
+  def self.sequence(name=Fabrication::Sequencer::DEFAULT, start=0, &block)
     Fabrication::Sequencer.sequence(name, start, &block)
   end
 

data/lib/fabrication/generator/mongoid.rb

@@ -0,0 +1,21 @@
+class Fabrication::Generator::Mongoid < Fabrication::Generator::Base
+  def self.supports?(klass)
+    defined?(Mongoid) && klass.ancestors.include?(Mongoid::Document)
+  end
+
+  def assign(method_name, options, raw_value=nil)
+    if options.has_key?(:count)
+      value = (1..options[:count]).map do |i|
+        block_given? ? yield(instance, i) : raw_value
+      end
+    else
+      value = block_given? ? yield(instance) : raw_value
+    end
+
+    if instance.respond_to?("#{method_name}=")
+      instance.send("#{method_name}=", value)
+    else
+      instance[method_name] = value
+    end
+  end
+end

data/lib/fabrication/schematic.rb

@@ -3,6 +3,7 @@ class Fabrication::Schematic
   GENERATORS = [
     Fabrication::Generator::ActiveRecord,
     Fabrication::Generator::Sequel,
+    Fabrication::Generator::Mongoid,
     Fabrication::Generator::Base
   ]
 
@@ -104,7 +105,7 @@ class Fabrication::Schematic
     method_name
   end
 
-  def sequence(name, start=0, &block)
+  def sequence(name=Fabrication::Sequencer::DEFAULT, start=0, &block)
     name = "#{self.klass.to_s.downcase.gsub(/::/, '_')}_#{name}"
     Fabrication::Sequencer.sequence(name, start, &block)
   end

data/lib/fabrication/sequencer.rb

@@ -1,19 +1,17 @@
 class Fabrication::Sequencer
 
-  class << self
+  DEFAULT = :_default
 
-    def sequence(name, start=0)
-      idx = sequences[name] ||= start
+  def self.sequence(name=DEFAULT, start=0)
+    idx = sequences[name] ||= start
 
-      (block_given? ? yield(idx) : idx).tap do
-        sequences[name] += 1
-      end
-    end
-
-    def sequences
-      @sequences ||= {}
+    (block_given? ? yield(idx) : idx).tap do
+      sequences[name] += 1
     end
+  end
 
+  def self.sequences
+    @sequences ||= {}
   end
 
 end

data/lib/fabrication/version.rb

@@ -1,3 +1,3 @@
 module Fabrication
-  VERSION = '1.0.1'
+  VERSION = '1.1.0'
 end

data/spec/fabrication/schematic_spec.rb

@@ -18,7 +18,7 @@ describe Fabrication::Schematic do
     end
     context "for a mongoid object" do
       it "uses the base generator" do
-        Fabrication::Schematic.new(Author).generator.should == Fabrication::Generator::Base
+        Fabrication::Schematic.new(Author).generator.should == Fabrication::Generator::Mongoid
       end
     end
     context "for a sequel object" do

data/spec/fabrication/{sequence_spec.rb → sequencer_spec.rb}

@@ -1,6 +1,15 @@
 require 'spec_helper'
 
-describe "Fabricate.sequence" do
+describe Fabrication::Sequencer do
+
+  context 'with no arguments' do
+    subject { Fabrication::Sequencer.sequence }
+
+    it { should == 0 }
+    it 'creates a default sequencer' do
+      Fabrication::Sequencer.sequences[:_default].should == 1
+    end
+  end
 
   context 'with only a name' do
 

data/spec/fabrication_spec.rb

@@ -10,6 +10,10 @@ shared_examples 'something fabricatable' do
     its(:string_field) { should == 'content' }
   end
 
+  context 'model callbacks are fired' do
+    its(:before_save_value) { should == 11 }
+  end
+
   context 'overriding at fabricate time' do
     subject do
       Fabricate(
@@ -206,6 +210,14 @@ describe Fabrication do
     it 'generates four books' do
       author.books.map(&:title).should == (1..4).map { |i| "book title #{i}" }
     end
+
+    it "sets dynamic fields" do
+      Fabricate(:special_author).mongoid_dynamic_field.should == 50
+    end
+
+    it "sets lazy dynamic fields" do
+      Fabricate(:special_author).lazy_dynamic_field.should == "foo"
+    end
   end
 
   context 'with multiple callbacks' do

data/spec/fabricators/mongoid_fabricator.rb

@@ -18,6 +18,11 @@ Fabricator(:author) do
   end
 end
 
+Fabricator(:special_author, :from => :author) do
+  mongoid_dynamic_field 50
+  lazy_dynamic_field { "foo" }
+end
+
 Fabricator(:hemingway, :from => :author) do
   name 'Ernest Hemingway'
 end

data/spec/fabricators/plain_old_ruby_object_fabricator.rb

@@ -49,13 +49,3 @@ end
 Fabricator("Something::Amazing") do
   stuff "cool"
 end
-
-Fabricator(:sequencer) do
-  simple_iterator { sequence(:simple_iterator) }
-  param_iterator  { sequence(:param_iterator, 9) }
-  block_iterator  { sequence(:block_iterator) { |i| "block#{i}" } }
-end
-
-Fabricator("Sequencer::Namespaced") do
-  iterator { sequence(:iterator) }
-end

data/spec/fabricators/sequencer_fabricator.rb

@@ -0,0 +1,9 @@
+Fabricator(:sequencer) do
+  simple_iterator { sequence }
+  param_iterator  { sequence(:param_iterator, 9) }
+  block_iterator  { sequence(:block_iterator) { |i| "block#{i}" } }
+end
+
+Fabricator("Sequencer::Namespaced") do
+  iterator { sequence(:iterator) }
+end

data/spec/support/active_record.rb

@@ -16,6 +16,7 @@ class TestMigration < ActiveRecord::Migration
     end
 
     create_table :parent_active_record_models, :force => true do |t|
+      t.column :before_save_value, :integer
       t.column :dynamic_field, :string
       t.column :nil_field, :string
       t.column :number_field, :integer
@@ -47,6 +48,10 @@ end
 
 class ParentActiveRecordModel < ActiveRecord::Base
   has_many :collection_field, :class_name => 'ChildActiveRecordModel'
+
+  before_save do
+    self.before_save_value = 11
+  end
 end
 
 class Company < ActiveRecord::Base

data/spec/support/mongoid.rb

@@ -20,12 +20,17 @@ end
 class ParentMongoidDocument
   include Mongoid::Document
 
+  field :before_save_value
   field :dynamic_field
   field :nil_field
   field :number_field
   field :string_field
 
-  references_many :collection_field, :class_name => 'ChildMongoidDocument'
+  references_many :collection_field, :class_name => 'ChildMongoidDocument', :inverse_of => :parent
+
+  before_save do
+    self.before_save_value = 11
+  end
 end
 
 class Author

data/spec/support/plain_old_ruby_objects.rb

@@ -1,11 +1,18 @@
+require 'ostruct'
+
 class ParentRubyObject
   attr_accessor \
+    :before_save_value,
     :collection_field,
     :dynamic_field,
     :nil_field,
     :number_field,
     :string_field
 
+  def initialize
+    self.before_save_value = 11
+  end
+
   def persisted?; true end
 end
 

data/spec/support/sequel.rb

@@ -14,4 +14,9 @@ class ParentSequelModel < Sequel::Model
   one_to_many :collection_field, :class => :ChildSequelModel, :key => :parent_sequel_model_id
 
   def persisted?; !new? end
+
+  def before_save
+    self.before_save_value = 11
+    super
+  end
 end

data/spec/support/sequel_migrations/001_create_tables.rb

@@ -8,6 +8,7 @@ Sequel.migration do
 
     create_table :parent_sequel_models do
       primary_key :id
+      Integer :before_save_value
       String :dynamic_field
       String :nil_field
       Integer :number_field

metadata

@@ -2,7 +2,7 @@
 name: fabrication
 version: !ruby/object:Gem::Version 
   prerelease: 
-  version: 1.0.1
+  version: 1.1.0
 platform: ruby
 authors: 
 - Paul Elliott
@@ -10,7 +10,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2011-07-02 00:00:00 Z
+date: 2011-08-26 00:00:00 Z
 dependencies: 
 - !ruby/object:Gem::Dependency 
   name: activerecord
@@ -86,7 +86,7 @@ dependencies:
     requirements: 
     - - ~>
       - !ruby/object:Gem::Version 
-        version: 2.0.2
+        version: 2.1.3
   type: :development
   version_requirements: *id007
 - !ruby/object:Gem::Dependency 
@@ -151,6 +151,7 @@ files:
 - lib/fabrication/fabricator.rb
 - lib/fabrication/generator/active_record.rb
 - lib/fabrication/generator/base.rb
+- lib/fabrication/generator/mongoid.rb
 - lib/fabrication/generator/sequel.rb
 - lib/fabrication/schematic.rb
 - lib/fabrication/sequencer.rb
@@ -167,7 +168,7 @@ files:
 - spec/fabrication/generator/active_record_spec.rb
 - spec/fabrication/generator/base_spec.rb
 - spec/fabrication/schematic_spec.rb
-- spec/fabrication/sequence_spec.rb
+- spec/fabrication/sequencer_spec.rb
 - spec/fabrication/support_spec.rb
 - spec/fabrication_spec.rb
 - spec/fabricators/active_record_fabricator.rb
@@ -175,6 +176,7 @@ files:
 - spec/fabricators/mongoid_fabricator.rb
 - spec/fabricators/plain_old_ruby_object_fabricator.rb
 - spec/fabricators/sequel_fabricator.rb
+- spec/fabricators/sequencer_fabricator.rb
 - spec/spec_helper.rb
 - spec/support/active_record.rb
 - spec/support/mongoid.rb
@@ -205,7 +207,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
 requirements: []
 
 rubyforge_project: 
-rubygems_version: 1.7.2
+rubygems_version: 1.8.6
 signing_key: 
 specification_version: 3
 summary: Fabrication provides a robust solution for test object generation.