dynamoid 0.3.0 → 0.3.1

data/Dynamoid.gemspec

@@ -5,7 +5,7 @@
 
 Gem::Specification.new do |s|
   s.name = "dynamoid"
-  s.version = "0.3.0"
+  s.version = "0.3.1"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Josh Symonds"]
@@ -86,6 +86,8 @@ Gem::Specification.new do |s|
     "lib/dynamoid/associations/has_and_belongs_to_many.rb",
     "lib/dynamoid/associations/has_many.rb",
     "lib/dynamoid/associations/has_one.rb",
+    "lib/dynamoid/associations/many_association.rb",
+    "lib/dynamoid/associations/single_association.rb",
     "lib/dynamoid/components.rb",
     "lib/dynamoid/config.rb",
     "lib/dynamoid/config/options.rb",

data/VERSION

@@ -1 +1 @@
-0.3.0
+0.3.1

data/lib/dynamoid/associations.rb

@@ -1,5 +1,7 @@
 # encoding: utf-8
 require 'dynamoid/associations/association'
+require 'dynamoid/associations/single_association'
+require 'dynamoid/associations/many_association'
 require 'dynamoid/associations/has_many'
 require 'dynamoid/associations/belongs_to'
 require 'dynamoid/associations/has_one'

data/lib/dynamoid/associations/association.rb

@@ -1,19 +1,15 @@
 # encoding: utf-8
 module Dynamoid #:nodoc:
 
-  # The base association module which all associations include. Every association has two very important components: the source and 
+  # The base association module which all associations include. Every association has two very important components: the source and
   # the target. The source is the object which is calling the association information. It always has the target_ids inside of an attribute on itself.
   # The target is the object which is referencing by this association.
   module Associations
     module Association
-      attr_accessor :name, :options, :source, :query
-      include Enumerable
-
-      # Delegate methods to the records the association represents.
-      delegate :first, :last, :empty?, :size, :to => :records
+      attr_accessor :name, :options, :source
 
       # Create a new association.
-      # 
+      #
       # @param [Class] source the source record of the association; that is, the record that you already have
       # @param [Symbol] name the name of the association
       # @param [Hash] options optional parameters for the association
@@ -28,158 +24,9 @@ module Dynamoid #:nodoc:
         @name = name
         @options = options
         @source = source
-        @query = {}
-      end
-      
-      # Alias convenience methods for the associations.
-      alias :nil? :empty?
-      alias :count :size
-      
-      # The records associated to the source.
-      # 
-      # @return the association records; depending on which association this is, either a single instance or an array
-      #
-      # @since 0.2.0
-      def records
-        results = Array(target_class.find(source_ids.to_a))
-
-        if query.empty?
-          results
-        else
-          results_with_query(results)
-        end
-      end
-      alias :all :records
-      
-      # Delegate include? to the records.
-      def include?(object)
-        records.include?(object)
-      end
-      
-      # @todo Improve the two methods below to not have quite so much duplicated code.
-      
-      # Deletes an object or array of objects from the association. This removes their records from the association field on the source, 
-      # and attempts to remove the source from the target association if it is detected to exist.
-      # 
-      # @param [Dynamoid::Document] object the object (or array of objects) to remove from the association
-      #
-      # @return [Dynamoid::Document] the deleted object
-      #
-      # @since 0.2.0
-      def delete(object)
-        source.update_attribute(source_attribute, source_ids - Array(object).collect(&:id))
-        Array(object).collect{|o| self.send(:disassociate_target, o)} if target_association
-        object
-      end
-
-      # Add an object or array of objects to an association. This preserves the current records in the association (if any)
-      # and adds the object to the target association if it is detected to exist.
-      # 
-      # @param [Dynamoid::Document] object the object (or array of objects) to add to the association
-      #
-      # @return [Dynamoid::Document] the added object
-      #
-      # @since 0.2.0      
-      def <<(object)
-        source.update_attribute(source_attribute, source_ids.merge(Array(object).collect(&:id)))
-        Array(object).collect{|o| self.send(:associate_target, o)} if target_association
-        object
-      end
-
-      # Replace an association with object or array of objects. This removes all of the existing associated records and replaces them with
-      # the passed object(s), and associates the target association if it is detected to exist.
-      # 
-      # @param [Dynamoid::Document] object the object (or array of objects) to add to the association
-      #
-      # @return [Dynamoid::Document] the added object
-      #
-      # @since 0.2.0            
-      def setter(object)
-        records.each {|o| delete(o)}
-        self << (object)
-        object
-      end
-      
-      # Create a new instance of the target class and add it directly to the association.
-      # 
-      # @param [Hash] attribute hash for the new object
-      #
-      # @return [Dynamoid::Document] the newly-created object
-      #
-      # @since 0.2.0            
-      def create(attributes = {})
-        self << target_class.create(attributes)
-      end
-      
-      # Create a new instance of the target class and add it directly to the association. If the create fails an exception will be raised.
-      # 
-      # @param [Hash] attribute hash for the new object
-      #
-      # @return [Dynamoid::Document] the newly-created object
-      #
-      # @since 0.2.0            
-      def create!(attributes = {})
-        self << target_class.create!(attributes)
-      end
-      
-      
-      # Naive association filtering.
-      # 
-      # @param [Hash] A hash of attributes; each must match every returned object's attribute exactly.
-      #
-      # @return [Dynamoid::Association] the association this method was called on (for chaining purposes)
-      #
-      # @since 0.2.0            
-      def where(args)
-        args.each {|k, v| query[k] = v}
-        self
-      end
-      
-      # Create a new instance of the target class and add it directly to the association. If the create fails an exception will be raised.
-      # 
-      # @param [Hash] attribute hash for the new object
-      #
-      # @return [Dynamoid::Document] the newly-created object
-      #
-      # @since 0.2.0            
-      def each(&block)
-        records.each(&block)
-      end
-
-      # Destroys all members of the association and removes them from the association.
-      # 
-      # @since 0.2.0
-      def destroy_all
-        objs = records
-        source.update_attribute(source_attribute, nil)
-        objs.each(&:destroy)
       end
 
-      # Deletes all members of the association and removes them from the association.
-      # 
-      # @since 0.2.0
-      def delete_all
-        objs = records
-        source.update_attribute(source_attribute, nil)
-        objs.each(&:delete)
-      end
-      
       private
-      
-      # If a query exists, filter all existing results based on that query.
-      #
-      # @param [Array] results the raw results for the association
-      #
-      # @return [Array] the filtered results for the query
-      # 
-      # @since 0.2.0
-      def results_with_query(results)
-        results.find_all do |result|
-          query.all? do |attribute, value|
-            result.send(attribute) == value
-          end
-        end
-      end
 
       # The target class name, either inferred through the association's name or specified in options.
       #
@@ -194,7 +41,7 @@ module Dynamoid #:nodoc:
       def target_class
         options[:class] || target_class_name.constantize
       end
-      
+
       # The target attribute: that is, the attribute on each object of the association that should reference the source.
       #
       # @since 0.2.0
@@ -204,25 +51,25 @@ module Dynamoid #:nodoc:
 
       # The ids in the target association.
       #
-      # @since 0.2.0      
+      # @since 0.2.0
       def target_ids
         target.send(target_attribute) || Set.new
       end
 
       # The ids in the target association.
       #
-      # @since 0.2.0            
+      # @since 0.2.0
       def source_class
         source.class
       end
-      
+
       # The source's association attribute: the name of the association with _ids afterwards, like "users_ids".
       #
       # @since 0.2.0
       def source_attribute
         "#{name}_ids".to_sym
       end
-      
+
       # The ids in the source association.
       #
       # @since 0.2.0
@@ -232,5 +79,5 @@ module Dynamoid #:nodoc:
 
     end
   end
-  
+
 end

