notahat-machinist 0.2.1 → 0.2.2

data/lib/machinist/active_record.rb

@@ -0,0 +1,107 @@
+module Machinist
+  
+  module ActiveRecord
+    
+    # This method takes care of converting any associated objects,
+    # in the hash returned by Lathe#assigned_attributed, into their
+    # object ids.
+    #
+    # For example, let's say we have blueprints like this:
+    #
+    #   Post.blueprint { }
+    #   Comment.blueprint { post }
+    #
+    # Lathe#assigned_attributes will return { :post => ... }, but
+    # we want to pass { :post_id => 1 } to a controller.
+    #
+    # This method takes care of cleaning this up.
+    def self.assigned_attributes_without_associations(lathe)
+      attributes = {}
+      lathe.assigned_attributes.each_pair do |attribute, value|
+        association = lathe.object.class.reflect_on_association(attribute)
+        if association && association.macro == :belongs_to
+          attributes[association.primary_key_name.to_sym] = value.id
+        else
+          attributes[attribute] = value
+        end
+      end
+      attributes
+    end
+    
+    # This sets a flag that stops make from saving objects, so
+    # that calls to make from within a blueprint don't create
+    # anything inside make_unsaved.
+    def self.with_save_nerfed
+      begin
+        @@nerfed = true
+        yield
+      ensure
+        @@nerfed = false
+      end
+    end
+
+    @@nerfed = false
+    def self.nerfed?
+      @@nerfed
+    end
+    
+    module Extensions
+      def self.included(base)
+        base.extend(ClassMethods)
+      end
+    
+      module ClassMethods
+        def blueprint(&blueprint)
+          @blueprint = blueprint if block_given?
+          @blueprint
+        end
+  
+        def make(attributes = {}, &block)
+          lathe = Lathe.run(self.new, attributes)
+          unless Machinist::ActiveRecord.nerfed?
+            lathe.object.save!
+            lathe.object.reload
+          end
+          lathe.object(&block)
+        end
+
+        def make_unsaved(attributes = {})
+          returning(Machinist::ActiveRecord.with_save_nerfed { make(attributes) }) do |object|
+            yield object if block_given?
+          end
+        end
+          
+        def plan(attributes = {})
+          lathe = Lathe.run(self.new, attributes)
+          Machinist::ActiveRecord.assigned_attributes_without_associations(lathe)
+        end
+      end
+    end
+  
+    module BelongsToExtensions
+      def make(attributes = {}, &block)
+        lathe = Lathe.run(self.build, attributes)
+        unless Machinist::ActiveRecord.nerfed?
+          lathe.object.save!
+          lathe.object.reload
+        end
+        lathe.object(&block)
+      end
+
+      def plan(attributes = {})
+        lathe = Lathe.run(self.build, attributes)
+        Machinist::ActiveRecord.assigned_attributes_without_associations(lathe)
+      end
+    end
+  
+  end
+end
+
+
+class ActiveRecord::Base
+  include Machinist::ActiveRecord::Extensions
+end
+
+class ActiveRecord::Associations::BelongsToAssociation
+  include Machinist::ActiveRecord::BelongsToExtensions
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: notahat-machinist
 version: !ruby/object:Gem::Version 
-  version: 0.2.1
+  version: 0.2.2
 platform: ruby
 authors: 
 - Pete Yandell
@@ -24,6 +24,7 @@ extra_rdoc_files: []
 files: 
 - lib/machinist.rb
 - lib/sham.rb
+- lib/machinist/active_record.rb
 has_rdoc: false
 homepage: http://github.com/notahat/machinist
 post_install_message: