reset dirty attributes after save

lib/couch_potato/persistence.rb

@@ -13,7 +13,8 @@ module CouchPotato
   module Persistence
 
     def self.included(base)
-      base.send :include, Properties, Callbacks, Validatable, Json, DirtyAttributes, CouchPotato::View::CustomViews
+      base.send :include, Properties, Callbacks, Validatable, Json, CouchPotato::View::CustomViews
+      base.send :include, DirtyAttributes
       base.send :include, MagicTimestamps
       base.class_eval do
         attr_accessor :_id, :_rev, :_attachments, :_deleted

lib/couch_potato/persistence/dirty_attributes.rb

@@ -1,11 +1,26 @@
 module CouchPotato
   module Persistence
     module DirtyAttributes
+
+      def self.included(base)
+        base.class_eval do
+          after_save :reset_dirty_attributes
+        end
+      end
+
       def dirty?
         new? || self.class.properties.inject(false) do |res, property|
           res || property.dirty?(self)
         end
       end
+
+      private
+
+      def reset_dirty_attributes
+        self.class.properties.each do |property|
+          instance_variable_set("@#{property.name}_was", send(property.name))
+        end
+      end
     end
   end
 end

spec/unit/dirty_attributes_spec.rb

@@ -88,7 +88,12 @@ class Plate
 
   describe "after save" do
     it "should reset all attributes to not dirty" do
-      pending
+      couchrest_db = stub('database', :get => {'_id' => '1', '_rev' => '2', 'food' => 'sushi', 'ruby_class' => 'Plate'}, :save_doc => {})
+      db = CouchPotato::Database.new(couchrest_db)
+      @plate = db.load_document '1'
+      @plate.food = 'burger'
+      db.save! @plate
+      @plate.should_not be_food_changed
     end
   end