data/lib/dynamoid/associations/belongs_to.rb

@@ -5,39 +5,11 @@ module Dynamoid #:nodoc:
   # object to which the association object is associated.
   module Associations
     class BelongsTo
-      include Dynamoid::Associations::Association
-      
-      # Is this object equal to the association's target?
-      #
-      # @return [Boolean] true/false
-      #
-      # @since 0.2.0
-      def ==(other)
-        target == other
-      end
-      
-      # Delegate methods we don't find directly to the target.
-      #
-      # @since 0.2.0
-      def method_missing(method, *args)
-        if target.respond_to?(method)
-          target.send(method, *args)
-        else
-          super
-        end
-      end
-      
+      include Association
+      include SingleAssociation
+
       private
-      
-      # Find the target of the belongs_to association.
-      #
-      # @return [Dynamoid::Document] the found target (or nil if nothing)
-      #
-      # @since 0.2.0
-      def target
-        records.first
-      end
-      
+
       # Find the target association, either has_many or has_one. Uses either options[:inverse_of] or the source class name and default parsing to
       # return the most likely name for the target association.
       #

data/lib/dynamoid/associations/has_and_belongs_to_many.rb

@@ -4,31 +4,12 @@ module Dynamoid #:nodoc:
   # The has and belongs to many association.
   module Associations
     class HasAndBelongsToMany
-      include Dynamoid::Associations::Association
-      
-      # Is this array equal to the association's records?
-      #
-      # @return [Boolean] true/false
-      #
-      # @since 0.2.0
-      def ==(other)
-        records == Array(other)
-      end
-      
-      # Delegate methods we don't find directly to the records array.
-      #
-      # @since 0.2.0
-      def method_missing(method, *args)
-        if records.respond_to?(method)
-          records.send(method, *args)
-        else
-          super
-        end
-      end
-      
+      include Association
+      include ManyAssociation
+
       private
-      
-      # Find the target association, always another :has_and_belongs_to_many association. Uses either options[:inverse_of] or the source class name 
+
+      # Find the target association, always another :has_and_belongs_to_many association. Uses either options[:inverse_of] or the source class name
       # and default parsing to return the most likely name for the target association.
       #
       # @since 0.2.0
@@ -38,15 +19,15 @@ module Dynamoid #:nodoc:
         return nil if guess.nil? || guess[:type] != :has_and_belongs_to_many
         key_name
       end
-            
+
       # Associate a source object to this association.
       #
-      # @since 0.2.0      
+      # @since 0.2.0
       def associate_target(object)
         ids = object.send(target_attribute) || Set.new
         object.update_attribute(target_attribute, ids.merge(Array(source.id)))
       end
-      
+
       # Disassociate a source object from this association.
       #
       # @since 0.2.0
@@ -56,5 +37,5 @@ module Dynamoid #:nodoc:
       end
     end
   end
-  
+
 end

data/lib/dynamoid/associations/has_many.rb

@@ -4,27 +4,8 @@ module Dynamoid #:nodoc:
   # The has_many association.
   module Associations
     class HasMany
-      include Dynamoid::Associations::Association
-      
-      # Is this array equal to the association's records?
-      #
-      # @return [Boolean] true/false
-      #
-      # @since 0.2.0
-      def ==(other)
-        records == Array(other)
-      end
-
-      # Delegate methods we don't find directly to the records array.
-      #
-      # @since 0.2.0
-      def method_missing(method, *args)
-        if records.respond_to?(method)
-          records.send(method, *args)
-        else
-          super
-        end
-      end
+      include Association
+      include ManyAssociation
 
       private
 

data/lib/dynamoid/associations/has_one.rb

@@ -4,64 +4,36 @@ module Dynamoid #:nodoc:
   # The HasOne association.
   module Associations
     class HasOne
-      include Dynamoid::Associations::Association
-      
-      # Is this object equal to the association's target?
-      #
-      # @return [Boolean] true/false
-      #
-      # @since 0.2.0
-      def ==(other)
-        target == other
-      end
+      include Association
+      include SingleAssociation
 
