Added tests for validators

[rails.git] / app / models / diary_comment.rb

diff --git a/app/models/diary_comment.rb b/app/models/diary_comment.rb
index 075d288edb8e0b644970b7b515a4f6967506abbe..475872b26bbeba365613ba30bd9819831d08c59f 100644 (file)
--- a/app/models/diary_comment.rb
+++ b/app/models/diary_comment.rb
@@ -1,16 +1,40 @@
+# == Schema Information
+#
+# Table name: diary_comments
+#
+#  id             :integer          not null, primary key
+#  diary_entry_id :integer          not null
+#  user_id        :integer          not null
+#  body           :text             not null
+#  created_at     :datetime         not null
+#  updated_at     :datetime         not null
+#  visible        :boolean          default(TRUE), not null
+#  body_format    :enum             default("markdown"), not null
+#
+# Indexes
+#
+#  diary_comment_user_id_created_at_index  (user_id,created_at)
+#  diary_comments_entry_id_idx             (diary_entry_id,id) UNIQUE
+#
+# Foreign Keys
+#
+#  diary_comments_diary_entry_id_fkey  (diary_entry_id => diary_entries.id)
+#  diary_comments_user_id_fkey         (user_id => users.id)
+#
+
 class DiaryComment < ActiveRecord::Base
   belongs_to :user
   belongs_to :diary_entry
 
-  validates_presence_of :body
-  validates_associated :diary_entry
+  scope :visible, -> { where(:visible => true) }
 
-  attr_accessible :body
+  validates :body, :presence => true, :invalid_chars => true
+  validates :diary_entry, :user, :associated => true
 
-  after_initialize :set_defaults
+  after_save :spam_check
 
   def body
-    RichText.new(read_attribute(:body_format), read_attribute(:body))
+    RichText.new(self[:body_format], self[:body])
   end
 
   def digest
@@ -22,9 +46,9 @@ class DiaryComment < ActiveRecord::Base
     md5.hexdigest
   end
 
-private
+  private
 
-  def set_defaults
-    self.body_format = "markdown" unless self.attribute_present?(:body_format)
+  def spam_check
+    user.spam_check
   end
 end