diff --git a/app/models/user.rb b/app/models/user.rb --- a/app/models/user.rb +++ b/app/models/user.rb @@ -6,6 +6,16 @@ has_many :test_requests, :order => "submitted_at DESC" + has_many :messages, + :class_name => "Message", + :foreign_key => "sender_id", + :order => 'created_at DESC' + + has_many :replied_messages, + :class_name => "Message", + :foreign_key => "receiver_id", + :order => 'created_at DESC' + belongs_to :site validates_presence_of :login