-      # Delegate methods we don't find directly to the target.
-      #
-      # @since 0.2.0
-      def method_missing(method, *args)
-        if target.respond_to?(method)
-          target.send(method, *args)
-        else
-          super
-        end
-      end
-      
       private
-      
-      # Find the target of the has_one association.
-      #
-      # @return [Dynamoid::Document] the found target (or nil if nothing)
-      #
-      # @since 0.2.0
-      def target
-        records.first
-      end
 
-      # Find the target association, always a :belongs_to association. Uses either options[:inverse_of] or the source class name 
+      # Find the target association, always a :belongs_to association. Uses either options[:inverse_of] or the source class name
       # and default parsing to return the most likely name for the target association.
       #
-      # @since 0.2.0      
+      # @since 0.2.0
       def target_association
         key_name = options[:inverse_of] || source.class.to_s.singularize.underscore.to_sym
         guess = target_class.associations[key_name]
         return nil if guess.nil? || guess[:type] != :belongs_to
         key_name
       end
-      
+
       # Associate a source object to this association.
       #
-      # @since 0.2.0      
+      # @since 0.2.0
       def associate_target(object)
         object.update_attribute(target_attribute, Set[source.id])
       end
-      
+
       # Disassociate a source object from this association.
       #
-      # @since 0.2.0      
+      # @since 0.2.0
       def disassociate_target(object)
         source.update_attribute(source_attribute, nil)
       end
     end
   end
-  
+
 end

data/lib/dynamoid/associations/many_association.rb

@@ -0,0 +1,188 @@
+# encoding: utf-8
+module Dynamoid #:nodoc:
+
+  module Associations
+    module ManyAssociation
+
+      attr_accessor :query
+
+      def initialize(*args)
+        @query = {}
+        super
+      end
+
+      include Enumerable
+      # Delegate methods to the records the association represents.
+      delegate :first, :last, :empty?, :size, :to => :records
+
+      # The records associated to the source.
+      #
+      # @return the association records; depending on which association this is, either a single instance or an array
+      #
+      # @since 0.2.0
+      def records
+        results = Array(target_class.find(source_ids.to_a))
+
+        if query.empty?
+          results
+        else
+          results_with_query(results)
+        end
+      end
+
+      # Alias convenience methods for the associations.
+      alias :all :records
+      alias :count :size
+      alias :nil? :empty?
+
+      # Delegate include? to the records.
+      def include?(object)
+        records.include?(object)
+      end
+
+      # Deletes an object or array of objects from the association. This removes their records from the association field on the source,
+      # and attempts to remove the source from the target association if it is detected to exist.
+      #
+      # @param [Dynamoid::Document] object the object (or array of objects) to remove from the association
+      #
+      # @return [Dynamoid::Document] the deleted object
+      #
+      # @since 0.2.0
+      def delete(object)
+        source.update_attribute(source_attribute, source_ids - Array(object).collect(&:id))
+        Array(object).each {|o| self.send(:disassociate_target, o)} if target_association
+        object
+      end
+
+
+      # Add an object or array of objects to an association. This preserves the current records in the association (if any)
+      # and adds the object to the target association if it is detected to exist.
+      #
+      # @param [Dynamoid::Document] object the object (or array of objects) to add to the association
+      #
+      # @return [Dynamoid::Document] the added object
+      #
+      # @since 0.2.0
+      def <<(object)
+        source.update_attribute(source_attribute, source_ids.merge(Array(object).collect(&:id)))
+        Array(object).each {|o| self.send(:associate_target, o)} if target_association
+        object
+      end
+
+      # Replace an association with object or array of objects. This removes all of the existing associated records and replaces them with
+      # the passed object(s), and associates the target association if it is detected to exist.
+      #
+      # @param [Dynamoid::Document] object the object (or array of objects) to add to the association
+      #
+      # @return [Dynamoid::Document] the added object
+      #
+      # @since 0.2.0
+      def setter(object)
+        records.each {|o| delete(o)}
+        self << (object)
+        object
+      end
+
+      # Create a new instance of the target class and add it directly to the association. If the create fails an exception will be raised.
+      #
+      # @param [Hash] attribute hash for the new object
+      #
+      # @return [Dynamoid::Document] the newly-created object
+      #
+      # @since 0.2.0
+      def create!(attributes = {})
+        self << target_class.create!(attributes)
+      end
+
+      # Create a new instance of the target class and add it directly to the association.
+      #
+      # @param [Hash] attribute hash for the new object
+      #
+      # @return [Dynamoid::Document] the newly-created object
+      #
+      # @since 0.2.0
+      def create(attributes = {})
+        self << target_class.create(attributes)
+      end
+
+      # Create a new instance of the target class and add it directly to the association. If the create fails an exception will be raised.
+      #
+      # @param [Hash] attribute hash for the new object
+      #
+      # @return [Dynamoid::Document] the newly-created object
+      #
+      # @since 0.2.0
+      def each(&block)
+        records.each(&block)
+      end
+
+      # Destroys all members of the association and removes them from the association.
+      #
+      # @since 0.2.0
+      def destroy_all
+        objs = records
+        source.update_attribute(source_attribute, nil)
+        objs.each(&:destroy)
+      end
+
+      # Deletes all members of the association and removes them from the association.
+      #
+      # @since 0.2.0
+      def delete_all
+        objs = records
+        source.update_attribute(source_attribute, nil)
+        objs.each(&:delete)
+      end
+
+      # Naive association filtering.
+      #
+      # @param [Hash] A hash of attributes; each must match every returned object's attribute exactly.
+      #
+      # @return [Dynamoid::Association] the association this method was called on (for chaining purposes)
+      #
+      # @since 0.2.0
+      def where(args)
+        args.each {|k, v| query[k] = v}
+        self
+      end
+
+      # Is this array equal to the association's records?
+      #
+      # @return [Boolean] true/false
+      #
+      # @since 0.2.0
+      def ==(other)
+        records == Array(other)
+      end
+
+      # Delegate methods we don't find directly to the records array.
+      #
+      # @since 0.2.0
+      def method_missing(method, *args)
+        if records.respond_to?(method)
+          records.send(method, *args)
+        else
+          super
+        end
+      end
+
+      private
+
+      # If a query exists, filter all existing results based on that query.
+      #
+      # @param [Array] results the raw results for the association
+      #
+      # @return [Array] the filtered results for the query
+      #
+      # @since 0.2.0
+      def results_with_query(results)
+        results.find_all do |result|
+          query.all? do |attribute, value|
+            result.send(attribute) == value
+          end
+        end
+      end
+
+    end
+  end
+end

data/lib/dynamoid/associations/single_association.rb

@@ -0,0 +1,66 @@
+# encoding: utf-8
+module Dynamoid #:nodoc:
+
+  module Associations
+    module SingleAssociation
+
+
+      def setter(object)
+        delete
+        source.update_attribute(source_attribute, Set[object.id])
+        self.send(:associate_target, object) if target_association
+        object
+      end
+
+      def delete
+        source.update_attribute(source_attribute, nil)
+        self.send(:disassociate_target, target) if target && target_association
+        target
+      end
+
+      def create!(attributes = {})
+        setter(target_class.create!(attributes))
+      end
+
+      def create(attributes = {})
+        setter(target_class.create!(attributes))
+      end
+
+
+      # Is this object equal to the association's target?
+      #
+      # @return [Boolean] true/false
+      #
+      # @since 0.2.0
+      def ==(other)
+        target == other
+      end
+
+      # Delegate methods we don't find directly to the target.
+      #
+      # @since 0.2.0
+      def method_missing(method, *args)
+        if target.respond_to?(method)
+          target.send(method, *args)
+        else
+          super
+        end
+      end
+
+      def nil?
+        target.nil?
+      end
+
+      private
+
+      # Find the target of the has_one association.
+      #
+      # @return [Dynamoid::Document] the found target (or nil if nothing)
+      #
+      # @since 0.2.0
+      def target
+        target_class.find(source_ids.first)
+      end
+    end
+  end
+end

data/lib/dynamoid/document.rb

@@ -21,11 +21,7 @@ module Dynamoid #:nodoc:
       #
       # @since 0.2.0
       def create(attrs = {})
-        obj = self.new(attrs)
-        obj.run_callbacks(:create) do
-          obj.save
-        end
-        obj
+        new(attrs).tap(&:save)
       end
 
       # Initialize a new object and immediately save it to the database. Raise an exception if persistence failed.
@@ -36,11 +32,7 @@ module Dynamoid #:nodoc:
       #
       # @since 0.2.0
       def create!(attrs = {})
-        obj = self.new(attrs)
-        obj.run_callbacks(:create) do
-          obj.save!
-        end
-        obj
+        new(attrs).tap(&:save!)
       end
       
       # Initialize a new object.
@@ -51,7 +43,7 @@ module Dynamoid #:nodoc:
       #
       # @since 0.2.0
       def build(attrs = {})
-        self.new(attrs)
+        new(attrs)
       end
 
       # Does this object exist?

data/lib/dynamoid/fields.rb

@@ -29,16 +29,12 @@ module Dynamoid #:nodoc:
       def field(name, type = :string, options = {})
         named = name.to_s
         self.attributes[name] = {:type => type}.merge(options)
-        define_method(named) do
-          read_attribute(named)
-        end
-        define_method("#{named}=") do |value|
-          write_attribute(named, value)
-        end
-        define_method("#{named}?") do
-          !read_attribute(named).nil?
-        end
-        define_attribute_methods(self.attributes.keys)
+
+        define_method(named) { read_attribute(named) }
+        define_method("#{named}?") { !read_attribute(named).nil? }
+        define_method("#{named}=") {|value| write_attribute(named, value) }
+
+        define_attribute_method(name)
       end
     end
     
@@ -53,7 +49,7 @@ module Dynamoid #:nodoc:
     #
     # @since 0.2.0
     def write_attribute(name, value)
-      self.send("#{name}_will_change!".to_sym) unless self.read_attribute(name) == value
+      attribute_will_change!(name) unless self.read_attribute(name) == value
       attributes[name.to_sym] = value
     end
     alias :[]= :write_attribute
@@ -107,4 +103,4 @@ module Dynamoid #:nodoc:
     
   end
   
-end
+end

data/lib/dynamoid/persistence.rb

@@ -100,17 +100,13 @@ module Dynamoid
     # @since 0.2.0
     def save(options = {})
       @previously_changed = changes
-      if self.new_record?
-        run_callbacks(:create) do
-          run_callbacks(:save) do
-            persist
-          end
-        end
+
+      if new_record?
+        run_callbacks(:create) { persist }
       else
-        run_callbacks(:save) do
-          persist
-        end
+        persist
       end
+
       self
     end
 
@@ -177,10 +173,12 @@ module Dynamoid
     #
     # @since 0.2.0
     def persist
-      self.id = SecureRandom.uuid if self.id.nil? || self.id.blank?
-      Dynamoid::Adapter.write(self.class.table_name, self.dump)
-      save_indexes
-      @new_record = false
+      run_callbacks(:save) do
+        self.id = SecureRandom.uuid if self.id.nil? || self.id.blank?
+        Dynamoid::Adapter.write(self.class.table_name, self.dump)
+        save_indexes
+        !(@new_record = false)
+      end
     end
         
   end

data/spec/dynamoid/associations/belongs_to_spec.rb

@@ -19,14 +19,13 @@ describe "Dynamoid::Associations::BelongsTo" do
   
     it 'delegates equality to its source record' do
       @magazine = @subscription.magazine.create
