X-Git-Url: https://git.openstreetmap.org/rails.git/blobdiff_plain/9c5c71f2d7413c280c046f45d9b0565981c7bbab..2939aa9398694d304067a2fd409104ed53afc57d:/app/models/message.rb

diff --git a/app/models/message.rb b/app/models/message.rb
index 590e3e357..ec712be25 100644
--- a/app/models/message.rb
+++ b/app/models/message.rb
@@ -1,2 +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
+
+  validates_presence_of :title, :body, :sent_on, :sender, :recipient
+  validates_inclusion_of :message_read, :in => [ true, false ]
+  validates_associated :sender, :recipient
 end