Minor Document#undelete! improvement

yrashk · May 4, 2008 · 18b624b · 18b624b
1 parent e88d174
commit 18b624b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/lib/strokedb/document/delete.rb b/lib/strokedb/document/delete.rb
@@ -11,7 +11,7 @@ class DocumentDeletionError < StandardError
     def undelete!
       deleted_version = versions.previous
       store.save_as_head!(deleted_version)
-      self.class.find(uuid)
+      store.find(uuid)
     end
   end
 

diff --git a/spec/lib/strokedb/document/document_spec.rb b/spec/lib/strokedb/document/document_spec.rb
@@ -474,10 +474,9 @@
   end
 
   it "should be undeletable" do
-    @document = @document.undelete!
-    @document.should_not be_a_kind_of(DeletedDocument)
-    @document.should be_mutable
-    @store.find(@document.uuid).should == @document
+    undeleted_document = @document.undelete!
+    @document.should_not be_mutable
+    undeleted_document.should_not be_a_kind_of(DeletedDocument)
   end