test/unread_test.rb in unread-0.0.4 vs test/unread_test.rb in unread-0.0.5

@@ -17,9 +17,17 @@
   end
   
   def test_schema_has_loaded_correctly
     assert_equal [@email1, @email2], Email.all
   end
+  
+  def test_readable_classes
+    assert_equal [ Email ], ReadMark.readable_classes
+  end
+  
+  def test_reader_class
+    assert_equal User, ReadMark.reader_class
+  end
 
   def test_scope
     assert_equal [@email1, @email2], Email.unread_by(@user)
     assert_equal [@email1, @email2], Email.unread_by(@other_user)
     
\ No newline at end of file