Add Relationship validations and fix tests (No issue opened) (#2078)

* Refactor code for user's social links * Fix Relationship tests * Add validations to Relationship model
publiclab · Feb 1, 2018 · e682250 · e682250
1 parent 69e0a39
commit e682250
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/app/models/relationship.rb b/app/models/relationship.rb
@@ -2,4 +2,6 @@ class Relationship < ActiveRecord::Base
   attr_accessible :followed_id, :follower_id
   belongs_to :follower, class_name: 'User'
   belongs_to :followed, class_name: 'User'
+  validates :followed_id, presence: true
+  validates :follower_id, presence: true
 end
diff --git a/test/unit/relationship_test.rb b/test/unit/relationship_test.rb
@@ -13,11 +13,11 @@ def setup
 
   test 'should require follower_id' do
     @relationship.follower_id = nil
-    assert @relationship.valid?
+    assert_not @relationship.valid?
   end
 
   test 'should require followed_id' do
     @relationship.followed_id = nil
-    assert @relationship.valid?
+    assert_not @relationship.valid?
   end
 end