fabrication 0.0.2 → 0.0.3

data/lib/fabrication.rb

@@ -4,6 +4,7 @@ module Fabrication
 
   module Generator
     autoload :ActiveRecord, 'fabrication/generator/active_record'
+    autoload :Mongoid,      'fabrication/generator/mongoid'
     autoload :Base,         'fabrication/generator/base'
   end
 

data/lib/fabrication/fabricator.rb

@@ -1,18 +1,26 @@
 class Fabrication::Fabricator
 
+  GENERATORS = [
+    Fabrication::Generator::ActiveRecord,
+    Fabrication::Generator::Mongoid,
+    Fabrication::Generator::Base
+  ]
+
+  attr_accessor :fabricator
+
   def initialize(class_name, &block)
     klass = class_for(class_name)
-    if klass.ancestors.include?(ActiveRecord::Base)
-      @fabricator = Fabrication::Generator::ActiveRecord.new(klass, &block)
-    else
-      @fabricator = Fabrication::Generator::Base.new(klass, &block)
-    end
+    self.fabricator = GENERATORS.detect do |generator|
+      generator.supports?(klass)
+    end.new(klass, &block)
   end
 
   def fabricate(options={})
-    @fabricator.generate(options)
+    fabricator.generate(options)
   end
 
+  protected
+
   #Stolen directly from factory_girl. Thanks thoughtbot!
   def class_for(class_or_to_s)
     if class_or_to_s.respond_to?(:to_sym)

data/lib/fabrication/generator/active_record.rb

@@ -7,6 +7,10 @@ class Fabrication::Generator::ActiveRecord < Fabrication::Generator::Base
     @instance
   end
 
+  def self.supports?(klass)
+    defined?(ActiveRecord) && klass.ancestors.include?(ActiveRecord::Base)
+  end
+
   def method_missing(method_name, *args, &block)
     method_name = method_name.to_s
     unless @options.include?(method_name.to_sym)

data/lib/fabrication/generator/base.rb

@@ -16,6 +16,8 @@ class Fabrication::Generator::Base
     assign(@instance, method_name.to_s, block_given? ? yield : args.first)
   end
 
+  def self.supports?(klass); true end
+
   private
 
   def assign(instance, method_name, value)

data/lib/fabrication/generator/mongoid.rb

@@ -0,0 +1,14 @@
+class Fabrication::Generator::Mongoid < Fabrication::Generator::Base
+
+  def generate(options)
+    @options = options
+    @instance = super
+    @instance.save
+    @instance
+  end
+
+  def self.supports?(klass)
+    defined?(Mongoid) && klass.ancestors.include?(Mongoid::Document)
+  end
+
+end

data/lib/fabrication/version.rb

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

data/spec/fabricator_spec.rb

@@ -38,4 +38,18 @@ describe Fabrication::Fabricator do
 
   end
 
+  context 'with a mongoid document' do
+
+    let(:fabricator) { Fabrication::Fabricator.new(:author) { name "Seth Godin" } }
+
+    it 'fabricates a Author instance' do
+      fabricator.fabricate.instance_of?(Author).should be_true
+    end
+
+    it 'uses the activerecord generator' do
+      fabricator.instance_variable_get(:@fabricator).instance_of?(Fabrication::Generator::Mongoid).should be_true
+    end
+
+  end
+
 end

data/spec/generator/mongoid_spec.rb

@@ -0,0 +1,24 @@
+require 'spec_helper'
+
+describe Fabrication::Generator::Mongoid do
+
+  before(:all) { TestMigration.up }
+  after(:all) { TestMigration.down }
+
+  context 'mongoid object' do
+
+    let(:company) do
+      Fabrication::Generator::Mongoid.new(Author) do
+        name 'Name'
+      end.generate({:name => 'Something'})
+    end
+
+    before { company }
+
+    it 'persists the author upon creation' do
+      Author.where(:name => 'Something').first.should be
+    end
+
+  end
+
+end

data/spec/spec_helper.rb

@@ -7,3 +7,10 @@ require 'spec/autorun'
 require 'fabrication'
 require 'ffaker'
 require 'active_record'
+require 'mongoid'
+
+Spec::Runner.configure do |config|
+  config.before(:all) do
+    Author.delete_all
+  end
+end

data/spec/support/mongoid_setup.rb

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

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: fabrication
 version: !ruby/object:Gem::Version 
-  hash: 27
+  hash: 25
   prerelease: false
   segments: 
   - 0
   - 0
-  - 2
-  version: 0.0.2
+  - 3
+  version: 0.0.3
 platform: ruby
 authors: 
 - Paul Elliott
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-06-16 00:00:00 -04:00
+date: 2010-06-17 00:00:00 -04:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -82,6 +82,22 @@ dependencies:
         version: 1.3.0
   type: :development
   version_requirements: *id004
+- !ruby/object:Gem::Dependency 
+  name: mongoid
+  prerelease: false
+  requirement: &id005 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 49
+        segments: 
+        - 1
+        - 9
+        - 1
+        version: 1.9.1
+  type: :development
+  version_requirements: *id005
 description: Fabrication is an object generation framework that lazily generated attributes as they are used
 email: 
 - paul@hashrocket.com
@@ -95,6 +111,7 @@ files:
 - lib/fabrication/fabricator.rb
 - lib/fabrication/generator/active_record.rb
 - lib/fabrication/generator/base.rb
+- lib/fabrication/generator/mongoid.rb
 - lib/fabrication/version.rb
 - lib/fabrication.rb
 - spec/block_generated_fields_spec.rb
@@ -102,12 +119,14 @@ files:
 - 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/database_setup.rb
+- spec/support/active_record_setup.rb
 - spec/support/helper_objects.rb
+- spec/support/mongoid_setup.rb
 - README.markdown
 has_rdoc: true
 homepage: http://github.com/paulelliott/fabrication