Copy daniel's tests from his PR

lib/zombie_record/restorable.rb

@@ -77,7 +77,7 @@ def associate_method(association, klass = nil)
           when :has_one, :belongs_to
             (klass || association.klass).unscoped { super() }
           when :has_many
-            super().deleted
+            super().with_deleted
           end
         else
           super()

spec/restorable_spec.rb

@@ -124,16 +124,46 @@
   end
 
   describe "#associate_with_deleted" do
-    it "removes default scope from associations when deleted" do
-      cover = Cover.create!
-      book = Book.create!(cover: cover)
+    it "allows accessing a deleted has_one association" do
+      book = Book.create!
+      cover = Cover.create!(book: book)
 
       book.destroy
-      Book.find_by_id(book.id).should be_nil
-      Cover.find_by_id(cover.id).should be_nil
+      book = Book.deleted.find(book.id)
 
-      book = Book.deleted.first
-      book.cover.should == cover
+      book.cover.should == cover.reload
+    end
+
+    it "allows accessing deleted belongs_to associations" do
+      book = Book.create!
+      chapter = book.chapters.create!
+
+      book.destroy
+      chapter = Chapter.deleted.find(chapter.id)
+
+      chapter.book.should == book
+    end
+
+    it "ensures deleted associations themselves allow access to deleted records" do
+      book = Book.create!
+      chapter = book.chapters.create!
+      bookmark = book.bookmarks.create!
+
+      book.destroy
+      bookmark = Bookmark.deleted.find(bookmark.id)
+
+      bookmark.book.chapters.should == [chapter]
+      chapter = bookmark.book.chapters.first
+
+      chapter.book.should == book
+    end
+
+    it "forwards normal method calls" do
+      book = Book.create!(title: "The Odyssey")
+      book.destroy
+      book = Book.deleted.find(book.id)
+
+      book.title.should == "The Odyssey"
     end
   end
 end

spec/spec_helper.rb

@@ -80,6 +80,7 @@ class Library < ActiveRecord::Base
         t.integer :author_id
         t.integer :library_id
         t.timestamps
+        t.string :title
         t.timestamp :deleted_at
       end