Add extra validation to messages and diary entries.

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

diff --git a/app/models/message.rb b/app/models/message.rb
index 039c8d30b9e3f31e2f0e4096eb84d57935dfa40c..97e411192b0df5cedbd89d7255f236ac7fb2cf35 100644 (file)
--- a/app/models/message.rb
+++ b/app/models/message.rb
@@ -1,4 +1,8 @@
 class Message < ActiveRecord::Base
   belongs_to :sender, :class_name => "User", :foreign_key => :from_user_id
   belongs_to :recipient, :class_name => "User", :foreign_key => :to_user_id
 class Message < ActiveRecord::Base
   belongs_to :sender, :class_name => "User", :foreign_key => :from_user_id
   belongs_to :recipient, :class_name => "User", :foreign_key => :to_user_id

+
+  validates_presence_of :title, :body, :sent_on
+  validates_inclusion_of :message_read, :in => [ true, false ]
+  validates_associated :sender, :recipient

 end
 end