-    
+
       @subscription.magazine.should == @magazine
     end
   
     it 'associates has_many automatically' do
       @magazine = @subscription.magazine.create
     
-      @magazine.subscriptions.size.should == 1
       @magazine.subscriptions.should include @subscription
 
       @magazine = Magazine.create
@@ -63,7 +62,6 @@ describe "Dynamoid::Associations::BelongsTo" do
     it 'associates has_one automatically' do
       @magazine = @sponsor.magazine.create
       
-      @magazine.sponsor.size.should == 1
       @magazine.sponsor.should == @sponsor
 
       @user = @subscription.customer.create

data/spec/dynamoid/associations/has_one_spec.rb

@@ -37,7 +37,6 @@ describe "Dynamoid::Associations::HasOne" do
   it 'associates belongs_to automatically' do
     @sponsor = @magazine.sponsor.create
     @sponsor.magazine.should == @magazine
-    @magazine.sponsor.size.should == 1
     @magazine.sponsor.should == @sponsor
 
     @subscription = @user.monthly.create

data/spec/dynamoid/document_spec.rb

@@ -8,6 +8,14 @@ describe "Dynamoid::Document" do
     @address.new_record.should be_true
     @address.attributes.should == {:id=>nil, :created_at=>nil, :updated_at=>nil, :city=>nil, :options=>nil}
   end
+
+  it 'responds to will_change! methods for all fields' do
+    @address = Address.new
+    @address.should respond_to(:id_will_change!)
+    @address.should respond_to(:options_will_change!)
+    @address.should respond_to(:created_at_will_change!)
+    @address.should respond_to(:updated_at_will_change!)
+  end
   
   it 'initializes a new document with attributes' do
     @address = Address.new(:city => 'Chicago')

data/spec/dynamoid/persistence_spec.rb

@@ -1,12 +1,14 @@
 require File.expand_path(File.dirname(__FILE__) + '/../spec_helper')
 
 describe "Dynamoid::Persistence" do
-  
+
+  let(:document_class) { Class.new.send :include, Dynamoid::Document }
+
   before do
     Random.stubs(:rand).with(Dynamoid::Config.partition_size).returns(0)
     @address = Address.new
   end
-  
+
   context 'without AWS keys' do
     unless ENV['ACCESS_KEY'] && ENV['SECRET_KEY']
       before do
@@ -102,6 +104,30 @@ describe "Dynamoid::Persistence" do
     User.undump(@hash)[:name].should == 'Josh'
     User.undump(@hash)[:created_at].to_f == @time.to_f
   end
+
+  it 'runs the before_create callback only once' do
+    document_class.before_create { doing_before_create }
+
+    document_class.any_instance.expects(:doing_before_create)
+
+    document_class.create
+  end
+
+  it 'runs after save callbacks when doing #create' do
+    document_class.after_create { doing_after_create }
+
+    document_class.any_instance.expects(:doing_after_create)
+
+    document_class.create
+  end
+
+  it 'runs after save callbacks when doing #save' do
+    document_class.after_create { doing_after_create }
+
+    document_class.any_instance.expects(:doing_after_create)
+
+    document_class.new.save
+  end
   
   it 'tracks previous changes on save or update' do
     @address.city = 'Chicago'

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: dynamoid
 version: !ruby/object:Gem::Version
-  version: 0.3.0
+  version: 0.3.1
   prerelease: 
 platform: ruby
 authors:
@@ -282,6 +282,8 @@ files:
 - lib/dynamoid/associations/has_and_belongs_to_many.rb
 - lib/dynamoid/associations/has_many.rb
 - lib/dynamoid/associations/has_one.rb
+- lib/dynamoid/associations/many_association.rb
+- lib/dynamoid/associations/single_association.rb
 - lib/dynamoid/components.rb
 - lib/dynamoid/config.rb
 - lib/dynamoid/config/options.rb
@@ -337,7 +339,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: 1952467845042018091
+      hash: 3564635141636457738
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements: