specs and tweaks for yaml serialization

lib/delayed/backend/active_record.rb

@@ -5,6 +5,8 @@ class ActiveRecord::Base
 
   def self.yaml_new(klass, tag, val)
     klass.find(val['attributes']['id'])
+  rescue ActiveRecord::RecordNotFound
+    nil
   end
 
   def to_yaml_properties

lib/delayed/backend/data_mapper.rb

@@ -1,7 +1,18 @@
 require 'dm-core'
 require 'dm-observer'
 require 'dm-aggregates'
-require 'dm-serializer'
+
+DataMapper::Resource.class_eval do
+  yaml_as "tag:ruby.yaml.org,2002:DataMapper"
+
+  def self.yaml_new(klass, tag, val)
+    klass.find(val['id'])
+  end
+
+  def to_yaml_properties
+    ['@id']
+  end
+end
 
 module Delayed
   module Backend

lib/delayed/backend/mongo_mapper.rb

@@ -4,9 +4,7 @@
   yaml_as "tag:ruby.yaml.org,2002:MongoMapper"
 
   def self.yaml_new(klass, tag, val)
-    klass.find!(val['_id'])
-  rescue MongoMapper::DocumentNotFound
-    nil
+    klass.find(val['_id'])
   end
 
   def to_yaml_properties

spec/backend/shared_backend_spec.rb

@@ -270,4 +270,22 @@ def create_job(opts = {})
       @job.id.should_not be_nil
     end
   end
+
+  describe "yaml serialization" do
+    it "should reload changed attributes" do
+      job = @backend.enqueue SimpleJob.new
+      yaml = job.to_yaml
+      job.priority = 99
+      job.save
+      YAML.load(yaml).priority.should == 99
+    end
+
+    it "should ignore destroyed records" do
+      job = @backend.enqueue SimpleJob.new
+      yaml = job.to_yaml
+      job.destroy
+      lambda { YAML.load(yaml).should be_nil }.should_not raise_error
+    end
+  end
+
 end